Schist queries for Account Transfers

3 files changed, 112 insertions, 0 deletions

diff --git a/rust/schist_queries/src/account_transfers.rs b/rust/schist_queries/src/account_transfers.rs
new file mode 100644
index 0000000..070bdc3
--- /dev/null
+++ b/rust/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/rust/schist_queries/src/lib.rs b/rust/schist_queries/src/lib.rs
index 941b219..a1a5c20 100644
--- a/rust/schist_queries/src/lib.rs
+++ b/rust/schist_queries/src/lib.rs
@@ -1,4 +1,5 @@
 pub mod accounts;
+pub mod account_transfers;
 pub mod budget_drips;
 pub mod categories;
 pub mod category_transfers;
diff --git a/rust/schist_queries/tests/account_transfers.rs b/rust/schist_queries/tests/account_transfers.rs
new file mode 100644
index 0000000..13f7db7
--- /dev/null
+++ b/rust/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);
+}