BatchedEvents - SubscribeAndAddTwoThenDisposeAllAndComplete

projects/DirectoryWatcherSample/DirectoryWatcherSample.Core/BatchedEvents.cs

@@ -23,6 +23,7 @@ public BatchedEvents(Func<Task> delay)
 
         public void Dispose()
         {
+            this.batches.Clear();
         }
 
         public IDisposable Subscribe(T item, Action<T> callback)

projects/DirectoryWatcherSample/DirectoryWatcherSample.Test/BatchedEventsTest.cs

@@ -180,5 +180,35 @@ public void SubscribeAndAddTwoThenDisposeOneAndComplete()
 
             batches.Should().ContainSingle().Which.Should().Be("B:item2");
         }
+
+        [TestMethod]
+        public void SubscribeAndAddTwoThenDisposeAllAndComplete()
+        {
+            Queue<TaskCompletionSource<bool>> pending = new Queue<TaskCompletionSource<bool>>();
+            BatchedEvents<string> events = new BatchedEvents<string>(() => pending.Dequeue().Task);
+            List<string> batches = new List<string>();
+            TaskCompletionSource<bool> pending1 = new TaskCompletionSource<bool>();
+            TaskCompletionSource<bool> pending2 = new TaskCompletionSource<bool>();
+            pending.Enqueue(pending1);
+            pending.Enqueue(pending2);
+
+            IDisposable sub1 = events.Subscribe("item1", i => batches.Add("A:" + i));
+            events.Subscribe("item2", i => batches.Add("B:" + i));
+
+            events.Add("item1", new TimePoint(1));
+            events.Add("item2", new TimePoint(1));
+
+            batches.Should().BeEmpty();
+
+            events.Dispose();
+
+            pending1.SetResult(true);
+
+            batches.Should().BeEmpty();
+
+            pending2.SetResult(true);
+
+            batches.Should().BeEmpty();
+        }
     }
 }