Integration tests for basic CRD queries

1 files changed, 86 insertions, 0 deletions

diff --git a/backend/core/tests/queries/transaction_categorisations.rs b/backend/core/tests/queries/transaction_categorisations.rs
new file mode 100644
index 0000000..3c2171d
--- /dev/null
+++ b/backend/core/tests/queries/transaction_categorisations.rs
@@ -0,0 +1,86 @@
+use budgeting_app_core::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 insert_prerequisite_data(connection: &mut SqliteConnection) {
+    let accounts = [make_fake_account(0)];
+    let categories = [make_fake_category(0)];
+    let transactions = [
+        make_fake_transaction_builder(0).account_id(0).build().unwrap(),
+        make_fake_transaction_builder(1).account_id(0).build().unwrap(),
+    ];
+    let _ = queries::accounts::insert_accounts(&accounts, connection);
+    let _ = queries::categories::insert_categories(&categories, connection);
+    let _ = queries::transactions::insert_transactions(&transactions, connection);
+}
+
+#[test]
+fn when_insert_transaction_categorisations_then_returns_num_transaction_categorisations() {
+    let context = TestContext::new();
+    let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
+    insert_prerequisite_data(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();
+    insert_prerequisite_data(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();
+    insert_prerequisite_data(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();
+    insert_prerequisite_data(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);
+}