diff options
| author | Joe Carstairs <jcarstairs@scottlogic.com> | 2025-07-28 13:22:53 +0100 |
|---|---|---|
| committer | Joe Carstairs <jcarstairs@scottlogic.com> | 2025-07-28 13:30:27 +0100 |
| commit | 2361adeb0ca77904138e3466d51c2f44cf6df613 (patch) | |
| tree | f6e7192908f99fab62061913d8a1e9d2b918414b /schist_core | |
| parent | c7ed1e61ce2218be0b9e60577454124c4e43a366 (diff) | |
Tidier folders
Diffstat (limited to 'schist_core')
62 files changed, 3040 insertions, 0 deletions
diff --git a/schist_core/schist_fakes/Cargo.toml b/schist_core/schist_fakes/Cargo.toml new file mode 100644 index 0000000..c69a28e --- /dev/null +++ b/schist_core/schist_fakes/Cargo.toml @@ -0,0 +1,8 @@ +[package] +name = "schist_fakes" +version = "0.1.0" +edition = "2021" + +[dependencies] +schist_models = { path = "../schist_models" } +schist_traits = { path = "../schist_traits" } diff --git a/schist_core/schist_fakes/src/account.rs b/schist_core/schist_fakes/src/account.rs new file mode 100644 index 0000000..1547eb7 --- /dev/null +++ b/schist_core/schist_fakes/src/account.rs @@ -0,0 +1,11 @@ +use schist_models::{account::Account, date_utc::DateUtc}; +use schist_traits::nowlike::Nowlike; + +pub fn make_fake_account(id: i32) -> Account { + Account { + id, + name: String::from("Joe Bloggs"), + opening_balance: 0, + opening_date: DateUtc::now(), + } +} diff --git a/schist_core/schist_fakes/src/account_transfer.rs b/schist_core/schist_fakes/src/account_transfer.rs new file mode 100644 index 0000000..f762c2f --- /dev/null +++ b/schist_core/schist_fakes/src/account_transfer.rs @@ -0,0 +1,18 @@ +use schist_models::{date_utc::DateUtc, account_transfer::{AccountTransfer, AccountTransferBuilder}}; +use schist_traits::nowlike::Nowlike; + +pub fn make_fake_account_transfer(id: i32) -> AccountTransfer { + make_fake_account_transfer_builder(id).build().unwrap() +} + +pub fn make_fake_account_transfer_builder(id: i32) -> AccountTransferBuilder { + let mut builder = AccountTransferBuilder::default(); + builder + .id(id) + .date(DateUtc::now()) + .description(String::from("Fake account transfer")) + .quantity(0) + .from_account_id(0) + .to_account_id(1); + builder +} diff --git a/schist_core/schist_fakes/src/budget_drip.rs b/schist_core/schist_fakes/src/budget_drip.rs new file mode 100644 index 0000000..f653565 --- /dev/null +++ b/schist_core/schist_fakes/src/budget_drip.rs @@ -0,0 +1,16 @@ +use schist_models::{budget_drip::{BudgetDrip, BudgetDripBuilder}, date_utc::DateUtc}; +use schist_traits::nowlike::Nowlike; + +pub fn make_fake_budget_drip(id: i32) -> BudgetDrip { + make_fake_budget_drip_builder(id).build().unwrap() +} + +pub fn make_fake_budget_drip_builder(id: i32) -> BudgetDripBuilder { + let mut builder = BudgetDripBuilder::default(); + builder + .id(id) + .category_id(0) + .date(DateUtc::now()) + .quantity(0); + builder +} diff --git a/schist_core/schist_fakes/src/category.rs b/schist_core/schist_fakes/src/category.rs new file mode 100644 index 0000000..c64818d --- /dev/null +++ b/schist_core/schist_fakes/src/category.rs @@ -0,0 +1,19 @@ +use schist_models::{budget_period_unit::BudgetPeriodUnit, category::{Category, CategoryBuilder}, date_utc::DateUtc}; +use schist_traits::nowlike::Nowlike; + +pub fn make_fake_category(id: i32) -> Category { + make_fake_category_builder(id).build().unwrap() +} + +pub fn make_fake_category_builder(id: i32) -> CategoryBuilder { + let mut builder = CategoryBuilder::default(); + builder + .id(id) + .name(String::from("Fake category")) + .balance(0) + .balance_date(DateUtc::now()) + .budget_period(0) + .budget_period_unit(BudgetPeriodUnit::Day) + .budget_quantity(0); + builder +} diff --git a/schist_core/schist_fakes/src/category_transfer.rs b/schist_core/schist_fakes/src/category_transfer.rs new file mode 100644 index 0000000..c28dffb --- /dev/null +++ b/schist_core/schist_fakes/src/category_transfer.rs @@ -0,0 +1,16 @@ +use schist_models::category_transfer::{CategoryTransfer, CategoryTransferBuilder}; + +pub fn make_fake_category_transfer(id: i32) -> CategoryTransfer { + make_fake_category_transfer_builder(id).build().unwrap() +} + +pub fn make_fake_category_transfer_builder(id: i32) -> CategoryTransferBuilder { + let mut builder = CategoryTransferBuilder::default(); + builder + .id(id) + .description(String::from("Fake category transfer")) + .quantity(0) + .from_category_id(0) + .to_category_id(1); + builder +} diff --git a/schist_core/schist_fakes/src/lib.rs b/schist_core/schist_fakes/src/lib.rs new file mode 100644 index 0000000..6e20c31 --- /dev/null +++ b/schist_core/schist_fakes/src/lib.rs @@ -0,0 +1,7 @@ +pub mod account; +pub mod account_transfer; +pub mod budget_drip; +pub mod category; +pub mod category_transfer; +pub mod transaction; +pub mod transaction_categorisation; diff --git a/schist_core/schist_fakes/src/transaction.rs b/schist_core/schist_fakes/src/transaction.rs new file mode 100644 index 0000000..30d1788 --- /dev/null +++ b/schist_core/schist_fakes/src/transaction.rs @@ -0,0 +1,18 @@ +use schist_models::{date_utc::DateUtc, transaction::{Transaction, TransactionBuilder}}; +use schist_traits::nowlike::Nowlike; + +pub fn make_fake_transaction(id: i32) -> Transaction { + make_fake_transaction_builder(id).build().unwrap() +} + +pub fn make_fake_transaction_builder(id: i32) -> TransactionBuilder { + let mut builder = TransactionBuilder::default(); + builder + .id(id) + .description(String::from("Fake category transfer")) + .payee(String::from("Fake payee")) + .quantity(0) + .date(DateUtc::now()) + .account_id(0); + builder +} diff --git a/schist_core/schist_fakes/src/transaction_categorisation.rs b/schist_core/schist_fakes/src/transaction_categorisation.rs new file mode 100644 index 0000000..ce9460c --- /dev/null +++ b/schist_core/schist_fakes/src/transaction_categorisation.rs @@ -0,0 +1,16 @@ +use schist_models::transaction_categorisation::{TransactionCategorisation, TransactionCategorisationBuilder}; + +pub fn make_fake_transaction_categorisation(id: i32) -> TransactionCategorisation { + make_fake_transaction_categorisation_builder(id).build().unwrap() +} + +pub fn make_fake_transaction_categorisation_builder(id: i32) -> TransactionCategorisationBuilder { + let mut builder = TransactionCategorisationBuilder::default(); + builder + .id(id) + .description(String::from("Fake category transfer")) + .quantity(0) + .transaction_id(0) + .category_id(0); + builder +} diff --git a/schist_core/schist_models/Cargo.toml b/schist_core/schist_models/Cargo.toml new file mode 100644 index 0000000..51ac324 --- /dev/null +++ b/schist_core/schist_models/Cargo.toml @@ -0,0 +1,17 @@ +[package] +name = "schist_models" +version = "0.1.0" +edition = "2021" + +[dependencies] +anyhow = { workspace = true } +chrono = { workspace = true, features = ["serde"] } +derive_builder = { workspace = true } +diesel = { workspace = true, features = ["sqlite"] } +diesel_migrations = { workspace = true } +schist_traits = { path = "../schist_traits" } +serde = { workspace = true, features = ["derive"] } + +[dev-dependencies] +libsqlite3-sys = { workspace = true, features = ["bundled"] } +schist_queries = { path = "../schist_queries" } diff --git a/schist_core/schist_models/diesel.toml b/schist_core/schist_models/diesel.toml new file mode 100644 index 0000000..2b8ebd8 --- /dev/null +++ b/schist_core/schist_models/diesel.toml @@ -0,0 +1,9 @@ +# For documentation on how to configure this file, +# see https://diesel.rs/guides/configuring-diesel-cli + +[print_schema] +file = "src/schema.rs" +custom_type_derives = ["diesel::query_builder::QueryId", "Clone"] + +[migrations_directory] +dir = "/home/joeac/src/schist/rust/schist_models/migrations" diff --git a/schist_core/schist_models/migrations/2024-08-31-084439_initial_setup/down.sql b/schist_core/schist_models/migrations/2024-08-31-084439_initial_setup/down.sql new file mode 100644 index 0000000..ba65012 --- /dev/null +++ b/schist_core/schist_models/migrations/2024-08-31-084439_initial_setup/down.sql @@ -0,0 +1,7 @@ +DROP TABLE accounts; +DROP TABLE account_transfers; +DROP TABLE budget_drips; +DROP TABLE categories; +DROP TABLE transaction_categorisations; +DROP TABLE category_transfers; +DROP TABLE transactions; diff --git a/schist_core/schist_models/migrations/2024-08-31-084439_initial_setup/up.sql b/schist_core/schist_models/migrations/2024-08-31-084439_initial_setup/up.sql new file mode 100644 index 0000000..a09bf7b --- /dev/null +++ b/schist_core/schist_models/migrations/2024-08-31-084439_initial_setup/up.sql @@ -0,0 +1,93 @@ +PRAGMA foreign_keys = ON; + +CREATE TABLE accounts( + id INTEGER NOT NULL PRIMARY KEY, + name TEXT NOT NULL, + opening_balance INTEGER NOT NULL, + opening_date TEXT NOT NULL +); + +CREATE TABLE account_transfers( + id INTEGER NOT NULL PRIMARY KEY, + date TEXT NOT NULL, + description TEXT NOT NULL, + quantity INTEGER NOT NULL, + from_account_id INTEGER NOT NULL, + to_account_id INTEGER NOT NULL, + FOREIGN KEY (from_account_id) + REFERENCES accounts (id) + ON UPDATE CASCADE + ON DELETE RESTRICT, + FOREIGN KEY (to_account_id) + REFERENCES accounts (id) + ON UPDATE CASCADE + ON DELETE RESTRICT +); + + +CREATE TABLE budget_drips( + id INTEGER NOT NULL PRIMARY KEY, + category_id INTEGER NOT NULL, + date TEXT NOT NULL, + quantity INTEGER NOT NULL, + FOREIGN KEY (category_id) + REFERENCES categories (id) + ON UPDATE CASCADE + ON DELETE RESTRICT, + UNIQUE(category_id, date) +); + +CREATE TABLE categories( + id INTEGER NOT NULL PRIMARY KEY, + name TEXT NOT NULL, + balance INTEGER NOT NULL, + balance_date TEXT NOT NULL, + budget_period INTEGER NOT NULL, + budget_period_unit TEXT NOT NULL, + budget_quantity INTEGER NOT NULL +); + +CREATE TABLE category_transfers( + id INTEGER NOT NULL PRIMARY KEY, + description TEXT NOT NULL, + quantity INTEGER NOT NULL, + from_category_id INTEGER NOT NULL, + to_category_id INTEGER NOT NULL, + FOREIGN KEY (from_category_id) + REFERENCES categories (id) + ON UPDATE CASCADE + ON DELETE RESTRICT, + FOREIGN KEY (to_category_id) + REFERENCES categories (id) + ON UPDATE CASCADE + ON DELETE RESTRICT +); + +CREATE TABLE transactions( + id INTEGER NOT NULL PRIMARY KEY, + description TEXT NOT NULL, + payee TEXT NOT NULL, + quantity INTEGER NOT NULL, + date TEXT NOT NULL, + account_id INTEGER NOT NULL, + FOREIGN KEY (account_id) + REFERENCES accounts (id) + ON UPDATE CASCADE + ON DELETE RESTRICT +); + +CREATE TABLE transaction_categorisations( + id INTEGER NOT NULL PRIMARY KEY, + description TEXT NOT NULL, + quantity INTEGER NOT NULL, + transaction_id INTEGER NOT NULL, + category_id INTEGER NOT NULL, + FOREIGN KEY (transaction_id) + REFERENCES transactions (id) + ON UPDATE CASCADE + ON DELETE RESTRICT, + FOREIGN KEY (category_id) + REFERENCES categories (id) + ON UPDATE CASCADE + ON DELETE RESTRICT +); diff --git a/schist_core/schist_models/src/account.rs b/schist_core/schist_models/src/account.rs new file mode 100644 index 0000000..05dceae --- /dev/null +++ b/schist_core/schist_models/src/account.rs @@ -0,0 +1,15 @@ +use diesel::prelude::*; +use serde::{Deserialize, Serialize}; + +use crate::date_utc::DateUtc; + +#[derive( + Insertable, Queryable, Identifiable, Selectable, Debug, PartialEq, Clone, Serialize, Deserialize, +)] +#[diesel(table_name = crate::schema::accounts)] +pub struct Account { + pub id: i32, + pub name: String, + pub opening_balance: i32, + pub opening_date: DateUtc, +} diff --git a/schist_core/schist_models/src/account_transfer.rs b/schist_core/schist_models/src/account_transfer.rs new file mode 100644 index 0000000..06c51ac --- /dev/null +++ b/schist_core/schist_models/src/account_transfer.rs @@ -0,0 +1,15 @@ +use derive_builder::Builder; +use diesel::prelude::{Identifiable, Insertable, Queryable, Selectable}; + +use crate::date_utc::DateUtc; + +#[derive(Builder, Queryable, Identifiable, Selectable, Debug, PartialEq, Insertable)] +#[diesel(table_name = crate::schema::account_transfers)] +pub struct AccountTransfer { + pub id: i32, + pub date: DateUtc, + pub description: String, + pub quantity: i32, + pub from_account_id: i32, + pub to_account_id: i32, +} diff --git a/schist_core/schist_models/src/budget_drip.rs b/schist_core/schist_models/src/budget_drip.rs new file mode 100644 index 0000000..2cc2c63 --- /dev/null +++ b/schist_core/schist_models/src/budget_drip.rs @@ -0,0 +1,14 @@ +use derive_builder::Builder; +use diesel::prelude::{Identifiable, Insertable, Queryable, Selectable}; +use serde::{Deserialize, Serialize}; + +use crate::date_utc::DateUtc; + +#[derive(Builder, Clone, Copy, Queryable, Identifiable, Selectable, Debug, PartialEq, Serialize, Deserialize, Insertable)] +#[diesel(table_name = crate::schema::budget_drips)] +pub struct BudgetDrip { + pub id: i32, + pub category_id: i32, + pub date: DateUtc, + pub quantity: i32, +} diff --git a/schist_core/schist_models/src/budget_period_unit.rs b/schist_core/schist_models/src/budget_period_unit.rs new file mode 100644 index 0000000..77d8641 --- /dev/null +++ b/schist_core/schist_models/src/budget_period_unit.rs @@ -0,0 +1,112 @@ +use core::fmt; +use std::str::FromStr; + +use anyhow::{Result, bail}; +use diesel::{backend::Backend, deserialize::{self, FromSql, FromSqlRow}, expression::AsExpression, serialize::{self, ToSql}, sql_types::Text, sqlite::Sqlite}; +use serde::{Deserialize, Serialize}; + +#[derive( + Eq, + PartialEq, + Serialize, + Deserialize, + Clone, + Debug, + AsExpression, + FromSqlRow, +)] +#[diesel(sql_type=Text)] +pub enum BudgetPeriodUnit { + Day, + Month, + Year, +} + +impl fmt::Display for BudgetPeriodUnit { + fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result { + match self { + BudgetPeriodUnit::Day => "DAY".fmt(f), + BudgetPeriodUnit::Month => "MONTH".fmt(f), + BudgetPeriodUnit::Year => "YEAR".fmt(f), + } + } +} + +impl FromStr for BudgetPeriodUnit { + type Err = anyhow::Error; + + fn from_str(s: &str) -> Result<Self> { + match s { + "DAY" => Ok(BudgetPeriodUnit::Day), + "MONTH" => Ok(BudgetPeriodUnit::Month), + "YEAR" => Ok(BudgetPeriodUnit::Year), + _ => bail!("\"{}\" is not a valid budget period unit. Valid values are: DAY, MONTH, YEAR", s), + } + } +} + +impl ToSql<Text, Sqlite> for BudgetPeriodUnit { + fn to_sql<'b>( + &'b self, + out: &mut diesel::serialize::Output<'b, '_, Sqlite>, + ) -> serialize::Result { + out.set_value(self.to_string()); + Ok(serialize::IsNull::No) + } +} + +impl FromSql<Text, Sqlite> for BudgetPeriodUnit { + fn from_sql(bytes: <Sqlite as Backend>::RawValue<'_>) -> deserialize::Result<BudgetPeriodUnit> { + let str = <String as FromSql<Text, Sqlite>>::from_sql(bytes)?; + Ok(str.parse()?) + } +} + +#[cfg(test)] +mod test { + use anyhow::Result; + + use super::BudgetPeriodUnit; + + #[test] + fn display_when_day_then_day() { + assert_eq!("DAY", format!("{}", BudgetPeriodUnit::Day)); + } + + #[test] + fn display_when_month_then_month() { + assert_eq!("MONTH", format!("{}", BudgetPeriodUnit::Month)); + } + + #[test] + fn display_when_year_then_year() { + assert_eq!("YEAR", format!("{}", BudgetPeriodUnit::Year)); + } + + #[test] + fn from_str_when_day_then_day() { + let result: Result<BudgetPeriodUnit> = "DAY".parse(); + assert!(result.is_ok()); + assert_eq!(BudgetPeriodUnit::Day, result.unwrap()); + } + + #[test] + fn from_str_when_month_then_month() { + let result: Result<BudgetPeriodUnit> = "MONTH".parse(); + assert!(result.is_ok()); + assert_eq!(BudgetPeriodUnit::Month, result.unwrap()); + } + + #[test] + fn from_str_when_year_then_year() { + let result: Result<BudgetPeriodUnit> = "YEAR".parse(); + assert!(result.is_ok()); + assert_eq!(BudgetPeriodUnit::Year, result.unwrap()); + } + + #[test] + fn from_str_when_sentence_case_day_then_err() { + let result: Result<BudgetPeriodUnit> = "Day".parse(); + assert!(result.is_err()); + } +} diff --git a/schist_core/schist_models/src/category.rs b/schist_core/schist_models/src/category.rs new file mode 100644 index 0000000..e3c9938 --- /dev/null +++ b/schist_core/schist_models/src/category.rs @@ -0,0 +1,17 @@ +use derive_builder::Builder; +use diesel::prelude::*; +use serde::{Deserialize, Serialize}; + +use crate::{budget_period_unit::BudgetPeriodUnit, date_utc::DateUtc}; + +#[derive(Builder, Clone, Queryable, Identifiable, Selectable, Debug, PartialEq, Serialize, Deserialize, Insertable)] +#[diesel(table_name = crate::schema::categories)] +pub struct Category { + pub id: i32, + pub name: String, + pub balance: i32, + pub balance_date: DateUtc, + pub budget_period: i32, + pub budget_period_unit: BudgetPeriodUnit, + pub budget_quantity: i32, +} diff --git a/schist_core/schist_models/src/category_transfer.rs b/schist_core/schist_models/src/category_transfer.rs new file mode 100644 index 0000000..7b36054 --- /dev/null +++ b/schist_core/schist_models/src/category_transfer.rs @@ -0,0 +1,12 @@ +use derive_builder::Builder; +use diesel::prelude::*; + +#[derive(Builder, Queryable, Identifiable, Selectable, Debug, PartialEq, Insertable)] +#[diesel(table_name = crate::schema::category_transfers)] +pub struct CategoryTransfer { + pub id: i32, + pub description: String, + pub quantity: i32, + pub from_category_id: i32, + pub to_category_id: i32, +} diff --git a/schist_core/schist_models/src/date_utc.rs b/schist_core/schist_models/src/date_utc.rs new file mode 100644 index 0000000..2b36ec0 --- /dev/null +++ b/schist_core/schist_models/src/date_utc.rs @@ -0,0 +1,547 @@ +use std::{borrow::Borrow, fmt, str::FromStr}; + +use anyhow::{Context, Error, Result}; +use chrono::{DateTime, Datelike, NaiveDate, NaiveTime, Utc}; +use diesel::{ + backend::Backend, + deserialize::{self, FromSql, FromSqlRow}, + expression::AsExpression, + serialize::{self, ToSql}, + sql_types::Text, + sqlite::Sqlite, +}; +use serde::{Deserialize, Serialize}; + +use schist_traits::{dateable::Dateable, nowlike::Nowlike}; + +use crate::datetime_utc::DatetimeUtc; + +#[derive( + Eq, + PartialEq, + Ord, + PartialOrd, + Serialize, + Deserialize, + Clone, + Copy, + Debug, + AsExpression, + FromSqlRow, +)] +#[diesel(sql_type=Text)] +pub struct DateUtc { + chrono_datetime_utc: DateTime<Utc>, +} + +impl DateUtc { + pub fn from_ymd(year: i32, month: u32, day: u32) -> Result<Self> { + Ok(Self { + chrono_datetime_utc: NaiveDate::from_ymd_opt(year, month, day) + .with_context(|| format!("failed to construct date from ymd {:04}-{:02}-{:02}", year, month, day))? + .and_hms_opt(0, 0, 0).unwrap() + .and_utc() + }) + } + + pub fn with_hms(&mut self, hour: u32, min: u32, sec: u32) -> Result<DatetimeUtc> { + DatetimeUtc::from_ymd_and_hms( + self.year(), self.month(), self.day(), hour, min, sec, + ) + } + + pub fn day_diff(&self, other: &DateUtc) -> u32 { + self.chrono_datetime_utc + .num_days_from_ce() + .abs_diff(other.chrono_datetime_utc.num_days_from_ce()) + } + + pub fn days_in_month(&self) -> u32 { + self.first_day_in_next_month().day_diff(&self.first_day_in_month()) + } + + pub fn add_days(&self, days: u64) -> Result<Self> { + let chrono_datetime_utc = self + .chrono_datetime_utc + .checked_add_days(chrono::Days::new(days)) + .with_context(|| format!("failed to add {} days to {}", days, self)) + .unwrap(); + Ok(Self { chrono_datetime_utc }) + } + + pub fn sub_days(&self, days: u64) -> Result<Self> { + let chrono_datetime_utc = self + .chrono_datetime_utc + .checked_sub_days(chrono::Days::new(days)) + .with_context(|| format!("failed to subtract {} days from {}", days, self)) + .unwrap(); + Ok(Self { chrono_datetime_utc }) + } + + fn is_day_str(str: &str) -> bool { + if str.len() != 10 { + return false; + } + for (ix, ch) in str.chars().enumerate() { + if ix == 4 || ix == 7 { + if ch != '-' { + return false; + } + } else { + if ch < '0' || ch > '9' { + return false; + } + } + } + return true; + } + + fn first_day_in_next_month(&self) -> Self { + let month = self.month(); + let mut curr_guess: Option<DateTime<Utc>> = None; + let mut curr_add = 1; + while curr_guess.is_none() || curr_guess.unwrap().month() == month { + curr_guess = self.chrono_datetime_utc.checked_add_days(chrono::Days::new(curr_add)); + curr_add = curr_add + 1; + }; + return Self { chrono_datetime_utc: curr_guess.unwrap() }; + } + + fn first_day_in_month(&self) -> Self { + let month = self.month(); + let mut prev_guess = self.clone(); + let mut curr_guess = self.clone(); + let mut curr_subtractor = 1; + while curr_guess.month() == month { + prev_guess = curr_guess.clone(); + + let mut next: Option<DateTime<Utc>> = None; + while next.is_none() { + if curr_subtractor > 31 { + return prev_guess; + } + next = self.chrono_datetime_utc.checked_sub_days(chrono::Days::new(curr_subtractor)); + curr_subtractor = curr_subtractor + 1; + } + let next = next.unwrap(); + + curr_guess = Self { + chrono_datetime_utc: next, + }; + }; + return prev_guess; + } +} + +impl Nowlike for DateUtc { + fn now() -> Self { + let now = Utc::now(); + Self::from_ymd( + now.year(), + now.month(), + now.day(), + ).unwrap() + } +} + +impl Dateable for DateUtc { + fn year(&self) -> i32 { + self.chrono_datetime_utc.year() + } + + fn month(&self) -> u32 { + self.chrono_datetime_utc.month() + } + + fn day(&self) -> u32 { + self.chrono_datetime_utc.day() + } +} + +impl From<DatetimeUtc> for DateUtc { + fn from(datetime_utc: DatetimeUtc) -> Self { + Self::from(&datetime_utc) + } +} + +impl From<&DatetimeUtc> for DateUtc { + fn from(datetime_utc: &DatetimeUtc) -> Self { + Self { + chrono_datetime_utc: datetime_utc + .to_string() + .parse::<DateTime<Utc>>().unwrap() + .with_time(NaiveTime::from_hms_opt(0, 0, 0).unwrap()).unwrap(), + } + } +} + +impl<T> From<T> for DateUtc where T: Borrow<DateTime<Utc>> { + fn from(chrono_datetime_utc: T) -> Self { + Self { + chrono_datetime_utc: chrono_datetime_utc + .borrow() + .with_time(NaiveTime::from_hms_opt(0, 0, 0).unwrap()).unwrap(), + } + } +} + +impl fmt::Display for DateUtc { + /// Converts to a day string, YYYY-MM-DD, in UTC + /// Example: 10.30am 24 Oct 2024 UTC -> "2024-10-24" + fn fmt(&self, f: &mut fmt::Formatter) -> Result<(), fmt::Error> { + let date_str = self.chrono_datetime_utc.to_rfc3339(); + let day_str = date_str + .split_at_checked(10) + .unwrap() + .0; + write!(f, "{}", day_str) + } +} + +impl FromStr for DateUtc { + type Err = anyhow::Error; + + /// Constructs DatetimeUtc from a day string, YYYY-MM-DD, in UTC + /// Example: "2024-10-24" -> Midnight 24 Oct 2024 UTC + fn from_str(date_str: &str) -> Result<Self> { + if !Self::is_day_str(&date_str) { + Err(Error::msg(format!("{} is not a day string in YYYY-MM-DD format", date_str))) + } else { + let mut datetime_str = String::from(date_str); + datetime_str.push_str("T00:00:00.000Z"); + Ok(Self { + chrono_datetime_utc: datetime_str.parse()?, + }) + } + } +} + +impl ToSql<Text, Sqlite> for DateUtc { + fn to_sql<'b>( + &'b self, + out: &mut diesel::serialize::Output<'b, '_, Sqlite>, + ) -> serialize::Result { + out.set_value(self.to_string()); + Ok(serialize::IsNull::No) + } +} + +impl FromSql<Text, Sqlite> for DateUtc { + fn from_sql(bytes: <Sqlite as Backend>::RawValue<'_>) -> deserialize::Result<DateUtc> { + let str = <String as FromSql<Text, Sqlite>>::from_sql(bytes)?; + Ok(str.parse()?) + } +} + +#[cfg(test)] +mod test { + use schist_traits::timeable::Timeable; + + use super::*; + + #[test] + fn when_from_ymd_then_ymd_are_as_expected() { + let date_utc = DateUtc::from_ymd(1970, 11, 23); + + assert!(date_utc.is_ok()); + let date_utc = date_utc.unwrap(); + assert_eq!(date_utc.year(), 1970); + assert_eq!(date_utc.month(), 11); + assert_eq!(date_utc.day(), 23); + } + + #[test] + fn when_from_ymd_then_to_string_as_expected() { + let date_utc = DateUtc::from_ymd(1970, 11, 23); + + assert!(date_utc.is_ok()); + assert_eq!(date_utc.unwrap().to_string(), "1970-11-23"); + } + + #[test] + fn when_from_invalid_ymd_then_err() { + let date_utc_1 = DateUtc::from_ymd(1970, 0, 23); + let date_utc_2 = DateUtc::from_ymd(1970, 13, 23); + let date_utc_3 = DateUtc::from_ymd(1970, 11, 0); + let date_utc_4 = DateUtc::from_ymd(1970, 11, 31); + let date_utc_5 = DateUtc::from_ymd(1971, 2, 29); + + assert!(date_utc_1.is_err()); + assert_eq!(date_utc_1.unwrap_err().to_string(), "failed to construct date from ymd 1970-00-23"); + assert!(date_utc_2.is_err()); + assert_eq!(date_utc_2.unwrap_err().to_string(), "failed to construct date from ymd 1970-13-23"); + assert!(date_utc_3.is_err()); + assert_eq!(date_utc_3.unwrap_err().to_string(), "failed to construct date from ymd 1970-11-00"); + assert!(date_utc_4.is_err()); + assert_eq!(date_utc_4.unwrap_err().to_string(), "failed to construct date from ymd 1970-11-31"); + assert!(date_utc_5.is_err()); + assert_eq!(date_utc_5.unwrap_err().to_string(), "failed to construct date from ymd 1971-02-29"); + } + + #[test] + fn when_from_string_then_ymd_are_as_expected() { + let date_utc: Result<DateUtc> = "1970-11-23".parse(); + + assert!(date_utc.is_ok()); + let date_utc = date_utc.unwrap(); + assert_eq!(date_utc.year(), 1970); + assert_eq!(date_utc.month(), 11); + assert_eq!(date_utc.day(), 23); + } + + #[test] + fn when_from_string_then_to_string_as_expected() { + let date_utc: Result<DateUtc> = "1970-11-23".parse(); + + assert!(date_utc.is_ok()); + assert_eq!(date_utc.unwrap().to_string(), "1970-11-23"); + } + + #[test] + fn when_from_invalid_string_then_err() { + let date_utc_1: Result<DateUtc> = "1970-00-23".parse(); + let date_utc_2: Result<DateUtc> = "1970-13-23".parse(); + let date_utc_3: Result<DateUtc> = "1970-11-0".parse(); + let date_utc_4: Result<DateUtc> = "1970-11-31".parse(); + let date_utc_5: Result<DateUtc> = "1971-02-29".parse(); + + assert!(date_utc_1.is_err()); + assert!(date_utc_2.is_err()); + assert!(date_utc_3.is_err()); + assert!(date_utc_4.is_err()); + assert!(date_utc_5.is_err()); + } + + #[test] + fn with_with_hms_then_ymd_hms_are_as_expected() { + let datetime_utc = DateUtc::from_ymd(1970, 11, 23).unwrap().with_hms(12, 55, 16); + + assert!(datetime_utc.is_ok()); + let datetime_utc = datetime_utc.unwrap(); + assert_eq!(datetime_utc.year(), 1970); + assert_eq!(datetime_utc.month(), 11); + assert_eq!(datetime_utc.day(), 23); + assert_eq!(datetime_utc.hour(), 12); + assert_eq!(datetime_utc.minute(), 55); + assert_eq!(datetime_utc.second(), 16); + } + + #[test] + fn when_with_invalid_hms_then_err() { + let datetime_utc_1 = DateUtc::from_ymd(1970, 11, 23).unwrap().with_hms(25, 0, 0); + let datetime_utc_2 = DateUtc::from_ymd(1970, 11, 23).unwrap().with_hms(0, 61, 0); + let datetime_utc_3 = DateUtc::from_ymd(1970, 11, 23).unwrap().with_hms(0, 0, 61); + + assert!(datetime_utc_1.is_err()); + assert!(datetime_utc_2.is_err()); + assert!(datetime_utc_3.is_err()); + } + + #[test] + fn when_from_datetime_utc_at_different_times_on_same_day_then_eq() { + let datetime_utc1 = DateUtc::from_ymd(1970, 11, 23).unwrap().with_hms(0, 0, 0).unwrap(); + let datetime_utc2 = DateUtc::from_ymd(1970, 11, 23).unwrap().with_hms(23, 59, 59).unwrap(); + let date_utc1 = DateUtc::from(datetime_utc1); + let date_utc2 = DateUtc::from(datetime_utc2); + + assert_eq!(date_utc1, date_utc2); + } + + #[test] + fn given_today_when_day_diff_today_then_return_0() { + let start = DateUtc::from_ymd(2000, 1, 1).unwrap(); + let end = DateUtc::from_ymd(2000, 1, 1).unwrap(); + + let day_diff = start.day_diff(&end); + let neg_day_diff = end.day_diff(&start); + + assert_eq!(day_diff, 0); + assert_eq!(day_diff, neg_day_diff); + } + + #[test] + fn given_today_when_day_diff_tomorrow_then_return_1() { + let start = DateUtc::from_ymd(2000, 1, 1).unwrap(); + let end = DateUtc::from_ymd(2000, 1, 2).unwrap(); + + let day_diff = start.day_diff(&end); + let neg_day_diff = end.day_diff(&start); + + assert_eq!(day_diff, 1); + assert_eq!(day_diff, neg_day_diff); + } + + #[test] + fn when_any_date_in_2024_then_first_day_in_month_is_correct() { + let jan_1_2024: DateUtc = "2024-01-01".parse().unwrap(); + for date in get_dates_in_jan_2024() { + assert_eq!(jan_1_2024, date.first_day_in_month()); + } + + let feb_1_2024: DateUtc = "2024-02-01".parse().unwrap(); + for date in get_dates_in_feb_2024() { + assert_eq!(feb_1_2024, date.first_day_in_month()); + } + + let mar_1_2024: DateUtc = "2024-03-01".parse().unwrap(); + for date in get_dates_in_mar_2024() { + assert_eq!(mar_1_2024, date.first_day_in_month()); + } + + let apr_1_2024: DateUtc = "2024-04-01".parse().unwrap(); + for date in get_dates_in_apr_2024() { + assert_eq!(apr_1_2024, date.first_day_in_month()); + } + + let may_1_2024: DateUtc = "2024-05-01".parse().unwrap(); + for date in get_dates_in_may_2024() { + assert_eq!(may_1_2024, date.first_day_in_month()); + } + + let jun_1_2024: DateUtc = "2024-06-01".parse().unwrap(); + for date in get_dates_in_jun_2024() { + assert_eq!(jun_1_2024, date.first_day_in_month()); + } + + let jul_1_2024: DateUtc = "2024-07-01".parse().unwrap(); + for date in get_dates_in_jul_2024() { + assert_eq!(jul_1_2024, date.first_day_in_month()); + } + + let aug_1_2024: DateUtc = "2024-08-01".parse().unwrap(); + for date in get_dates_in_aug_2024() { + assert_eq!(aug_1_2024, date.first_day_in_month()); + } + + let sep_1_2024: DateUtc = "2024-09-01".parse().unwrap(); + for date in get_dates_in_sep_2024() { + assert_eq!(sep_1_2024, date.first_day_in_month()); + } + + let oct_1_2024: DateUtc = "2024-10-01".parse().unwrap(); + for date in get_dates_in_oct_2024() { + assert_eq!(oct_1_2024, date.first_day_in_month()); + } + + let nov_1_2024: DateUtc = "2024-11-01".parse().unwrap(); + for date in get_dates_in_nov_2024() { + assert_eq!(nov_1_2024, date.first_day_in_month()); + } + + let dec_1_2024: DateUtc = "2024-12-01".parse().unwrap(); + for date in get_dates_in_dec_2024() { + assert_eq!(dec_1_2024, date.first_day_in_month()); + } + } + + #[test] + fn when_any_date_in_2024_then_first_day_in_next_month_is_correct() { + let feb_1_2024: DateUtc = "2024-02-01".parse().unwrap(); + for date in get_dates_in_jan_2024() { + assert_eq!(feb_1_2024, date.first_day_in_next_month()); + } + + let mar_1_2024: DateUtc = "2024-03-01".parse().unwrap(); + for date in get_dates_in_feb_2024() { + assert_eq!(mar_1_2024, date.first_day_in_next_month()); + } + + let apr_1_2024: DateUtc = "2024-04-01".parse().unwrap(); + for date in get_dates_in_mar_2024() { + assert_eq!(apr_1_2024, date.first_day_in_next_month()); + } + + let may_1_2024: DateUtc = "2024-05-01".parse().unwrap(); + for date in get_dates_in_apr_2024() { + assert_eq!(may_1_2024, date.first_day_in_next_month()); + } + + let jun_1_2024: DateUtc = "2024-06-01".parse().unwrap(); + for date in get_dates_in_may_2024() { + assert_eq!(jun_1_2024, date.first_day_in_next_month()); + } + + let jul_1_2024: DateUtc = "2024-07-01".parse().unwrap(); + for date in get_dates_in_jun_2024() { + assert_eq!(jul_1_2024, date.first_day_in_next_month()); + } + + let aug_1_2024: DateUtc = "2024-08-01".parse().unwrap(); + for date in get_dates_in_jul_2024() { + assert_eq!(aug_1_2024, date.first_day_in_next_month()); + } + + let sep_1_2024: DateUtc = "2024-09-01".parse().unwrap(); + for date in get_dates_in_aug_2024() { + assert_eq!(sep_1_2024, date.first_day_in_next_month()); + } + + let oct_1_2024: DateUtc = "2024-10-01".parse().unwrap(); + for date in get_dates_in_sep_2024() { + assert_eq!(oct_1_2024, date.first_day_in_next_month()); + } + + let nov_1_2024: DateUtc = "2024-11-01".parse().unwrap(); + for date in get_dates_in_oct_2024() { + assert_eq!(nov_1_2024, date.first_day_in_next_month()); + } + + let dec_1_2024: DateUtc = "2024-12-01".parse().unwrap(); + for date in get_dates_in_nov_2024() { + assert_eq!(dec_1_2024, date.first_day_in_next_month()); + } + + let jan_1_2025: DateUtc = "2025-01-01".parse().unwrap(); + for date in get_dates_in_dec_2024() { + assert_eq!(jan_1_2025, date.first_day_in_next_month()); + } + } + + fn get_dates_in_jan_2024() -> Vec<DateUtc> { + (1..31).map(|day| format!("2024-01-{:02}", day).parse::<DateUtc>().unwrap()).collect::<Vec<DateUtc>>() + } + + fn get_dates_in_feb_2024() -> Vec<DateUtc> { + (1..29).map(|day| format!("2024-02-{:02}", day).parse::<DateUtc>().unwrap()).collect::<Vec<DateUtc>>() + } + + fn get_dates_in_mar_2024() -> Vec<DateUtc> { + (1..31).map(|day| format!("2024-03-{:02}", day).parse::<DateUtc>().unwrap()).collect::<Vec<DateUtc>>() + } + + fn get_dates_in_apr_2024() -> Vec<DateUtc> { + (1..30).map(|day| format!("2024-04-{:02}", day).parse::<DateUtc>().unwrap()).collect::<Vec<DateUtc>>() + } + + fn get_dates_in_may_2024() -> Vec<DateUtc> { + (1..31).map(|day| format!("2024-05-{:02}", day).parse::<DateUtc>().unwrap()).collect::<Vec<DateUtc>>() + } + + fn get_dates_in_jun_2024() -> Vec<DateUtc> { + (1..30).map(|day| format!("2024-06-{:02}", day).parse::<DateUtc>().unwrap()).collect::<Vec<DateUtc>>() + } + + fn get_dates_in_jul_2024() -> Vec<DateUtc> { + (1..31).map(|day| format!("2024-07-{:02}", day).parse::<DateUtc>().unwrap()).collect::<Vec<DateUtc>>() + } + + fn get_dates_in_aug_2024() -> Vec<DateUtc> { + (1..31).map(|day| format!("2024-08-{:02}", day).parse::<DateUtc>().unwrap()).collect::<Vec<DateUtc>>() + } + + fn get_dates_in_sep_2024() -> Vec<DateUtc> { + (1..30).map(|day| format!("2024-09-{:02}", day).parse::<DateUtc>().unwrap()).collect::<Vec<DateUtc>>() + } + + fn get_dates_in_oct_2024() -> Vec<DateUtc> { + (1..31).map(|day| format!("2024-10-{:02}", day).parse::<DateUtc>().unwrap()).collect::<Vec<DateUtc>>() + } + + fn get_dates_in_nov_2024() -> Vec<DateUtc> { + (1..30).map(|day| format!("2024-11-{:02}", day).parse::<DateUtc>().unwrap()).collect::<Vec<DateUtc>>() + } + + fn get_dates_in_dec_2024() -> Vec<DateUtc> { + (1..31).map(|day| format!("2024-12-{:02}", day).parse::<DateUtc>().unwrap()).collect::<Vec<DateUtc>>() + } +} diff --git a/schist_core/schist_models/src/datetime_utc.rs b/schist_core/schist_models/src/datetime_utc.rs new file mode 100644 index 0000000..9b63145 --- /dev/null +++ b/schist_core/schist_models/src/datetime_utc.rs @@ -0,0 +1,223 @@ +use std::{fmt::Display, str::FromStr}; + +use anyhow::{Context, Result}; +use chrono::{DateTime, Datelike as ChronoDatelike, NaiveDate, Timelike as ChronoTimelike, Utc}; +use diesel::{ + backend::Backend, + deserialize::{self, FromSql, FromSqlRow}, + expression::AsExpression, + serialize::{self, ToSql}, + sql_types::Text, + sqlite::Sqlite, +}; +use serde::{Deserialize, Serialize}; + +use schist_traits::{dateable::Dateable, nowlike::Nowlike, timeable::Timeable}; + +#[derive( + PartialEq, + Eq, + PartialOrd, + Ord, + Serialize, + Deserialize, + Clone, + Debug, + AsExpression, + FromSqlRow, +)] +#[diesel(sql_type=Text)] +pub struct DatetimeUtc { + chrono_datetime_utc: DateTime<Utc>, +} + +impl DatetimeUtc { + pub fn from_ymd_and_hms(year: i32, month: u32, day: u32, hour: u32, min: u32, sec: u32) -> Result<Self> { + Ok(Self { chrono_datetime_utc: NaiveDate + ::from_ymd_opt(year, month, day) + .with_context(|| format!("failed to construct datetime from ymd {:04}-{:02}-{:02}", year, month, day))? + .and_hms_opt(hour, min, sec) + .with_context(|| format!("failed to construct datetime from hms {:02}:{:02}:{:02}", hour, min, sec))? + .and_utc() + }) + } +} + +impl FromStr for DatetimeUtc { + type Err = anyhow::Error; + + /// Constructs DatetimeUtc from a date string in ISO format, in UTC + /// Example: "2024-10-24T10:30:00.000" -> 10.30am 24 Oct 2024 UTC + fn from_str(datetime_str: &str) -> Result<Self> { + let chrono_datetime_utc = datetime_str.parse::<DateTime<Utc>>()?; + Ok(Self { + chrono_datetime_utc, + }) + } +} + +impl Display for DatetimeUtc { + /// Converts to a date string in ISO format, in UTC + /// Example: 10.30am 24 Oct 2024 UTC -> "2024-10-24T10:30:00.000Z" + fn fmt(&self, f: &mut std::fmt::Formatter) -> std::fmt::Result { + write!(f, "{}", self.chrono_datetime_utc.to_rfc3339_opts(chrono::SecondsFormat::Millis, true)) + } +} + +impl Nowlike for DatetimeUtc { + fn now() -> Self { + let now = Utc::now(); + Self::from_ymd_and_hms( + now.year(), + now.month(), + now.day(), + now.hour(), + now.minute(), + now.second(), + ).unwrap() + } +} + +impl Dateable for DatetimeUtc { + fn year(&self) -> i32 { + self.chrono_datetime_utc.year() + } + + fn month(&self) -> u32 { + self.chrono_datetime_utc.month() + } + + fn day(&self) -> u32 { + self.chrono_datetime_utc.day() + } +} + +impl Timeable for DatetimeUtc { + fn hour(&self) -> u32 { + self.chrono_datetime_utc.hour() + } + + fn minute(&self) -> u32 { + self.chrono_datetime_utc.minute() + } + + fn second(&self) -> u32 { + self.chrono_datetime_utc.second() + } +} + +impl ToSql<Text, Sqlite> for DatetimeUtc { + fn to_sql<'b>( + &'b self, + out: &mut diesel::serialize::Output<'b, '_, Sqlite>, + ) -> serialize::Result { + out.set_value::<String>(self.to_string()); + Ok(serialize::IsNull::No) + } +} + +impl FromSql<Text, Sqlite> for DatetimeUtc { + fn from_sql(bytes: <Sqlite as Backend>::RawValue<'_>) -> deserialize::Result<DatetimeUtc> { + let str = <String as FromSql<Text, Sqlite>>::from_sql(bytes)?; + Ok(str.parse()?) + } +} + +#[cfg(test)] +mod test { + use super::*; + + #[test] + fn when_from_ymd_and_hms_then_ymd_and_hms_are_as_expected() { + let datetime_utc = DatetimeUtc::from_ymd_and_hms(1970, 11, 23, 12, 55, 16); + + assert!(datetime_utc.is_ok()); + let date_utc = datetime_utc.unwrap(); + assert_eq!(date_utc.year(), 1970); + assert_eq!(date_utc.month(), 11); + assert_eq!(date_utc.day(), 23); + assert_eq!(date_utc.hour(), 12); + assert_eq!(date_utc.minute(), 55); + assert_eq!(date_utc.second(), 16); + } + + #[test] + fn when_from_ymd_and_hms_then_to_string_as_expected() { + let date_utc = DatetimeUtc::from_ymd_and_hms(1970, 11, 23, 12, 55, 16); + + assert!(date_utc.is_ok()); + assert_eq!(date_utc.unwrap().to_string(), "1970-11-23T12:55:16.000Z") + } + + #[test] + fn when_with_invalid_ymd_and_hms_then_err() { + let datetime_utc_1 = DatetimeUtc::from_ymd_and_hms(1970, 0, 23, 12, 55, 16); + let datetime_utc_2 = DatetimeUtc::from_ymd_and_hms(1970, 13, 23, 12, 55, 16); + let datetime_utc_3 = DatetimeUtc::from_ymd_and_hms(1970, 11, 0, 12, 55, 16); + let datetime_utc_4 = DatetimeUtc::from_ymd_and_hms(1970, 11, 31, 12, 55, 16); + let datetime_utc_5 = DatetimeUtc::from_ymd_and_hms(1971, 2, 29, 12, 55, 16); + let datetime_utc_6 = DatetimeUtc::from_ymd_and_hms(1970, 11, 23, 25, 0, 0); + let datetime_utc_7 = DatetimeUtc::from_ymd_and_hms(1970, 11, 23, 0, 61, 0); + let datetime_utc_8 = DatetimeUtc::from_ymd_and_hms(1970, 11, 23, 0, 0, 61); + + assert!(datetime_utc_1.is_err()); + assert_eq!(datetime_utc_1.unwrap_err().to_string(), "failed to construct datetime from ymd 1970-00-23"); + assert!(datetime_utc_2.is_err()); + assert_eq!(datetime_utc_2.unwrap_err().to_string(), "failed to construct datetime from ymd 1970-13-23"); + assert!(datetime_utc_3.is_err()); + assert_eq!(datetime_utc_3.unwrap_err().to_string(), "failed to construct datetime from ymd 1970-11-00"); + assert!(datetime_utc_4.is_err()); + assert_eq!(datetime_utc_4.unwrap_err().to_string(), "failed to construct datetime from ymd 1970-11-31"); + assert!(datetime_utc_5.is_err()); + assert_eq!(datetime_utc_5.unwrap_err().to_string(), "failed to construct datetime from ymd 1971-02-29"); + assert!(datetime_utc_6.is_err()); + assert_eq!(datetime_utc_6.unwrap_err().to_string(), "failed to construct datetime from hms 25:00:00"); + assert!(datetime_utc_7.is_err()); + assert_eq!(datetime_utc_7.unwrap_err().to_string(), "failed to construct datetime from hms 00:61:00"); + assert!(datetime_utc_8.is_err()); + assert_eq!(datetime_utc_8.unwrap_err().to_string(), "failed to construct datetime from hms 00:00:61"); + } + + #[test] + fn when_from_string_then_ymd_and_hms_are_as_expected() { + let date_utc: Result<DatetimeUtc> = "1970-11-23T12:55:16.000Z".parse(); + + assert!(date_utc.is_ok()); + let datetime_utc = date_utc.unwrap(); + assert_eq!(datetime_utc.year(), 1970); + assert_eq!(datetime_utc.month(), 11); + assert_eq!(datetime_utc.day(), 23); + assert_eq!(datetime_utc.hour(), 12); + assert_eq!(datetime_utc.minute(), 55); + assert_eq!(datetime_utc.second(), 16); + } + + #[test] + fn when_from_string_then_to_string_as_expected() { + let datetime_utc: Result<DatetimeUtc> = "1970-11-23T12:55:16.000Z".parse(); + + assert!(datetime_utc.is_ok()); + assert_eq!(datetime_utc.unwrap().to_string(), "1970-11-23T12:55:16.000Z") + } + + #[test] + fn when_from_invalid_string_then_err() { + let datetime_utc_1: Result<DatetimeUtc> = "1970-00-23T00:00:00.000Z".parse(); + let datetime_utc_2: Result<DatetimeUtc> = "1970-13-23T00:00:00.000Z".parse(); + let datetime_utc_3: Result<DatetimeUtc> = "1970-11-0T00:00:00.000Z".parse(); + let datetime_utc_4: Result<DatetimeUtc> = "1970-11-31T00:00:00.000Z".parse(); + let datetime_utc_5: Result<DatetimeUtc> = "1971-02-29T00:00:00.000Z".parse(); + let datetime_utc_6: Result<DatetimeUtc> = "1970-11-23T25:00:00.000Z".parse(); + let datetime_utc_7: Result<DatetimeUtc> = "1970-11-23T00:61:00.000Z".parse(); + let datetime_utc_8: Result<DatetimeUtc> = "1970-11-23T00:00:61.000Z".parse(); + + assert!(datetime_utc_1.is_err()); + assert!(datetime_utc_2.is_err()); + assert!(datetime_utc_3.is_err()); + assert!(datetime_utc_4.is_err()); + assert!(datetime_utc_5.is_err()); + assert!(datetime_utc_6.is_err()); + assert!(datetime_utc_7.is_err()); + assert!(datetime_utc_8.is_err()); + } +} diff --git a/schist_core/schist_models/src/lib.rs b/schist_core/schist_models/src/lib.rs new file mode 100644 index 0000000..d9e93c4 --- /dev/null +++ b/schist_core/schist_models/src/lib.rs @@ -0,0 +1,12 @@ +pub mod account; +pub mod account_transfer; +pub mod budget_drip; +pub mod budget_period_unit; +pub mod category; +pub mod category_transfer; +pub mod date_utc; +pub mod datetime_utc; +pub mod migrations; +pub mod transaction; +pub mod transaction_categorisation; +pub mod schema; diff --git a/schist_core/schist_models/src/migrations.rs b/schist_core/schist_models/src/migrations.rs new file mode 100644 index 0000000..d10cb83 --- /dev/null +++ b/schist_core/schist_models/src/migrations.rs @@ -0,0 +1,3 @@ +use diesel_migrations::{embed_migrations, EmbeddedMigrations}; + +pub const MIGRATIONS: EmbeddedMigrations = embed_migrations!("migrations"); diff --git a/schist_core/schist_models/src/schema.rs b/schist_core/schist_models/src/schema.rs new file mode 100644 index 0000000..1dd1391 --- /dev/null +++ b/schist_core/schist_models/src/schema.rs @@ -0,0 +1,88 @@ +// @generated automatically by Diesel CLI. + +diesel::table! { + account_transfers (id) { + id -> Integer, + date -> Text, + description -> Text, + quantity -> Integer, + from_account_id -> Integer, + to_account_id -> Integer, + } +} + +diesel::table! { + accounts (id) { + id -> Integer, + name -> Text, + opening_balance -> Integer, + opening_date -> Text, + } +} + +diesel::table! { + budget_drips (id) { + id -> Integer, + category_id -> Integer, + date -> Text, + quantity -> Integer, + } +} + +diesel::table! { + categories (id) { + id -> Integer, + name -> Text, + balance -> Integer, + balance_date -> Text, + budget_period -> Integer, + budget_period_unit -> Text, + budget_quantity -> Integer, + } +} + +diesel::table! { + category_transfers (id) { + id -> Integer, + description -> Text, + quantity -> Integer, + from_category_id -> Integer, + to_category_id -> Integer, + } +} + +diesel::table! { + transaction_categorisations (id) { + id -> Integer, + description -> Text, + quantity -> Integer, + transaction_id -> Integer, + category_id -> Integer, + } +} + +diesel::table! { + transactions (id) { + id -> Integer, + description -> Text, + payee -> Text, + quantity -> Integer, + date -> Text, + account_id -> Integer, + } +} + +diesel::joinable!(budget_drips -> categories (category_id)); +diesel::joinable!(transaction_categorisations -> categories (category_id)); +diesel::joinable!(transaction_categorisations -> transactions (transaction_id)); +diesel::joinable!(transactions -> accounts (account_id)); + +diesel::allow_tables_to_appear_in_same_query!( + account_transfers, + accounts, + budget_drips, + categories, + category_transfers, + transaction_categorisations, + transactions, +); diff --git a/schist_core/schist_models/src/transaction.rs b/schist_core/schist_models/src/transaction.rs new file mode 100644 index 0000000..9e983f5 --- /dev/null +++ b/schist_core/schist_models/src/transaction.rs @@ -0,0 +1,16 @@ +use derive_builder::Builder; +use diesel::prelude::{Associations, Identifiable, Insertable, Queryable, Selectable}; + +use crate::{account::Account, date_utc::DateUtc}; + +#[derive(Builder, Clone, Queryable, Identifiable, Selectable, Insertable, Associations, Debug, PartialEq)] +#[diesel(table_name = crate::schema::transactions)] +#[diesel(belongs_to(Account))] +pub struct Transaction { + pub id: i32, + pub description: String, + pub payee: String, + pub quantity: i32, + pub date: DateUtc, + pub account_id: i32, +} diff --git a/schist_core/schist_models/src/transaction_categorisation.rs b/schist_core/schist_models/src/transaction_categorisation.rs new file mode 100644 index 0000000..c6ac6ea --- /dev/null +++ b/schist_core/schist_models/src/transaction_categorisation.rs @@ -0,0 +1,12 @@ +use derive_builder::Builder; +use diesel::prelude::*; + +#[derive(Builder, Queryable, Identifiable, Selectable, Insertable, Debug, PartialEq)] +#[diesel(table_name = crate::schema::transaction_categorisations)] +pub struct TransactionCategorisation { + pub id: i32, + pub description: String, + pub quantity: i32, + pub transaction_id: i32, + pub category_id: i32, +} diff --git a/schist_core/schist_models/tests/budget_period_unit.rs b/schist_core/schist_models/tests/budget_period_unit.rs new file mode 100644 index 0000000..6e8928a --- /dev/null +++ b/schist_core/schist_models/tests/budget_period_unit.rs @@ -0,0 +1,86 @@ +mod common; + +use anyhow::Result; +use common::test_context::TestContext; +use diesel::{sql_query, Connection, ExpressionMethods, QueryDsl, RunQueryDsl, Selectable, SelectableHelper, SqliteConnection}; +use diesel::prelude::{Insertable, Queryable}; +use schist_models::budget_period_unit::BudgetPeriodUnit; + +diesel::table! { + budget_period_unit_test (id) { + id -> Integer, + budget_period_unit -> Text, + } +} + +#[derive(Insertable, Queryable, Selectable, Clone)] +#[diesel(table_name = budget_period_unit_test)] +pub struct BudgetPeriodUnitTest { + pub id: i32, + pub budget_period_unit: BudgetPeriodUnit, +} + +fn new_test_context() -> Result<TestContext> { + let test_context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&test_context.db_url)?; + let query = "CREATE TABLE budget_period_unit_test(id INTEGER PRIMARY KEY, budget_period_unit TEXT);"; + sql_query(query).execute(connection)?; + Ok(test_context) +} + +fn insert_budget_period_unit(id: i32, budget_period_unit: &BudgetPeriodUnit, connection: &mut SqliteConnection) -> Result<usize> { + let model = BudgetPeriodUnitTest { + id, + budget_period_unit: budget_period_unit.clone(), + }; + let num_rows_inserted = diesel::insert_into(budget_period_unit_test::dsl::budget_period_unit_test) + .values(&[model]) + .execute(connection)?; + Ok(num_rows_inserted) +} + +fn get_budget_period_unit(id: i32, connection: &mut SqliteConnection) -> Result<BudgetPeriodUnitTest> { + let budget_period_unit: Vec<BudgetPeriodUnitTest> = budget_period_unit_test::dsl::budget_period_unit_test + .select(BudgetPeriodUnitTest::as_select()) + .filter(budget_period_unit_test::id.eq(id)) + .load(connection)?; + Ok(budget_period_unit[0].clone()) +} + +#[test] +fn when_insert_day_and_select_then_returns_day() { + let test_context = new_test_context().unwrap(); + let connection = &mut SqliteConnection::establish(&test_context.db_url).unwrap(); + + let budget_period_unit = BudgetPeriodUnit::Day; + let num_rows_inserted = insert_budget_period_unit(0, &budget_period_unit, connection); + + assert!(num_rows_inserted.is_ok()); + let num_rows_inserted = num_rows_inserted.unwrap(); + assert_eq!(num_rows_inserted, 1); + + let budget_period_unit_returned = get_budget_period_unit(0, connection); + + assert!(budget_period_unit_returned.is_ok()); + let budget_period_unit_returned = budget_period_unit_returned.unwrap(); + assert_eq!(BudgetPeriodUnit::Day, budget_period_unit_returned.budget_period_unit); +} + +#[test] +fn when_insert_month_and_select_then_returns_month() { + let test_context = new_test_context().unwrap(); + let connection = &mut SqliteConnection::establish(&test_context.db_url).unwrap(); + + let budget_period_unit = BudgetPeriodUnit::Month; + let num_rows_inserted = insert_budget_period_unit(0, &budget_period_unit, connection); + + assert!(num_rows_inserted.is_ok()); + let num_rows_inserted = num_rows_inserted.unwrap(); + assert_eq!(num_rows_inserted, 1); + + let budget_period_unit_returned = get_budget_period_unit(0, connection); + + assert!(budget_period_unit_returned.is_ok()); + let budget_period_unit_returned = budget_period_unit_returned.unwrap(); + assert_eq!(BudgetPeriodUnit::Month, budget_period_unit_returned.budget_period_unit); +} diff --git a/schist_core/schist_models/tests/common/mod.rs b/schist_core/schist_models/tests/common/mod.rs new file mode 100644 index 0000000..5ff67da --- /dev/null +++ b/schist_core/schist_models/tests/common/mod.rs @@ -0,0 +1 @@ +pub mod test_context; diff --git a/schist_core/schist_models/tests/common/test_context.rs b/schist_core/schist_models/tests/common/test_context.rs new file mode 100644 index 0000000..fb4112d --- /dev/null +++ b/schist_core/schist_models/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_queries::clear::clear; +use schist_models::migrations::MIGRATIONS; + +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/schist_core/schist_models/tests/date_utc.rs b/schist_core/schist_models/tests/date_utc.rs new file mode 100644 index 0000000..c5ad82e --- /dev/null +++ b/schist_core/schist_models/tests/date_utc.rs @@ -0,0 +1,61 @@ +mod common; + +use anyhow::Result; +use common::test_context::TestContext; +use diesel::{sql_query, Connection, ExpressionMethods, QueryDsl, RunQueryDsl, Selectable, SelectableHelper, SqliteConnection}; +use diesel::prelude::{Insertable, Queryable}; +use schist_models::date_utc::DateUtc; +use schist_traits::nowlike::Nowlike; + +diesel::table! { + date_utc_test (id) { + id -> Integer, + date_utc -> Text, + } +} + +#[derive(Insertable, Queryable, Selectable, Clone)] +#[diesel(table_name = date_utc_test)] +pub struct DateUtcTest { + pub id: i32, + pub date_utc: DateUtc, +} + +fn new_test_context() -> Result<TestContext> { + let test_context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&test_context.db_url)?; + let query = "CREATE TABLE date_utc_test(id INTEGER PRIMARY KEY, date_utc TEXT);"; + sql_query(query).execute(connection)?; + Ok(test_context) +} + +fn insert_date_utc(id: i32, date_utc: &DateUtc, connection: &mut SqliteConnection) -> Result<usize> { + let model = DateUtcTest { + id, + date_utc: date_utc.clone(), + }; + let num_rows_inserted = diesel::insert_into(date_utc_test::dsl::date_utc_test) + .values(&[model]) + .execute(connection)?; + Ok(num_rows_inserted) +} + +fn get_date_utc(id: i32, connection: &mut SqliteConnection) -> Result<DateUtcTest> { + let date_utc: Vec<DateUtcTest> = date_utc_test::dsl::date_utc_test + .select(DateUtcTest::as_select()) + .filter(date_utc_test::id.eq(id)) + .load(connection)?; + Ok(date_utc[0].clone()) +} + +#[test] +fn when_insert_date_utc_and_select_then_returns_original_value() -> Result<()> { + let date_utc = DateUtc::now(); + let test_context = new_test_context()?; + let connection = &mut SqliteConnection::establish(&test_context.db_url)?; + let num_rows_inserted = insert_date_utc(0, &date_utc, connection)?; + assert_eq!(num_rows_inserted, 1); + let date_utc_returned = get_date_utc(0, connection)?; + assert_eq!(date_utc_returned.date_utc, date_utc); + Ok(()) +} diff --git a/schist_core/schist_models/tests/datetime_utc.rs b/schist_core/schist_models/tests/datetime_utc.rs new file mode 100644 index 0000000..27182ff --- /dev/null +++ b/schist_core/schist_models/tests/datetime_utc.rs @@ -0,0 +1,62 @@ +mod common; + +use anyhow::Result; +use common::test_context::TestContext; +use diesel::{sql_query, Connection, ExpressionMethods, QueryDsl, RunQueryDsl, Selectable, SelectableHelper, SqliteConnection}; +use diesel::prelude::{Insertable, Queryable}; +use schist_models::datetime_utc::DatetimeUtc; +use schist_traits::nowlike::Nowlike; + + +diesel::table! { + datetime_utc_test (id) { + id -> Integer, + datetime_utc -> Text, + } +} + +#[derive(Insertable, Queryable, Selectable, Clone)] +#[diesel(table_name = datetime_utc_test)] +pub struct DatetimeUtcTest { + pub id: i32, + pub datetime_utc: DatetimeUtc, +} + +fn new_test_context() -> Result<TestContext> { + let test_context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&test_context.db_url)?; + let query = "CREATE TABLE datetime_utc_test(id INTEGER PRIMARY KEY, datetime_utc TEXT);"; + sql_query(query).execute(connection)?; + Ok(test_context) +} + +fn insert_datetime_utc(id: i32, datetime_utc: &DatetimeUtc, connection: &mut SqliteConnection) -> Result<usize> { + let model = DatetimeUtcTest { + id, + datetime_utc: datetime_utc.clone(), + }; + let num_rows_inserted = diesel::insert_into(datetime_utc_test::dsl::datetime_utc_test) + .values(&[model]) + .execute(connection)?; + Ok(num_rows_inserted) +} + +fn get_datetime_utc(id: i32, connection: &mut SqliteConnection) -> Result<DatetimeUtcTest> { + let datetime_utc: Vec<DatetimeUtcTest> = datetime_utc_test::dsl::datetime_utc_test + .select(DatetimeUtcTest::as_select()) + .filter(datetime_utc_test::id.eq(id)) + .load(connection)?; + Ok(datetime_utc[0].clone()) +} + +#[test] +fn when_insert_datetime_utc_and_select_then_returns_original_value() -> Result<()> { + let datetime_utc = DatetimeUtc::now(); + let test_context = new_test_context()?; + let connection = &mut SqliteConnection::establish(&test_context.db_url)?; + let num_rows_inserted = insert_datetime_utc(0, &datetime_utc, connection)?; + assert_eq!(num_rows_inserted, 1); + let datetime_utc_returned = get_datetime_utc(0, connection)?; + assert_eq!(datetime_utc_returned.datetime_utc, datetime_utc); + Ok(()) +} diff --git a/schist_core/schist_models/user_data.sqlite b/schist_core/schist_models/user_data.sqlite Binary files differnew file mode 100644 index 0000000..a428ff0 --- /dev/null +++ b/schist_core/schist_models/user_data.sqlite diff --git a/schist_core/schist_queries/Cargo.toml b/schist_core/schist_queries/Cargo.toml new file mode 100644 index 0000000..2cd0c03 --- /dev/null +++ b/schist_core/schist_queries/Cargo.toml @@ -0,0 +1,17 @@ +[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" } + +[dev-dependencies] +diesel_migrations = { workspace = true } +libsqlite3-sys = { workspace = true, features = ["bundled"] } +schist_fakes = { path = "../schist_fakes" } diff --git a/schist_core/schist_queries/src/account_transfers.rs b/schist_core/schist_queries/src/account_transfers.rs new file mode 100644 index 0000000..070bdc3 --- /dev/null +++ b/schist_core/schist_queries/src/account_transfers.rs @@ -0,0 +1,40 @@ +use anyhow::{Context, Result}; +use diesel::{QueryDsl, RunQueryDsl, SelectableHelper, SqliteConnection}; +use schist_models::account_transfer::AccountTransfer; +use schist_schema::schema::account_transfers::dsl::account_transfers as account_transfers_table; + +pub fn delete_all_account_transfers(connection: &mut SqliteConnection) -> Result<usize> { + let num_rows_deleted = diesel::delete(account_transfers_table) + .execute(connection) + .with_context(|| "failed to delete all account_transfers")?; + Ok(num_rows_deleted) +} + +pub fn get_all_account_transfers(connection: &mut SqliteConnection) -> Result<Vec<AccountTransfer>> { + let all_account_transfers = account_transfers_table.select(AccountTransfer::as_select()) + .load(connection) + .with_context(|| "failed to get all account transfers")?; + Ok(all_account_transfers) +} + +pub fn insert_account_transfers( + account_transfers: &[AccountTransfer], + connection: &mut SqliteConnection, +) -> Result<usize> { + let num_account_transfers_inserted = diesel::insert_into(account_transfers_table) + .values(account_transfers) + .execute(connection) + .with_context(|| insert_err_msg(account_transfers))?; + Ok(num_account_transfers_inserted) +} + +fn insert_err_msg(account_transfers: &[AccountTransfer]) -> String { + format!( + "failed to insert account transfers: [{}]", + account_transfers + .iter() + .map(|a| a.id.to_string()) + .collect::<Vec<String>>() + .join(", ") + ) +} diff --git a/schist_core/schist_queries/src/accounts.rs b/schist_core/schist_queries/src/accounts.rs new file mode 100644 index 0000000..ead281a --- /dev/null +++ b/schist_core/schist_queries/src/accounts.rs @@ -0,0 +1,40 @@ +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/schist_core/schist_queries/src/budget_drips.rs b/schist_core/schist_queries/src/budget_drips.rs new file mode 100644 index 0000000..b6f9372 --- /dev/null +++ b/schist_core/schist_queries/src/budget_drips.rs @@ -0,0 +1,40 @@ +use anyhow::{Context, Result}; +use diesel::{QueryDsl, RunQueryDsl, SelectableHelper, SqliteConnection}; +use schist_models::budget_drip::BudgetDrip; +use schist_schema::schema::budget_drips::dsl::budget_drips as budget_drips_table; + +pub fn delete_all_budget_drips(connection: &mut SqliteConnection) -> Result<usize> { + let num_rows_deleted = diesel::delete(budget_drips_table) + .execute(connection) + .with_context(|| "failed to delete all budget drips")?; + Ok(num_rows_deleted) +} + +pub fn get_all_budget_drips(connection: &mut SqliteConnection) -> Result<Vec<BudgetDrip>> { + let all_budget_drips = budget_drips_table.select(BudgetDrip::as_select()) + .load(connection) + .with_context(|| "failed to get all budget drips")?; + Ok(all_budget_drips) +} + +pub fn insert_budget_drips( + budget_drips: &[BudgetDrip], + connection: &mut SqliteConnection, +) -> Result<usize> { + let num_budget_drips_inserted = diesel::insert_into(budget_drips_table) + .values(budget_drips) + .execute(connection) + .with_context(|| insert_err_msg(budget_drips))?; + Ok(num_budget_drips_inserted) +} + +fn insert_err_msg(budget_drips: &[BudgetDrip]) -> String { + format!( + "failed to insert budget drips: [{}]", + budget_drips + .iter() + .map(|a| a.id.to_string()) + .collect::<Vec<String>>() + .join(", ") + ) +} diff --git a/schist_core/schist_queries/src/categories.rs b/schist_core/schist_queries/src/categories.rs new file mode 100644 index 0000000..f4a9747 --- /dev/null +++ b/schist_core/schist_queries/src/categories.rs @@ -0,0 +1,49 @@ +use anyhow::{Context, Result}; +use diesel::{QueryDsl, RunQueryDsl, SelectableHelper, SqliteConnection}; +use schist_models::category::Category; +use schist_schema::schema::categories::{self as categories_schema, dsl::categories as categories_table}; + +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_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 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/schist_core/schist_queries/src/category_transfers.rs b/schist_core/schist_queries/src/category_transfers.rs new file mode 100644 index 0000000..33a53ea --- /dev/null +++ b/schist_core/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/schist_core/schist_queries/src/clear.rs b/schist_core/schist_queries/src/clear.rs new file mode 100644 index 0000000..7cee699 --- /dev/null +++ b/schist_core/schist_queries/src/clear.rs @@ -0,0 +1,16 @@ +use anyhow::Result; +use diesel::SqliteConnection; + +use crate::{ + accounts::delete_all_accounts, budget_drips::delete_all_budget_drips, categories::delete_all_categories, category_transfers::delete_all_category_transfers, transaction_categorisations::delete_all_transaction_categorisations, transactions::delete_all_transactions +}; + +pub fn clear(connection: &mut SqliteConnection) -> Result<()> { + delete_all_accounts(connection)?; + delete_all_budget_drips(connection)?; + delete_all_categories(connection)?; + delete_all_category_transfers(connection)?; + delete_all_transactions(connection)?; + delete_all_transaction_categorisations(connection)?; + Ok(()) +} diff --git a/schist_core/schist_queries/src/lib.rs b/schist_core/schist_queries/src/lib.rs new file mode 100644 index 0000000..a1a5c20 --- /dev/null +++ b/schist_core/schist_queries/src/lib.rs @@ -0,0 +1,10 @@ +pub mod accounts; +pub mod account_transfers; +pub mod budget_drips; +pub mod categories; +pub mod category_transfers; +pub mod clear; +pub mod transactions; +pub mod transaction_categorisations; + +mod utils; diff --git a/schist_core/schist_queries/src/transaction_categorisations.rs b/schist_core/schist_queries/src/transaction_categorisations.rs new file mode 100644 index 0000000..320782d --- /dev/null +++ b/schist_core/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/schist_core/schist_queries/src/transactions.rs b/schist_core/schist_queries/src/transactions.rs new file mode 100644 index 0000000..70b14e7 --- /dev/null +++ b/schist_core/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/schist_core/schist_queries/src/utils/calculate_budgets_accrual.rs b/schist_core/schist_queries/src/utils/calculate_budgets_accrual.rs new file mode 100644 index 0000000..ca82953 --- /dev/null +++ b/schist_core/schist_queries/src/utils/calculate_budgets_accrual.rs @@ -0,0 +1,72 @@ +use anyhow::Result; +use schist_models::{ + budget_drip::BudgetDrip, + category::Category, +}; + +pub fn calculate_budgets_accrual(category: &Category, budget_drips: &[BudgetDrip]) -> Result<i64> { + Ok(budget_drips + .iter() + .filter(|bd| bd.category_id == category.id) + .map(|bd| Into::<i64>::into(bd.quantity)) + .sum() + ) +} + +#[cfg(test)] +mod test { + use schist_fakes::{budget_drip::make_fake_budget_drip_builder, category::make_fake_category}; + + use super::calculate_budgets_accrual; + + #[test] + fn when_no_budget_drips_then_return_zero() { + let category = make_fake_category(0); + let budget_drips = []; + + let result = calculate_budgets_accrual(&category, &budget_drips); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 0_i64); + } + + #[test] + fn when_one_budget_drip_for_another_category_then_return_zero() { + let category = make_fake_category(0); + let budget_drips = [ + make_fake_budget_drip_builder(0).quantity(100).category_id(1).build().unwrap(), + ]; + + let result = calculate_budgets_accrual(&category, &budget_drips); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 0_i64); + } + + #[test] + fn when_one_budget_drip_then_return_quantity() { + let category = make_fake_category(0); + let budget_drips = [ + make_fake_budget_drip_builder(0).quantity(100).category_id(0).build().unwrap(), + ]; + + let result = calculate_budgets_accrual(&category, &budget_drips); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 100); + } + + #[test] + fn when_two_budget_drips_then_return_sum() { + let category = make_fake_category(0); + let budget_drips = [ + make_fake_budget_drip_builder(0).quantity(100).category_id(0).build().unwrap(), + make_fake_budget_drip_builder(1).quantity(120).category_id(0).build().unwrap(), + ]; + + let result = calculate_budgets_accrual(&category, &budget_drips); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 220); + } +} diff --git a/schist_core/schist_queries/src/utils/calculate_category_balance.rs b/schist_core/schist_queries/src/utils/calculate_category_balance.rs new file mode 100644 index 0000000..73e80cd --- /dev/null +++ b/schist_core/schist_queries/src/utils/calculate_category_balance.rs @@ -0,0 +1,130 @@ +use anyhow::Result; +use schist_models::{budget_drip::BudgetDrip, category::Category}; + +use super::{calculate_budgets_accrual, 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_drips: &[BudgetDrip], +) -> 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_drips)?; + + Ok( + transaction_sum + + budget_accruals_sum + + category_transfer_to_sum + - category_transfer_from_sum + ) +} + +#[cfg(test)] +mod test { + use schist_fakes::{budget_drip::make_fake_budget_drip_builder, category::make_fake_category}; + + use super::calculate_category_balance; + + #[test] + fn when_no_data_then_returns_zero() { + let category = make_fake_category(0); + + 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 = make_fake_category(0); + let transaction_sums = [(1, 100)]; + let category_transfer_from_sums = [(1, 91)]; + let category_transfer_to_sums = [(2, 79)]; + let budget_drips = [ + make_fake_budget_drip_builder(0).quantity(100).category_id(1).build().unwrap(), + ]; + + let result = calculate_category_balance( + &category, + &transaction_sums, + &category_transfer_from_sums, + &category_transfer_to_sums, + &budget_drips, + ); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 0_i64); + } + + #[test] + fn when_transaction_sum_is_zero_then_returns_zero() { + let category = make_fake_category(0); + let transaction_sums = [(0, 0)]; + let category_transfer_from_sums = []; + let category_transfer_to_sums = []; + let budget_drips = []; + + let result = calculate_category_balance( + &category, + &transaction_sums, + &category_transfer_from_sums, + &category_transfer_to_sums, + &budget_drips, + ); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 0_i64); + } + + #[test] + fn when_category_transfers_are_zero_then_returns_zero() { + let category = make_fake_category(0); + let transaction_sums = []; + let category_transfer_from_sums = [(0, 0)]; + let category_transfer_to_sums = [(0, 0)]; + let budget_drips = []; + + let result = calculate_category_balance( + &category, + &transaction_sums, + &category_transfer_from_sums, + &category_transfer_to_sums, + &budget_drips, + ); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 0_i64); + } + + #[test] + fn when_some_data_then_calculates_as_expected() { + let category = make_fake_category(0); + let transaction_sums = [(0, 100)]; + let category_transfer_from_sums = [(0, 91)]; + let category_transfer_to_sums = [(0, 79)]; + let budget_drips = [ + make_fake_budget_drip_builder(0).quantity(77).category_id(0).build().unwrap(), + make_fake_budget_drip_builder(0).quantity(77).category_id(0).build().unwrap(), + make_fake_budget_drip_builder(0).quantity(77).category_id(0).build().unwrap(), + make_fake_budget_drip_builder(0).quantity(77).category_id(0).build().unwrap(), + make_fake_budget_drip_builder(0).quantity(77).category_id(0).build().unwrap(), + ]; + + let result = calculate_category_balance( + &category, + &transaction_sums, + &category_transfer_from_sums, + &category_transfer_to_sums, + &budget_drips, + ); + + let expected_result = 100 + (-91) + 79 + 77 * 5; + assert!(result.is_ok()); + assert_eq!(result.unwrap(), expected_result); + } +} diff --git a/schist_core/schist_queries/src/utils/find_by_id_or.rs b/schist_core/schist_queries/src/utils/find_by_id_or.rs new file mode 100644 index 0000000..8335e5c --- /dev/null +++ b/schist_core/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/schist_core/schist_queries/src/utils/mod.rs b/schist_core/schist_queries/src/utils/mod.rs new file mode 100644 index 0000000..f2653d8 --- /dev/null +++ b/schist_core/schist_queries/src/utils/mod.rs @@ -0,0 +1,8 @@ +mod calculate_budgets_accrual; +mod calculate_category_balance; +mod find_by_id_or; + +use calculate_budgets_accrual::calculate_budgets_accrual; + +pub use calculate_category_balance::calculate_category_balance; +pub use find_by_id_or::find_by_id_or; diff --git a/schist_core/schist_queries/tests/account.rs b/schist_core/schist_queries/tests/account.rs new file mode 100644 index 0000000..e3bb7db --- /dev/null +++ b/schist_core/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/schist_core/schist_queries/tests/account_transfers.rs b/schist_core/schist_queries/tests/account_transfers.rs new file mode 100644 index 0000000..13f7db7 --- /dev/null +++ b/schist_core/schist_queries/tests/account_transfers.rs @@ -0,0 +1,71 @@ +mod common; + +use common::test_context::TestContext; +use diesel::{Connection, SqliteConnection}; +use schist_fakes::{account::make_fake_account, account_transfer::make_fake_account_transfer}; +use schist_queries as queries; + +#[test] +fn when_insert_account_transfers_then_returns_num_account_transfers() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let account_transfers = [make_fake_account_transfer(0), make_fake_account_transfer(1)]; + let accounts = [make_fake_account(0), make_fake_account(1)]; + let _ = queries::accounts::insert_accounts(&accounts, connection).unwrap(); + + let num_account_transfers_inserted = queries::account_transfers::insert_account_transfers( + &account_transfers, + connection, + ).unwrap(); + + assert_eq!(num_account_transfers_inserted, 2); +} + +#[test] +fn given_account_transfers_when_get_all_account_transfers_then_return_account_transfers() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let account_transfers = [make_fake_account_transfer(0), make_fake_account_transfer(1)]; + let accounts = [make_fake_account(0), make_fake_account(1)]; + let _ = queries::accounts::insert_accounts(&accounts, connection).unwrap(); + let _ = queries::account_transfers::insert_account_transfers(&account_transfers, connection).unwrap(); + + let account_transfers_in_db = queries::account_transfers::get_all_account_transfers(connection).unwrap(); + + assert_eq!(account_transfers_in_db.len(), account_transfers.len()); + for account_transfer in &account_transfers { + assert!(account_transfers_in_db.iter().any(|account_transfer_in_db| account_transfer_in_db.eq(&account_transfer))); + } + for account_transfer_in_db in account_transfers_in_db { + assert!(account_transfers.iter().any(|account_transfer| account_transfer.eq(&account_transfer_in_db))); + } +} + +#[test] +fn given_account_transfers_when_delete_all_account_transfers_then_return_num_of_account_transfers() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let account_transfers = [make_fake_account_transfer(0), make_fake_account_transfer(1)]; + let accounts = [make_fake_account(0), make_fake_account(1)]; + let _ = queries::accounts::insert_accounts(&accounts, connection).unwrap(); + let _ = queries::account_transfers::insert_account_transfers(&account_transfers, connection).unwrap(); + + let num_account_transfers_deleted = queries::account_transfers::delete_all_account_transfers(connection).unwrap(); + + assert_eq!(num_account_transfers_deleted, account_transfers.len()); +} + +#[test] +fn given_deleted_account_transfers_when_get_all_account_transfers_then_return_empty() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let account_transfers = [make_fake_account_transfer(0), make_fake_account_transfer(1)]; + let accounts = [make_fake_account(0), make_fake_account(1)]; + let _ = queries::accounts::insert_accounts(&accounts, connection).unwrap(); + let _ = queries::account_transfers::insert_account_transfers(&account_transfers, connection).unwrap(); + let _ = queries::account_transfers::delete_all_account_transfers(connection).unwrap(); + + let account_transfers_returned = queries::account_transfers::get_all_account_transfers(connection).unwrap(); + + assert_eq!(account_transfers_returned.len(), 0); +} diff --git a/schist_core/schist_queries/tests/budget_drips.rs b/schist_core/schist_queries/tests/budget_drips.rs new file mode 100644 index 0000000..ac33885 --- /dev/null +++ b/schist_core/schist_queries/tests/budget_drips.rs @@ -0,0 +1,186 @@ +mod common; + +use common::test_context::TestContext; +use diesel::{Connection, SqliteConnection}; +use schist_fakes::{budget_drip::make_fake_budget_drip_builder, category::make_fake_category}; +use schist_models::{category::Category, date_utc::DateUtc}; +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); + categories +} + +#[test] +fn when_insert_budget_drips_then_returns_num_budget_drips() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let _categories = given_categories(connection); + let budget_drips = [ + make_fake_budget_drip_builder(0) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .category_id(0) + .build().unwrap(), + make_fake_budget_drip_builder(1) + .date(DateUtc::from_ymd(2020, 12, 26).unwrap()) + .category_id(0) + .build().unwrap(), + ]; + + let num_budget_drips_inserted = queries::budget_drips::insert_budget_drips(&budget_drips, connection).unwrap(); + + assert_eq!(num_budget_drips_inserted, 2); +} + +#[test] +fn given_budget_drip_when_insert_budget_drip_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_drips_1 = [ + make_fake_budget_drip_builder(0) + .category_id(0) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .build().unwrap() + ]; + let _ = queries::budget_drips::insert_budget_drips(&budget_drips_1, connection); + + let budget_drips_2 = [ + make_fake_budget_drip_builder(1) + .category_id(0) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .build().unwrap() + ]; + let result = queries::budget_drips::insert_budget_drips(&budget_drips_2, connection); + + assert!(result.is_err()); +} + +#[test] +fn given_budget_drip_when_insert_budget_drip_different_category_and_same_date_then_return_num_budget_drips_inserted() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let _categories = given_categories(connection); + + let budget_drips_1 = [ + make_fake_budget_drip_builder(0) + .category_id(0) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .build().unwrap() + ]; + let _ = queries::budget_drips::insert_budget_drips(&budget_drips_1, connection); + + let budget_drips_2 = [ + make_fake_budget_drip_builder(1) + .category_id(1) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .build().unwrap() + ]; + let result = queries::budget_drips::insert_budget_drips(&budget_drips_2, connection); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 1); +} + +#[test] +fn given_budget_drip_when_insert_budget_drip_same_category_and_different_date_then_return_num_budget_drips_inserted() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let _categories = given_categories(connection); + + let budget_drips_1 = [ + make_fake_budget_drip_builder(0) + .category_id(0) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .build().unwrap() + ]; + let _ = queries::budget_drips::insert_budget_drips(&budget_drips_1, connection); + + let budget_drips_2 = [ + make_fake_budget_drip_builder(1) + .category_id(0) + .date(DateUtc::from_ymd(2020, 12, 26).unwrap()) + .build().unwrap() + ]; + let result = queries::budget_drips::insert_budget_drips(&budget_drips_2, connection); + + assert!(result.is_ok()); + assert_eq!(result.unwrap(), 1); +} + +#[test] +fn given_budget_drips_when_get_all_budget_drips_then_return_budget_drips() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let _categories = given_categories(connection); + let budget_drips = [ + make_fake_budget_drip_builder(0) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .category_id(0) + .build().unwrap(), + make_fake_budget_drip_builder(1) + .date(DateUtc::from_ymd(2020, 12, 26).unwrap()) + .category_id(0) + .build().unwrap(), + ]; + let _ = queries::budget_drips::insert_budget_drips(&budget_drips, connection).unwrap(); + + let budget_drips_in_db = queries::budget_drips::get_all_budget_drips(connection).unwrap(); + + assert_eq!(budget_drips_in_db.len(), budget_drips.len()); + assert_eq!(budget_drips_in_db[0], budget_drips[0]); + assert_eq!(budget_drips_in_db[1], budget_drips[1]); + for budget_drip in &budget_drips { + assert!(budget_drips_in_db.iter().any(|budget_drip_in_db| budget_drip_in_db.eq(&budget_drip))); + } + for budget_drip_in_db in budget_drips_in_db { + assert!(budget_drips.iter().any(|budget_drip| budget_drip.eq(&budget_drip_in_db))); + } +} + +#[test] +fn given_budget_drips_when_delete_all_budget_drips_then_return_num_of_budget_drips() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let _categories = given_categories(connection); + let budget_drips = [ + make_fake_budget_drip_builder(0) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .category_id(0) + .build().unwrap(), + make_fake_budget_drip_builder(1) + .date(DateUtc::from_ymd(2020, 12, 26).unwrap()) + .category_id(0) + .build().unwrap(), + ]; + let _ = queries::budget_drips::insert_budget_drips(&budget_drips, connection).unwrap(); + + let num_budget_drips_deleted = queries::budget_drips::delete_all_budget_drips(connection).unwrap(); + + assert_eq!(num_budget_drips_deleted, budget_drips.len()); +} + +#[test] +fn given_deleted_budget_drips_when_get_all_budget_drips_then_return_empty() { + let context = TestContext::new(); + let connection = &mut SqliteConnection::establish(&context.db_url).unwrap(); + let _categories = given_categories(connection); + let budget_drips = [ + make_fake_budget_drip_builder(0) + .date(DateUtc::from_ymd(2020, 12, 25).unwrap()) + .category_id(0) + .build().unwrap(), + make_fake_budget_drip_builder(1) + .date(DateUtc::from_ymd(2020, 12, 26).unwrap()) + .category_id(0) + .build().unwrap(), + ]; + let _ = queries::budget_drips::insert_budget_drips(&budget_drips, connection).unwrap(); + let _ = queries::budget_drips::delete_all_budget_drips(connection).unwrap(); + + let budget_drips_returned = queries::budget_drips::get_all_budget_drips(connection).unwrap(); + + assert_eq!(budget_drips_returned.len(), 0); +} diff --git a/schist_core/schist_queries/tests/categories.rs b/schist_core/schist_queries/tests/categories.rs new file mode 100644 index 0000000..915e853 --- /dev/null +++ b/schist_core/schist_queries/tests/categories.rs @@ -0,0 +1,81 @@ +mod common; + +use common::test_context::TestContext; +use diesel::{Connection, SqliteConnection}; +use schist_fakes::category::make_fake_category; +use schist_models::category::Category; +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 +} + +#[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_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)); + } +} diff --git a/schist_core/schist_queries/tests/category_transfers.rs b/schist_core/schist_queries/tests/category_transfers.rs new file mode 100644 index 0000000..f37a668 --- /dev/null +++ b/schist_core/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/schist_core/schist_queries/tests/common/mod.rs b/schist_core/schist_queries/tests/common/mod.rs new file mode 100644 index 0000000..5ff67da --- /dev/null +++ b/schist_core/schist_queries/tests/common/mod.rs @@ -0,0 +1 @@ +pub mod test_context; diff --git a/schist_core/schist_queries/tests/common/test_context.rs b/schist_core/schist_queries/tests/common/test_context.rs new file mode 100644 index 0000000..183cd9b --- /dev/null +++ b/schist_core/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/schist_core/schist_queries/tests/transaction_categorisations.rs b/schist_core/schist_queries/tests/transaction_categorisations.rs new file mode 100644 index 0000000..13864b4 --- /dev/null +++ b/schist_core/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/schist_core/schist_queries/tests/transactions.rs b/schist_core/schist_queries/tests/transactions.rs new file mode 100644 index 0000000..5e57074 --- /dev/null +++ b/schist_core/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); +} diff --git a/schist_core/schist_schema/Cargo.toml b/schist_core/schist_schema/Cargo.toml new file mode 100644 index 0000000..0af3470 --- /dev/null +++ b/schist_core/schist_schema/Cargo.toml @@ -0,0 +1,7 @@ +[package] +name = "schist_schema" +version = "0.1.0" +edition = "2021" + +[dependencies] +schist_models = { path = "../schist_models" } diff --git a/schist_core/schist_schema/src/lib.rs b/schist_core/schist_schema/src/lib.rs new file mode 100644 index 0000000..c60d7d4 --- /dev/null +++ b/schist_core/schist_schema/src/lib.rs @@ -0,0 +1,2 @@ +pub use schist_models::schema; +pub use schist_models::migrations; diff --git a/schist_core/schist_traits/Cargo.toml b/schist_core/schist_traits/Cargo.toml new file mode 100644 index 0000000..1515eda --- /dev/null +++ b/schist_core/schist_traits/Cargo.toml @@ -0,0 +1,4 @@ +[package] +name = "schist_traits" +version = "0.1.0" +edition = "2021" diff --git a/schist_core/schist_traits/src/dateable.rs b/schist_core/schist_traits/src/dateable.rs new file mode 100644 index 0000000..05f2d74 --- /dev/null +++ b/schist_core/schist_traits/src/dateable.rs @@ -0,0 +1,5 @@ +pub trait Dateable { + fn year(&self) -> i32; + fn month(&self) -> u32; + fn day(&self) -> u32; +} diff --git a/schist_core/schist_traits/src/lib.rs b/schist_core/schist_traits/src/lib.rs new file mode 100644 index 0000000..ab5caca --- /dev/null +++ b/schist_core/schist_traits/src/lib.rs @@ -0,0 +1,3 @@ +pub mod dateable; +pub mod nowlike; +pub mod timeable; diff --git a/schist_core/schist_traits/src/nowlike.rs b/schist_core/schist_traits/src/nowlike.rs new file mode 100644 index 0000000..bae2309 --- /dev/null +++ b/schist_core/schist_traits/src/nowlike.rs @@ -0,0 +1,3 @@ +pub trait Nowlike { + fn now() -> Self; +} diff --git a/schist_core/schist_traits/src/timeable.rs b/schist_core/schist_traits/src/timeable.rs new file mode 100644 index 0000000..f6f9244 --- /dev/null +++ b/schist_core/schist_traits/src/timeable.rs @@ -0,0 +1,5 @@ +pub trait Timeable { + fn hour(&self) -> u32; + fn minute(&self) -> u32; + fn second(&self) -> u32; +} |
