test: add authentication tests

brianheineman · brianheineman · commit e7a21be51f17 · 2024-05-02T18:02:23.000-06:00
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/postgresql_embedded/src/postgresql.rs b/postgresql_embedded/src/postgresql.rs
@@ -338,7 +338,6 @@ impl PostgreSQL {
             self.settings.port
         );
         let psql = PsqlBuilder::from(&self.settings)
-            .program_dir(self.settings.binary_dir())
             .command(format!(
                 "SELECT 1 FROM pg_database WHERE datname='{}'",
                 database_name.as_ref()
@@ -366,7 +365,6 @@ impl PostgreSQL {
             self.settings.port
         );
         let psql = PsqlBuilder::from(&self.settings)
-            .program_dir(self.settings.binary_dir())
             .command(format!(
                 "DROP DATABASE IF EXISTS \"{}\"",
                 database_name.as_ref()
diff --git a/postgresql_embedded/tests/postgresql.rs b/postgresql_embedded/tests/postgresql.rs
@@ -1,5 +1,7 @@
 use anyhow::bail;
 use postgresql_archive::LATEST;
+use postgresql_commands::psql::PsqlBuilder;
+use postgresql_commands::CommandBuilder;
 use postgresql_embedded::{PostgreSQL, Result, Settings, Status};
 use std::fs::{remove_dir_all, remove_file};
 use test_log::test;
@@ -144,11 +146,63 @@ async fn postgres_concurrency() -> anyhow::Result<()> {
     Ok(())
 }
 
+#[test(tokio::test)]
+async fn test_authentication_success() -> Result<()> {
+    let mut postgresql = PostgreSQL::default();
+    postgresql.setup().await?;
+    postgresql.start().await?;
+
+    let mut psql = PsqlBuilder::from(postgresql.settings())
+        .command("SELECT 1")
+        .no_psqlrc()
+        .tuples_only()
+        .build();
+
+    let output = psql.output()?;
+    assert!(output.status.success());
+    Ok(())
+}
+
+#[test(tokio::test)]
+async fn test_authentication_invalid_username() -> Result<()> {
+    let mut postgresql = PostgreSQL::default();
+    postgresql.setup().await?;
+    postgresql.start().await?;
+
+    let mut psql = PsqlBuilder::from(postgresql.settings())
+        .command("SELECT 1")
+        .username("invalid")
+        .no_psqlrc()
+        .tuples_only()
+        .build();
+
+    let output = psql.output()?;
+    assert!(!output.status.success());
+    Ok(())
+}
+
+#[test(tokio::test)]
+async fn test_authentication_invalid_password() -> Result<()> {
+    let mut postgresql = PostgreSQL::default();
+    postgresql.setup().await?;
+    postgresql.start().await?;
+
+    let mut psql = PsqlBuilder::from(postgresql.settings())
+        .command("SELECT 1")
+        .pg_password("invalid")
+        .no_psqlrc()
+        .tuples_only()
+        .build();
+
+    let output = psql.output()?;
+    assert!(!output.status.success());
+    Ok(())
+}
+
 #[test(tokio::test)]
 async fn test_username_setting() -> Result<()> {
-    let username = "admin".to_string();
     let settings = Settings {
-        username,
+        username: "admin".to_string(),
         ..Default::default()
     };
     let mut postgresql = PostgreSQL::new(LATEST, settings);
