test improvements

Cargo.lock

@@ -696,7 +696,6 @@ dependencies = [
  "rand",
  "reqwest",
  "serde",
- "serde_json",
  "serial_test",
  "sha2",
  "tokio",

Cargo.toml

@@ -19,7 +19,6 @@ env_logger = { version = "0.10.1", default-features = false }
 futures = "0.3.29"
 rand = "0.8.5"
 serde = "1.0.192"
-serde_json = "1.0.108"
 sha2 = "0.10.8"
 tokio = { version = "1.34.0", features = ["full"] }
 tracing = "0.1.40"

tests/common.rs

@@ -12,6 +12,7 @@ use serde::ser::Serialize;
 use std::env::current_dir;
 use std::process::{Command, Stdio};
 use std::sync::Once;
+use std::thread::spawn;
 use tokio::task::JoinHandle;
 use tracing::log::LevelFilter;
 use url::Url;
@@ -19,9 +20,9 @@ use url::Url;
 pub static CLIENT: Lazy<Client> = Lazy::new(Client::new);
 
 pub struct TestData {
-    pub alpha: Instance,
-    pub beta: Instance,
-    pub gamma: Instance,
+    pub alpha: FediwikiInstance,
+    pub beta: FediwikiInstance,
+    pub gamma: FediwikiInstance,
 }
 
 impl TestData {
@@ -35,51 +36,56 @@ impl TestData {
                 .init();
         });
 
-        // initialize postgres databases in parallel because its slow
-        let (alpha_db_path, alpha_db_thread) = start_temporary_database("alpha");
-        let (beta_db_path, beta_db_thread) = start_temporary_database("beta");
-        let (gamma_db_path, gamma_db_thread) = start_temporary_database("gamma");
+        let alpha_db_path = generate_db_path("alpha");
+        let beta_db_path = generate_db_path("beta");
+        let gamma_db_path = generate_db_path("gamma");
 
-        alpha_db_thread.join().unwrap();
-        beta_db_thread.join().unwrap();
-        gamma_db_thread.join().unwrap();
+        // initialize postgres databases in parallel because its slow
+        for j in [
+            FediwikiInstance::prepare_db(alpha_db_path.clone()),
+            FediwikiInstance::prepare_db(beta_db_path.clone()),
+            FediwikiInstance::prepare_db(gamma_db_path.clone()),
+        ] {
+            j.join().unwrap();
+        }
 
         Self {
-            alpha: Instance::start(alpha_db_path, 8131),
-            beta: Instance::start(beta_db_path, 8132),
-            gamma: Instance::start(gamma_db_path, 8133),
+            alpha: FediwikiInstance::start(alpha_db_path, 8131),
+            beta: FediwikiInstance::start(beta_db_path, 8132),
+            gamma: FediwikiInstance::start(gamma_db_path, 8133),
         }
     }
 
     pub fn stop(self) -> MyResult<()> {
-        self.alpha.stop();
-        self.beta.stop();
-        self.gamma.stop();
+        for j in [self.alpha.stop(), self.beta.stop(), self.gamma.stop()] {
+            j.join().unwrap();
+        }
         Ok(())
     }
 }
 
-fn start_temporary_database(name: &'static str) -> (String, std::thread::JoinHandle<()>) {
-    let db_path = format!("{}/target/test_db/{name}", current_dir().unwrap().display());
-    let db_path_ = db_path.clone();
-    let db_thread = std::thread::spawn(move || {
+fn generate_db_path(name: &'static str) -> String {
+    format!("{}/target/test_db/{name}", current_dir().unwrap().display())
+}
+
+pub struct FediwikiInstance {
+    pub hostname: String,
+    db_path: String,
+    db_handle: JoinHandle<()>,
+}
+
+impl FediwikiInstance {
+    fn prepare_db(db_path: String) -> std::thread::JoinHandle<()> {
+        spawn(move || {
             Command::new("./tests/scripts/start_dev_db.sh")
-            .arg(&db_path_)
+                .arg(&db_path)
                 .stdout(Stdio::null())
                 .stderr(Stdio::null())
                 .output()
                 .unwrap();
-    });
-    (db_path, db_thread)
+        })
     }
 
-pub struct Instance {
-    db_path: String,
-    pub hostname: String,
-    handle: JoinHandle<()>,
-}
-
-impl Instance {
     fn start(db_path: String, port: i32) -> Self {
         let db_url = format!("postgresql://lemmy:password@/lemmy?host={db_path}");
         let hostname = format!("localhost:{port}");
@@ -90,18 +96,20 @@ impl Instance {
         Self {
             db_path,
             hostname,
-            handle,
+            db_handle: handle,
         }
     }
 
-    fn stop(self) {
-        self.handle.abort();
+    fn stop(self) -> std::thread::JoinHandle<()> {
+        self.db_handle.abort();
+        spawn(move || {
             Command::new("./tests/scripts/stop_dev_db.sh")
                 .arg(&self.db_path)
                 .stdout(Stdio::null())
                 .stderr(Stdio::null())
                 .output()
                 .unwrap();
+        })
     }
 }
 

tests/scripts/start_dev_db.sh

@@ -4,15 +4,17 @@ set -e
 export PGHOST=$1
 export PGDATA="$1/dev_pgdata"
 
-# If cluster exists, stop the server and delete the cluster
+# If cluster exists, stop the server
 if [ -d $PGDATA ]
 then
   # Prevent `stop` from failing if server already stopped
   pg_ctl restart > /dev/null
   pg_ctl stop
-  rm -rf $PGDATA
 fi
 
+# Remove any leftover data from revious run
+rm -rf $PGDATA
+
 # Create cluster
 initdb --username=postgres --auth=trust --no-instructions