theseus-rs · brianheineman · May 3, 2024 · May 3, 2024 · May 3, 2024
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);