openiddict-core/test/OpenIddict.Validation.Tests/OpenIddictValidationEventServiceTests.cs

/*
 * Licensed under the Apache License, Version 2.0 (http://www.apache.org/licenses/LICENSE-2.0)
 * See https://github.com/openiddict/openiddict-core for more information concerning
 * the license and the contributors participating to this project.
 */

using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using Moq;
using Xunit;

namespace OpenIddict.Validation.Tests
{
    public class OpenIddictValidationEventServiceTests
    {
        [Fact]
        public async Task PublishAsync_ThrowsAnExceptionForNullNotification()
        {
            // Arrange
            var provider = Mock.Of<IServiceProvider>();
            var service = new OpenIddictValidationEventService(provider);

            // Act and assert
            var exception = await Assert.ThrowsAsync<ArgumentNullException>(()
                => service.PublishAsync<Event>(notification: null));

            Assert.Equal("notification", exception.ParamName);
        }

        [Fact]
        public async Task PublishAsync_InvokesHandlers()
        {
            // Arrange
            var handlers = new List<IOpenIddictValidationEventHandler<Event>>
            {
                Mock.Of<IOpenIddictValidationEventHandler<Event>>(),
                Mock.Of<IOpenIddictValidationEventHandler<Event>>()
            };

            var provider = new Mock<IServiceProvider>();
            provider.Setup(mock => mock.GetService(typeof(IEnumerable<IOpenIddictValidationEventHandler<Event>>)))
                .Returns(handlers);

            var service = new OpenIddictValidationEventService(provider.Object);

            var notification = new Event();

            // Act
            await service.PublishAsync(notification);

            // Assert
            Mock.Get(handlers[0]).Verify(mock => mock.HandleAsync(notification), Times.Once());
            Mock.Get(handlers[1]).Verify(mock => mock.HandleAsync(notification), Times.Once());
        }

        [Fact]
        public async Task PublishAsync_StopsInvokingHandlersWhenHandledIsReturned()
        {
            // Arrange
            var handlers = new List<IOpenIddictValidationEventHandler<Event>>
            {
                Mock.Of<IOpenIddictValidationEventHandler<Event>>(
                    mock => mock.HandleAsync(It.IsAny<Event>()) == Task.FromResult(OpenIddictValidationEventState.Unhandled)),
                Mock.Of<IOpenIddictValidationEventHandler<Event>>(
                    mock => mock.HandleAsync(It.IsAny<Event>()) == Task.FromResult(OpenIddictValidationEventState.Unhandled)),
                Mock.Of<IOpenIddictValidationEventHandler<Event>>(
                    mock => mock.HandleAsync(It.IsAny<Event>()) == Task.FromResult(OpenIddictValidationEventState.Handled)),
                Mock.Of<IOpenIddictValidationEventHandler<Event>>()
            };

            var provider = new Mock<IServiceProvider>();
            provider.Setup(mock => mock.GetService(typeof(IEnumerable<IOpenIddictValidationEventHandler<Event>>)))
                .Returns(handlers);

            var service = new OpenIddictValidationEventService(provider.Object);

            var notification = new Event();

            // Act
            await service.PublishAsync(notification);

            // Assert
            Mock.Get(handlers[0]).Verify(mock => mock.HandleAsync(notification), Times.Once());
            Mock.Get(handlers[1]).Verify(mock => mock.HandleAsync(notification), Times.Once());
            Mock.Get(handlers[2]).Verify(mock => mock.HandleAsync(notification), Times.Once());
            Mock.Get(handlers[3]).Verify(mock => mock.HandleAsync(notification), Times.Never());
        }

        public class Event : IOpenIddictValidationEvent { }
    }
}