From ce3aa6be742365ddd365d54870910f5482c79571 Mon Sep 17 00:00:00 2001
From: Michael Chung <michael.chung@databites.ca>
Date: Thu, 23 Nov 2023 01:09:17 -0500
Subject: [PATCH] model_weights delete need tests

---
 .../queries/model_weights/delete_weights.rs   | 59 +++++++++----------
 .../src/queries/model_weights/mod.rs          |  2 +-
 2 files changed, 29 insertions(+), 32 deletions(-)

diff --git a/crates/lib/mysql_queries/src/queries/model_weights/delete_weights.rs b/crates/lib/mysql_queries/src/queries/model_weights/delete_weights.rs
index b36c9de645..f2d66739ca 100644
--- a/crates/lib/mysql_queries/src/queries/model_weights/delete_weights.rs
+++ b/crates/lib/mysql_queries/src/queries/model_weights/delete_weights.rs
@@ -6,8 +6,9 @@ pub async fn delete_weights_as_user(
     weight_token: &str,
     mysql_pool: &MySqlPool
 ) -> AnyhowResult<()> {
-    let _r = sqlx::query!(
-        r#"
+    let _r = sqlx
+        ::query!(
+            r#"
 UPDATE model_weights
 SET
   user_deleted_at = CURRENT_TIMESTAMP
@@ -15,30 +16,26 @@ WHERE
   token = ?
 LIMIT 1
         "#,
-      weight_token,
-    )
-        .execute(mysql_pool)
-        .await?;
+            weight_token
+        )
+        .execute(mysql_pool).await?;
     Ok(())
 }
 
-pub async fn delete_voice_as_mod(
-    weight_token: &str,
-    mysql_pool: &MySqlPool
-) -> AnyhowResult<()> {
-    let _r = sqlx::query!(
-        r#"
+pub async fn delete_voice_as_mod(weight_token: &str, mysql_pool: &MySqlPool) -> AnyhowResult<()> {
+    let _r = sqlx
+        ::query!(
+            r#"
 UPDATE model_weights
 SET
-  mod_deleted_at = CURRENT_TIMESTAMP,
+  mod_deleted_at = CURRENT_TIMESTAMP
 WHERE
   token = ?
 LIMIT 1
         "#,
-      weight_token,
-    )
-        .execute(mysql_pool)
-        .await?;
+            weight_token
+        )
+        .execute(mysql_pool).await?;
     Ok(())
 }
 
@@ -46,8 +43,9 @@ pub async fn undelete_weights_as_user(
     weight_token: &str,
     mysql_pool: &MySqlPool
 ) -> AnyhowResult<()> {
-    let _r = sqlx::query!(
-        r#"
+    let _r = sqlx
+        ::query!(
+            r#"
 UPDATE model_weights
 SET
   user_deleted_at = NULL
@@ -55,10 +53,9 @@ WHERE
   token = ?
 LIMIT 1
         "#,
-      weight_token,
-    )
-        .execute(mysql_pool)
-        .await?;
+            weight_token
+        )
+        .execute(mysql_pool).await?;
     Ok(())
 }
 
@@ -66,18 +63,18 @@ pub async fn undelete_weights_as_mod(
     weight_token: &str,
     mysql_pool: &MySqlPool
 ) -> AnyhowResult<()> {
-    let _r = sqlx::query!(
-        r#"
+    let _r = sqlx
+        ::query!(
+            r#"
 UPDATE model_weights
 SET
-  mod_deleted_at = NULL,
+  mod_deleted_at = NULL
 WHERE
   token = ?
 LIMIT 1
         "#,
-      weight_token,
-    )
-        .execute(mysql_pool)
-        .await?;
+            weight_token
+        )
+        .execute(mysql_pool).await?;
     Ok(())
-}
\ No newline at end of file
+}
diff --git a/crates/lib/mysql_queries/src/queries/model_weights/mod.rs b/crates/lib/mysql_queries/src/queries/model_weights/mod.rs
index c4c1262251..a78c3438dd 100644
--- a/crates/lib/mysql_queries/src/queries/model_weights/mod.rs
+++ b/crates/lib/mysql_queries/src/queries/model_weights/mod.rs
@@ -1,4 +1,4 @@
 pub mod create_weight;
-//pub mod delete_weight;
+pub mod delete_weights;
 pub mod get_weight;
 pub mod update_weight;
\ No newline at end of file