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