// Licensed to the .NET Foundation under one or more agreements.
// The .NET Foundation licenses this file to you under the MIT license.
 
using System.Security.Claims;
using Microsoft.AspNetCore.Http;
using Microsoft.Extensions.DependencyInjection;
 
namespace Microsoft.AspNetCore.Authentication.Core.Test;
 
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());
    }
 
    [Fact]
    public async Task GetTokenWorksWithDefaultAuthenticateScheme()
    {
        var context = new DefaultHttpContext();
        var services = new ServiceCollection().AddOptions()
            .AddAuthenticationCore(o =>
            {
                o.DefaultScheme = "simple";
                o.AddScheme("simple", s => s.HandlerType = typeof(SimpleAuth));
            });
        context.RequestServices = services.BuildServiceProvider();
 
        Assert.Equal("1", await context.GetTokenAsync("One"));
        Assert.Equal("2", await context.GetTokenAsync("Two"));
        Assert.Equal("3", await context.GetTokenAsync("Three"));
    }
 
    [Fact]
    public async Task GetTokenWorksWithExplicitScheme()
    {
        var context = new DefaultHttpContext();
        var services = new ServiceCollection().AddOptions()
            .AddAuthenticationCore(o => o.AddScheme("simple", s => s.HandlerType = typeof(SimpleAuth)));
        context.RequestServices = services.BuildServiceProvider();
 
        Assert.Equal("1", await context.GetTokenAsync("simple", "One"));
        Assert.Equal("2", await context.GetTokenAsync("simple", "Two"));
        Assert.Equal("3", await context.GetTokenAsync("simple", "Three"));
    }
 
    private class SimpleAuth : IAuthenticationHandler
    {
        public Task<AuthenticateResult> AuthenticateAsync()
        {
            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);
            return Task.FromResult(AuthenticateResult.Success(new AuthenticationTicket(new ClaimsPrincipal(), props, "simple")));
        }
 
        public Task ChallengeAsync(AuthenticationProperties? properties)
        {
            throw new NotImplementedException();
        }
 
        public Task ForbidAsync(AuthenticationProperties? properties)
        {
            throw new NotImplementedException();
        }
 
        public Task InitializeAsync(AuthenticationScheme scheme, HttpContext context)
        {
            return Task.FromResult(0);
        }
 
        public Task SignInAsync(ClaimsPrincipal user, AuthenticationProperties properties)
        {
            throw new NotImplementedException();
        }
 
        public Task SignOutAsync(AuthenticationProperties properties)
        {
            throw new NotImplementedException();
        }
    }
 
}