Skip to content

Commit 91bf23d

Browse files
committed
worker/jobs/downloads/update_metadata: Read downloads from crate_downloads table
1 parent 8032a18 commit 91bf23d

File tree

1 file changed

+10
-7
lines changed

1 file changed

+10
-7
lines changed

src/worker/jobs/downloads/update_metadata.rs

Lines changed: 10 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -176,8 +176,9 @@ mod tests {
176176
use super::*;
177177
use crate::email::Emails;
178178
use crate::models::{Crate, NewCrate, NewUser, NewVersion, User, Version};
179-
use crate::schema::{crates, versions};
179+
use crate::schema::{crate_downloads, crates, versions};
180180
use crate::test_util::test_db_connection;
181+
use chrono::NaiveDateTime;
181182
use std::collections::BTreeMap;
182183

183184
fn user(conn: &mut PgConnection) -> User {
@@ -235,9 +236,9 @@ mod tests {
235236
.select(versions::downloads)
236237
.first(conn);
237238
assert_eq!(version_downloads, Ok(1));
238-
let crate_downloads = crates::table
239+
let crate_downloads = crate_downloads::table
239240
.find(krate.id)
240-
.select(crates::downloads)
241+
.select(crate_downloads::downloads)
241242
.first(conn);
242243
assert_eq!(crate_downloads, Ok(1));
243244
super::update(conn).unwrap();
@@ -332,17 +333,19 @@ mod tests {
332333
.unwrap();
333334

334335
let version_before: Version = versions::table.find(version.id).first(conn).unwrap();
335-
let krate_before: (NaiveDateTime, i32) = crates::table
336-
.select((crates::updated_at, crates::downloads))
336+
let krate_before: (NaiveDateTime, i64) = crates::table
337+
.inner_join(crate_downloads::table)
338+
.select((crates::updated_at, crate_downloads::downloads))
337339
.filter(crates::id.eq(krate.id))
338340
.first(conn)
339341
.unwrap();
340342
super::update(conn).unwrap();
341343
let version2: Version = versions::table.find(version.id).first(conn).unwrap();
342344
assert_eq!(version2.downloads, 2);
343345
assert_eq!(version2.updated_at, version_before.updated_at);
344-
let krate2: (NaiveDateTime, i32) = crates::table
345-
.select((crates::updated_at, crates::downloads))
346+
let krate2: (NaiveDateTime, i64) = crates::table
347+
.inner_join(crate_downloads::table)
348+
.select((crates::updated_at, crate_downloads::downloads))
346349
.filter(crates::id.eq(krate.id))
347350
.first(conn)
348351
.unwrap();

0 commit comments

Comments
 (0)