butane_test_helper: Enhancements to improve API & testing (#363)

Author: jayvdb

Cargo.lock

@@ -13,6 +13,7 @@ documentation = "https://docs.rs/butane/"
 rust-version.workspace = true
 
 [features]
+bin = ["ctrlc"]
 default = ["sqlite", "pg"]
 sqlite = ["butane_core/sqlite", "butane_core/async-adapter"]
 sqlite-bundled = ["butane_core/sqlite-bundled"]
@@ -21,15 +22,24 @@ pg = ["butane_core/pg", "tokio-postgres"]
 [dependencies]
 block-id = "0.2"
 butane_core = { workspace = true }
+ctrlc = { version = "3.4", optional = true }
 env_logger.workspace = true
 libc = "0.2"
 log.workspace = true
 maybe-async-cfg.workspace = true
 nonempty.workspace = true
+thiserror.workspace = true
 tokio-postgres = { features = ["with-geo-types-0_7"], optional = true, workspace = true }
 rand.workspace = true
 tempfile.workspace = true
 uuid = { features = ["v4"], workspace = true }
 
 [package.metadata.release]
 release = false
+
+[[bin]]
+bench = false
+name = "pg_tmp_server"
+path = "src/bin/pg_tmp_server.rs"
+required-features = ["bin", "pg"]
+test = false

butane_test_helper/Cargo.toml

@@ -0,0 +1,30 @@
+use std::sync::atomic::{AtomicBool, Ordering};
+use std::sync::Arc;
+
+use butane_test_helper::{pg_tmp_server_create, PgServerOptions};
+
+fn main() {
+    // Start the PostgreSQL server
+    let server = pg_tmp_server_create(PgServerOptions {
+        port: Some(5432),
+        ..Default::default()
+    })
+    .unwrap();
+    // Print the connection string
+    println!(
+        "Running temporary PostgreSQL server\nUnix socket dir: {}\nUser: postgres",
+        server.sockdir.path().display()
+    );
+
+    let running = Arc::new(AtomicBool::new(true));
+    let r = running.clone();
+
+    ctrlc::set_handler(move || {
+        r.store(false, Ordering::SeqCst);
+    })
+    .expect("Error setting Ctrl-C handler");
+
+    println!("Waiting for Ctrl-C...");
+    while running.load(Ordering::SeqCst) {}
+    println!("Exiting...");
+}