ensure writes succeed

Signed-off-by: razzle <harry@razzle.cloud>
zarf-dev · May 23, 2024 · d38e690 · d38e690
1 parent 6aa9329
commit d38e690
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/src/pkg/message/pausable_test.go b/src/pkg/message/pausable_test.go
@@ -15,19 +15,25 @@ func TestPausableWriter(t *testing.T) {
 
 	pw := NewPausableWriter(&buf)
 
-	pw.Write([]byte("foo"))
+	n, err := pw.Write([]byte("foo"))
+	require.NoError(t, err)
+	require.Equal(t, 3, n)
 
 	require.Equal(t, "foo", buf.String())
 
 	pw.Pause()
 
-	pw.Write([]byte("bar"))
+	n, err = pw.Write([]byte("bar"))
+	require.NoError(t, err)
+	require.Equal(t, 3, n)
 
 	require.Equal(t, "foo", buf.String())
 
 	pw.Resume()
 
-	pw.Write([]byte("baz"))
+	n, err = pw.Write([]byte("baz"))
+	require.NoError(t, err)
+	require.Equal(t, 3, n)
 
 	require.Equal(t, "foobaz", buf.String())
 }