1 files changed, 20 insertions, 0 deletions
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 {