Umbraco-CMS/tests/Umbraco.Tests.UnitTests/Umbraco.Core/ClaimsIdentityExtensionsTests.cs

// Copyright (c) Umbraco.
// See LICENSE for more details.

using System.Collections.Generic;
using System.Security.Claims;
using NUnit.Framework;
using Umbraco.Extensions;

namespace Umbraco.Cms.Tests.UnitTests.Umbraco.Core;

public class ClaimsIdentityExtensionsTests
{
    [Test]
    public void FindFirstValue_WhenIdentityIsNull_ExpectArgumentNullException()
    {
        ClaimsIdentity identity = null;
        Assert.Throws<ArgumentNullException>(() => identity.FindFirstValue("test"));
    }

    [Test]
    public void FindFirstValue_WhenClaimNotPresent_ExpectNull()
    {
        var identity = new ClaimsIdentity(new List<Claim>());
        var value = identity.FindFirstValue("test");
        Assert.IsNull(value);
    }

    [Test]
    public void FindFirstValue_WhenMatchingClaimPresent_ExpectCorrectValue()
    {
        var expectedClaim = new Claim("test", "123", "string", "Umbraco");
        var identity = new ClaimsIdentity(new List<Claim> { expectedClaim });

        var value = identity.FindFirstValue("test");

        Assert.AreEqual(expectedClaim.Value, value);
    }

    [Test]
    public void FindFirstValue_WhenMultipleMatchingClaimsPresent_ExpectFirstValue()
    {
        var expectedClaim = new Claim("test", "123", "string", "Umbraco");
        var dupeClaim = new Claim(expectedClaim.Type, Guid.NewGuid().ToString());
        var identity = new ClaimsIdentity(new List<Claim> { expectedClaim, dupeClaim });

        var value = identity.FindFirstValue("test");

        Assert.AreEqual(expectedClaim.Value, value);
    }
}