fix: tests and api

geofmureithi · Jul 4, 2024 · e8c900d · e8c900d
1 parent 6c145a3
commit e8c900d
Show file tree

Hide file tree

Showing 4 changed files with 21 additions and 6 deletions.
diff --git a/packages/apalis-redis/src/storage.rs b/packages/apalis-redis/src/storage.rs
@@ -987,7 +987,11 @@ mod tests {
         let job_id = &job.get::<Context>().unwrap().id;
 
         storage
-            .ack(&worker_id, &job_id)
+            .ack(AckResponse {
+                acknowledger: job_id.clone(),
+                result: "Success".to_string(),
+                worker: worker_id.clone(),
+            })
             .await
             .expect("failed to acknowledge the job");
 

diff --git a/packages/apalis-sql/src/mysql.rs b/packages/apalis-sql/src/mysql.rs
@@ -627,7 +627,11 @@ mod tests {
         let job_id = ctx.id();
 
         storage
-            .ack(&worker_id, job_id)
+            .ack(AckResponse {
+                acknowledger: job_id.clone(),
+                result: "Success".to_string(),
+                worker: worker_id.clone(),
+            })
             .await
             .expect("failed to acknowledge the job");
 

diff --git a/packages/apalis-sql/src/postgres.rs b/packages/apalis-sql/src/postgres.rs
@@ -585,7 +585,6 @@ mod tests {
 
     use super::*;
     use email_service::Email;
-    use futures::StreamExt;
     use sqlx::types::chrono::Utc;
 
     /// migrate DB and return a storage instance.
@@ -638,7 +637,7 @@ mod tests {
         storage: &mut PostgresStorage<Email>,
         worker_id: &WorkerId,
     ) -> Request<Email> {
-        let mut req = storage.fetch_next(worker_id).await;
+        let req = storage.fetch_next(worker_id).await;
         req.unwrap()[0].clone()
     }
 
@@ -699,7 +698,11 @@ mod tests {
         let job_id = ctx.id();
 
         storage
-            .ack(&worker_id, job_id)
+            .ack(AckResponse {
+                acknowledger: job_id.clone(),
+                result: "Success".to_string(),
+                worker: worker_id.clone(),
+            })
             .await
             .expect("failed to acknowledge the job");
 

diff --git a/packages/apalis-sql/src/sqlite.rs b/packages/apalis-sql/src/sqlite.rs
@@ -607,7 +607,11 @@ mod tests {
         let job_id = ctx.id();
 
         storage
-            .ack(&worker_id, job_id)
+            .ack(AckResponse {
+                acknowledger: job_id.clone(),
+                result: "Success".to_string(),
+                worker: worker_id.clone(),
+            })
             .await
             .expect("failed to acknowledge the job");