dotnet-sdk/test/Dapr.Actors.Test/ApiTokenTests.cs

// ------------------------------------------------------------
// Copyright (c) Microsoft Corporation.
// Licensed under the MIT License.
// ------------------------------------------------------------

using System;
using System.Threading;
using System.Threading.Tasks;
using Dapr.Actors.Client;
using FluentAssertions;
using Xunit;

namespace Dapr.Actors.Test
{
    public class ApiTokenTests
    {
        [Fact]
        public async Task CreateProxyWithRemoting_WithApiToken()
        {
            await using var client = TestClient.CreateForMessageHandler();

            var actorId = new ActorId("abc");
            var options = new ActorProxyOptions
            {
                DaprApiToken = "test_token",
            };

            var request = await client.CaptureHttpRequestAsync(async handler =>
            {
                var factory = new ActorProxyFactory(options, handler);
                var proxy = factory.CreateActorProxy<ITestActor>(actorId, "TestActor");
                await proxy.SetCountAsync(1, new CancellationToken());
            });

            request.Dismiss();

            var headerValues = request.Request.Headers.GetValues("dapr-api-token");
            headerValues.Should().Contain("test_token");
        }

        [Fact]
        public async Task CreateProxyWithRemoting_WithNoApiToken()
        {
            await using var client = TestClient.CreateForMessageHandler();

            var actorId = new ActorId("abc");

            var request = await client.CaptureHttpRequestAsync(async handler =>
            {
                var factory = new ActorProxyFactory(null, handler);
                var proxy = factory.CreateActorProxy<ITestActor>(actorId, "TestActor");
                await proxy.SetCountAsync(1, new CancellationToken());
            });

            request.Dismiss();

            Assert.Throws<InvalidOperationException>(() =>
            {
                request.Request.Headers.GetValues("dapr-api-token");
            });
        }

        [Fact]
        public async Task CreateProxyWithNoRemoting_WithApiToken()
        {
            await using var client = TestClient.CreateForMessageHandler();

            var actorId = new ActorId("abc");
            var options = new ActorProxyOptions
            {
                DaprApiToken = "test_token",
            };

            var request = await client.CaptureHttpRequestAsync(async handler =>
            {
                var factory = new ActorProxyFactory(options, handler);
                var proxy = factory.Create(actorId, "TestActor");
                await proxy.InvokeMethodAsync("SetCountAsync", 1, new CancellationToken());
            });

            request.Dismiss();

            var headerValues = request.Request.Headers.GetValues("dapr-api-token");
            headerValues.Should().Contain("test_token");
        }

        [Fact]
        public async Task CreateProxyWithNoRemoting_WithNoApiToken()
        {
            await using var client = TestClient.CreateForMessageHandler();

            var actorId = new ActorId("abc");

            var request = await client.CaptureHttpRequestAsync(async handler =>
            {
                var factory = new ActorProxyFactory(null, handler);
                var proxy = factory.Create(actorId, "TestActor");
                await proxy.InvokeMethodAsync("SetCountAsync", 1, new CancellationToken());
            });

            request.Dismiss();

            Assert.Throws<InvalidOperationException>(() =>
            {
                request.Request.Headers.GetValues("dapr-api-token");
            });
        }
    }
}