Updated check for log messages in the Cancelled Context test

testcontainers · Mar 8, 2024 · beb3de6 · beb3de6
1 parent c953106
commit beb3de6
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/logconsumer_test.go b/logconsumer_test.go
@@ -602,9 +602,10 @@ func Test_MultiContainerLogConsumer_CancelledContext(t *testing.T) {
 	// Deliberately calling context cancel
 	cancel()
 
-	// Check the log messages
-	assert.Equal(t, []string{"ready\n", "echo hello1\n", "echo there1\n"}, first.Msgs)
-	assert.Equal(t, []string{"ready\n", "echo hello2\n", "echo there2\n"}, second.Msgs)
+	// We check log size due to context cancellation causing
+	// varying message counts, leading to test failure.
+	assert.GreaterOrEqual(t, len(first.Msgs), 2)
+	assert.GreaterOrEqual(t, len(second.Msgs), 2)
 
 	// Restore stderr
 	w.Close()