Tests category_transfers queries

1 files changed, 40 insertions, 4 deletions

diff --git a/backend/core/tests/queries/category_transfers.rs b/backend/core/tests/queries/category_transfers.rs
index 80ec70f..0df144f 100644
--- a/backend/core/tests/queries/category_transfers.rs
+++ b/backend/core/tests/queries/category_transfers.rs
@@ -78,11 +78,47 @@ fn given_deleted_category_transfers_when_get_all_category_transfers_then_return_
 }
 
 #[test]
-fn todo() {
-    todo!("test sum_quantity_per_from_category_id");
+fn given_category_transfers_when_sum_quantity_per_from_category_id_then_returns_sum_per_from_category_id() {
+    let context = TestContext::new();
+    let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
+    let categories = [make_fake_category(0), make_fake_category(1)];
+    let category_transfers = [
+        make_fake_category_transfer_builder(0).from_category_id(0).to_category_id(1).quantity(1).build().unwrap(),
+        make_fake_category_transfer_builder(1).from_category_id(0).to_category_id(1).quantity(2).build().unwrap(),
+        make_fake_category_transfer_builder(2).from_category_id(1).to_category_id(0).quantity(4).build().unwrap(),
+    ];
+    let _ = queries::categories::insert_categories(&categories, connection);
+    let _ = queries::category_transfers::insert_category_transfers(&category_transfers, connection).unwrap();
+
+    let sums = queries::category_transfers::sum_category_transfer_quantity_per_from_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);
 }
 
 #[test]
-fn todo2() {
-    todo!("test sum_quantity_per_from_category_id");
+fn given_category_transfers_when_sum_quantity_per_to_category_id_then_returns_sum_per_to_category_id() {
+    let context = TestContext::new();
+    let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
+    let categories = [make_fake_category(0), make_fake_category(1)];
+    let category_transfers = [
+        make_fake_category_transfer_builder(0).from_category_id(0).to_category_id(1).quantity(1).build().unwrap(),
+        make_fake_category_transfer_builder(1).from_category_id(0).to_category_id(1).quantity(2).build().unwrap(),
+        make_fake_category_transfer_builder(2).from_category_id(1).to_category_id(0).quantity(4).build().unwrap(),
+    ];
+    let _ = queries::categories::insert_categories(&categories, connection);
+    let _ = queries::category_transfers::insert_category_transfers(&category_transfers, connection).unwrap();
+
+    let sums = queries::category_transfers::sum_category_transfer_quantity_per_to_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!(4, sums[0].1);
+    assert_eq!(3, sums[1].1);
 }