Unit test cleanup.

Jroland · Apr 5, 2015 · 92b3988 · 92b3988
1 parent d3dcc04
commit 92b3988
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/src/kafka-tests/Unit/NagleBlockingCollectionTests.cs b/src/kafka-tests/Unit/NagleBlockingCollectionTests.cs
@@ -195,7 +195,7 @@ public void TakeAsyncShouldPlayNiceWithTPL()
 
             dataTask.ContinueWith(x => exit = true);
 
-            Parallel.ForEach(Enumerable.Range(0, max),
+            Parallel.ForEach(Enumerable.Range(0, max).ToList(),
                    new ParallelOptions { MaxDegreeOfParallelism = 20 },
                    async x =>
                    {
@@ -228,7 +228,7 @@ public void TakeAsyncShouldBeThreadSafe()
             take3.ContinueWith(t => Console.WriteLine("Take3 done..."));
             Task.WhenAll(take1, take2, take3).ContinueWith(x => exit = true);
 
-            Parallel.ForEach(Enumerable.Range(0, max),
+            Parallel.ForEach(Enumerable.Range(0, max).ToList(),
                    new ParallelOptions { MaxDegreeOfParallelism = 20 },
                    async x =>
                    {

diff --git a/src/kafka-tests/Unit/ProducerTests.cs b/src/kafka-tests/Unit/ProducerTests.cs
@@ -84,7 +84,8 @@ public void ProducerShouldReportCorrectAmountOfAsyncRequests()
                 semaphore.Release();
                 sendTask.Wait(TimeSpan.FromMilliseconds(500));
 
-                Assert.That(producer.AsyncCount, Is.EqualTo(0), "Async should have completed.");
+                Assert.That(sendTask.IsCompleted, Is.True, "Send task should be marked as completed.");
+                Assert.That(producer.AsyncCount, Is.EqualTo(0), "Async should now show zero count.");
             }
         }