146 lines
4.0 KiB
C#
146 lines
4.0 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Threading;
|
|
using FluentAssertions;
|
|
using Moq;
|
|
using NUnit.Framework;
|
|
using NzbDrone.Common.Messaging;
|
|
using NzbDrone.Core.Messaging;
|
|
using NzbDrone.Test.Common;
|
|
|
|
namespace NzbDrone.Common.Test.MessagingTests
|
|
{
|
|
[TestFixture]
|
|
public class MessageAggregatorEventTests : TestBase<MessageAggregator>
|
|
{
|
|
private Mock<IHandle<EventA>> HandlerA1;
|
|
private Mock<IHandle<EventA>> HandlerA2;
|
|
|
|
private Mock<IHandle<EventB>> HandlerB1;
|
|
private Mock<IHandle<EventB>> HandlerB2;
|
|
|
|
private Mock<IHandleAsync<EventA>> AsyncHandlerA1;
|
|
|
|
|
|
[SetUp]
|
|
public void Setup()
|
|
{
|
|
HandlerA1 = new Mock<IHandle<EventA>>();
|
|
HandlerA2 = new Mock<IHandle<EventA>>();
|
|
HandlerB1 = new Mock<IHandle<EventB>>();
|
|
HandlerB2 = new Mock<IHandle<EventB>>();
|
|
|
|
AsyncHandlerA1 = new Mock<IHandleAsync<EventA>>();
|
|
|
|
Mocker.GetMock<IServiceFactory>()
|
|
.Setup(c => c.BuildAll<IHandle<EventA>>())
|
|
.Returns(new List<IHandle<EventA>> { HandlerA1.Object, HandlerA2.Object });
|
|
|
|
Mocker.GetMock<IServiceFactory>()
|
|
.Setup(c => c.BuildAll<IHandle<EventB>>())
|
|
.Returns(new List<IHandle<EventB>> { HandlerB1.Object, HandlerB2.Object });
|
|
|
|
}
|
|
|
|
[Test]
|
|
public void should_publish_event_to_handlers()
|
|
{
|
|
var eventA = new EventA();
|
|
|
|
Subject.PublishEvent(eventA);
|
|
|
|
HandlerA1.Verify(c => c.Handle(eventA), Times.Once());
|
|
HandlerA2.Verify(c => c.Handle(eventA), Times.Once());
|
|
}
|
|
|
|
[Test]
|
|
public void should_not_publish_to_incompatible_handlers()
|
|
{
|
|
var eventA = new EventA();
|
|
|
|
|
|
Subject.PublishEvent(eventA);
|
|
|
|
HandlerA1.Verify(c => c.Handle(eventA), Times.Once());
|
|
HandlerA2.Verify(c => c.Handle(eventA), Times.Once());
|
|
|
|
HandlerB1.Verify(c => c.Handle(It.IsAny<EventB>()), Times.Never());
|
|
HandlerB2.Verify(c => c.Handle(It.IsAny<EventB>()), Times.Never());
|
|
}
|
|
|
|
|
|
[Test]
|
|
public void broken_handler_should_not_effect_others_handler()
|
|
{
|
|
var eventA = new EventA();
|
|
|
|
|
|
HandlerA1.Setup(c => c.Handle(It.IsAny<EventA>()))
|
|
.Throws(new NotImplementedException());
|
|
|
|
Subject.PublishEvent(eventA);
|
|
|
|
HandlerA1.Verify(c => c.Handle(eventA), Times.Once());
|
|
HandlerA2.Verify(c => c.Handle(eventA), Times.Once());
|
|
|
|
ExceptionVerification.ExpectedErrors(1);
|
|
}
|
|
|
|
|
|
[Test]
|
|
public void should_queue_multiple_async_events()
|
|
{
|
|
var eventA = new EventA();
|
|
|
|
|
|
|
|
var handlers = new List<IHandleAsync<EventA>>
|
|
{
|
|
AsyncHandlerA1.Object,
|
|
AsyncHandlerA1.Object,
|
|
AsyncHandlerA1.Object,
|
|
AsyncHandlerA1.Object,
|
|
AsyncHandlerA1.Object,
|
|
AsyncHandlerA1.Object,
|
|
AsyncHandlerA1.Object,
|
|
};
|
|
|
|
Mocker.GetMock<IServiceFactory>()
|
|
.Setup(c => c.BuildAll<IHandle<EventA>>())
|
|
.Returns(new List<IHandle<EventA>>());
|
|
|
|
Mocker.GetMock<IServiceFactory>()
|
|
.Setup(c => c.BuildAll<IHandleAsync<EventA>>())
|
|
.Returns(handlers);
|
|
|
|
var counter = new ConcurrencyCounter(handlers.Count);
|
|
|
|
|
|
AsyncHandlerA1.Setup(c => c.HandleAsync(It.IsAny<EventA>()))
|
|
.Callback<EventA>(c =>
|
|
{
|
|
var id = counter.Start();
|
|
Thread.Sleep(1000);
|
|
counter.Stop(id);
|
|
});
|
|
|
|
Subject.PublishEvent(eventA);
|
|
|
|
counter.WaitForAllItems();
|
|
|
|
counter.MaxThreads.Should().Be(3);
|
|
}
|
|
}
|
|
|
|
|
|
public class EventA : IEvent
|
|
{
|
|
|
|
}
|
|
|
|
public class EventB : IEvent
|
|
{
|
|
|
|
}
|
|
|
|
} |