src/Security/Authentication/test/TokenExtensionTests.cs - main - githubmirror/aspnetcore

src/Security/Authentication/test/TokenExtensionTests.cs
176 строк6 KB
Shreyas Jejurkar
chore : Remove unwanted `using` statement (#39176)
02 янв 2022, 17:39
02 янв 2022, 17:39c85baf8
// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.

namespace Microsoft.AspNetCore.Authentication;

public class TokenExtensionTests
{
    [Fact]
    public void CanStoreMultipleTokens()
    {
        var props = new AuthenticationProperties();
        var tokens = new List<AuthenticationToken>();
        var tok1 = new AuthenticationToken { Name = "One", Value = "1" };
        var tok2 = new AuthenticationToken { Name = "Two", Value = "2" };
        var tok3 = new AuthenticationToken { Name = "Three", Value = "3" };
        tokens.Add(tok1);
        tokens.Add(tok2);
        tokens.Add(tok3);
        props.StoreTokens(tokens);

        Assert.Equal("1", props.GetTokenValue("One"));
        Assert.Equal("2", props.GetTokenValue("Two"));
        Assert.Equal("3", props.GetTokenValue("Three"));
        Assert.Equal(3, props.GetTokens().Count());
    }

    [Fact]
    public void SubsequentStoreTokenDeletesPreviousTokens()
    {
        var props = new AuthenticationProperties();
        var tokens = new List<AuthenticationToken>();
        var tok1 = new AuthenticationToken { Name = "One", Value = "1" };
        var tok2 = new AuthenticationToken { Name = "Two", Value = "2" };
        var tok3 = new AuthenticationToken { Name = "Three", Value = "3" };
        tokens.Add(tok1);
        tokens.Add(tok2);
        tokens.Add(tok3);

        props.StoreTokens(tokens);

        props.StoreTokens(new[] { new AuthenticationToken { Name = "Zero", Value = "0" } });

        Assert.Equal("0", props.GetTokenValue("Zero"));
        Assert.Null(props.GetTokenValue("One"));
        Assert.Null(props.GetTokenValue("Two"));
        Assert.Null(props.GetTokenValue("Three"));
        Assert.Single(props.GetTokens());
    }

    [Fact]
    public void CanUpdateTokens()
    {
        var props = new AuthenticationProperties();
        var tokens = new List<AuthenticationToken>();
        var tok1 = new AuthenticationToken { Name = "One", Value = "1" };
        var tok2 = new AuthenticationToken { Name = "Two", Value = "2" };
        var tok3 = new AuthenticationToken { Name = "Three", Value = "3" };
        tokens.Add(tok1);
        tokens.Add(tok2);
        tokens.Add(tok3);
        props.StoreTokens(tokens);

        tok1.Value = ".1";
        tok2.Value = ".2";
        tok3.Value = ".3";
        props.StoreTokens(tokens);

        Assert.Equal(".1", props.GetTokenValue("One"));
        Assert.Equal(".2", props.GetTokenValue("Two"));
        Assert.Equal(".3", props.GetTokenValue("Three"));
        Assert.Equal(3, props.GetTokens().Count());
    }

    [Fact]
    public void CanUpdateTokenValues()
    {
        var props = new AuthenticationProperties();
        var tokens = new List<AuthenticationToken>();
        var tok1 = new AuthenticationToken { Name = "One", Value = "1" };
        var tok2 = new AuthenticationToken { Name = "Two", Value = "2" };
        var tok3 = new AuthenticationToken { Name = "Three", Value = "3" };
        tokens.Add(tok1);
        tokens.Add(tok2);
        tokens.Add(tok3);
        props.StoreTokens(tokens);

        Assert.True(props.UpdateTokenValue("One", ".11"));
        Assert.True(props.UpdateTokenValue("Two", ".22"));
        Assert.True(props.UpdateTokenValue("Three", ".33"));

        Assert.Equal(".11", props.GetTokenValue("One"));
        Assert.Equal(".22", props.GetTokenValue("Two"));
        Assert.Equal(".33", props.GetTokenValue("Three"));
        Assert.Equal(3, props.GetTokens().Count());
    }

    [Fact]
    public void UpdateTokenValueReturnsFalseForUnknownToken()
    {
        var props = new AuthenticationProperties();
        var tokens = new List<AuthenticationToken>();
        var tok1 = new AuthenticationToken { Name = "One", Value = "1" };
        var tok2 = new AuthenticationToken { Name = "Two", Value = "2" };
        var tok3 = new AuthenticationToken { Name = "Three", Value = "3" };
        tokens.Add(tok1);
        tokens.Add(tok2);
        tokens.Add(tok3);
        props.StoreTokens(tokens);

        Assert.False(props.UpdateTokenValue("ONE", ".11"));
        Assert.False(props.UpdateTokenValue("Jigglypuff", ".11"));

        Assert.Null(props.GetTokenValue("ONE"));
        Assert.Null(props.GetTokenValue("Jigglypuff"));
        Assert.Equal(3, props.GetTokens().Count());

    }

    //public class TestAuthHandler : IAuthenticationHandler
    //{
    //    private readonly AuthenticationProperties _props;
    //    public TestAuthHandler(AuthenticationProperties props)
    //    {
    //        _props = props;
    //    }

    //    public Task AuthenticateAsync(AuthenticateContext context)
    //    {
    //        context.Authenticated(new ClaimsPrincipal(), _props.Items, new Dictionary<string, object>());
    //        return Task.FromResult(0);
    //    }

    //    public Task ChallengeAsync(AuthenticationProperties properties)
    //    {
    //        throw new NotImplementedException();
    //    }

    //    public void GetDescriptions(DescribeSchemesContext context)
    //    {
    //        throw new NotImplementedException();
    //    }

    //    public Task SignInAsync(ClaimsPrincipal principal, AuthenticationProperties properties)
    //    {
    //        throw new NotImplementedException();
    //    }

    //    public Task SignOutAsync(AuthenticationProperties properties)
    //    {
    //        throw new NotImplementedException();
    //    }
    //}

    //[Fact]
    //public async Task CanGetTokenFromContext()
    //{
    //    var props = new AuthenticationProperties();
    //    var tokens = new List<AuthenticationToken>();
    //    var tok1 = new AuthenticationToken { Name = "One", Value = "1" };
    //    var tok2 = new AuthenticationToken { Name = "Two", Value = "2" };
    //    var tok3 = new AuthenticationToken { Name = "Three", Value = "3" };
    //    tokens.Add(tok1);
    //    tokens.Add(tok2);
    //    tokens.Add(tok3);
    //    props.StoreTokens(tokens);

    //    var context = new DefaultHttpContext();
    //    var handler = new TestAuthHandler(props);
    //    context.Features.Set<IHttpAuthenticationFeature>(new HttpAuthenticationFeature() { Handler = handler });

    //    Assert.Equal("1", await context.GetTokenAsync("One"));
    //    Assert.Equal("2", await context.GetTokenAsync("Two"));
    //    Assert.Equal("3", await context.GetTokenAsync("Three"));
    //}

}