From a5bca1852ba397da7fda9d15972006eb4247ef8d Mon Sep 17 00:00:00 2001
From: Bartosz Sypytkowski <b.sypytkowski@gmail.com>
Date: Tue, 18 Jun 2024 08:53:02 +0200
Subject: [PATCH] chore: change test query to macro

---
 services/appflowy-indexer/src/collab_handle.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/services/appflowy-indexer/src/collab_handle.rs b/services/appflowy-indexer/src/collab_handle.rs
index 4b67489d..28ac33d6 100644
--- a/services/appflowy-indexer/src/collab_handle.rs
+++ b/services/appflowy-indexer/src/collab_handle.rs
@@ -363,14 +363,14 @@ mod test {
 
     assert_eq!(contents.len(), 1);
 
-    let tokens: i64 = sqlx::query(
+    let tokens = sqlx::query_scalar!(
       "SELECT index_tokens_consumed from af_workspace_ai_usage WHERE workspace_id = $1",
+      workspace_id
     )
-    .bind(workspace_id)
     .fetch_one(&db)
     .await
     .unwrap()
-    .get(0);
+    .unwrap_or(0);
     assert_ne!(tokens, 0);
   }
 }