test: NetworkOutput error

keboola · Feb 11, 2025 · c619d25 · c619d25
1 parent 7ecc321
commit c619d25
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/internal/pkg/service/stream/storage/level/local/encoding/pipeline_test.go b/internal/pkg/service/stream/storage/level/local/encoding/pipeline_test.go
@@ -667,6 +667,7 @@ func (w *dummyEncoder) Close() error {
 
 type dummyOutput struct {
 	bytes      bytes.Buffer
+	WriteError error
 	SyncError  error
 	CloseError error
 }
@@ -684,6 +685,9 @@ func (o *dummyOutput) IsReady() bool {
 }
 
 func (o *dummyOutput) Write(ctx context.Context, aligned bool, p []byte) (n int, err error) {
+	if o.WriteError != nil {
+		return 0, o.WriteError
+	}
 	return o.bytes.Write(p)
 }