src/Middleware/RateLimiting/test/RateLimitingOptionsTests.cs - main - githubmirror/aspnetcore

src/Middleware/RateLimiting/test/RateLimitingOptionsTests.cs

48 строк2 KB

dotnet-maestro[bot]

[main] Update dependencies from dotnet/runtime dotnet/efcore (#42688)

14 июл 2022, 05:48

14 июл 2022, 05:48ca77aff

// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

using System.Threading.RateLimiting;
using Castle.DynamicProxy.Generators.Emitters.SimpleAST;
using Microsoft.AspNetCore.Http;

namespace Microsoft.AspNetCore.RateLimiting;

public class RateLimitingOptionsTests
{
    [Fact]
    public void AddPolicy_ThrowsOnNullPolicyName()
    {
        var options = new RateLimiterOptions();
        Assert.Throws<ArgumentNullException>(() => options.AddPolicy<string>(null, context => RateLimitPartition.GetNoLimiter<string>("myKey")));
    }

    [Fact]
    public void AddPolicy_ThrowsOnNullPartitioner()
    {
        var options = new RateLimiterOptions();
        Assert.Throws<ArgumentNullException>(() => options.AddPolicy<string>("myKey", partitioner: null));
    }

    [Fact]
    public void AddPolicy_ThrowsOnNullPolicy()
    {
        var options = new RateLimiterOptions();
        Assert.Throws<ArgumentNullException>(() => options.AddPolicy<string>("myKey", policy: null));
    }

    [Fact]
    public void AddPolicy_ThrowsOnDuplicateName()
    {
        var options = new RateLimiterOptions();
        options.AddPolicy<string>("myKey", context => RateLimitPartition.GetNoLimiter<string>("myKey"));
        Assert.Throws<ArgumentException>(() => options.AddPolicy<string>("myKey", context => RateLimitPartition.GetNoLimiter<string>("yourKey")));
    }

    [Fact]
    public void AddPolicy_Generic_ThrowsOnDuplicateName()
    {
        var options = new RateLimiterOptions();
        options.AddPolicy<string>("myKey", context => RateLimitPartition.GetNoLimiter<string>("myKey"));
        Assert.Throws<ArgumentException>(() => options.AddPolicy<string, TestRateLimiterPolicy>("myKey"));
    }
}