BasicUtilsTests.cs

// Copyright (c) Mihir Dilip. All rights reserved.
// Licensed under the MIT License. See License.txt in the project root for license information.

using System.Security.Claims;
using Xunit;

namespace AspNetCore.Authentication.Basic.Tests
{
    public class BasicUtilsTests
    {
        [Fact]
        public static void BuildClaimsPrincipal_null_ownerName_no_exception()
        {
            var claimsPrincipal = BasicUtils.BuildClaimsPrincipal(null, "Test", "Test", null);
            Assert.NotNull(claimsPrincipal);
        }

        [Fact]
        public static void BuildClaimsPrincipal_null_schemeName_throws_ArgumentNullException()
        {
            var exception = Assert.Throws<ArgumentNullException>(() => BasicUtils.BuildClaimsPrincipal(null, null!, null, null));
            Assert.Contains("schemeName", exception.Message);
        }

        [Fact]
        public static void BuildClaimsPrincipal_null_claimsIssuer_no_exception()
        {
            var claimsPrincipal = BasicUtils.BuildClaimsPrincipal(null, "Test", null, null);
            Assert.NotNull(claimsPrincipal);
        }

        [Fact]
        public static void BuildClaimsPrincipal_null_claims_no_exception()
        {
            var claimsPrincipal = BasicUtils.BuildClaimsPrincipal(null, "Test", null, null);
            Assert.NotNull(claimsPrincipal);
        }

        [Fact]
        public static void BuildClaimsPrincipal_adds_single_identity_without_claims()
        {
            var schemeName = "Test";
            var claimsPrincipal = BasicUtils.BuildClaimsPrincipal(null, schemeName, null, null);
            Assert.NotNull(claimsPrincipal);
            Assert.Single(claimsPrincipal.Identities);
            Assert.NotNull(claimsPrincipal.Identity);
            Assert.Equal(schemeName, claimsPrincipal.Identity.AuthenticationType);
            Assert.Empty(claimsPrincipal.Claims);
        }

        [Fact]
        public static void BuildClaimsPrincipal_adds_single_identity_with_claims()
        {
            var schemeName = "Test";
            var claims = new List<Claim> 
            { 
                new(ClaimTypes.Email, "abc@xyz.com") ,
                new(ClaimTypes.Role, "admin")
            };
            var claimsPrincipal = BasicUtils.BuildClaimsPrincipal(null, schemeName, null, claims);
            Assert.NotNull(claimsPrincipal);
            Assert.Single(claimsPrincipal.Identities);
            Assert.NotNull(claimsPrincipal.Identity);
            Assert.Equal(schemeName, claimsPrincipal.Identity.AuthenticationType);
            Assert.NotEmpty(claimsPrincipal.Claims);
            Assert.Equal(claims.Count, claimsPrincipal.Claims.Count());
        }

        [Fact]
        public static void BuildClaimsPrincipal_ownerName_adds_Name_and_NameIdentifier_claims()
        {
            var ownerName = "Test";
            var schemeName = "Test";
            var claimsPrincipal = BasicUtils.BuildClaimsPrincipal(ownerName, schemeName, null, null);
            Assert.NotNull(claimsPrincipal);
            Assert.NotEmpty(claimsPrincipal.Claims);
            Assert.Equal(2, claimsPrincipal.Claims.Count());
            Assert.Contains(claimsPrincipal.Claims, c => c.Type == ClaimTypes.NameIdentifier && c.Value == ownerName);
            Assert.Contains(claimsPrincipal.Claims, c => c.Type == ClaimTypes.Name && c.Value == ownerName);
        }

        [Fact]
        public static void BuildClaimsPrincipal_ownerName_adds_Name_and_NameIdentifier_claims2()
        {
            var ownerName = "Test";
            var schemeName = "Test";
            var claims = new List<Claim>
            {
                new(ClaimTypes.Email, "abc@xyz.com") ,
                new(ClaimTypes.Role, "admin")
            };
            var claimsPrincipal = BasicUtils.BuildClaimsPrincipal(ownerName, schemeName, null, claims);
            Assert.NotNull(claimsPrincipal);
            Assert.NotEmpty(claimsPrincipal.Claims);
            Assert.NotEqual(claims.Count, claimsPrincipal.Claims.Count());
            Assert.Equal(claims.Count + 2, claimsPrincipal.Claims.Count());
            Assert.Contains(claimsPrincipal.Claims, c => c.Type == ClaimTypes.NameIdentifier && c.Value == ownerName);
            Assert.Contains(claimsPrincipal.Claims, c => c.Type == ClaimTypes.Name && c.Value == ownerName);
        }

        [Fact]
        public static void BuildClaimsPrincipal_ownerName_adds_Name_and_NameIdentifier_claims_if_not_already_exists()
        {
            var ownerName = "Test";
            var schemeName = "Test";
            var claims = new List<Claim>
            {
                new(ClaimTypes.Name, "Admin"),
                new(ClaimTypes.Role, "admin")
            };
            var claimsPrincipal = BasicUtils.BuildClaimsPrincipal(ownerName, schemeName, null, claims);
            Assert.NotNull(claimsPrincipal);
            Assert.NotEmpty(claimsPrincipal.Claims);
            Assert.NotEqual(claims.Count, claimsPrincipal.Claims.Count());
            Assert.Equal(claims.Count + 1, claimsPrincipal.Claims.Count());
            Assert.Contains(claimsPrincipal.Claims, c => c.Type == ClaimTypes.NameIdentifier && c.Value == ownerName);
            Assert.Contains(claimsPrincipal.Claims, c => c.Type == ClaimTypes.Name && c.Value != ownerName);
            Assert.Contains(claimsPrincipal.Claims, c => c.Type == ClaimTypes.Name && c.Value == "Admin");
        }
    }
}