Fix tests to reflect changes

mostafa · Apr 26, 2024 · d4031f7 · d4031f7
1 parent 70b2465
commit d4031f7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/reader_test.go b/reader_test.go
@@ -33,7 +33,7 @@ func TestConsumerMaxWaitExceeded(t *testing.T) {
 		require.NoError(t, test.moveToVUCode())
 
 		// Consume a message in the VU function.
-		assert.NotPanics(t, func() {
+		assert.Panics(t, func() {
 			messages := test.module.Kafka.consume(reader, &ConsumeConfig{Limit: 1})
 			assert.Empty(t, messages)
 		})
@@ -231,7 +231,7 @@ func TestConsumerContextCancelled(t *testing.T) {
 		test.cancelContext()
 
 		// Consume a message in the VU function.
-		assert.NotPanics(t, func() {
+		assert.Panics(t, func() {
 			messages := test.module.Kafka.consume(reader, &ConsumeConfig{Limit: 1})
 			assert.Empty(t, messages)
 		})