benzin 0.4.0

An async extension for Diesel the safe, extensible ORM and Query Builder
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

#[cfg(feature = "bb8")]
mod bb8 {
    #[tokio::test]
    async fn test_basic_insert_and_load() {
        use benzin::{
            RunQueryDsl,
            pooled_connection::{AsyncDieselConnectionManager, bb8::Pool},
        };
        use diesel::prelude::*;

        use super::super::users;

        let config = AsyncDieselConnectionManager::<benzin::TestConnection>::new("test");
        let pool = Pool::builder().max_size(1).build(config).await.unwrap();

        let conn = pool.get().await.unwrap();
        drop(conn);
        let mut conn = pool.get().await.unwrap();
        conn.expected_return_count(1);
        diesel::insert_into(users::table)
            .values(users::name.eq("John"))
            .execute(&mut *conn)
            .await
            .unwrap();
        conn.expected_load(vec![vec![String::from("John")]]);
        let all_names = users::table
            .select(users::name)
            .load::<String>(&mut *conn)
            .await
            .unwrap();
        assert_eq!(all_names, vec!["John"]);
        drop(conn);
    }
}

#[cfg(feature = "deadpool")]
mod deadpool {
    #[tokio::test]
    async fn test_basic_insert_and_load() {
        use benzin::{
            RunQueryDsl,
            pooled_connection::{AsyncDieselConnectionManager, deadpool::Pool},
        };
        use diesel::prelude::*;

        use super::super::users;

        let config = AsyncDieselConnectionManager::<benzin::TestConnection>::new("test");
        let pool = Pool::builder(config).max_size(1).build().unwrap();

        let conn = pool.get().await.unwrap();
        drop(conn);
        let mut conn = pool.get().await.unwrap();
        conn.expected_return_count(1);
        diesel::insert_into(users::table)
            .values(users::name.eq("John"))
            .execute(&mut *conn)
            .await
            .unwrap();
        conn.expected_load(vec![vec![String::from("John")]]);
        let all_names = users::table
            .select(users::name)
            .load::<String>(&mut *conn)
            .await
            .unwrap();
        assert_eq!(all_names, vec!["John"]);
        drop(conn);
    }
}