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
123
124
125
126
127
|
use budgeting_app_core::{models::{account::Account, category::Category, transaction::Transaction}, queries};
use diesel::{Connection, SqliteConnection};
use crate::common::{fakes::{account::make_fake_account, category::make_fake_category, transaction::make_fake_transaction_builder, transaction_categorisation::make_fake_transaction_categorisation_builder}, test_context::TestContext};
fn given_accounts(connection: &mut SqliteConnection) -> Vec<Account> {
let accounts = vec![make_fake_account(0)];
let _ = queries::accounts::insert_accounts(&accounts, connection);
accounts
}
fn given_categories(connection: &mut SqliteConnection) -> Vec<Category> {
let categories = vec![make_fake_category(0), make_fake_category(1)];
let _ = queries::categories::insert_categories(&categories, connection);
categories
}
fn given_transactions(connection: &mut SqliteConnection) -> Vec<Transaction> {
let transactions = vec![
make_fake_transaction_builder(0).account_id(0).build().unwrap(),
make_fake_transaction_builder(1).account_id(0).build().unwrap(),
];
let _ = queries::transactions::insert_transactions(&transactions, connection);
transactions
}
#[test]
fn when_insert_transaction_categorisations_then_returns_num_transaction_categorisations() {
let context = TestContext::new();
let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
given_accounts(connection);
given_categories(connection);
given_transactions(connection);
let transaction_categorisations = [
make_fake_transaction_categorisation_builder(0).category_id(0).build().unwrap(),
make_fake_transaction_categorisation_builder(1).category_id(0).build().unwrap(),
];
let num_transaction_categorisations_inserted = queries::transaction_categorisations::insert_transaction_categorisations(&transaction_categorisations, connection).unwrap();
assert_eq!(num_transaction_categorisations_inserted, 2);
}
#[test]
fn given_transaction_categorisations_when_get_all_transaction_categorisations_then_return_transaction_categorisations() {
let context = TestContext::new();
let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
given_accounts(connection);
given_categories(connection);
given_transactions(connection);
let transaction_categorisations = [
make_fake_transaction_categorisation_builder(0).category_id(0).build().unwrap(),
make_fake_transaction_categorisation_builder(1).category_id(0).build().unwrap(),
];
let _ = queries::transaction_categorisations::insert_transaction_categorisations(&transaction_categorisations, connection);
let transaction_categorisations_in_db = queries::transaction_categorisations::get_all_transaction_categorisations(connection).unwrap();
assert_eq!(transaction_categorisations_in_db.len(), transaction_categorisations.len());
for transaction_categorisation in &transaction_categorisations {
assert!(transaction_categorisations_in_db.iter().any(|transaction_categorisation_in_db| transaction_categorisation_in_db.eq(&transaction_categorisation)));
}
for transaction_categorisation_in_db in transaction_categorisations_in_db {
assert!(transaction_categorisations.iter().any(|transaction_categorisation| transaction_categorisation.eq(&transaction_categorisation_in_db)));
}
}
#[test]
fn given_transaction_categorisations_when_delete_all_transaction_categorisations_then_return_num_of_transaction_categorisations() {
let context = TestContext::new();
let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
given_accounts(connection);
given_categories(connection);
given_transactions(connection);
let transaction_categorisations = [
make_fake_transaction_categorisation_builder(0).category_id(0).build().unwrap(),
make_fake_transaction_categorisation_builder(1).category_id(0).build().unwrap(),
];
let _ = queries::transaction_categorisations::insert_transaction_categorisations(&transaction_categorisations, connection);
let num_transaction_categorisations_deleted = queries::transaction_categorisations::delete_all_transaction_categorisations(connection).unwrap();
assert_eq!(num_transaction_categorisations_deleted, transaction_categorisations.len());
}
#[test]
fn given_deleted_transaction_categorisations_when_get_all_transaction_categorisations_then_return_empty() {
let context = TestContext::new();
let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
given_accounts(connection);
given_categories(connection);
given_transactions(connection);
let transaction_categorisations = [
make_fake_transaction_categorisation_builder(0).category_id(0).build().unwrap(),
make_fake_transaction_categorisation_builder(1).category_id(0).build().unwrap(),
];
let _ = queries::transaction_categorisations::insert_transaction_categorisations(&transaction_categorisations, connection);
let _ = queries::transaction_categorisations::delete_all_transaction_categorisations(connection);
let transaction_categorisations_returned = queries::transaction_categorisations::get_all_transaction_categorisations(connection).unwrap();
assert_eq!(transaction_categorisations_returned.len(), 0);
}
#[test]
fn given_transaction_categorisations_when_sum_quantity_per_category_id_then_returns_sum_quantity_per_category_id() {
let context = TestContext::new();
let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
given_accounts(connection);
let categories = given_categories(connection);
given_transactions(connection);
let transaction_categorisations = [
make_fake_transaction_categorisation_builder(0).category_id(0).quantity(1).build().unwrap(),
make_fake_transaction_categorisation_builder(1).category_id(0).quantity(2).build().unwrap(),
make_fake_transaction_categorisation_builder(2).category_id(1).quantity(4).build().unwrap(),
];
let _ = queries::transaction_categorisations::insert_transaction_categorisations(&transaction_categorisations, connection);
let sums = queries::transaction_categorisations::sum_transaction_categorisation_quantity_per_category_id(connection);
assert!(sums.is_ok());
let mut sums = sums.unwrap();
sums.sort_by_key(|sum| sum.0);
assert_eq!(categories.len(), sums.len());
assert_eq!(3, sums[0].1);
assert_eq!(4, sums[1].1);
}
|
