fix?: use test db in tests

Author: juleswritescode

Cargo.lock

@@ -19,8 +19,10 @@ serde.workspace      = true
 serde_json.workspace = true
 pg_diagnostics.workspace = true
 pg_console.workspace = true
-
 sqlx.workspace = true
 
+[dev-dependencies]
+pg_test_utils.workspace = true
+
 [lib]
 doctest = false

crates/pg_schema_cache/Cargo.toml

@@ -75,17 +75,16 @@ pub trait SchemaCacheItem {
 
 #[cfg(test)]
 mod tests {
-    use sqlx::PgPool;
+    use async_std::task::block_on;
+    use pg_test_utils::test_database::get_new_test_db;
 
     use crate::SchemaCache;
 
     #[test]
     fn test_schema_cache() {
-        let conn_string = std::env::var("DATABASE_URL").unwrap();
+        let test_db = block_on(get_new_test_db());
 
-        let pool = async_std::task::block_on(PgPool::connect(conn_string.as_str())).unwrap();
-
-        async_std::task::block_on(SchemaCache::load(&pool)).expect("Couldn't load Schema Cache");
+        block_on(SchemaCache::load(&test_db)).expect("Couldn't load Schema Cache");
 
         assert!(true);
     }

crates/pg_schema_cache/src/schema_cache.rs

@@ -24,6 +24,7 @@ async-std = "1.12.0"
 
 
 [dev-dependencies]
+pg_test_utils.workspace = true
 
 [lib]
 doctest = false

crates/pg_typecheck/Cargo.toml

@@ -85,23 +85,21 @@ pub async fn check_sql<'a>(params: TypecheckerParams<'a>) -> Vec<TypeError> {
 #[cfg(test)]
 mod tests {
     use async_std::task::block_on;
-    use sqlx::PgPool;
+    use pg_test_utils::test_database::get_new_test_db;
 
     use crate::{check_sql, TypecheckerParams};
 
     #[test]
-    fn test_check_sql() {
+    fn test_basic_type() {
         let input = "select id, unknown from contact;";
 
-        let conn_string = std::env::var("DATABASE_URL").unwrap();
-
-        let pool = block_on(PgPool::connect(conn_string.as_str())).unwrap();
+        let test_db = block_on(get_new_test_db());
 
         let root = pg_query_ext::parse(input).unwrap();
         let ast = pg_syntax::parse_syntax(input, &root).ast;
 
         let errs = block_on(check_sql(TypecheckerParams {
-            conn: &pool,
+            conn: &test_db,
             sql: input,
             ast: &root,
             enriched_ast: Some(&ast),
@@ -111,6 +109,6 @@ mod tests {
 
         let e = &errs[0];
 
-        assert_eq!(&input[e.range.unwrap()], "unknown");
+        assert_eq!(&input[e.range.unwrap()], "contact");
     }
 }