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
|
use anyhow::Result;
use budgeting_app_core::models::date_utc::DateUtc;
use budgeting_app_core::traits::nowlike::Nowlike;
use diesel::{sql_query, Connection, ExpressionMethods, QueryDsl, RunQueryDsl, Selectable, SelectableHelper, SqliteConnection};
use diesel::prelude::{Insertable, Queryable};
use crate::common::test_context::TestContext;
diesel::table! {
date_utc_test (id) {
id -> Integer,
date_utc -> Text,
}
}
#[derive(Insertable, Queryable, Selectable, Clone)]
#[diesel(table_name = date_utc_test)]
pub struct DateUtcTest {
pub id: i32,
pub date_utc: DateUtc,
}
fn new_test_context() -> Result<TestContext> {
let test_context = TestContext::new();
let connection = &mut SqliteConnection::establish(&test_context.db_url)?;
let query = "CREATE TABLE date_utc_test(id INTEGER PRIMARY KEY, date_utc TEXT);";
sql_query(query).execute(connection)?;
Ok(test_context)
}
fn insert_date_utc(id: i32, date_utc: &DateUtc, connection: &mut SqliteConnection) -> Result<usize> {
let model = DateUtcTest {
id,
date_utc: date_utc.clone(),
};
let num_rows_inserted = diesel::insert_into(date_utc_test::dsl::date_utc_test)
.values(&[model])
.execute(connection)?;
Ok(num_rows_inserted)
}
fn get_date_utc(id: i32, connection: &mut SqliteConnection) -> Result<DateUtcTest> {
let date_utc: Vec<DateUtcTest> = date_utc_test::dsl::date_utc_test
.select(DateUtcTest::as_select())
.filter(date_utc_test::id.eq(id))
.load(connection)?;
Ok(date_utc[0].clone())
}
#[test]
fn when_insert_date_utc_and_select_then_returns_original_value() -> Result<()> {
let date_utc = DateUtc::now();
let test_context = new_test_context()?;
let connection = &mut SqliteConnection::establish(&test_context.db_url)?;
let num_rows_inserted = insert_date_utc(0, &date_utc, connection)?;
assert_eq!(num_rows_inserted, 1);
let date_utc_returned = get_date_utc(0, connection)?;
assert_eq!(date_utc_returned.date_utc, date_utc);
Ok(())
}
|
