persistence/Persistence.IntegrationTests/Controllers/TimeSeriesBaseControllerTest.cs

using Mapster;
using Persistence.Database.Model;
using Persistence.IntegrationTests.Clients;
using System.Net;
using Xunit;

namespace Persistence.IntegrationTests.Controllers;
public abstract class TimeSeriesBaseControllerTest<TEntity, TDto> : BaseIntegrationTest
    where TEntity : class, ITimestampedData, new()
    where TDto : class, new()
{
    private ITimeSeriesClient<TDto> client;

    public TimeSeriesBaseControllerTest(WebAppFactoryFixture factory) : base(factory)
    {
        dbContext.CleanupDbSet<TEntity>();

        Task.Run(async () =>
        {
            client = await factory.GetAuthorizedHttpClient<ITimeSeriesClient<TDto>>(string.Empty);
        }).Wait();
    }

    public async Task InsertRangeSuccess(TDto dto)
    {
        //arrange
        var expected = dto.Adapt<TDto>();

        //act
        var response = await client.InsertRange(new TDto[] { expected });

        //assert
        Assert.Equal(HttpStatusCode.OK, response.StatusCode);
        Assert.Equal(1, response.Content);
    }

    public async Task GetSuccess(DateTimeOffset beginDate, DateTimeOffset endDate, TEntity entity)
    {
        //arrange
        var dbset = dbContext.Set<TEntity>();

        dbset.Add(entity);

        dbContext.SaveChanges();

        var response = await client.Get(beginDate, endDate);

        //assert
        Assert.Equal(HttpStatusCode.OK, response.StatusCode);
        Assert.NotNull(response.Content);
        Assert.Single(response.Content);
    }

    public async Task GetDatesRangeSuccess(TEntity entity)
    {
        //arrange
        var datesRangeExpected = 30;

        var entity2 = entity.Adapt<TEntity>();
        entity2.Date = entity.Date.AddDays(datesRangeExpected);

        var dbset = dbContext.Set<TEntity>();
        dbset.Add(entity);
        dbset.Add(entity2);

        dbContext.SaveChanges();

        var response = await client.GetDatesRange();

        //assert
        Assert.Equal(HttpStatusCode.OK, response.StatusCode);
        Assert.NotNull(response.Content);

        var datesRangeActual = (response.Content.To - response.Content.From).Days;
        Assert.Equal(datesRangeExpected, datesRangeActual);
    }

    public async Task GetResampledDataSuccess(TEntity entity)
    {
        //arrange
        var approxPointsCount = 10;
        var differenceBetweenStartAndEndDays = 50;

        var entities = new List<TEntity>();
        for (var i = 1; i <= differenceBetweenStartAndEndDays; i++)
        {
            var entity2 = entity.Adapt<TEntity>();
            entity2.Date = entity.Date.AddDays(i - 1);

            entities.Add(entity2);
        }

        var dbset = dbContext.Set<TEntity>();
        dbset.AddRange(entities);

        dbContext.SaveChanges();

        var response = await client.GetResampledData(entity.Date.AddMinutes(-1), differenceBetweenStartAndEndDays * 24 * 60 * 60 + 60, approxPointsCount);

        //assert
        Assert.Equal(HttpStatusCode.OK, response.StatusCode);
        Assert.NotNull(response.Content);

        var ratio = entities.Count() / approxPointsCount;
        if (ratio > 1)
        {
            var expectedResampledCount = entities
                .Where((_, index) => index % ratio == 0)
                .Count();

            Assert.Equal(expectedResampledCount, response.Content.Count());
        }
        else
        {
            Assert.Equal(entities.Count(), response.Content.Count());
        }


    }


}