summaryrefslogtreecommitdiff
path: root/schist_core
diff options
context:
space:
mode:
authorJoe Carstairs <jcarstairs@scottlogic.com>2025-07-28 13:22:53 +0100
committerJoe Carstairs <jcarstairs@scottlogic.com>2025-07-28 13:30:27 +0100
commit2361adeb0ca77904138e3466d51c2f44cf6df613 (patch)
treef6e7192908f99fab62061913d8a1e9d2b918414b /schist_core
parentc7ed1e61ce2218be0b9e60577454124c4e43a366 (diff)
Tidier folders
Diffstat (limited to 'schist_core')
-rw-r--r--schist_core/schist_fakes/Cargo.toml8
-rw-r--r--schist_core/schist_fakes/src/account.rs11
-rw-r--r--schist_core/schist_fakes/src/account_transfer.rs18
-rw-r--r--schist_core/schist_fakes/src/budget_drip.rs16
-rw-r--r--schist_core/schist_fakes/src/category.rs19
-rw-r--r--schist_core/schist_fakes/src/category_transfer.rs16
-rw-r--r--schist_core/schist_fakes/src/lib.rs7
-rw-r--r--schist_core/schist_fakes/src/transaction.rs18
-rw-r--r--schist_core/schist_fakes/src/transaction_categorisation.rs16
-rw-r--r--schist_core/schist_models/Cargo.toml17
-rw-r--r--schist_core/schist_models/diesel.toml9
-rw-r--r--schist_core/schist_models/migrations/2024-08-31-084439_initial_setup/down.sql7
-rw-r--r--schist_core/schist_models/migrations/2024-08-31-084439_initial_setup/up.sql93
-rw-r--r--schist_core/schist_models/src/account.rs15
-rw-r--r--schist_core/schist_models/src/account_transfer.rs15
-rw-r--r--schist_core/schist_models/src/budget_drip.rs14
-rw-r--r--schist_core/schist_models/src/budget_period_unit.rs112
-rw-r--r--schist_core/schist_models/src/category.rs17
-rw-r--r--schist_core/schist_models/src/category_transfer.rs12
-rw-r--r--schist_core/schist_models/src/date_utc.rs547
-rw-r--r--schist_core/schist_models/src/datetime_utc.rs223
-rw-r--r--schist_core/schist_models/src/lib.rs12
-rw-r--r--schist_core/schist_models/src/migrations.rs3
-rw-r--r--schist_core/schist_models/src/schema.rs88
-rw-r--r--schist_core/schist_models/src/transaction.rs16
-rw-r--r--schist_core/schist_models/src/transaction_categorisation.rs12
-rw-r--r--schist_core/schist_models/tests/budget_period_unit.rs86
-rw-r--r--schist_core/schist_models/tests/common/mod.rs1
-rw-r--r--schist_core/schist_models/tests/common/test_context.rs44
-rw-r--r--schist_core/schist_models/tests/date_utc.rs61
-rw-r--r--schist_core/schist_models/tests/datetime_utc.rs62
-rw-r--r--schist_core/schist_models/user_data.sqlitebin0 -> 45056 bytes
-rw-r--r--schist_core/schist_queries/Cargo.toml17
-rw-r--r--schist_core/schist_queries/src/account_transfers.rs40
-rw-r--r--schist_core/schist_queries/src/accounts.rs40
-rw-r--r--schist_core/schist_queries/src/budget_drips.rs40
-rw-r--r--schist_core/schist_queries/src/categories.rs49
-rw-r--r--schist_core/schist_queries/src/category_transfers.rs74
-rw-r--r--schist_core/schist_queries/src/clear.rs16
-rw-r--r--schist_core/schist_queries/src/lib.rs10
-rw-r--r--schist_core/schist_queries/src/transaction_categorisations.rs59
-rw-r--r--schist_core/schist_queries/src/transactions.rs41
-rw-r--r--schist_core/schist_queries/src/utils/calculate_budgets_accrual.rs72
-rw-r--r--schist_core/schist_queries/src/utils/calculate_category_balance.rs130
-rw-r--r--schist_core/schist_queries/src/utils/find_by_id_or.rs41
-rw-r--r--schist_core/schist_queries/src/utils/mod.rs8
-rw-r--r--schist_core/schist_queries/tests/account.rs60
-rw-r--r--schist_core/schist_queries/tests/account_transfers.rs71
-rw-r--r--schist_core/schist_queries/tests/budget_drips.rs186
-rw-r--r--schist_core/schist_queries/tests/categories.rs81
-rw-r--r--schist_core/schist_queries/tests/category_transfers.rs126
-rw-r--r--schist_core/schist_queries/tests/common/mod.rs1
-rw-r--r--schist_core/schist_queries/tests/common/test_context.rs44
-rw-r--r--schist_core/schist_queries/tests/transaction_categorisations.rs130
-rw-r--r--schist_core/schist_queries/tests/transactions.rs80
-rw-r--r--schist_core/schist_schema/Cargo.toml7
-rw-r--r--schist_core/schist_schema/src/lib.rs2
-rw-r--r--schist_core/schist_traits/Cargo.toml4
-rw-r--r--schist_core/schist_traits/src/dateable.rs5
-rw-r--r--schist_core/schist_traits/src/lib.rs3
-rw-r--r--schist_core/schist_traits/src/nowlike.rs3
-rw-r--r--schist_core/schist_traits/src/timeable.rs5
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
new file mode 100644
index 0000000..a428ff0
--- /dev/null
+++ b/schist_core/schist_models/user_data.sqlite
Binary files differ
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;
+}