From 19b05f27ceeeaa0ed827d3652c504f45dafc1328 Mon Sep 17 00:00:00 2001
From: Vo Van Nghia <git@vnghia.com>
Date: Tue, 9 Jan 2024 23:38:34 +0700
Subject: [PATCH] os(user): add test for setup

---
 src/open_subsonic/user/setup.rs | 67 +++++++++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)

diff --git a/src/open_subsonic/user/setup.rs b/src/open_subsonic/user/setup.rs
index 961d8489..18603677 100644
--- a/src/open_subsonic/user/setup.rs
+++ b/src/open_subsonic/user/setup.rs
@@ -47,3 +47,70 @@ pub async fn setup_handler(
     .await?;
     Ok(SetupBody::default().into())
 }
+
+#[cfg(test)]
+mod tests {
+    use fake::{faker::internet::en::FreeEmail, Fake};
+
+    use super::*;
+    use crate::utils::test::{db::TemporaryDatabase, user::create_key_user_token};
+
+    #[tokio::test]
+    async fn test_setup_no_user() {
+        let db = TemporaryDatabase::new_from_env().await;
+        let (key, username, password, _, _) = create_key_user_token();
+
+        let state = State(ServerState {
+            conn: db.get_conn().clone(),
+            encryption_key: key,
+        });
+        let form = Form(SetupParams {
+            username,
+            password,
+            email: FreeEmail().fake(),
+        });
+
+        assert_eq!(
+            setup_handler(state, form).await.unwrap().0,
+            SetupBody::default().into()
+        );
+
+        db.async_drop().await;
+    }
+
+    #[tokio::test]
+    async fn test_setup_with_user() {
+        let db = TemporaryDatabase::new_from_env().await;
+        let (_, current_username, current_password, _, _) = create_key_user_token();
+        let (key, username, password, _, _) = create_key_user_token();
+
+        let state = State(ServerState {
+            conn: db.get_conn().clone(),
+            encryption_key: key,
+        });
+        let form = Form(SetupParams {
+            username,
+            password,
+            email: FreeEmail().fake(),
+        });
+
+        create_user(
+            &state.conn,
+            &state.encryption_key,
+            CreateUserParams {
+                username: current_username,
+                password: current_password,
+                ..Default::default()
+            },
+        )
+        .await
+        .unwrap();
+
+        assert!(matches!(
+            setup_handler(state, form).await,
+            Err(OpenSubsonicError::Forbidden { message: _ })
+        ));
+
+        db.async_drop().await;
+    }
+}