1 files changed, 38 insertions, 0 deletions
diff --git a/workhorse/internal/upload/destination/filestore/filestore_test.go b/workhorse/internal/upload/destination/filestore/filestore_test.go
new file mode 100644
index 00000000000..ec67eae96b9
--- /dev/null
+++ b/workhorse/internal/upload/destination/filestore/filestore_test.go
@@ -0,0 +1,38 @@
+package filestore
+
+import (
+	"context"
+	"io/ioutil"
+	"os"
+	"strings"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+)
+
+func TestConsume(t *testing.T) {
+	f, err := ioutil.TempFile("", "filestore-local-file")
+	if f != nil {
+		defer os.Remove(f.Name())
+	}
+	require.NoError(t, err)
+	defer f.Close()
+
+	localFile := &LocalFile{File: f}
+
+	ctx, cancel := context.WithCancel(context.Background())
+	defer cancel()
+
+	content := "file content"
+	reader := strings.NewReader(content)
+	var deadline time.Time
+
+	n, err := localFile.Consume(ctx, reader, deadline)
+	require.NoError(t, err)
+	require.Equal(t, int64(len(content)), n)
+
+	consumedContent, err := ioutil.ReadFile(f.Name())
+	require.NoError(t, err)
+	require.Equal(t, content, string(consumedContent))
+}