1 files changed, 130 insertions, 0 deletions
diff --git a/schist_queries/tests/transaction_categorisations.rs b/schist_queries/tests/transaction_categorisations.rs
new file mode 100644
index 0000000..13864b4
--- /dev/null
+++ b/schist_queries/tests/transaction_categorisations.rs
@@ -0,0 +1,130 @@
+mod common;
+
+use common::test_context::TestContext;
+use diesel::{Connection, SqliteConnection};
+use schist_fakes::{account::make_fake_account, category::make_fake_category, transaction::make_fake_transaction_builder, transaction_categorisation::make_fake_transaction_categorisation_builder};
+use schist_models::{account::Account, category::Category, transaction::Transaction};
+use schist_queries as queries;
+
+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);
+}