refactor: Split-up pgqueueerror into multiple

PostHog · Feb 5, 2024 · 2adfd31 · 2adfd31
1 parent d5fd180
commit 2adfd31
Show file tree

Hide file tree

Showing 7 changed files with 103 additions and 118 deletions.
diff --git a/hook-api/src/handlers/app.rs b/hook-api/src/handlers/app.rs
@@ -30,9 +30,7 @@ mod tests {
 
     #[sqlx::test(migrations = "../migrations")]
     async fn index(db: PgPool) {
-        let pg_queue = PgQueue::new_from_pool("test_index", db)
-            .await
-            .expect("failed to construct pg_queue");
+        let pg_queue = PgQueue::new_from_pool("test_index", db).await;
 
         let app = add_routes(Router::new(), pg_queue);
 

diff --git a/hook-api/src/handlers/webhook.rs b/hook-api/src/handlers/webhook.rs
@@ -127,9 +127,7 @@ mod tests {
 
     #[sqlx::test(migrations = "../migrations")]
     async fn webhook_success(db: PgPool) {
-        let pg_queue = PgQueue::new_from_pool("test_index", db)
-            .await
-            .expect("failed to construct pg_queue");
+        let pg_queue = PgQueue::new_from_pool("test_index", db).await;
 
         let app = add_routes(Router::new(), pg_queue);
 
@@ -172,9 +170,7 @@ mod tests {
 
     #[sqlx::test(migrations = "../migrations")]
     async fn webhook_bad_url(db: PgPool) {
-        let pg_queue = PgQueue::new_from_pool("test_index", db)
-            .await
-            .expect("failed to construct pg_queue");
+        let pg_queue = PgQueue::new_from_pool("test_index", db).await;
 
         let app = add_routes(Router::new(), pg_queue);
 
@@ -212,9 +208,7 @@ mod tests {
 
     #[sqlx::test(migrations = "../migrations")]
     async fn webhook_payload_missing_fields(db: PgPool) {
-        let pg_queue = PgQueue::new_from_pool("test_index", db)
-            .await
-            .expect("failed to construct pg_queue");
+        let pg_queue = PgQueue::new_from_pool("test_index", db).await;
 
         let app = add_routes(Router::new(), pg_queue);
 
@@ -235,9 +229,7 @@ mod tests {
 
     #[sqlx::test(migrations = "../migrations")]
     async fn webhook_payload_not_json(db: PgPool) {
-        let pg_queue = PgQueue::new_from_pool("test_index", db)
-            .await
-            .expect("failed to construct pg_queue");
+        let pg_queue = PgQueue::new_from_pool("test_index", db).await;
 
         let app = add_routes(Router::new(), pg_queue);
 
@@ -258,9 +250,7 @@ mod tests {
 
     #[sqlx::test(migrations = "../migrations")]
     async fn webhook_payload_body_too_large(db: PgPool) {
-        let pg_queue = PgQueue::new_from_pool("test_index", db)
-            .await
-            .expect("failed to construct pg_queue");
+        let pg_queue = PgQueue::new_from_pool("test_index", db).await;
 
         let app = add_routes(Router::new(), pg_queue);