fix: flaky test

leg100 · Dec 24, 2024 · d4bfbf4 · d4bfbf4
1 parent 30a3793
commit d4bfbf4
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/internal/task/task_test.go b/internal/task/task_test.go
@@ -43,8 +43,12 @@ func TestTask_NewReader(t *testing.T) {
 	require.NoError(t, err)
 	// stderr and stdout streams can be interwoven, so we use assert.Contains
 	// rather than assert.Equal
-	assert.Contains(t, string(got), "foo\nbar\nbaz\nbye\n")
-	assert.Contains(t, string(got), "err")
+	assert.Len(t, strings.Fields(string(got)), 5)
+	assert.Contains(t, strings.Fields(string(got)), "foo")
+	assert.Contains(t, strings.Fields(string(got)), "bar")
+	assert.Contains(t, strings.Fields(string(got)), "bye")
+	assert.Contains(t, strings.Fields(string(got)), "baz")
+	assert.Contains(t, strings.Fields(string(got)), "err")
 }
 
 func TestTask_cancel(t *testing.T) {