testfixtures 0.1.2

A library for preparing test data from yaml files in Rust
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

use crate::loader::Loader;
use crate::postgresql::helper;
use chrono::{Offset, TimeZone};
use sqlx::{PgConnection, Postgres};
use std::fmt::Display;

// TODO: Complete this type.
/// An alias for [Loader](crate::loader::Loader), specialized for **PostgreSQL**.
pub(crate) type PglLoader<O, Tz> = Loader<Postgres, PgConnection, O, Tz>;

impl<O, Tz> PostgresLoader<O, Tz>
where
    O: Offset + Display + Send + Sync + 'static,
    Tz: TimeZone<Offset = O> + Sync + Send + 'static,
{
    pub async fn new<F>(options: F) -> anyhow::Result<PostgresLoader<O, Tz>>
    where
        F: FnOnce(&mut PostgresLoader<O, Tz>),
    {
        let mut loader = Self::default();
        options(&mut loader);
        loader.helper = Some(Box::new(helper::PostgreSql::default()));
        loader.build_insert_sqls();
        loader
            .helper
            .as_mut()
            .unwrap()
            .init(loader.pool.as_ref().unwrap())
            .await?;
        Ok(loader)
    }
}