diff options
Diffstat (limited to 'rust/schist_queries')
21 files changed, 1878 insertions, 0 deletions
diff --git a/rust/schist_queries/Cargo.toml b/rust/schist_queries/Cargo.toml new file mode 100644 index 0000000..9822847 --- /dev/null +++ b/rust/schist_queries/Cargo.toml @@ -0,0 +1,18 @@ +[package] +name = "schist_queries" +version = "0.1.0" +edition = "2021" + +[dependencies] +anyhow = { workspace = true } +chrono = { workspace = true, features = ["serde"] } +diesel = { workspace = true, features = ["sqlite"] } +schist_models = { path = "../schist_models" } +schist_schema = { path = "../schist_schema" } +schist_traits = { path = "../schist_traits" } +serde = { workspace = true, features = ["derive"] } + +[dev-dependencies] +diesel_migrations = { workspace = true } +libsqlite3-sys = { workspace = true, features = ["bundled"] } +schist_fakes = { path = "../schist_fakes" } diff --git a/rust/schist_queries/src/accounts.rs b/rust/schist_queries/src/accounts.rs new file mode 100644 index 0000000..e8a4f53 --- /dev/null +++ b/rust/schist_queries/src/accounts.rs @@ -0,0 +1,41 @@ +use anyhow::{Context, Result}; +use diesel::{QueryDsl, RunQueryDsl, SelectableHelper, SqliteConnection}; + +use schist_models::account::Account; +use schist_schema::schema::accounts::dsl::accounts as accounts_table; + +pub fn delete_all_accounts(connection: &mut SqliteConnection) -> Result<usize> { + let num_rows_deleted = diesel::delete(accounts_table) + .execute(connection) + .with_context(|| "failed to delete all accounts")?; + Ok(num_rows_deleted) +} + +pub fn get_all_accounts(connection: &mut SqliteConnection) -> Result<Vec<Account>> { + let all_accounts = accounts_table.select(Account::as_select()) + .load(connection) + .with_context(|| "failed to get all accounts")?; + Ok(all_accounts) +} + +pub fn insert_accounts( + accounts: &[Account], + connection: &mut SqliteConnection, +) -> Result<usize> { + let num_accounts_inserted = diesel::insert_into(accounts_table) + .values(accounts) + .execute(connection) + .with_context(|| insert_err_msg(accounts))?; + Ok(num_accounts_inserted) +} + +fn insert_err_msg(accounts: &[Account]) -> String { + format!( + "failed to insert accounts: [{}]", + accounts + .iter() + .map(|a| format!("\"{}\": \"{}\"", a.id, a.name)) + .collect::<Vec<String>>() + .join(", ") + ) +} diff --git a/rust/schist_queries/src/budget_updates.rs b/rust/schist_queries/src/budget_updates.rs new file mode 100644 index 0000000..07d4808 --- /dev/null +++ b/rust/schist_queries/src/budget_updates.rs @@ -0,0 +1,74 @@ +use anyhow::{Context, Result}; +use diesel::{dsl::max, ExpressionMethods, OptionalExtension, QueryDsl, RunQueryDsl, SelectableHelper, SqliteConnection}; +use schist_models::{budget_update::BudgetUpdate, date_utc::DateUtc}; +use schist_schema::schema::budget_updates::{ + self as budget_updates_schema, dsl::budget_updates as budget_updates_table, +}; +use schist_traits::nowlike::Nowlike; + +pub fn delete_all_budget_updates(connection: &mut SqliteConnection) -> Result<usize> { + let num_rows_deleted = diesel::delete(budget_updates_table) + .execute(connection) + .with_context(|| "failed to delete all budget updates")?; + Ok(num_rows_deleted) +} + +pub fn get_all_budget_updates( + connection: &mut SqliteConnection, +) -> Result<Vec<BudgetUpdate>> { + let all_budget_updates = budget_updates_table + .select(BudgetUpdate::as_select()) + .load::<BudgetUpdate>(connection) + .with_context(|| "failed to get all budget updates")?; + Ok(all_budget_updates) +} + +pub fn get_first_budget_update_with_date_and_category_id( + date: &DateUtc, + category_id: i32, + connection: &mut SqliteConnection, +) -> Result<Option<BudgetUpdate>> { + let first_budget_update = budget_updates_table + .select(BudgetUpdate::as_select()) + .filter(budget_updates_schema::date.eq(date.to_string())) + .filter(budget_updates_schema::category_id.eq(category_id)) + .first(connection) + .optional() + .with_context(|| format!("failed to get first budget update with date {}", date.to_string()))?; + Ok(first_budget_update) +} + +pub fn get_most_recent_budget_update_date_with_category_id( + category_id: i32, + connection: &mut SqliteConnection, +) -> Result<Option<DateUtc>> { + let most_recent_date = budget_updates_table + .select(max(budget_updates_schema::date)) + .filter(budget_updates_schema::category_id.eq(category_id)) + .filter(budget_updates_schema::date.le(DateUtc::now())) + .first(connection) + .with_context(|| format!("failed to get most recent budget update date with category ID {}", category_id))?; + Ok(most_recent_date) +} + +pub fn insert_budget_updates( + budget_updates: &[BudgetUpdate], + connection: &mut SqliteConnection, +) -> Result<usize> { + let num_rows_inserted = diesel::insert_into(budget_updates_table) + .values(budget_updates) + .execute(connection) + .with_context(|| insert_err_msg(budget_updates))?; + Ok(num_rows_inserted) +} + +fn insert_err_msg(budget_updates: &[BudgetUpdate]) -> String { + format!( + "failed to insert budget updates: [{}]", + budget_updates + .iter() + .map(|a| a.id.to_string()) + .collect::<Vec<String>>() + .join(", ") + ) +} diff --git a/rust/schist_queries/src/categories.rs b/rust/schist_queries/src/categories.rs new file mode 100644 index 0000000..f54373c --- /dev/null +++ b/rust/schist_queries/src/categories.rs @@ -0,0 +1,155 @@ +use anyhow::{Context, Result}; +use diesel::{QueryDsl, RunQueryDsl, SelectableHelper, SqliteConnection}; +use schist_models::{budget::Budget, category::Category, category_balance::CategoryBalance, category_budget::CategoryBudget, date_utc::DateUtc}; +use schist_schema::schema::categories::{self as categories_schema, dsl::categories as categories_table}; + +use crate::{ + budget_updates::{ + get_all_budget_updates, + get_first_budget_update_with_date_and_category_id, + get_most_recent_budget_update_date_with_category_id, + }, + category_transfers::{ + sum_category_transfer_quantity_per_from_category_id, + sum_category_transfer_quantity_per_to_category_id, + }, + transaction_categorisations::sum_transaction_categorisation_quantity_per_category_id, + utils::calculate_category_balance, +}; + +pub fn delete_all_categories(connection: &mut SqliteConnection) -> Result<usize> { + let num_rows_deleted = diesel::delete(categories_table) + .execute(connection) + .with_context(|| "failed to delete all categories")?; + Ok(num_rows_deleted) +} + +pub fn get_all_categories(connection: &mut SqliteConnection) -> Result<Vec<Category>> { + let all_categories = categories_table + .select(Category::as_select()) + .load(connection) + .with_context(|| "failed to get all categories")?; + Ok(all_categories) +} + +pub fn get_all_category_balances( + connection: &mut SqliteConnection, +) -> Result<Vec<CategoryBalance>> { + get_all_category_balances_without_context(connection) + .with_context(|| "failed to get all category balances") +} + +fn get_all_category_balances_without_context( + connection: &mut SqliteConnection, +) -> Result<Vec<CategoryBalance>> { + let categories = get_all_categories(connection)?; + let transaction_sums = sum_transaction_categorisation_quantity_per_category_id(connection)?; + let category_transfer_from_sums = + sum_category_transfer_quantity_per_from_category_id(connection)?; + let category_transfer_to_sums = sum_category_transfer_quantity_per_to_category_id(connection)?; + let budget_updates = get_all_budget_updates(connection)?; + + let mut balances = Vec::<CategoryBalance>::new(); + for category in categories { + let balance = calculate_category_balance( + &category, + &transaction_sums, + &category_transfer_from_sums, + &category_transfer_to_sums, + &budget_updates, + )?; + + balances.push(CategoryBalance { + category_id: category.id, + balance: balance.try_into().unwrap(), + }) + } + + Ok(balances) +} + +pub fn get_all_category_ids(connection: &mut SqliteConnection) -> Result<Vec<i32>> { + let all_category_ids = categories_table + .select(categories_schema::id) + .load(connection) + .with_context(|| "failed to get all category IDs")?; + Ok(all_category_ids) +} + +pub fn get_most_recent_category_budget_per_category_id( + connection: &mut SqliteConnection, +) -> Result<Vec<CategoryBudget>> { + get_most_recent_category_budget_per_category_id_without_context(connection) + .with_context(|| "failed to get most recent category budget per category ID") +} + +fn get_most_recent_category_budget_per_category_id_without_context( + connection: &mut SqliteConnection, +) -> Result<Vec<CategoryBudget>> { + let category_ids: Vec<i32> = get_all_category_ids(connection)?; + + let mut budgets = Vec::<CategoryBudget>::new(); + for category_id in category_ids { + let most_recent_budget_update_date = + get_most_recent_budget_update_date_with_category_id(category_id, connection)?; + + budgets.push(match most_recent_budget_update_date { + None => CategoryBudget { + category_id, + budget: Option::None, + }, + Some(most_recent_budget_update_date) => + get_most_recent_category_budget_with_date_and_category_id( + &most_recent_budget_update_date, + category_id, + connection, + )?, + }) + } + + Ok(budgets) +} + +fn get_most_recent_category_budget_with_date_and_category_id( + most_recent_budget_update_date: &DateUtc, + category_id: i32, + connection: &mut SqliteConnection +) -> Result<CategoryBudget> { + let most_recent_budget_update = get_first_budget_update_with_date_and_category_id( + &most_recent_budget_update_date, + category_id, + connection, + )?; + + let budget = match most_recent_budget_update { + None => None, + Some(most_recent_budget_update) => Some(Budget::from(&most_recent_budget_update)), + }; + + Ok(CategoryBudget { + category_id, + budget, + }) +} + +pub fn insert_categories( + categories: &[Category], + connection: &mut SqliteConnection, +) -> Result<usize> { + let num_categories_inserted = diesel::insert_into(categories_table) + .values(categories) + .execute(connection) + .with_context(|| insert_err_msg(categories))?; + Ok(num_categories_inserted) +} + +fn insert_err_msg(categories: &[Category]) -> String { + format!( + "failed to insert categories: [{}]", + categories + .iter() + .map(|c| format!("\"{}\": \"{}\"", c.id, c.name)) + .collect::<Vec<String>>() + .join(", ") + ) +} diff --git a/rust/schist_queries/src/category_transfers.rs b/rust/schist_queries/src/category_transfers.rs new file mode 100644 index 0000000..33a53ea --- /dev/null +++ b/rust/schist_queries/src/category_transfers.rs @@ -0,0 +1,74 @@ +use anyhow::{Context, Result}; +use diesel::{dsl::sum, QueryDsl, RunQueryDsl, SelectableHelper, SqliteConnection}; +use schist_schema::schema::category_transfers::{ + self as category_transfers_schema, + dsl::category_transfers as category_transfers_table, +}; +use schist_models::category_transfer::CategoryTransfer; + +pub fn delete_all_category_transfers(connection: &mut SqliteConnection) -> Result<usize> { + let num_rows_deleted = diesel::delete(category_transfers_table) + .execute(connection) + .with_context(|| "failed to delete all category transfers")?; + Ok(num_rows_deleted) +} + +pub fn get_all_category_transfers(connection: &mut SqliteConnection) -> Result<Vec<CategoryTransfer>> { + let all_category_transfers = category_transfers_table + .select(CategoryTransfer::as_select()) + .load(connection) + .with_context(|| "failed to get all category transfers")?; + Ok(all_category_transfers) +} + +pub fn insert_category_transfers( + category_transfers: &[CategoryTransfer], + connection: &mut SqliteConnection, +) -> Result<usize> { + let num_rows_inserted = diesel::insert_into(category_transfers_table) + .values(category_transfers) + .execute(connection) + .with_context(|| insert_err_msg(&category_transfers))?; + Ok(num_rows_inserted) +} + +fn insert_err_msg(category_transfers: &[CategoryTransfer]) -> String { + format!( + "failed to insert category transfers: [{}]", + category_transfers + .iter() + .map(|ct| ct.id.to_string()) + .collect::<Vec<String>>() + .join(", ") + ) +} + +pub fn sum_category_transfer_quantity_per_from_category_id( + connection: &mut SqliteConnection, +) -> Result<Vec<(i32, i64)>> { + let sum = category_transfers_table + .group_by(category_transfers_schema::from_category_id) + .select(( + category_transfers_schema::from_category_id, + sum(category_transfers_schema::quantity), + )) + .load::<(i32, Option<i64>)>(connection) + .map(|result| result.iter().map(|sum| (sum.0, sum.1.unwrap_or(0))).collect()) + .with_context(|| "failed to sum category transfer quantity per from category ID")?; + Ok(sum) +} + +pub fn sum_category_transfer_quantity_per_to_category_id( + connection: &mut SqliteConnection, +) -> Result<Vec<(i32, i64)>> { + let sum = category_transfers_table + .group_by(category_transfers_schema::to_category_id) + .select(( + category_transfers_schema::to_category_id, + sum(category_transfers_schema::quantity), + )) + .load::<(i32, Option<i64>)>(connection) + .map(|vec| vec.iter().map(|(id, sum)| (*id, sum.unwrap_or(0))).collect()) + .with_context(|| "failed to sum category transfer quantity per to category ID")?; + Ok(sum) +} diff --git a/rust/schist_queries/src/clear.rs b/rust/schist_queries/src/clear.rs new file mode 100644 index 0000000..c823bcc --- /dev/null +++ b/rust/schist_queries/src/clear.rs @@ -0,0 +1,21 @@ +use anyhow::Result; +use diesel::SqliteConnection; + +use crate::{ + accounts::delete_all_accounts, + budget_updates::delete_all_budget_updates, + categories::delete_all_categories, + category_transfers::delete_all_category_transfers, + transactions::delete_all_transactions, + transaction_categorisations::delete_all_transaction_categorisations, +}; + +pub fn clear(connection: &mut SqliteConnection) -> Result<()> { + delete_all_accounts(connection)?; + delete_all_budget_updates(connection)?; + delete_all_categories(connection)?; + delete_all_category_transfers(connection)?; + delete_all_transactions(connection)?; + delete_all_transaction_categorisations(connection)?; + Ok(()) +} diff --git a/rust/schist_queries/src/lib.rs b/rust/schist_queries/src/lib.rs new file mode 100644 index 0000000..3cae464 --- /dev/null +++ b/rust/schist_queries/src/lib.rs @@ -0,0 +1,9 @@ +pub mod accounts; +pub mod budget_updates; +pub mod categories; +pub mod category_transfers; +pub mod clear; +pub mod transactions; +pub mod transaction_categorisations; + +mod utils; diff --git a/rust/schist_queries/src/transaction_categorisations.rs b/rust/schist_queries/src/transaction_categorisations.rs new file mode 100644 index 0000000..320782d --- /dev/null +++ b/rust/schist_queries/src/transaction_categorisations.rs @@ -0,0 +1,59 @@ +use anyhow::{Context, Result}; +use diesel::{dsl::sum, QueryDsl, RunQueryDsl, SelectableHelper, SqliteConnection}; +use schist_models::transaction_categorisation::TransactionCategorisation; +use schist_schema::schema::transaction_categorisations::{ + self as transaction_categorisations_schema, + dsl::transaction_categorisations as transaction_categorisations_table, +}; + +pub fn delete_all_transaction_categorisations(connection: &mut SqliteConnection) -> Result<usize> { + let num_rows_deleted = diesel::delete(transaction_categorisations_table) + .execute(connection) + .with_context(|| "failed to delete all transaction categorisations")?; + Ok(num_rows_deleted) +} + +pub fn get_all_transaction_categorisations(connection: &mut SqliteConnection) -> Result<Vec<TransactionCategorisation>> { + let all_transaction_categorisations = transaction_categorisations_table + .select(TransactionCategorisation::as_select()) + .load(connection) + .with_context(|| "failed to get all transaction categorisations")?; + Ok(all_transaction_categorisations) +} + +pub fn insert_transaction_categorisations( + transaction_categorisations: &[TransactionCategorisation], + connection: &mut SqliteConnection, +) -> Result<usize> { + let num_rows_inserted = diesel::insert_into(transaction_categorisations_table) + .values(transaction_categorisations) + .execute(connection) + .with_context(|| insert_err_msg(&transaction_categorisations))?; + Ok(num_rows_inserted) +} + +fn insert_err_msg(transaction_categorisations: &[TransactionCategorisation]) -> String { + format!( + "failed to insert transaction categorisations: [{}]", + transaction_categorisations + .iter() + .map(|tc| tc.id.to_string()) + .collect::<Vec<String>>() + .join(", ") + ) +} + +pub fn sum_transaction_categorisation_quantity_per_category_id( + connection: &mut SqliteConnection, +) -> Result<Vec<(i32, i64)>> { + let sum = transaction_categorisations_table + .group_by(transaction_categorisations_schema::category_id) + .select(( + transaction_categorisations_schema::category_id, + sum(transaction_categorisations_schema::quantity), + )) + .load::<(i32, Option<i64>)>(connection) + .map(|vec| vec.iter().map(|(category_id, sum)| (*category_id, sum.unwrap_or(0))).collect()) + .with_context(|| "failed to sum transaction categorisation quantity per category ID")?; + Ok(sum) +} diff --git a/rust/schist_queries/src/transactions.rs b/rust/schist_queries/src/transactions.rs new file mode 100644 index 0000000..70b14e7 --- /dev/null +++ b/rust/schist_queries/src/transactions.rs @@ -0,0 +1,41 @@ +use anyhow::{Context, Result}; +use diesel::{QueryDsl, RunQueryDsl, SelectableHelper, SqliteConnection}; +use schist_models::transaction::Transaction; +use schist_schema::schema::transactions::dsl::transactions as transactions_table; + +pub fn delete_all_transactions(connection: &mut SqliteConnection) -> Result<usize> { + let num_rows_deleted = diesel::delete(transactions_table) + .execute(connection) + .with_context(|| "failed to delete all transactions")?; + Ok(num_rows_deleted) +} + +pub fn get_all_transactions(connection: &mut SqliteConnection) -> Result<Vec<Transaction>> { + let all_transactions = transactions_table + .select(Transaction::as_select()) + .load(connection) + .with_context(|| "failed to get all category transfers")?; + Ok(all_transactions) +} + +pub fn insert_transactions( + transactions: &[Transaction], + connection: &mut SqliteConnection, +) -> Result<usize> { + let num_rows_inserted = diesel::insert_into(transactions_table) + .values(transactions) + .execute(connection) + .with_context(|| insert_err_msg(&transactions))?; + Ok(num_rows_inserted) +} + +fn insert_err_msg(transactions: &[Transaction]) -> String { + format!( + "failed to insert transactions: [{}]", + transactions + .iter() + .map(|t| t.id.to_string()) + .collect::<Vec<String>>() + .join(", ") + ) +} diff --git a/rust/schist_queries/src/utils/calculate_budgets_accrual.rs b/rust/schist_queries/src/utils/calculate_budgets_accrual.rs new file mode 100644 index 0000000..1d4ed02 --- /dev/null +++ b/rust/schist_queries/src/utils/calculate_budgets_accrual.rs @@ -0,0 +1,156 @@ +use anyhow::Result; +use schist_models::{ + budget_update::BudgetUpdate, category::Category, date_utc::DateUtc, +}; +use schist_traits::nowlike::Nowlike; + +pub fn calculate_budgets_accrual(category: &Category, budget_updates: &[BudgetUpdate]) -> Result<i64> { + let mut budget_updates = budget_updates + .iter() + .filter(|bu| bu.category_id == category.id) + .collect::<Vec<&BudgetUpdate>>(); + budget_updates.sort_by_key(|bu| bu.date.clone()); + budget_updates.reverse(); + + let mut budgets_accrual = 0.0_f64; + let mut calculated_back_to_date = DateUtc::now(); + + for budget_update in budget_updates { + let days_on_this_budget = calculated_back_to_date.day_diff(&budget_update.date); + if days_on_this_budget == 0_u32 { + return Err(anyhow::Error::msg(format!( + "failed to calculate budgets accrual since two budget updates had the same date ({})", + budget_update.date, + ))); + } + let budget_per_day = budget_update.new_budget as f64 / budget_update.new_period as f64; + budgets_accrual += days_on_this_budget as f64 * budget_per_day; + calculated_back_to_date = budget_update.date.clone(); + } + + Ok(budgets_accrual.floor() as i64) +} + +#[cfg(test)] +mod test { + use chrono::{Days, Utc}; + use schist_models::{budget_update::BudgetUpdate, category::Category, date_utc::DateUtc}; + use schist_traits::nowlike::Nowlike; + + use super::calculate_budgets_accrual; + + #[test] + fn when_no_budget_updates_then_return_zero() { + let category = Category { + id: 0, + name: String::from("Groceries"), + }; + let budget_updates = []; + + let result = calculate_budgets_accrual(&category, &budget_updates); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 0_i64); + } + + #[test] + fn when_one_budget_update_for_another_category_then_return_zero() { + let category = Category { + id: 0, + name: String::from("Groceries"), + }; + let budget_updates = [BudgetUpdate { + id: 0, + category_id: 1, + date: DateUtc::now(), + new_budget: 100, + new_period: 1, + }]; + + let result = calculate_budgets_accrual(&category, &budget_updates); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 0_i64); + } + + #[test] + fn when_one_budget_update_then_return_quantity_times_days_in_past() { + let category = Category { + id: 0, + name: String::from("Groceries"), + }; + let budget_updates = [BudgetUpdate { + id: 0, + category_id: 0, + date: DateUtc::from(Utc::now().checked_sub_days(Days::new(120)).unwrap()), + new_budget: 100, + new_period: 1, + }]; + + let result = calculate_budgets_accrual(&category, &budget_updates); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 12_000); + } + + #[test] + fn when_two_budget_updates_then_calculates_as_expected() { + let category = Category { + id: 0, + name: String::from("Groceries"), + }; + let budget_updates = [ + BudgetUpdate { + id: 0, + category_id: 0, + date: DateUtc::from(&Utc::now().checked_sub_days(Days::new(60)).unwrap()), + new_budget: 200, + new_period: 1, + }, + BudgetUpdate { + id: 0, + category_id: 0, + date: DateUtc::from(&Utc::now().checked_sub_days(Days::new(120)).unwrap()), + new_budget: 100, + new_period: 1, + }, + ]; + + let result = calculate_budgets_accrual(&category, &budget_updates); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 18_000); + } + + #[test] + fn when_two_budget_updates_on_same_day_then_returns_err() { + let category = Category { + id: 0, + name: String::from("Groceries"), + }; + let budget_updates = [ + BudgetUpdate { + id: 0, + category_id: 0, + date: DateUtc::from_ymd(2020, 12, 25).unwrap(), + new_budget: 200, + new_period: 1, + }, + BudgetUpdate { + id: 0, + category_id: 0, + date: DateUtc::from_ymd(2020, 12, 25).unwrap(), + new_budget: 100, + new_period: 1, + }, + ]; + let mut budget_updates_reversed = budget_updates.clone(); + budget_updates_reversed.reverse(); + + let result_leftways = calculate_budgets_accrual(&category, &budget_updates); + let result_rightways = calculate_budgets_accrual(&category, &budget_updates_reversed); + + assert!(result_leftways.is_err()); + assert!(result_rightways.is_err()); + } +} diff --git a/rust/schist_queries/src/utils/calculate_category_balance.rs b/rust/schist_queries/src/utils/calculate_category_balance.rs new file mode 100644 index 0000000..acb56fb --- /dev/null +++ b/rust/schist_queries/src/utils/calculate_category_balance.rs @@ -0,0 +1,150 @@ +use anyhow::Result; +use schist_models::{budget_update::BudgetUpdate, category::Category}; + +use super::{calculate_budgets_accrual::calculate_budgets_accrual, find_by_id_or::find_by_id_or}; + +pub fn calculate_category_balance( + category: &Category, + transaction_sums: &[(i32, i64)], + category_transfer_from_sums: &[(i32, i64)], + category_transfer_to_sums: &[(i32, i64)], + budget_updates: &[BudgetUpdate], +) -> Result<i64> { + let transaction_sum = find_by_id_or(transaction_sums, category.id, 0); + let category_transfer_from_sum = find_by_id_or(category_transfer_from_sums, category.id, 0); + let category_transfer_to_sum = find_by_id_or(category_transfer_to_sums, category.id, 0); + let budget_accruals_sum = calculate_budgets_accrual(category, budget_updates)?; + + Ok( + transaction_sum + + budget_accruals_sum + + category_transfer_to_sum + - category_transfer_from_sum + ) +} + +#[cfg(test)] +mod test { + use chrono::{Days, Utc}; + use schist_models::{budget_update::BudgetUpdate, category::Category, date_utc::DateUtc}; + + use super::calculate_category_balance; + + #[test] + fn when_no_data_then_returns_zero() { + let category = Category { + id: 0, + name: String::new(), + }; + + let result = calculate_category_balance(&category, &[], &[], &[], &[]); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 0_i64); + } + + #[test] + fn when_data_only_for_another_category_then_returns_zero() { + let category = Category { + id: 0, + name: String::new(), + }; + let transaction_sums = [(1, 100)]; + let category_transfer_from_sums = [(1, 91)]; + let category_transfer_to_sums = [(2, 79)]; + let budget_updates = [BudgetUpdate { + id: 0, + category_id: 1, + date: DateUtc::from(Utc::now().checked_sub_days(Days::new(120)).unwrap()), + new_budget: 77, + new_period: 1, + }]; + + let result = calculate_category_balance( + &category, + &transaction_sums, + &category_transfer_from_sums, + &category_transfer_to_sums, + &budget_updates, + ); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 0_i64); + } + + #[test] + fn when_transaction_sum_is_zero_then_returns_zero() { + let category = Category { + id: 0, + name: String::new(), + }; + let transaction_sums = [(0, 0)]; + let category_transfer_from_sums = []; + let category_transfer_to_sums = []; + let budget_updates = []; + + let result = calculate_category_balance( + &category, + &transaction_sums, + &category_transfer_from_sums, + &category_transfer_to_sums, + &budget_updates, + ); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 0_i64); + } + + #[test] + fn when_category_transfers_are_zero_then_returns_zero() { + let category = Category { + id: 0, + name: String::new(), + }; + let transaction_sums = []; + let category_transfer_from_sums = [(0, 0)]; + let category_transfer_to_sums = [(0, 0)]; + let budget_updates = []; + + let result = calculate_category_balance( + &category, + &transaction_sums, + &category_transfer_from_sums, + &category_transfer_to_sums, + &budget_updates, + ); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 0_i64); + } + + #[test] + fn when_some_data_then_calculates_as_expected() { + let category = Category { + id: 0, + name: String::new(), + }; + let transaction_sums = [(0, 100)]; + let category_transfer_from_sums = [(0, 91)]; + let category_transfer_to_sums = [(0, 79)]; + let budget_updates = [BudgetUpdate { + id: 0, + category_id: 0, + date: DateUtc::from(&Utc::now().checked_sub_days(Days::new(120)).unwrap()), + new_budget: 77, + new_period: 1, + }]; + + let result = calculate_category_balance( + &category, + &transaction_sums, + &category_transfer_from_sums, + &category_transfer_to_sums, + &budget_updates, + ); + + let expected_result = 100 + (-91) + 79 + 77 * 120; + assert!(result.is_ok()); + assert_eq!(result.unwrap(), expected_result); + } +} diff --git a/rust/schist_queries/src/utils/find_by_id_or.rs b/rust/schist_queries/src/utils/find_by_id_or.rs new file mode 100644 index 0000000..8335e5c --- /dev/null +++ b/rust/schist_queries/src/utils/find_by_id_or.rs @@ -0,0 +1,41 @@ +pub fn find_by_id_or<Id: PartialEq, Val: Copy>( + arr: &[(Id, Val)], + id: Id, + default: Val, +) -> Val { + arr.iter() + .find(|(elem_id, _)| *elem_id == id) + .map_or(default, |row| row.1) +} + +#[cfg(test)] +mod test { + use super::find_by_id_or; + + #[test] + fn when_no_elements_then_return_default() { + let arr = []; + + let result = find_by_id_or(&arr, 1, 42); + + assert_eq!(result, 42); + } + + #[test] + fn when_id_not_in_elements_then_return_default() { + let arr = [(0, 100), (2, 200)]; + + let result = find_by_id_or(&arr, 1, 42); + + assert_eq!(result, 42); + } + + #[test] + fn when_id_in_elements_then_return_value() { + let arr = [(1, 67)]; + + let result = find_by_id_or(&arr, 1, 42); + + assert_eq!(result, 67); + } +} diff --git a/rust/schist_queries/src/utils/mod.rs b/rust/schist_queries/src/utils/mod.rs new file mode 100644 index 0000000..6949ce5 --- /dev/null +++ b/rust/schist_queries/src/utils/mod.rs @@ -0,0 +1,5 @@ +mod calculate_budgets_accrual; +mod calculate_category_balance; +mod find_by_id_or; + +pub use calculate_category_balance::calculate_category_balance; diff --git a/rust/schist_queries/tests/account.rs b/rust/schist_queries/tests/account.rs new file mode 100644 index 0000000..e3bb7db --- /dev/null +++ b/rust/schist_queries/tests/account.rs @@ -0,0 +1,60 @@ +mod common; + +use common::test_context::TestContext; +use diesel::{Connection, SqliteConnection}; +use schist_fakes::account::make_fake_account; +use schist_queries as queries; + +#[test] +fn when_insert_accounts_then_returns_num_accounts() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let accounts = [make_fake_account(0), make_fake_account(1)]; + + let num_accounts_inserted = queries::accounts::insert_accounts(&accounts, connection).unwrap(); + + assert_eq!(num_accounts_inserted, 2); +} + +#[test] +fn given_accounts_when_get_all_accounts_then_return_accounts() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let accounts = [make_fake_account(0), make_fake_account(1)]; + let _ = queries::accounts::insert_accounts(&accounts, connection).unwrap(); + + let accounts_in_db = queries::accounts::get_all_accounts(connection).unwrap(); + + assert_eq!(accounts_in_db.len(), accounts.len()); + for account in &accounts { + assert!(accounts_in_db.iter().any(|account_in_db| account_in_db.eq(&account))); + } + for account_in_db in accounts_in_db { + assert!(accounts.iter().any(|account| account.eq(&account_in_db))); + } +} + +#[test] +fn given_accounts_when_delete_all_accounts_then_return_num_of_accounts() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let accounts = [make_fake_account(0), make_fake_account(1)]; + let _ = queries::accounts::insert_accounts(&accounts, connection).unwrap(); + + let num_accounts_deleted = queries::accounts::delete_all_accounts(connection).unwrap(); + + assert_eq!(num_accounts_deleted, accounts.len()); +} + +#[test] +fn given_deleted_accounts_when_get_all_accounts_then_return_empty() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let accounts = [make_fake_account(0), make_fake_account(1)]; + let _ = queries::accounts::insert_accounts(&accounts, connection).unwrap(); + let _ = queries::accounts::delete_all_accounts(connection).unwrap(); + + let accounts_returned = queries::accounts::get_all_accounts(connection).unwrap(); + + assert_eq!(accounts_returned.len(), 0); +} diff --git a/rust/schist_queries/tests/budget_updates.rs b/rust/schist_queries/tests/budget_updates.rs new file mode 100644 index 0000000..bd84128 --- /dev/null +++ b/rust/schist_queries/tests/budget_updates.rs @@ -0,0 +1,358 @@ +mod common; + +use common::test_context::TestContext; +use diesel::{Connection, SqliteConnection}; +use schist_fakes::{budget_update::make_fake_budget_update_builder, category::make_fake_category}; +use schist_models::{category::Category, date_utc::DateUtc}; +use schist_queries as queries; +use schist_traits::nowlike::Nowlike; + +fn given_categories(connection: &mut SqliteConnection) -> Vec<Category> { + let categories = vec![make_fake_category(0), make_fake_category(1)]; + let _ = queries::categories::insert_categories(&categories, connection); + categories +} + +#[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 = given_categories(connection); + let budget_updates = [ + make_fake_budget_update_builder(0) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .category_id(0) + .build().unwrap(), + make_fake_budget_update_builder(1) + .date(DateUtc::from_ymd(2020, 12, 26).unwrap()) + .category_id(0) + .build().unwrap(), + ]; + + 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_update_when_insert_budget_update_same_category_and_same_date_then_return_err() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let _categories = given_categories(connection); + + let budget_updates_1 = [ + make_fake_budget_update_builder(0) + .category_id(0) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .build().unwrap() + ]; + let _ = queries::budget_updates::insert_budget_updates(&budget_updates_1, connection); + + let budget_updates_2 = [ + make_fake_budget_update_builder(1) + .category_id(0) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .build().unwrap() + ]; + let result = queries::budget_updates::insert_budget_updates(&budget_updates_2, connection); + + assert!(result.is_err()); +} + +#[test] +fn given_budget_update_when_insert_budget_update_different_category_and_same_date_then_return_num_budget_updates_inserted() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let _categories = given_categories(connection); + + let budget_updates_1 = [ + make_fake_budget_update_builder(0) + .category_id(0) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .build().unwrap() + ]; + let _ = queries::budget_updates::insert_budget_updates(&budget_updates_1, connection); + + let budget_updates_2 = [ + make_fake_budget_update_builder(1) + .category_id(1) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .build().unwrap() + ]; + let result = queries::budget_updates::insert_budget_updates(&budget_updates_2, connection); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 1); +} + +#[test] +fn given_budget_update_when_insert_budget_update_same_category_and_different_date_then_return_num_budget_updates_inserted() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let _categories = given_categories(connection); + + let budget_updates_1 = [ + make_fake_budget_update_builder(0) + .category_id(0) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .build().unwrap() + ]; + let _ = queries::budget_updates::insert_budget_updates(&budget_updates_1, connection); + + let budget_updates_2 = [ + make_fake_budget_update_builder(1) + .category_id(0) + .date(DateUtc::from_ymd(2020, 12, 26).unwrap()) + .build().unwrap() + ]; + let result = queries::budget_updates::insert_budget_updates(&budget_updates_2, connection); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 1); +} + +#[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 = given_categories(connection); + let budget_updates = [ + make_fake_budget_update_builder(0) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .category_id(0) + .build().unwrap(), + make_fake_budget_update_builder(1) + .date(DateUtc::from_ymd(2020, 12, 26).unwrap()) + .category_id(0) + .build().unwrap(), + ]; + 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()); + assert_eq!(budget_updates_in_db[0], budget_updates[0]); + assert_eq!(budget_updates_in_db[1], budget_updates[1]); + 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 = given_categories(connection); + let budget_updates = [ + make_fake_budget_update_builder(0) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .category_id(0) + .build().unwrap(), + make_fake_budget_update_builder(1) + .date(DateUtc::from_ymd(2020, 12, 26).unwrap()) + .category_id(0) + .build().unwrap(), + ]; + 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 = given_categories(connection); + let budget_updates = [ + make_fake_budget_update_builder(0) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .category_id(0) + .build().unwrap(), + make_fake_budget_update_builder(1) + .date(DateUtc::from_ymd(2020, 12, 26).unwrap()) + .category_id(0) + .build().unwrap(), + ]; + 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); +} + +#[test] +fn given_no_categories_when_get_first_with_date_and_category_id_then_return_none() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + + let first_with_date_and_category_id = + queries::budget_updates::get_first_budget_update_with_date_and_category_id( + &DateUtc::now(), + 0, + connection, + ).unwrap(); + + assert!(first_with_date_and_category_id.is_none()); +} + +#[test] +fn given_no_budget_updates_when_get_first_with_date_and_category_id_then_return_none() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let _categories = given_categories(connection); + + let first_with_date_and_category_id = + queries::budget_updates::get_first_budget_update_with_date_and_category_id( + &DateUtc::now(), + 0, + connection, + ).unwrap(); + + assert!(first_with_date_and_category_id.is_none()); +} + +#[test] +fn given_budget_update_when_get_first_with_date_then_return_budget_update() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let _categories = given_categories(connection); + let date: DateUtc = "2024-10-28".parse().unwrap(); + let budget_update = make_fake_budget_update_builder(0).date(date.clone()).build().unwrap(); + let _ = queries::budget_updates::insert_budget_updates(&[budget_update.clone()], connection).unwrap(); + + let first_with_date_and_category_id = + queries::budget_updates::get_first_budget_update_with_date_and_category_id( + &date, + 0, + connection, + ).unwrap(); + + assert!(first_with_date_and_category_id.is_some()); + assert_eq!(first_with_date_and_category_id.unwrap(), budget_update); +} + +#[test] +fn given_budget_update_when_get_first_with_date_and_different_category_id_then_return_none() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let _categories = given_categories(connection); + let date: DateUtc = "2024-10-28".parse().unwrap(); + let different_date: DateUtc = "2024-10-29".parse().unwrap(); + let budget_update = make_fake_budget_update_builder(0).date(date.clone()).build().unwrap(); + let _ = queries::budget_updates::insert_budget_updates(&[budget_update.clone()], connection).unwrap(); + + let first_with_date_and_category_id = + queries::budget_updates::get_first_budget_update_with_date_and_category_id( + &different_date, + 2, + connection + ).unwrap(); + + assert!(first_with_date_and_category_id.is_none()); +} + +#[test] +fn given_budget_update_when_get_first_with_different_date_and_category_id_then_return_none() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let _categories = given_categories(connection); + let date: DateUtc = "2024-10-28".parse().unwrap(); + let different_date: DateUtc = "2024-10-29".parse().unwrap(); + let budget_update = make_fake_budget_update_builder(0).date(date.clone()).build().unwrap(); + let _ = queries::budget_updates::insert_budget_updates(&[budget_update.clone()], connection).unwrap(); + + let first_with_date_and_category_id = + queries::budget_updates::get_first_budget_update_with_date_and_category_id( + &different_date, + 0, + connection + ).unwrap(); + + assert!(first_with_date_and_category_id.is_none()); +} + +#[test] +fn given_many_budget_updates_when_get_first_with_date_and_category_id_then_return_any_with_date() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let _categories = given_categories(connection); + let budget_updates = [ + make_fake_budget_update_builder(0).date("2024-12-25".parse().unwrap()).category_id(0).build().unwrap(), + make_fake_budget_update_builder(1).date("2024-12-25".parse().unwrap()).category_id(1).build().unwrap(), + make_fake_budget_update_builder(2).date("2020-12-26".parse().unwrap()).category_id(0).build().unwrap(), + make_fake_budget_update_builder(3).date("2020-12-26".parse().unwrap()).category_id(1).build().unwrap(), + make_fake_budget_update_builder(4).date("2020-12-27".parse().unwrap()).category_id(0).build().unwrap(), + make_fake_budget_update_builder(5).date("2020-12-27".parse().unwrap()).category_id(1).build().unwrap(), + ]; + let _ = queries::budget_updates::insert_budget_updates(&budget_updates, connection).unwrap(); + + let first_with_date_and_category_id = + queries::budget_updates::get_first_budget_update_with_date_and_category_id( + &"2024-12-25".parse().unwrap(), + 0, + connection, + ).unwrap(); + + assert!(first_with_date_and_category_id.is_some()); + assert_eq!(first_with_date_and_category_id.unwrap(), budget_updates[0]); +} + +#[test] +fn given_many_budget_updates_when_get_most_recent_with_category_id_then_return_correct_date() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let _categories = given_categories(connection); + let budget_updates = [ + make_fake_budget_update_builder(0) + .category_id(0) + .date("2024-12-25".parse().unwrap()) + .build().unwrap(), + make_fake_budget_update_builder(1) + .category_id(0) + .date("2020-12-26".parse().unwrap()) + .build().unwrap(), + make_fake_budget_update_builder(2) + .category_id(1) + .date("2020-12-27".parse().unwrap()) + .build().unwrap(), + ]; + let _ = queries::budget_updates::insert_budget_updates(&budget_updates, connection).unwrap(); + + let most_recent_with_category = queries::budget_updates::get_most_recent_budget_update_date_with_category_id( + 0, connection, + ); + + assert!(most_recent_with_category.is_ok()); + let most_recent_with_category = most_recent_with_category.unwrap(); + assert!(most_recent_with_category .is_some()); + let most_recent_with_category = most_recent_with_category.unwrap(); + assert_eq!(most_recent_with_category, budget_updates[1].date); +} + +#[test] +fn given_budget_update_when_get_most_recent_with_different_category_id_then_return_none() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let _categories = given_categories(connection); + let budget_updates = [ + make_fake_budget_update_builder(0) + .category_id(1) + .date("2020-12-27".parse().unwrap()) + .build().unwrap(), + ]; + let _ = queries::budget_updates::insert_budget_updates(&budget_updates, connection).unwrap(); + + let most_recent_with_category = queries::budget_updates::get_most_recent_budget_update_date_with_category_id( + 0, connection, + ); + + assert!(most_recent_with_category.is_ok()); + let most_recent_with_category = most_recent_with_category.unwrap(); + assert!(most_recent_with_category .is_none()); +} diff --git a/rust/schist_queries/tests/categories.rs b/rust/schist_queries/tests/categories.rs new file mode 100644 index 0000000..74c395d --- /dev/null +++ b/rust/schist_queries/tests/categories.rs @@ -0,0 +1,235 @@ +mod common; + +use chrono::{Days, Utc}; +use common::test_context::TestContext; +use diesel::{Connection, SqliteConnection}; +use schist_fakes::{ + account::make_fake_account, + budget_update::make_fake_budget_update_builder, + category::make_fake_category, + transaction::make_fake_transaction_builder, + transaction_categorisation::make_fake_transaction_categorisation_builder, +}; +use schist_models::{ + account::Account, + budget::Budget, + budget_update::BudgetUpdate, + category::Category, + date_utc::DateUtc, + transaction::Transaction, + transaction_categorisation::TransactionCategorisation, +}; +use schist_queries as queries; + +fn given_categories(connection: &mut SqliteConnection) -> Vec<Category> { + let categories = vec![make_fake_category(0), make_fake_category(1)]; + let _ = queries::categories::insert_categories(&categories, connection).unwrap(); + categories +} + +fn given_transactions(connection: &mut SqliteConnection) -> Vec<Transaction> { + let transactions = vec![ + make_fake_transaction_builder(0).quantity(1).build().unwrap(), + make_fake_transaction_builder(1).quantity(2).build().unwrap(), + make_fake_transaction_builder(2).quantity(4).build().unwrap(), + make_fake_transaction_builder(3).quantity(8).build().unwrap(), + ]; + let _ = queries::transactions::insert_transactions(&transactions, connection).unwrap(); + transactions +} + +fn given_accounts(connection: &mut SqliteConnection) -> Vec<Account> { + let accounts = vec![ + make_fake_account(0), + ]; + let _ = queries::accounts::insert_accounts(&accounts, connection); + accounts +} + +fn given_transaction_categorisations(connection: &mut SqliteConnection) -> Vec<TransactionCategorisation> { + let transaction_categorisations = vec![ + make_fake_transaction_categorisation_builder(0).quantity(1).transaction_id(0).category_id(0).build().unwrap(), + make_fake_transaction_categorisation_builder(1).quantity(1).transaction_id(1).category_id(0).build().unwrap(), + make_fake_transaction_categorisation_builder(2).quantity(1).transaction_id(1).category_id(0).build().unwrap(), + make_fake_transaction_categorisation_builder(3).quantity(4).transaction_id(2).category_id(1).build().unwrap(), + make_fake_transaction_categorisation_builder(4).quantity(3).transaction_id(3).category_id(1).build().unwrap(), + make_fake_transaction_categorisation_builder(5).quantity(5).transaction_id(3).category_id(1).build().unwrap(), + ]; + let _ = queries::transaction_categorisations::insert_transaction_categorisations(&transaction_categorisations, connection).unwrap(); + transaction_categorisations +} + +fn given_budget_updates(connection: &mut SqliteConnection) -> Vec<BudgetUpdate> { + let now = Utc::now(); + let twenty_days_ago = DateUtc::from(now.checked_sub_days(Days::new(20)).unwrap()); + let ten_days_ago = DateUtc::from(now.checked_sub_days(Days::new(10)).unwrap()); + let budget_updates = vec![ + make_fake_budget_update_builder(0).category_id(0).date(twenty_days_ago.clone()).new_budget(100).new_period(1).build().unwrap(), + make_fake_budget_update_builder(1).category_id(0).date(ten_days_ago.clone()).new_budget(50).new_period(2).build().unwrap(), + make_fake_budget_update_builder(2).category_id(1).date(twenty_days_ago.clone()).new_budget(90).new_period(4).build().unwrap(), + make_fake_budget_update_builder(3).category_id(1).date(ten_days_ago.clone()).new_budget(40).new_period(8).build().unwrap(), + ]; + let _ = queries::budget_updates::insert_budget_updates(&budget_updates, connection); + budget_updates +} + +#[test] +fn when_insert_categories_then_returns_num_categories() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let categories = [make_fake_category(0), make_fake_category(1)]; + + let num_categories_inserted = queries::categories::insert_categories(&categories, connection).unwrap(); + + assert_eq!(num_categories_inserted, 2); +} + +#[test] +fn given_categories_when_get_all_categories_then_return_categories() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let categories = given_categories(connection); + + let categories_in_db = queries::categories::get_all_categories(connection).unwrap(); + + assert_eq!(categories_in_db.len(), categories.len()); + for category in &categories { + assert!(categories_in_db.iter().any(|category_in_db| category_in_db.eq(&category))); + } + for category_in_db in categories_in_db { + assert!(categories.iter().any(|category| category.eq(&category_in_db))); + } +} + +#[test] +fn given_categories_when_delete_all_categories_then_return_num_of_categories() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let categories = given_categories(connection); + + let num_categories_deleted = queries::categories::delete_all_categories(connection).unwrap(); + + assert_eq!(num_categories_deleted, categories.len()); +} + +#[test] +fn given_deleted_categories_when_get_all_categories_then_return_empty() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + given_categories(connection); + let _ = queries::categories::delete_all_categories(connection).unwrap(); + + let categories_returned = queries::categories::get_all_categories(connection).unwrap(); + + assert_eq!(categories_returned.len(), 0); +} + +#[test] +fn given_categories_and_no_transactions_when_get_all_balances_then_return_all_zero() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + given_categories(connection); + + let balances = queries::categories::get_all_category_balances(connection); + + assert!(balances.is_ok()); + let mut balances = balances.unwrap(); + balances.sort_by(|b1, b2| b1.category_id.cmp(&b2.category_id)); + assert_eq!(balances.len(), 2); + for balance in balances { + assert_eq!(0, balance.balance); + } +} + +#[test] +fn given_categories_and_transaction_categorisations_when_get_all_balances_then_return_balances() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let categories = given_categories(connection); + let _accounts = given_accounts(connection); + let _transactions = given_transactions(connection); + let transaction_categorisations = given_transaction_categorisations(connection); + + let balances = queries::categories::get_all_category_balances(connection); + + assert!(balances.is_ok()); + let mut balances = balances.unwrap(); + balances.sort_by(|b1, b2| b1.category_id.cmp(&b2.category_id)); + assert_eq!(balances.len(), categories.len()); + for category in categories { + let balance = balances.iter().find(|b| b.category_id == category.id).unwrap(); + let expected_balance: i32 = transaction_categorisations + .iter() + .filter(|t| t.category_id == category.id) + .map(|t| t.quantity) + .sum(); + assert_eq!(expected_balance, balance.balance); + } +} + +#[test] +fn given_categories_when_get_all_ids_then_return_all_ids() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let categories = given_categories(connection); + + let ids = queries::categories::get_all_category_ids(connection).unwrap(); + + assert_eq!(categories.len(), ids.len()); + for category in &categories { + assert!(ids.iter().any(|id| category.id == *id)); + } + for id in ids { + assert!(categories.iter().any(|category| category.id == id)); + } +} + +#[test] +fn given_categories_and_no_budget_updates_when_get_most_recent_budget_per_category_id_then_return_none() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let categories = given_categories(connection); + + let most_recent_budget_per_category_id = queries::categories::get_most_recent_category_budget_per_category_id(connection); + + assert!(most_recent_budget_per_category_id.is_ok()); + let most_recent_budget_per_category_id = most_recent_budget_per_category_id.unwrap(); + assert_eq!(categories.len(), most_recent_budget_per_category_id.len()); + for category in categories { + let most_recent_budget = most_recent_budget_per_category_id.iter().find(|b| b.category_id == category.id); + assert!(most_recent_budget.is_some()); + let most_recent_budget = most_recent_budget.unwrap(); + assert!(most_recent_budget.budget.is_none()); + } +} + +#[test] +fn given_categories_and_budget_updates_when_get_most_recent_budget_per_category_id_then_return_most_recent_budget() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let categories = given_categories(connection); + let budget_updates = given_budget_updates(connection); + + let most_recent_budget_per_category_id = queries::categories::get_most_recent_category_budget_per_category_id(connection); + + assert!(most_recent_budget_per_category_id.is_ok()); + let most_recent_budget_per_category_id = most_recent_budget_per_category_id.unwrap(); + assert_eq!(categories.len(), most_recent_budget_per_category_id.len()); + for category in categories { + let most_recent_budget_update = budget_updates + .iter() + .filter(|bu| bu.category_id == category.id) + .max_by_key(|bu| bu.date.clone()) + .unwrap(); + + let most_recent_budget = most_recent_budget_per_category_id.iter().find(|b| b.category_id == category.id); + assert!(most_recent_budget.is_some()); + let most_recent_budget = most_recent_budget.unwrap(); + assert!(most_recent_budget.budget.is_some()); + let most_recent_budget = most_recent_budget.budget.unwrap(); + assert_eq!(most_recent_budget, Budget { + quantity: most_recent_budget_update.new_budget, + period: most_recent_budget_update.new_period, + }); + } +} diff --git a/rust/schist_queries/tests/category_transfers.rs b/rust/schist_queries/tests/category_transfers.rs new file mode 100644 index 0000000..f37a668 --- /dev/null +++ b/rust/schist_queries/tests/category_transfers.rs @@ -0,0 +1,126 @@ +mod common; + +use common::test_context::TestContext; +use diesel::{Connection, SqliteConnection}; +use schist_fakes::{category::make_fake_category, category_transfer::make_fake_category_transfer_builder}; +use schist_queries as queries; + +#[test] +fn when_insert_category_transfers_then_returns_num_category_transfers() { + 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).build().unwrap(), + make_fake_category_transfer_builder(1).from_category_id(0).to_category_id(1).build().unwrap(), + ]; + let _ = queries::categories::insert_categories(&categories, connection); + + let num_category_transfers_inserted = queries::category_transfers::insert_category_transfers(&category_transfers, connection).unwrap(); + + assert_eq!(num_category_transfers_inserted, 2); +} + +#[test] +fn given_category_transfers_when_get_all_category_transfers_then_return_category_transfers() { + 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).build().unwrap(), + make_fake_category_transfer_builder(1).from_category_id(0).to_category_id(1).build().unwrap(), + ]; + let _ = queries::categories::insert_categories(&categories, connection); + let _ = queries::category_transfers::insert_category_transfers(&category_transfers, connection).unwrap(); + + let category_transfers_in_db = queries::category_transfers::get_all_category_transfers(connection).unwrap(); + + assert_eq!(category_transfers_in_db.len(), category_transfers.len()); + for category_transfer in &category_transfers { + assert!(category_transfers_in_db.iter().any(|category_transfer_in_db| category_transfer_in_db.eq(&category_transfer))); + } + for category_transfer_in_db in category_transfers_in_db { + assert!(category_transfers.iter().any(|category_transfer| category_transfer.eq(&category_transfer_in_db))); + } +} + +#[test] +fn given_category_transfers_when_delete_all_category_transfers_then_return_num_of_category_transfers() { + 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).build().unwrap(), + make_fake_category_transfer_builder(1).from_category_id(0).to_category_id(1).build().unwrap(), + ]; + let _ = queries::categories::insert_categories(&categories, connection); + let _ = queries::category_transfers::insert_category_transfers(&category_transfers, connection).unwrap(); + + let num_category_transfers_deleted = queries::category_transfers::delete_all_category_transfers(connection).unwrap(); + + assert_eq!(num_category_transfers_deleted, category_transfers.len()); +} + +#[test] +fn given_deleted_category_transfers_when_get_all_category_transfers_then_return_empty() { + 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).build().unwrap(), + make_fake_category_transfer_builder(1).from_category_id(0).to_category_id(1).build().unwrap(), + ]; + let _ = queries::categories::insert_categories(&categories, connection); + let _ = queries::category_transfers::insert_category_transfers(&category_transfers, connection).unwrap(); + let _ = queries::category_transfers::delete_all_category_transfers(connection).unwrap(); + + let category_transfers_returned = queries::category_transfers::get_all_category_transfers(connection).unwrap(); + + assert_eq!(category_transfers_returned.len(), 0); +} + +#[test] +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 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); +} diff --git a/rust/schist_queries/tests/common/mod.rs b/rust/schist_queries/tests/common/mod.rs new file mode 100644 index 0000000..5ff67da --- /dev/null +++ b/rust/schist_queries/tests/common/mod.rs @@ -0,0 +1 @@ +pub mod test_context; diff --git a/rust/schist_queries/tests/common/test_context.rs b/rust/schist_queries/tests/common/test_context.rs new file mode 100644 index 0000000..183cd9b --- /dev/null +++ b/rust/schist_queries/tests/common/test_context.rs @@ -0,0 +1,44 @@ +use std::sync::atomic::{AtomicU64, Ordering}; + +use diesel::{Connection, SqliteConnection}; +use diesel_migrations::MigrationHarness; +use schist_models::migrations::MIGRATIONS; +use schist_queries::clear::clear; + +pub struct TestContext { + pub db_url: String, +} + +static ID_COUNTER: AtomicU64 = AtomicU64::new(0); + +impl TestContext { + pub fn new() -> Self { + let db_url = Self::next_id() + ".sqlite"; + + let connection = &mut SqliteConnection + ::establish(&db_url) + .expect("failed to connect to database"); + + connection + .run_pending_migrations(MIGRATIONS) + .expect("failed to run migrations"); + + clear(connection) + .expect("failed to clear database"); + + Self { + db_url, + } + } + + fn next_id() -> String { + ID_COUNTER.fetch_add(1, Ordering::Relaxed).to_string() + } +} + +impl Drop for TestContext { + fn drop(&mut self) { + std::fs::remove_file(&self.db_url) + .expect(format!("failed to delete database {}", self.db_url).as_str()); + } +} diff --git a/rust/schist_queries/tests/transaction_categorisations.rs b/rust/schist_queries/tests/transaction_categorisations.rs new file mode 100644 index 0000000..13864b4 --- /dev/null +++ b/rust/schist_queries/tests/transaction_categorisations.rs @@ -0,0 +1,130 @@ +mod common; + +use common::test_context::TestContext; +use diesel::{Connection, SqliteConnection}; +use schist_fakes::{account::make_fake_account, category::make_fake_category, transaction::make_fake_transaction_builder, transaction_categorisation::make_fake_transaction_categorisation_builder}; +use schist_models::{account::Account, category::Category, transaction::Transaction}; +use schist_queries as queries; + +fn given_accounts(connection: &mut SqliteConnection) -> Vec<Account> { + let accounts = vec![make_fake_account(0)]; + let _ = queries::accounts::insert_accounts(&accounts, connection); + accounts +} + +fn given_categories(connection: &mut SqliteConnection) -> Vec<Category> { + let categories = vec![make_fake_category(0), make_fake_category(1)]; + let _ = queries::categories::insert_categories(&categories, connection); + categories +} + +fn given_transactions(connection: &mut SqliteConnection) -> Vec<Transaction> { + let transactions = vec![ + make_fake_transaction_builder(0).account_id(0).build().unwrap(), + make_fake_transaction_builder(1).account_id(0).build().unwrap(), + ]; + let _ = queries::transactions::insert_transactions(&transactions, connection); + transactions +} + +#[test] +fn when_insert_transaction_categorisations_then_returns_num_transaction_categorisations() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + given_accounts(connection); + given_categories(connection); + given_transactions(connection); + let transaction_categorisations = [ + make_fake_transaction_categorisation_builder(0).category_id(0).build().unwrap(), + make_fake_transaction_categorisation_builder(1).category_id(0).build().unwrap(), + ]; + + let num_transaction_categorisations_inserted = queries::transaction_categorisations::insert_transaction_categorisations(&transaction_categorisations, connection).unwrap(); + + assert_eq!(num_transaction_categorisations_inserted, 2); +} + +#[test] +fn given_transaction_categorisations_when_get_all_transaction_categorisations_then_return_transaction_categorisations() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + given_accounts(connection); + given_categories(connection); + given_transactions(connection); + let transaction_categorisations = [ + make_fake_transaction_categorisation_builder(0).category_id(0).build().unwrap(), + make_fake_transaction_categorisation_builder(1).category_id(0).build().unwrap(), + ]; + let _ = queries::transaction_categorisations::insert_transaction_categorisations(&transaction_categorisations, connection); + + let transaction_categorisations_in_db = queries::transaction_categorisations::get_all_transaction_categorisations(connection).unwrap(); + + assert_eq!(transaction_categorisations_in_db.len(), transaction_categorisations.len()); + for transaction_categorisation in &transaction_categorisations { + assert!(transaction_categorisations_in_db.iter().any(|transaction_categorisation_in_db| transaction_categorisation_in_db.eq(&transaction_categorisation))); + } + for transaction_categorisation_in_db in transaction_categorisations_in_db { + assert!(transaction_categorisations.iter().any(|transaction_categorisation| transaction_categorisation.eq(&transaction_categorisation_in_db))); + } +} + +#[test] +fn given_transaction_categorisations_when_delete_all_transaction_categorisations_then_return_num_of_transaction_categorisations() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + given_accounts(connection); + given_categories(connection); + given_transactions(connection); + let transaction_categorisations = [ + make_fake_transaction_categorisation_builder(0).category_id(0).build().unwrap(), + make_fake_transaction_categorisation_builder(1).category_id(0).build().unwrap(), + ]; + let _ = queries::transaction_categorisations::insert_transaction_categorisations(&transaction_categorisations, connection); + + let num_transaction_categorisations_deleted = queries::transaction_categorisations::delete_all_transaction_categorisations(connection).unwrap(); + + assert_eq!(num_transaction_categorisations_deleted, transaction_categorisations.len()); +} + +#[test] +fn given_deleted_transaction_categorisations_when_get_all_transaction_categorisations_then_return_empty() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + given_accounts(connection); + given_categories(connection); + given_transactions(connection); + let transaction_categorisations = [ + make_fake_transaction_categorisation_builder(0).category_id(0).build().unwrap(), + make_fake_transaction_categorisation_builder(1).category_id(0).build().unwrap(), + ]; + let _ = queries::transaction_categorisations::insert_transaction_categorisations(&transaction_categorisations, connection); + let _ = queries::transaction_categorisations::delete_all_transaction_categorisations(connection); + + let transaction_categorisations_returned = queries::transaction_categorisations::get_all_transaction_categorisations(connection).unwrap(); + + assert_eq!(transaction_categorisations_returned.len(), 0); +} + +#[test] +fn given_transaction_categorisations_when_sum_quantity_per_category_id_then_returns_sum_quantity_per_category_id() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + given_accounts(connection); + let categories = given_categories(connection); + given_transactions(connection); + let transaction_categorisations = [ + make_fake_transaction_categorisation_builder(0).category_id(0).quantity(1).build().unwrap(), + make_fake_transaction_categorisation_builder(1).category_id(0).quantity(2).build().unwrap(), + make_fake_transaction_categorisation_builder(2).category_id(1).quantity(4).build().unwrap(), + ]; + let _ = queries::transaction_categorisations::insert_transaction_categorisations(&transaction_categorisations, connection); + + let sums = queries::transaction_categorisations::sum_transaction_categorisation_quantity_per_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); +} diff --git a/rust/schist_queries/tests/transactions.rs b/rust/schist_queries/tests/transactions.rs new file mode 100644 index 0000000..5e57074 --- /dev/null +++ b/rust/schist_queries/tests/transactions.rs @@ -0,0 +1,80 @@ +mod common; + +use common::test_context::TestContext; +use diesel::{Connection, SqliteConnection}; +use schist_fakes::{account::make_fake_account, transaction::make_fake_transaction_builder}; +use schist_queries as queries; + +#[test] +fn when_insert_transactions_then_returns_num_transactions() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let accounts = [make_fake_account(0)]; + let transactions = [ + make_fake_transaction_builder(0).account_id(0).build().unwrap(), + make_fake_transaction_builder(1).account_id(0).build().unwrap(), + ]; + let _ = queries::accounts::insert_accounts(&accounts, connection); + + let num_transactions_inserted = queries::transactions::insert_transactions(&transactions, connection).unwrap(); + + assert_eq!(num_transactions_inserted, 2); +} + +#[test] +fn given_transactions_when_get_all_transactions_then_return_transactions() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let accounts = [make_fake_account(0)]; + let transactions = [ + make_fake_transaction_builder(0).account_id(0).build().unwrap(), + make_fake_transaction_builder(1).account_id(0).build().unwrap(), + ]; + let _ = queries::accounts::insert_accounts(&accounts, connection); + let _ = queries::transactions::insert_transactions(&transactions, connection).unwrap(); + + let transactions_in_db = queries::transactions::get_all_transactions(connection).unwrap(); + + assert_eq!(transactions_in_db.len(), transactions.len()); + for transaction in &transactions { + assert!(transactions_in_db.iter().any(|transaction_in_db| transaction_in_db.eq(&transaction))); + } + for transaction_in_db in transactions_in_db { + assert!(transactions.iter().any(|transaction| transaction.eq(&transaction_in_db))); + } +} + +#[test] +fn given_transactions_when_delete_all_transactions_then_return_num_of_transactions() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let accounts = [make_fake_account(0)]; + let transactions = [ + make_fake_transaction_builder(0).account_id(0).build().unwrap(), + make_fake_transaction_builder(1).account_id(0).build().unwrap(), + ]; + let _ = queries::accounts::insert_accounts(&accounts, connection); + let _ = queries::transactions::insert_transactions(&transactions, connection).unwrap(); + + let num_transactions_deleted = queries::transactions::delete_all_transactions(connection).unwrap(); + + assert_eq!(num_transactions_deleted, transactions.len()); +} + +#[test] +fn given_deleted_transactions_when_get_all_transactions_then_return_empty() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let accounts = [make_fake_account(0)]; + let transactions = [ + make_fake_transaction_builder(0).account_id(0).build().unwrap(), + make_fake_transaction_builder(1).account_id(0).build().unwrap(), + ]; + let _ = queries::accounts::insert_accounts(&accounts, connection); + let _ = queries::transactions::insert_transactions(&transactions, connection).unwrap(); + let _ = queries::transactions::delete_all_transactions(connection).unwrap(); + + let transactions_returned = queries::transactions::get_all_transactions(connection).unwrap(); + + assert_eq!(transactions_returned.len(), 0); +} |