Tests budget drip transformation

1 files changed, 3 insertions, 8 deletions

diff --git a/rust/actualbudget_to_schist_transformer/tests/run.rs b/rust/actualbudget_to_schist_transformer/tests/run.rs
index 019b51a..998d8df 100644
--- a/rust/actualbudget_to_schist_transformer/tests/run.rs
+++ b/rust/actualbudget_to_schist_transformer/tests/run.rs
@@ -1,7 +1,7 @@
 pub mod common;
 
 use actualbudget_to_schist_transformer::run::run;
-use common::{get_account_matchers, get_account_transfer_matchers, get_budget_drip_matchers, get_category_matchers, get_category_transfer_matchers, get_transaction_categorisation_matchers, get_transaction_matchers, TestContext, TEST_IN_DB_URL, TEST_OUT_DB_URL};
+use common::{assert_budget_drip_matches, get_account_matchers, get_account_transfer_matchers, get_category_matchers, get_category_transfer_matchers, get_num_expected_budget_drips, get_transaction_categorisation_matchers, get_transaction_matchers, TestContext, TEST_IN_DB_URL, TEST_OUT_DB_URL};
 use diesel::{Connection, SqliteConnection};
 use schist_queries::{account_transfers::get_all_account_transfers, accounts::get_all_accounts, budget_drips::get_all_budget_drips, categories::get_all_categories, category_transfers::get_all_category_transfers, transaction_categorisations::get_all_transaction_categorisations, transactions::get_all_transactions};
 
@@ -38,14 +38,9 @@ fn given_test_database_when_run_then_exports_expected_schist_state() {
         );
     }
 
-    let budget_drip_matchers = get_budget_drip_matchers();
-    assert_eq!(budget_drip_matchers.len(), budget_drips.len());
+    assert_eq!(get_num_expected_budget_drips(), budget_drips.len());
     for budget_drip in budget_drips {
-        assert!(
-            budget_drip_matchers.iter().any(|m| m(&budget_drip)),
-            "no match found for budget_drip {:?}",
-            budget_drip,
-        );
+        assert_budget_drip_matches(&budget_drip, &categories);
     }
 
     let category_matchers = get_category_matchers();