Put database prep somewhere tests can call it.

1 files changed, 18 insertions, 0 deletions

diff --git a/src/repo/pool.rs b/src/repo/pool.rs
new file mode 100644
index 0000000..b4aa6fc
--- /dev/null
+++ b/src/repo/pool.rs
@@ -0,0 +1,18 @@
+use std::str::FromStr;
+
+use sqlx::sqlite::{SqliteConnectOptions, SqlitePool, SqlitePoolOptions};
+
+pub async fn prepare(url: &str) -> sqlx::Result<SqlitePool> {
+    let pool = create(url).await?;
+    sqlx::migrate!().run(&pool).await?;
+    Ok(pool)
+}
+
+async fn create(database_url: &str) -> sqlx::Result<SqlitePool> {
+    let options = SqliteConnectOptions::from_str(database_url)?
+        .create_if_missing(true)
+        .optimize_on_close(true, /* analysis_limit */ None);
+
+    let pool = SqlitePoolOptions::new().connect_with(options).await?;
+    Ok(pool)
+}