with_postgres_ready 0.1.1

Helper for running tests with a Postgres database in a Docker container that is ready to accept connections.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

use std::{future::Future, panic::UnwindSafe};

use crate::Runner;

/// Run a test with a postgres container.
/// The test will be passed a postgres connection url.
pub fn with_postgres_ready<T, Fut>(f: T)
where
    T: FnOnce(String) -> Fut + UnwindSafe,
    Fut: Future<Output = ()> + Send + 'static,
{
    Runner::new().run(f);
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test_log::test]
    fn it_works() {
        with_postgres_ready(|url| async move {
            let (client, connection) = tokio_postgres::connect(&url, tokio_postgres::NoTls)
                .await
                .unwrap();

            tokio::spawn(async move {
                if let Err(e) = connection.await {
                    eprintln!("connection error: {}", e);
                }
            });

            let rows = client
                .query("SELECT 1 + 2", &[])
                .await
                .expect("Error running query");

            assert_eq!(rows.len(), 1);

            let sum: i32 = rows[0].get(0);
            assert_eq!(sum, 3);
        });
    }
}