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