sqlb 0.4.0

Simple, expressive, and progressive SQL Builder for 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
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122

mod utils;

use serial_test::serial;
use sqlb::{Field, SqlxBindable};
use sqlx::{postgres::PgArguments, query::Query, Postgres};
use std::error::Error;
use utils::init_db;

// region:    Custom Type (enum)
#[derive(Eq, PartialEq, Hash, sqlx::Type, Debug, Clone)]
#[sqlx(type_name = "todo_status_enum")]
#[sqlx(rename_all = "lowercase")]
pub enum TodoStatus {
	New,
	Open,
	Done,
}

// NOTE: manual implementation, see test_rules for the macros alternative.
impl SqlxBindable for TodoStatus {
	fn bind_query<'q>(&self, query: Query<'q, Postgres, PgArguments>) -> Query<'q, sqlx::Postgres, PgArguments> {
		let query = query.bind(self.clone());
		query
	}
}

// endregion: Custom Type (enum)

// This is to test that the type above was undestood by Sqlx
#[serial]
#[tokio::test]
async fn sb_enum_direct_sqlx_ok() -> Result<(), Box<dyn Error>> {
	let db_pool = init_db().await?;

	let query = sqlx::query::<sqlx::Postgres>("INSERT INTO todo (title, status) VALUES ($1, $2)");
	let query = query.bind("test sb_enum_direct_sqlx_ok 01");
	let query = query.bind(TodoStatus::Done);
	let _r = query.execute(&db_pool).await?.rows_affected();

	Ok(())
}

#[serial]
#[tokio::test]
async fn sb_enum_direct_sqlb_field_ok() -> Result<(), Box<dyn Error>> {
	let db_pool = init_db().await?;

	// fixtures
	let title_1 = "test - sb_enum_direct_sqlb_field_ok";
	let status_1 = TodoStatus::Open;

	// DO the insert
	let data: Vec<Field> = vec![("title", title_1).into(), ("status", TodoStatus::Open).into()];
	let sb = sqlb::insert().table("todo").data(data).returning(&["id", "title", "status"]);
	let (id_1, title, status) = sb.fetch_one::<_, (i64, String, TodoStatus)>(&db_pool).await?;

	// CHECK the insert
	assert_eq!(title_1, title);
	assert_eq!(status_1, status);

	// DO the select
	let sb = sqlb::select()
		.table("todo")
		.columns(&["id", "title", "status"])
		.and_where_eq("id", id_1);
	let (id, title, status) = sb.fetch_one::<_, (i64, String, TodoStatus)>(&db_pool).await?;

	// CHECK the insert
	assert_eq!(id_1, id);
	assert_eq!(title_1, title);
	assert_eq!(status_1, status);

	Ok(())
}

#[serial]
#[tokio::test]
async fn sb_enum_option_sqlx_ok() -> Result<(), Box<dyn Error>> {
	let db_pool = init_db().await?;

	let status = Some(TodoStatus::Done);
	let query = sqlx::query::<sqlx::Postgres>("INSERT INTO todo (title, status) VALUES ($1, $2)");
	let query = query.bind("test sb_enum_option_sqlx_ok 01");
	let query = query.bind(status);
	let _r = query.execute(&db_pool).await?.rows_affected();

	Ok(())
}

#[serial]
#[tokio::test]
async fn sb_enum_option_sqlb_field_ok() -> Result<(), Box<dyn Error>> {
	let db_pool = init_db().await?;

	// fixtures
	let title_1 = "test - sb_enum_option_sqlb_field_ok";
	let status_1 = TodoStatus::Open;

	// DO the insert
	let status = Some(TodoStatus::Open);
	let data: Vec<Field> = vec![("title", title_1).into(), ("status", status).into()];
	let sb = sqlb::insert().table("todo").data(data).returning(&["id", "title", "status"]);
	let (id_1, title, status) = sb.fetch_one::<_, (i64, String, TodoStatus)>(&db_pool).await?;

	// CHECK the insert
	assert_eq!(title_1, title);
	assert_eq!(status_1, status);

	// DO the select
	let sb = sqlb::select()
		.table("todo")
		.columns(&["id", "title", "status"])
		.and_where_eq("id", id_1);
	let (id, title, status) = sb.fetch_one::<_, (i64, String, TodoStatus)>(&db_pool).await?;

	// CHECK the insert
	assert_eq!(id_1, id);
	assert_eq!(title_1, title);
	assert_eq!(status_1, status);

	Ok(())
}