1 files changed, 45 insertions, 0 deletions
diff --git a/actualbudget_queries/tests/common/get_reference_data.rs b/actualbudget_queries/tests/common/get_reference_data.rs
new file mode 100644
index 0000000..0d4f19c
--- /dev/null
+++ b/actualbudget_queries/tests/common/get_reference_data.rs
@@ -0,0 +1,45 @@
+use std::collections::HashMap;
+
+use actualbudget_models::{ActualbudgetAccount, ActualbudgetCategory};
+use actualbudget_queries::{actualbudget_accounts::get_all_actualbudget_accounts, actualbudget_categories::get_all_actualbudget_categories};
+use diesel::SqliteConnection;
+
+use super::{account_names::ACCOUNT_NAMES, category_names::CATEGORY_NAMES};
+
+pub fn get_accounts(connection: &mut SqliteConnection) -> HashMap<String, ActualbudgetAccount> {
+    let accounts = get_all_actualbudget_accounts(connection).expect("failed to get accounts");
+    let mut accounts_hash_map = HashMap::new();
+
+    for &account_name in ACCOUNT_NAMES {
+        let account = accounts
+            .clone()
+            .into_iter()
+            .find(|a| a.name.as_str() == account_name)
+            .expect(&format!("failed to get account: {}", account_name));
+        accounts_hash_map.insert(
+            String::from(account_name),
+            account,
+        );
+    }
+
+    accounts_hash_map
+}
+
+pub fn get_categories(connection: &mut SqliteConnection) -> HashMap<String, ActualbudgetCategory> {
+    let categories = get_all_actualbudget_categories(connection).expect("failed to get categories");
+    let mut categories_hash_map = HashMap::new();
+
+    for &category_name in CATEGORY_NAMES {
+        let category = categories
+            .clone()
+            .into_iter()
+            .find(|a| a.name.as_str() == category_name)
+            .expect(&format!("failed to get category: {}", category_name));
+        categories_hash_map.insert(
+            String::from(category_name),
+            category,
+        );
+    }
+
+    categories_hash_map
+}