From 18e760db885df0e896445dcc8e6c7e1a7a619e44 Mon Sep 17 00:00:00 2001
From: Sebastian Stehle <sebastian@squidex.io>
Date: Tue, 10 Oct 2017 08:12:14 +0200
Subject: [PATCH] Tests for exception handling

---
 .../CQRS/Events/PollingSubscriptionTests.cs   | 66 ++++++++++++++++---
 1 file changed, 57 insertions(+), 9 deletions(-)

diff --git a/tests/Squidex.Infrastructure.Tests/CQRS/Events/PollingSubscriptionTests.cs b/tests/Squidex.Infrastructure.Tests/CQRS/Events/PollingSubscriptionTests.cs
index 8cb1df42d..5056ccd0f 100644
--- a/tests/Squidex.Infrastructure.Tests/CQRS/Events/PollingSubscriptionTests.cs
+++ b/tests/Squidex.Infrastructure.Tests/CQRS/Events/PollingSubscriptionTests.cs
@@ -19,29 +19,75 @@ namespace Squidex.Infrastructure.CQRS.Events
         private readonly IEventStore eventStore = A.Fake<IEventStore>();
         private readonly IEventNotifier eventNotifier = new DefaultEventNotifier(new InMemoryPubSub());
         private readonly IEventSubscriber eventSubscriber = A.Fake<IEventSubscriber>();
-        private readonly PollingSubscription sut;
         private readonly string position = Guid.NewGuid().ToString();
 
-        public PollingSubscriptionTests()
+        [Fact]
+        public async Task Should_subscribe_on_start()
         {
-            sut = new PollingSubscription(eventStore, eventNotifier, eventSubscriber, "^my-stream", position);
+            var sut = new PollingSubscription(eventStore, eventNotifier, eventSubscriber, "^my-stream", position);
+
+            await WaitAndStopAsync(sut);
+
+            A.CallTo(() => eventStore.GetEventsAsync(A<Func<StoredEvent, Task>>.Ignored, A<CancellationToken>.Ignored, "^my-stream", position))
+                .MustHaveHappened(Repeated.Exactly.Once);
         }
 
         [Fact]
-        public async Task Should_subscribe_on_start()
+        public async Task Should_propagate_exception_to_subscriber()
         {
-            await WaitAndStopAsync();
+            var ex = new InvalidOperationException();
 
             A.CallTo(() => eventStore.GetEventsAsync(A<Func<StoredEvent, Task>>.Ignored, A<CancellationToken>.Ignored, "^my-stream", position))
-                .MustHaveHappened(Repeated.Exactly.Once);
+                .Throws(ex);
+
+            var sut = new PollingSubscription(eventStore, eventNotifier, eventSubscriber, "^my-stream", position);
+
+            await WaitAndStopAsync(sut);
+
+            A.CallTo(() => eventSubscriber.OnErrorAsync(sut, ex))
+                .MustHaveHappened();
+        }
+
+        [Fact]
+        public async Task Should_propagate_operation_cancelled_exception_to_subscriber()
+        {
+            var ex = new OperationCanceledException();
+
+            A.CallTo(() => eventStore.GetEventsAsync(A<Func<StoredEvent, Task>>.Ignored, A<CancellationToken>.Ignored, "^my-stream", position))
+                .Throws(ex);
+
+            var sut = new PollingSubscription(eventStore, eventNotifier, eventSubscriber, "^my-stream", position);
+
+            await WaitAndStopAsync(sut);
+
+            A.CallTo(() => eventSubscriber.OnErrorAsync(sut, ex))
+                .MustNotHaveHappened();
+        }
+
+        [Fact]
+        public async Task Should_propagate_aggregate_operation_cancelled_exception_to_subscriber()
+        {
+            var ex = new AggregateException(new OperationCanceledException());
+
+            A.CallTo(() => eventStore.GetEventsAsync(A<Func<StoredEvent, Task>>.Ignored, A<CancellationToken>.Ignored, "^my-stream", position))
+                .Throws(ex);
+
+            var sut = new PollingSubscription(eventStore, eventNotifier, eventSubscriber, "^my-stream", position);
+
+            await WaitAndStopAsync(sut);
+
+            A.CallTo(() => eventSubscriber.OnErrorAsync(sut, ex))
+                .MustNotHaveHappened();
         }
 
         [Fact]
         public async Task Should_not_subscribe_on_notify_when_stream_matches()
         {
+            var sut = new PollingSubscription(eventStore, eventNotifier, eventSubscriber, "^my-stream", position);
+
             eventNotifier.NotifyEventsStored("other-stream-123");
 
-            await WaitAndStopAsync();
+            await WaitAndStopAsync(sut);
 
             A.CallTo(() => eventStore.GetEventsAsync(A<Func<StoredEvent, Task>>.Ignored, A<CancellationToken>.Ignored, "^my-stream", position))
                 .MustHaveHappened(Repeated.Exactly.Once);
@@ -50,15 +96,17 @@ namespace Squidex.Infrastructure.CQRS.Events
         [Fact]
         public async Task Should_subscribe_on_notify_when_stream_matches()
         {
+            var sut = new PollingSubscription(eventStore, eventNotifier, eventSubscriber, "^my-stream", position);
+
             eventNotifier.NotifyEventsStored("my-stream-123");
 
-            await WaitAndStopAsync();
+            await WaitAndStopAsync(sut);
 
             A.CallTo(() => eventStore.GetEventsAsync(A<Func<StoredEvent, Task>>.Ignored, A<CancellationToken>.Ignored, "^my-stream", position))
                 .MustHaveHappened(Repeated.Exactly.Twice);
         }
 
-        private async Task WaitAndStopAsync()
+        private async Task WaitAndStopAsync(PollingSubscription sut)
         {
             await Task.Delay(200);