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

using Microsoft.AspNetCore.Hosting;
using Microsoft.Extensions.Hosting;
using System;
using System.Net.Http;
using System.Threading.Tasks;
using Xunit;

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

        //Данные в тестовой БД
        //select
        //  tw.id,	
        //	t_stat.minDate,
        //	t_stat.maxDate
        //from(
        //    select

        //        id_telemetry,
        //        count(1) as count,
        //        min("date") as minDate,
        //        max("date") as maxDate

        //    from t_telemetry_data_saub

        //    group by id_telemetry
        //) as t_stat
        //join t_well tw on tw.id_telemetry = t_stat.id_telemetry
        //where tw is not null
        //order by t_stat.count
        //limit 10;
        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 UserConnectionsLimitMiddlwareTest()
        {
            var host = Host.CreateDefaultBuilder(Array.Empty<string>())
                .ConfigureWebHostDefaults(webBuilder =>
                {
                    webBuilder.UseStartup<Startup>();
                })
                .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 response.Content.ReadAsStringAsync();
                    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;
        }
    }
    
}