Category names in AB transformer tests common module

2 files changed, 31 insertions, 1 deletions

diff --git a/rust/actualbudget_to_schist_transformer/tests/common/category_names.rs b/rust/actualbudget_to_schist_transformer/tests/common/category_names.rs
new file mode 100644
index 0000000..b51718c
--- /dev/null
+++ b/rust/actualbudget_to_schist_transformer/tests/common/category_names.rs
@@ -0,0 +1,27 @@
+pub const CATEGORY_NAME_FOOD: &str = "Food";
+pub const CATEGORY_NAME_GENERAL: &str = "General";
+pub const CATEGORY_NAME_BILLS: &str = "Bills";
+pub const CATEGORY_NAME_BILLS_FLEXIBLE: &str = "Bills (Flexible)";
+pub const CATEGORY_NAME_SAVINGS: &str = "Savings";
+pub const CATEGORY_NAME_STARTING_BALANCES: &str = "Starting Balances";
+pub const CATEGORY_NAME_INCOME: &str = "Income";
+
+pub struct CategoryNamesDict {
+    pub food: &'static str,
+    pub general: &'static str,
+    pub bills: &'static str,
+    pub bills_flexible: &'static str,
+    pub savings: &'static str,
+    pub starting_balances: &'static str,
+    pub income: &'static str,
+}
+
+pub const CATEGORY_NAMES_DICT: CategoryNamesDict = CategoryNamesDict {
+    food: CATEGORY_NAME_FOOD,
+    general: CATEGORY_NAME_GENERAL,
+    bills: CATEGORY_NAME_BILLS,
+    bills_flexible: CATEGORY_NAME_BILLS_FLEXIBLE,
+    savings: CATEGORY_NAME_SAVINGS,
+    starting_balances: CATEGORY_NAME_STARTING_BALANCES,
+    income: CATEGORY_NAME_INCOME,
+};
diff --git a/rust/actualbudget_to_schist_transformer/tests/common/mod.rs b/rust/actualbudget_to_schist_transformer/tests/common/mod.rs
index 66cc60c..d5a7d1d 100644
--- a/rust/actualbudget_to_schist_transformer/tests/common/mod.rs
+++ b/rust/actualbudget_to_schist_transformer/tests/common/mod.rs
@@ -3,6 +3,7 @@ mod account_names;
 mod account_transfer_matchers;
 mod budget_drip_matchers;
 mod category_matchers;
+mod category_names;
 mod category_transfer_matchers;
 mod db_url;
 mod test_context;
@@ -12,7 +13,9 @@ mod transaction_categorisation_matchers;
 pub use account_matchers::get_account_matchers;
 pub use account_names::ACCOUNT_NAMES_DICT;
 pub use account_transfer_matchers::get_account_transfer_matchers;
-pub use budget_drip_matchers::get_budget_drip_matchers;
+pub use budget_drip_matchers::assert_budget_drip_matches;
+pub use budget_drip_matchers::get_num_expected_budget_drips;
+pub use category_names::CATEGORY_NAMES_DICT;
 pub use category_matchers::get_category_matchers;
 pub use category_transfer_matchers::get_category_transfer_matchers;
 pub use db_url::TEST_IN_DB_URL;