From 82c50f9364329432b8fac0579332afa82e249d98 Mon Sep 17 00:00:00 2001
From: ridhozhr <ridho@nodeflux.io>
Date: Mon, 27 May 2019 13:30:57 +0700
Subject: [PATCH] add test file with prefix

---
 gcsstore/gcsstore_test.go | 29 +++++++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/gcsstore/gcsstore_test.go b/gcsstore/gcsstore_test.go
index bd66e4f..7ab0101 100644
--- a/gcsstore/gcsstore_test.go
+++ b/gcsstore/gcsstore_test.go
@@ -65,6 +65,35 @@ func TestNewUpload(t *testing.T) {
 	assert.Equal(id, mockID)
 }
 
+func TestNewUploadWithPrefix(t *testing.T) {
+	mockCtrl := gomock.NewController(t)
+	defer mockCtrl.Finish()
+	assert := assert.New(t)
+
+	service := NewMockGCSAPI(mockCtrl)
+	store := gcsstore.New(mockBucket, service)
+	store.ObjectPrefix = "/path/to/file"
+
+	assert.Equal(store.Bucket, mockBucket)
+
+	data, err := json.Marshal(mockTusdInfo)
+	assert.Nil(err)
+
+	r := bytes.NewReader(data)
+
+	params := gcsstore.GCSObjectParams{
+		Bucket: store.Bucket,
+		ID:     fmt.Sprintf("%s.info", "/path/to/file/"+mockID),
+	}
+
+	ctx := context.Background()
+	service.EXPECT().WriteObject(ctx, params, r).Return(int64(r.Len()), nil)
+
+	id, err := store.NewUpload(mockTusdInfo)
+	assert.Nil(err)
+	assert.Equal(id, mockID)
+}
+
 type MockGetInfoReader struct{}
 
 func (r MockGetInfoReader) Close() error {