| 18 | } |
| 19 | |
| 20 | func TestWriteSimple(t *testing.T) { |
| 21 | results := bytes.NewBuffer(nil) |
| 22 | expected := []byte("some test data") |
| 23 | w, _ := NewWriter(t.Context(), "s3://localhost/upload", nil) |
| 24 | w.uploader = mockUploader(func(_ context.Context, in *s3manager.UploadInput, _ ...func(*s3manager.Uploader)) (*s3manager.UploadOutput, error) { |
| 25 | _, err := io.Copy(results, in.Body) |
| 26 | return &s3manager.UploadOutput{}, err |
| 27 | }) |
| 28 | _, err := w.Write(expected) |
| 29 | require.NoError(t, err) |
| 30 | require.NoError(t, w.Close()) |
| 31 | require.Equal(t, expected, results.Bytes()) |
| 32 | |
| 33 | } |
| 34 | |
| 35 | func TestWriteImmediateError(t *testing.T) { |
| 36 | expected := errors.New("expected error") |