summaryrefslogtreecommitdiff
path: root/backend/core/tests/queries/budget_updates.rs
blob: 147d06fc75325617dffe90cb0b7593c2c68ea5ff (plain)
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
use budgeting_app_core::queries;
use diesel::{Connection, SqliteConnection};

use crate::common::{fakes::{budget_update::make_fake_budget_update_builder, category::make_fake_category}, test_context::TestContext};

#[test]
fn when_insert_budget_updates_then_returns_num_budget_updates() {
    let context = TestContext::new();
    let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
    let categories = [make_fake_category(0)];
    let budget_updates = [
        make_fake_budget_update_builder(0).category_id(0).build().unwrap(),
        make_fake_budget_update_builder(1).category_id(0).build().unwrap(),
    ];
    let _ = queries::categories::insert_categories(&categories, connection);

    let num_budget_updates_inserted = queries::budget_updates::insert_budget_updates(&budget_updates, connection).unwrap();

    assert_eq!(num_budget_updates_inserted, 2);
}

#[test]
fn given_budget_updates_when_get_all_budget_updates_then_return_budget_updates() {
    let context = TestContext::new();
    let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
    let categories = [make_fake_category(0)];
    let budget_updates = [
        make_fake_budget_update_builder(0).category_id(0).build().unwrap(),
        make_fake_budget_update_builder(1).category_id(0).build().unwrap(),
    ];
    let _ = queries::categories::insert_categories(&categories, connection);
    let _ = queries::budget_updates::insert_budget_updates(&budget_updates, connection).unwrap();

    let budget_updates_in_db = queries::budget_updates::get_all_budget_updates(connection).unwrap();

    assert_eq!(budget_updates_in_db.len(), budget_updates.len());
    for budget_update in &budget_updates {
        assert!(budget_updates_in_db.iter().any(|budget_update_in_db| budget_update_in_db.eq(&budget_update)));
    }
    for budget_update_in_db in budget_updates_in_db {
        assert!(budget_updates.iter().any(|budget_update| budget_update.eq(&budget_update_in_db)));
    }
}

#[test]
fn given_budget_updates_when_delete_all_budget_updates_then_return_num_of_budget_updates() {
    let context = TestContext::new();
    let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
    let categories = [make_fake_category(0)];
    let budget_updates = [
        make_fake_budget_update_builder(0).category_id(0).build().unwrap(),
        make_fake_budget_update_builder(1).category_id(0).build().unwrap(),
    ];
    let _ = queries::categories::insert_categories(&categories, connection);
    let _ = queries::budget_updates::insert_budget_updates(&budget_updates, connection).unwrap();

    let num_budget_updates_deleted = queries::budget_updates::delete_all_budget_updates(connection).unwrap();

    assert_eq!(num_budget_updates_deleted, budget_updates.len());
}

#[test]
fn given_deleted_budget_updates_when_get_all_budget_updates_then_return_empty() {
    let context = TestContext::new();
    let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
    let categories = [make_fake_category(0)];
    let budget_updates = [
        make_fake_budget_update_builder(0).category_id(0).build().unwrap(),
        make_fake_budget_update_builder(1).category_id(0).build().unwrap(),
    ];
    let _ = queries::categories::insert_categories(&categories, connection);
    let _ = queries::budget_updates::insert_budget_updates(&budget_updates, connection).unwrap();
    let _ = queries::budget_updates::delete_all_budget_updates(connection).unwrap();

    let budget_updates_returned = queries::budget_updates::get_all_budget_updates(connection).unwrap();

    assert_eq!(budget_updates_returned.len(), 0);
}