iframe-proxy | Sunbelt Computer Software

History

79 lines (62 loc) · 2.78 KB

Raw

namespace NpgsqlRestTests;

public static partial class Database

{

public static void RateLimiterTests()

{

script.Append("""

create function get_rate_unlimited1()

returns text

language sql as $$

select current_user;

$$;

create function get_rate_limited1()

returns text

language sql as $$

select current_user;

$$;

comment on function get_rate_limited1() is 'rate_limiter max 2 per second';

""");

}

[Collection("TestFixture")]

public class RateLimiterTests(TestFixture test)

{

[Fact]

public async Task Test_get_rate_unlimited1()

{

using var request1 = new HttpRequestMessage(HttpMethod.Get, "/api/get-rate-unlimited1");

using var result1 = await test.Client.SendAsync(request1);

var response1 = await result1.Content.ReadAsStringAsync();

using var request2 = new HttpRequestMessage(HttpMethod.Get, "/api/get-rate-unlimited1");

using var result2 = await test.Client.SendAsync(request2);

var response2 = await result2.Content.ReadAsStringAsync();

using var request3 = new HttpRequestMessage(HttpMethod.Get, "/api/get-rate-unlimited1");

using var result3 = await test.Client.SendAsync(request3);

var response3 = await result3.Content.ReadAsStringAsync();

result1.StatusCode.Should().Be(HttpStatusCode.OK);

response1.Should().Be("postgres");

result2.StatusCode.Should().Be(HttpStatusCode.OK);

response2.Should().Be("postgres");

result3.StatusCode.Should().Be(HttpStatusCode.OK);

response3.Should().Be("postgres");

}

[Fact]

public async Task Test_get_rate_limited1()

{

using var request1 = new HttpRequestMessage(HttpMethod.Get, "/api/get-rate-limited1");

using var result1 = await test.Client.SendAsync(request1);

var response1 = await result1.Content.ReadAsStringAsync();

using var request2 = new HttpRequestMessage(HttpMethod.Get, "/api/get-rate-limited1");

using var result2 = await test.Client.SendAsync(request2);

var response2 = await result2.Content.ReadAsStringAsync();

using var request3 = new HttpRequestMessage(HttpMethod.Get, "/api/get-rate-limited1");

using var result3 = await test.Client.SendAsync(request3);

var response3 = await result3.Content.ReadAsStringAsync();

result1.StatusCode.Should().Be(HttpStatusCode.OK);

response1.Should().Be("postgres");

result2.StatusCode.Should().Be(HttpStatusCode.OK);

response2.Should().Be("postgres");

result3.StatusCode.Should().Be(HttpStatusCode.TooManyRequests);

response3.Should().Be("Rate limit exceeded. Please try again later.");

}

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Uh oh!

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

Uh oh!

FilesExpand file tree

RateLimiterTests.cs

Latest commit

History

RateLimiterTests.cs

File metadata and controls