DD.WellWorkover.Cloud/AsbCloudWebApi.Tests/Middlware/UserConnectionsLimitMiddlwareTest.cs

using AsbCloudApp.Data.SAUB;
using AsbCloudApp.Services;
using Microsoft.AspNetCore.Hosting;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.DependencyInjection.Extensions;
using Microsoft.Extensions.Hosting;
using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Net.Http;
using System.Threading;
using System.Threading.Tasks;
using Xunit;

namespace AsbCloudWebApi.Tests.Middlware
{
    public class UserConnectionsLimitMiddlwareTest
    {
        const int iterations2Block = 8;

        private readonly (int, DateTime, DateTime)[] wells = new[]
        {
            (191, new DateTime(2022, 09, 01, 21, 43, 00, DateTimeKind.Utc), new DateTime(2022, 09, 04, 07, 37, 31, DateTimeKind.Utc)),
            (3  , new DateTime(2021, 09, 16, 06, 13, 33, DateTimeKind.Utc), new DateTime(2021, 09, 20, 00, 29, 28, DateTimeKind.Utc)),
            (199, new DateTime(2022, 09, 15, 11, 27, 18, DateTimeKind.Utc), new DateTime(2022, 09, 20, 14, 00, 23, DateTimeKind.Utc)),
            (6  , new DateTime(2021, 09, 20, 00, 35, 03, DateTimeKind.Utc), new DateTime(2021, 09, 25, 06, 46, 17, DateTimeKind.Utc)),
            (41 , new DateTime(2021, 12, 10, 00, 59, 52, DateTimeKind.Utc), new DateTime(2022, 10, 31, 15, 29, 24, DateTimeKind.Utc)),
            (100, new DateTime(2022, 04, 24, 03, 04, 05, DateTimeKind.Utc), new DateTime(2022, 04, 29, 11, 38, 36, DateTimeKind.Utc)),
            (154, new DateTime(2022, 03, 28, 10, 09, 14, DateTimeKind.Utc), new DateTime(2022, 06, 14, 15, 01, 12, DateTimeKind.Utc)),
            (5  , new DateTime(2021, 09, 25, 08, 09, 37, DateTimeKind.Utc), new DateTime(2021, 10, 01, 14, 39, 51, DateTimeKind.Utc)),
            (1  , new DateTime(2021, 09, 10, 01, 32, 42, DateTimeKind.Utc), new DateTime(2021, 09, 18, 00, 35, 22, DateTimeKind.Utc)),
            (112, new DateTime(2022, 04, 20, 16, 47, 51, DateTimeKind.Utc), new DateTime(2022, 04, 28, 15, 04, 33, DateTimeKind.Utc)),
        };

        public class TelemetryDataSaubService : ITelemetryDataSaubService
        {
            public async Task<IEnumerable<TelemetryDataSaubDto>?> GetAsync(int idWell, DateTime dateBegin = default, double intervalSec = 600, int approxPointsCount = 1024, CancellationToken token = default)
            {
                await Task.Delay(1000, token);
                return Enumerable.Empty<TelemetryDataSaubDto>();
            }

            public Task<IEnumerable<TelemetryDataSaubStatDto>> GetTelemetryDataStatAsync(int idTelemetry, CancellationToken token) => throw new NotImplementedException();

            public Task<Stream> GetZippedCsv(int idWell, DateTime beginDate, DateTime endDate, CancellationToken token)
            {
                throw new NotImplementedException();
            }

            public Task<int> UpdateDataAsync(string uid, IEnumerable<TelemetryDataSaubDto> dtos, CancellationToken token) => throw new NotImplementedException();
        }

        public UserConnectionsLimitMiddlwareTest()
        {
            var host = Host.CreateDefaultBuilder(Array.Empty<string>())
                .ConfigureWebHostDefaults(webBuilder =>
                {
                    webBuilder.UseStartup<Startup>();
                    webBuilder.ConfigureServices(serviceCollection => {
                        serviceCollection.ReplaceService<ITelemetryDataSaubService>(new TelemetryDataSaubService());
                    });
                })
                .Build();
            host.Start();
        }

        [Fact]
        public async Task Send_n_requests_and_get_blocked()
        {
            var i = 0;
            for (; i < iterations2Block; i++)
                _ = Task.Run(async () =>
                {
                    var well = wells[i];
                    var url = MakeUrl(well.Item1, well.Item2, well.Item3);
                    var response = await MakeHttpClient().GetAsync(url);
                    await Task.Delay(1000);
                });

            var well = wells[i];
            var url = MakeUrl(well.Item1, well.Item2, well.Item3);
            var response = await MakeHttpClient().GetAsync(url);
            Assert.Equal(System.Net.HttpStatusCode.TooManyRequests, response.StatusCode);
        }


        [Fact]
        public async Task Send_n_requests_and_get_blocked_then_restored()
        {
            var i = 0;
            var tasks = new Task[iterations2Block];
            for (; i < iterations2Block; i++)
                tasks[i] = Task.Run(async () =>
                {
                    var well = wells[i];
                    var url = MakeUrl(well.Item1, well.Item2, well.Item3);
                    var response = await MakeHttpClient().GetAsync(url);
                    await Task.Delay(1000);
                });

            var well = wells[i];
            var url = MakeUrl(well.Item1, well.Item2, well.Item3);
            var response = await MakeHttpClient().GetAsync(url);
            Assert.Equal(System.Net.HttpStatusCode.TooManyRequests, response.StatusCode);

            Task.WaitAll(tasks);
            response = await MakeHttpClient().GetAsync(url);
            Assert.Equal(System.Net.HttpStatusCode.OK, response.StatusCode);
        }

        private static string MakeUrl(int idWell, DateTime dateBegin, DateTime dateEnd)
        {
            var interval = (dateEnd - dateBegin).TotalSeconds;
            var dateBeginString = dateBegin.ToString("yyyy-MM-ddZ");
            var url = $"http://127.0.0.1:5000/api/TelemetryDataSaub/{idWell}?begin={dateBeginString}&intervalSec={interval}&approxPointsCount={interval}";
            return url;
        }

        private static HttpClient MakeHttpClient()
        {
            var httpClient = new HttpClient();
            httpClient.DefaultRequestHeaders.Authorization = new("Bearer", "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpZCI6IjEiLCJodHRwOi8vc2NoZW1hcy54bWxzb2FwLm9yZy93cy8yMDA1LzA1L2lkZW50aXR5L2NsYWltcy9uYW1lIjoiZGV2IiwiaWRDb21wYW55IjoiMSIsImh0dHA6Ly9zY2hlbWFzLm1pY3Jvc29mdC5jb20vd3MvMjAwOC8wNi9pZGVudGl0eS9jbGFpbXMvcm9sZSI6InJvb3QiLCJuYmYiOjE2NjY1ODY2MjAsImV4cCI6MTY5ODE0NDIyMCwiaXNzIjoiYSIsImF1ZCI6ImEifQ.zqBdR4nYB87-Xyzv025waasN47i43c9FJ23RfzIvUsM");
            return httpClient;
        }
    }
    
}