From 10f071447594f2bdcec84a461df9ad648f71f44b Mon Sep 17 00:00:00 2001
From: Joe Carstairs <me@joeac.net>
Date: Sun, 17 Nov 2024 09:29:32 +0000
Subject: Renames app to 'schist'

---
 rust/schist_core/tests/common/test_context.rs | 43 +++++++++++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 rust/schist_core/tests/common/test_context.rs

(limited to 'rust/schist_core/tests/common/test_context.rs')

diff --git a/rust/schist_core/tests/common/test_context.rs b/rust/schist_core/tests/common/test_context.rs
new file mode 100644
index 0000000..5dcaebc
--- /dev/null
+++ b/rust/schist_core/tests/common/test_context.rs
@@ -0,0 +1,43 @@
+use std::sync::atomic::{AtomicU64, Ordering};
+
+use schist_core::{queries::clear::clear, MIGRATIONS};
+use diesel::{Connection, SqliteConnection};
+use diesel_migrations::MigrationHarness;
+
+pub struct TestContext {
+    pub db_url: String,
+}
+
+static ID_COUNTER: AtomicU64 = AtomicU64::new(0);
+
+impl TestContext {
+    pub fn new() -> Self {
+        let db_url = Self::next_id() + ".sqlite";
+
+        let connection = &mut SqliteConnection
+            ::establish(&db_url)
+            .expect("failed to connect to database");
+
+        connection
+            .run_pending_migrations(MIGRATIONS)
+            .expect("failed to run migrations");
+
+        clear(connection)
+            .expect("failed to clear database");
+
+        Self {
+            db_url,
+        }
+    }
+
+    fn next_id() -> String {
+        ID_COUNTER.fetch_add(1, Ordering::Relaxed).to_string()
+    }
+}
+
+impl Drop for TestContext {
+    fn drop(&mut self) {
+        std::fs::remove_file(&self.db_url)
+            .expect(format!("failed to delete database {}", self.db_url).as_str());
+    }
+}
-- 
cgit v1.2.3