Merge pull request #50 from theseus-rs/add-command-integration-test

test: add command integration test

Author: brianheineman

Cargo.lock

@@ -7,6 +7,7 @@ use postgresql_commands::initdb::InitDbBuilder;
 use postgresql_commands::pg_ctl::Mode::{Start, Stop};
 use postgresql_commands::pg_ctl::PgCtlBuilder;
 use postgresql_commands::pg_ctl::ShutdownMode::Fast;
+use postgresql_commands::psql::PsqlBuilder;
 #[cfg(feature = "tokio")]
 use postgresql_commands::AsyncCommandExecutor;
 use postgresql_commands::CommandBuilder;
@@ -22,7 +23,6 @@ use std::str::FromStr;
 use tracing::{debug, instrument};
 
 use crate::Error::{CreateDatabaseError, DatabaseExistsError, DropDatabaseError};
-use postgresql_commands::psql::PsqlBuilder;
 
 #[cfg(feature = "bundled")]
 lazy_static::lazy_static! {

postgresql_embedded/src/postgresql.rs

@@ -0,0 +1,42 @@
+use postgresql_commands::pg_dump::PgDumpBuilder;
+use postgresql_commands::psql::PsqlBuilder;
+use postgresql_commands::{CommandBuilder, CommandExecutor};
+use postgresql_embedded::PostgreSQL;
+use std::fs;
+use tempfile::NamedTempFile;
+use test_log::test;
+
+#[test(tokio::test)]
+async fn dump_command() -> anyhow::Result<()> {
+    let mut postgresql = PostgreSQL::default();
+
+    postgresql.setup().await?;
+    postgresql.start().await?;
+    let settings = postgresql.settings();
+
+    let database_name = "test";
+    postgresql.create_database(database_name).await?;
+
+    let mut psql = PsqlBuilder::from(settings)
+        .command("CREATE TABLE person42 (id INTEGER, name VARCHAR(20))")
+        .dbname(database_name)
+        .no_psqlrc()
+        .no_align()
+        .tuples_only()
+        .build();
+    let (_stdout, _stderr) = psql.execute()?;
+
+    let temp_file = NamedTempFile::new()?;
+    let file = temp_file.as_ref();
+    let mut pgdump = PgDumpBuilder::from(settings)
+        .dbname(database_name)
+        .schema_only()
+        .file(file.to_string_lossy().to_string())
+        .build();
+    let (_stdout, _stderr) = pgdump.execute()?;
+
+    let contents = fs::read_to_string(file)?;
+    assert!(contents.contains("person42"));
+
+    Ok(())
+}