1 files changed, 78 insertions, 0 deletions
diff --git a/backend/core/tests/queries/budget_updates.rs b/backend/core/tests/queries/budget_updates.rs
new file mode 100644
index 0000000..147d06f
--- /dev/null
+++ b/backend/core/tests/queries/budget_updates.rs
@@ -0,0 +1,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);
+}