Backend Expose Test

examples/actix-web/src/main.rs

@@ -56,3 +56,28 @@ async fn main() -> Result<()> {
     future::try_join(http, worker).await?;
     Ok(())
 }
+
+// #[cfg(test)]
+// mod tests {
+
+//     async fn push_email_test() -> () {
+//         // curl an email
+//         // check redis if email job exists
+//         let client = reqwest::Client::new();
+//         let res = client
+//             .post("http://localhost:8000/emails/push")
+//             .body(
+//                 "{
+//                     'to': '[email protected]',
+//                     'subject': 'Message from Web',
+//                     'text': 'This is the text'
+//                 }",
+//             )
+//             .send()
+//             .await?;
+
+//         let redis_url = std::env::var("REDIS_URL").expect("Missing env variable REDIS_URL");
+//         let conn = apalis_redis::connect(redis_url);
+//         let exists: bool = conn
+//     }
+// }

packages/apalis-sql/src/sqlite.rs

@@ -584,7 +584,7 @@ impl<J: 'static + Serialize + DeserializeOwned + Unpin + Send + Sync> BackendExp
                             COUNT(1) FILTER (WHERE status = 'Killed') AS killed
                         FROM Jobs WHERE job_type = ?";
 
-        let res: (i64, i64, i64, i64, i64, i64) = sqlx::query_as(fetch_query)
+        let res: (i64, i64, i64, i64, i64) = sqlx::query_as(fetch_query)
             .bind(self.get_config().namespace())
             .fetch_one(self.pool())
             .await?;
@@ -883,4 +883,67 @@ mod tests {
         assert_eq!(*ctx.last_error(), Some("{\"Ok\":\"success\"}".to_owned()));
         assert_eq!(job.parts.attempt.current(), 1);
     }
+
+    #[tokio::test]
+    async fn test_list_workers() {
+        let mut storage = setup().await;
+        register_worker(&mut storage).await;
+
+        let workers = storage.list_workers().await.expect("no workers found!");
+        assert_eq!(workers.len(), 1);
+    }
+
+    /// Pushes email jobs and verifies they are listed correctly.
+    #[tokio::test]
+    async fn test_push_and_list_10_jobs() {
+        let mut storage = setup().await;
+
+        // Push 100 email jobs
+        for i in 0..10 {
+            push_email(
+                &mut storage,
+                Email {
+                    subject: format!("Test Subject {i}"),
+                    to: format!("user{i}@example.com"),
+                    text: format!("This is test email number {i}"),
+                },
+            )
+            .await;
+        }
+
+        // Verify the count of jobs
+        let jobs_num = storage
+            .list_jobs(&State::Pending, 1)
+            .await
+            .expect("Failed to get job Count");
+        assert_eq!(jobs_num.len(), 10);
+    }
+
+    #[tokio::test]
+    async fn test_stats() {
+        let mut storage = setup().await;
+        let worker = register_worker(&mut storage).await;
+
+        // Push 100 email jobs
+        for i in 0..10 {
+            push_email(
+                &mut storage,
+                Email {
+                    subject: format!("Test Subject {i}"),
+                    to: format!("user{i}@example.com"),
+                    text: format!("This is test email number {i}"),
+                },
+            )
+            .await;
+        }
+
+        let stats = storage.stats().await.expect("Failed to get stats");
+
+        assert_eq!(stats.pending, 10);
+
+        let job = consume_one(&mut storage, &worker).await;
+        let ctx = job.parts.context;
+        assert_eq!(stats.pending, 10);
+        assert_eq!(*ctx.status(), State::Running);
+    }
 }