Check/MagicCarpetProject/MagicCarpetTests/StoragesContractsTests/SalaryStorageContractTests.cs

using MagicCarpetContracts.DataModels;
using MagicCarpetDatabase.Implementations;
using MagicCarpetDatabase.Models;
using Microsoft.EntityFrameworkCore;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;

namespace MagicCarpetTests.StoragesContracts;

[TestFixture]
internal class SalaryStorageContractTests : BaseStorageContractTest
{
    private SalaryStorageContract _salaryStorageContract;
    private Employee _employee;

    [SetUp]
    public void SetUp()
    {
        _salaryStorageContract = new SalaryStorageContract(MagicCarpetDbContext);
        _employee = InsertEmployeeToDatabaseAndReturn();
    }

    [TearDown]
    public void TearDown()
    {
        MagicCarpetDbContext.Database.ExecuteSqlRaw("TRUNCATE \"Salaries\" CASCADE;");
        MagicCarpetDbContext.Database.ExecuteSqlRaw("TRUNCATE \"Employees\" CASCADE;");
    }

    [Test]
    public void Try_GetList_WhenHaveRecords_Test()
    {
        var salary = InsertSalaryToDatabaseAndReturn(_employee.Id, employeeSalary: 100);
        InsertSalaryToDatabaseAndReturn(_employee.Id);
        InsertSalaryToDatabaseAndReturn(_employee.Id);
        var list = _salaryStorageContract.GetList(DateTime.UtcNow.AddDays(-10), DateTime.UtcNow.AddDays(10));
        Assert.That(list, Is.Not.Null);
        Assert.That(list, Has.Count.EqualTo(3));
        AssertElement(list.First(), salary);
    }

    [Test]
    public void Try_GetList_WhenNoRecords_Test()
    {
        var list = _salaryStorageContract.GetList(DateTime.UtcNow.AddDays(-10), DateTime.UtcNow.AddDays(10));
        Assert.That(list, Is.Not.Null);
        Assert.That(list, Is.Empty);
    }

    [Test]
    public void Try_GetList_OnlyInDatePeriod_Test()
    {
        InsertSalaryToDatabaseAndReturn(_employee.Id, salaryDate: DateTime.UtcNow.AddDays(-2));
        InsertSalaryToDatabaseAndReturn(_employee.Id, salaryDate: DateTime.UtcNow.AddDays(-1).AddMinutes(-5));
        InsertSalaryToDatabaseAndReturn(_employee.Id, salaryDate: DateTime.UtcNow.AddDays(-1).AddMinutes(5));
        InsertSalaryToDatabaseAndReturn(_employee.Id, salaryDate: DateTime.UtcNow.AddDays(1).AddMinutes(-5));
        InsertSalaryToDatabaseAndReturn(_employee.Id, salaryDate: DateTime.UtcNow.AddDays(1).AddMinutes(5));
        InsertSalaryToDatabaseAndReturn(_employee.Id, salaryDate: DateTime.UtcNow.AddDays(-2));
        var list = _salaryStorageContract.GetList(DateTime.UtcNow.AddDays(-1), DateTime.UtcNow.AddDays(1));
        Assert.That(list, Is.Not.Null);
        Assert.Multiple(() =>
        {
            Assert.That(list, Has.Count.EqualTo(2));
        });
    }

    [Test]
    public void Try_GetList_ByEmployee_Test()
    {
        var employee = InsertEmployeeToDatabaseAndReturn("name 2");
        InsertSalaryToDatabaseAndReturn(_employee.Id);
        InsertSalaryToDatabaseAndReturn(_employee.Id);
        InsertSalaryToDatabaseAndReturn(employee.Id);
        var list = _salaryStorageContract.GetList(DateTime.UtcNow.AddDays(-1), DateTime.UtcNow.AddDays(1), _employee.Id);
        Assert.That(list, Is.Not.Null);
        Assert.Multiple(() =>
        {
            Assert.That(list, Has.Count.EqualTo(2));
            Assert.That(list.All(x => x.EmployeeId == _employee.Id));
        });
    }

    [Test]
    public void Try_GetList_ByEmployeeOnlyInDatePeriod_Test()
    {
        var employee = InsertEmployeeToDatabaseAndReturn("name 2");
        InsertSalaryToDatabaseAndReturn(_employee.Id, salaryDate: DateTime.UtcNow.AddDays(-2));
        InsertSalaryToDatabaseAndReturn(_employee.Id, salaryDate: DateTime.UtcNow.AddDays(-1).AddMinutes(5));
        InsertSalaryToDatabaseAndReturn(employee.Id, salaryDate: DateTime.UtcNow.AddDays(-1).AddMinutes(5));
        InsertSalaryToDatabaseAndReturn(_employee.Id, salaryDate: DateTime.UtcNow.AddDays(1).AddMinutes(-5));
        InsertSalaryToDatabaseAndReturn(employee.Id, salaryDate: DateTime.UtcNow.AddDays(1).AddMinutes(-5));
        InsertSalaryToDatabaseAndReturn(_employee.Id, salaryDate: DateTime.UtcNow.AddDays(-2));
        var list = _salaryStorageContract.GetList(DateTime.UtcNow.AddDays(-1), DateTime.UtcNow.AddDays(1), _employee.Id);
        Assert.That(list, Is.Not.Null);
        Assert.Multiple(() =>
        {
            Assert.That(list, Has.Count.EqualTo(2));
            Assert.That(list.All(x => x.EmployeeId == _employee.Id));
        });
    }

    [Test]
    public void Try_AddElement_Test()
    {
        var salary = CreateModel(_employee.Id);
        _salaryStorageContract.AddElement(salary);
        AssertElement(GetSalaryFromDatabaseByEmployeeId(_employee.Id), salary);
    }

    private Employee InsertEmployeeToDatabaseAndReturn(string employeeFIO = "fio", string employeeEmail = "abc@mail.ru")
    {
        var employee = new Employee() { Id = Guid.NewGuid().ToString(), PostId = Guid.NewGuid().ToString(), FIO = employeeFIO, Email = employeeEmail, IsDeleted = false };
        MagicCarpetDbContext.Employees.Add(employee);
        MagicCarpetDbContext.SaveChanges();
        return employee;
    }

    private Salary InsertSalaryToDatabaseAndReturn(string employeeId, double employeeSalary = 1, DateTime? salaryDate = null)
    {
        var salary = new Salary() { EmployeeId = employeeId, EmployeeSalary = employeeSalary, SalaryDate = salaryDate ?? DateTime.UtcNow };
        MagicCarpetDbContext.Salaries.Add(salary);
        MagicCarpetDbContext.SaveChanges();
        return salary;
    }

    private static void AssertElement(SalaryDataModel? actual, Salary expected)
    {
        Assert.That(actual, Is.Not.Null);
        Assert.Multiple(() =>
        {
            Assert.That(actual.EmployeeId, Is.EqualTo(expected.EmployeeId));
            Assert.That(actual.Salary, Is.EqualTo(expected.EmployeeSalary));
        });
    }

    private static SalaryDataModel CreateModel(string employeeId, double employeeSalary = 1, DateTime? salaryDate = null)
        => new(employeeId, salaryDate ?? DateTime.UtcNow, employeeSalary);

    private Salary? GetSalaryFromDatabaseByEmployeeId(string id) => MagicCarpetDbContext.Salaries.FirstOrDefault(x => x.EmployeeId == id);

    private static void AssertElement(Salary? actual, SalaryDataModel expected)
    {
        Assert.That(actual, Is.Not.Null);
        Assert.Multiple(() =>
        {
            Assert.That(actual.EmployeeId, Is.EqualTo(expected.EmployeeId));
            Assert.That(actual.EmployeeSalary, Is.EqualTo(expected.Salary));
        });
    }
}