Tidier folders

1 files changed, 80 insertions, 0 deletions

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);
+}