updated CommandExecutedEvent to be handled sync.

2026-04-18 21:35:51 -04:00 · 2013-09-16 21:52:02 -07:00
parent eb7dda0629
commit 48c06de098
6 changed files with 12 additions and 16 deletions
@@ -0,0 +1,146 @@
+using System;
+using System.Collections.Generic;
+using System.Threading;
+using FluentAssertions;
+using Moq;
+using NUnit.Framework;
+using NzbDrone.Common;
+using NzbDrone.Common.Messaging;
+using NzbDrone.Core.Messaging.Events;
+using NzbDrone.Test.Common;
+
+namespace NzbDrone.Core.Test.Messaging.Events
+{
+    [TestFixture]
+    public class EventAggregatorFixture : TestBase<EventAggregator>
+    {
+        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
+    {
+
+    }
+}
@@ -150,6 +150,7 @@
    <Compile Include="Messaging\Commands\CommandEqualityComparerFixture.cs" />
    <Compile Include="Messaging\Commands\CommandExecutorFixture.cs" />
    <Compile Include="Messaging\Commands\CommandFixture.cs" />
+    <Compile Include="Messaging\Events\EventAggregatorFixture.cs" />
    <Compile Include="MetadataSourceTests\TraktProxyFixture.cs" />
    <Compile Include="NotificationTests\NotificationServiceFixture.cs" />
    <Compile Include="NotificationTests\Xbmc\GetJsonVersionFixture.cs" />