From 7494db93b9262c3d8330fd11631e711a1642b8fc Mon Sep 17 00:00:00 2001
From: Joel Klinghed <the_jk@spawned.biz>
Date: Fri, 3 Jan 2025 01:28:54 +0100
Subject: Add initital tests

Also add /users endpoint.
---
 server/src/auth.rs | 20 ++++++++++++++++++++
 1 file changed, 20 insertions(+)

(limited to 'server/src/auth.rs')

diff --git a/server/src/auth.rs b/server/src/auth.rs
index f1b8f70..db3a6a0 100644
--- a/server/src/auth.rs
+++ b/server/src/auth.rs
@@ -248,8 +248,10 @@ fn unauthorized() -> Json<api_model::StatusResponse> {
     Json(STATUS_UNAUTHORIZED)
 }
 
+#[cfg(not(test))]
 async fn run_import(rocket: Rocket<Build>) -> fairing::Result {
     match Db::fetch(&rocket) {
+        // TODO: Replace with ldap
         Some(db) => match sqlx::query!("INSERT IGNORE INTO users (username) VALUES (?)", "user")
             .execute(&**db)
             .await
@@ -261,6 +263,24 @@ async fn run_import(rocket: Rocket<Build>) -> fairing::Result {
     }
 }
 
+#[cfg(test)]
+async fn run_import(rocket: Rocket<Build>) -> fairing::Result {
+    match Db::fetch(&rocket) {
+        Some(db) => match sqlx::query!(
+            "INSERT IGNORE INTO users (username) VALUES (?), (?)",
+            "user",
+            "other",
+        )
+        .execute(&**db)
+        .await
+        {
+            Ok(_) => Ok(rocket),
+            Err(_) => Err(rocket),
+        },
+        None => Err(rocket),
+    }
+}
+
 pub fn stage(basepath: &str) -> AdHoc {
     let l_basepath = basepath.to_string();
     AdHoc::on_ignite("Auth Stage", |rocket| async {
-- 
cgit v1.2.3-70-g09d2