address review comments (#791)

nats-io · Jun 20, 2023 · d621b05 · d621b05
1 parent 0fd815a
commit d621b05
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/Tests/IntegrationTests/TestSimplification.cs b/src/Tests/IntegrationTests/TestSimplification.cs
@@ -302,11 +302,12 @@ public void TestIterableConsumer()
                     pubThread.Join();
 
                     Assert.True(count.Read() > 500);
-
-                    // coverage
-                    consumerContext.consume(ConsumeOptions.DefaultConsumeOptions);
-                    Assert.Throws<ArgumentException>(() => consumerContext.consume((ConsumeOptions)null));
                 }
+
+                // coverage
+                IIterableConsumer consumer2 = consumerContext.consume(ConsumeOptions.DefaultConsumeOptions);
+                consumer2.Dispose();
+                Assert.Throws<ArgumentException>(() => consumerContext.consume((ConsumeOptions)null));
             });
         }