From 92b3988c7e2336ca5724b1eef77e66dcb8834c3d Mon Sep 17 00:00:00 2001 From: James Roland Date: Sat, 4 Apr 2015 23:25:56 -0700 Subject: [PATCH] Unit test cleanup. --- src/kafka-tests/Unit/NagleBlockingCollectionTests.cs | 4 ++-- src/kafka-tests/Unit/ProducerTests.cs | 3 ++- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/src/kafka-tests/Unit/NagleBlockingCollectionTests.cs b/src/kafka-tests/Unit/NagleBlockingCollectionTests.cs index e653d819..8ff9bc5e 100644 --- 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 index e34217cf..54c34b0d 100644 --- 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."); } }