1 files changed, 56 insertions, 0 deletions
diff --git a/workhorse/internal/objectstore/gocloud_object_test.go b/workhorse/internal/objectstore/gocloud_object_test.go
new file mode 100644
index 00000000000..4dc9d2d75cc
--- /dev/null
+++ b/workhorse/internal/objectstore/gocloud_object_test.go
@@ -0,0 +1,56 @@
+package objectstore_test
+
+import (
+	"context"
+	"fmt"
+	"strings"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+
+	"gitlab.com/gitlab-org/gitlab-workhorse/internal/objectstore"
+	"gitlab.com/gitlab-org/gitlab-workhorse/internal/objectstore/test"
+	"gitlab.com/gitlab-org/gitlab-workhorse/internal/testhelper"
+)
+
+func TestGoCloudObjectUpload(t *testing.T) {
+	mux, _, cleanup := test.SetupGoCloudFileBucket(t, "azuretest")
+	defer cleanup()
+
+	ctx, cancel := context.WithCancel(context.Background())
+	deadline := time.Now().Add(testTimeout)
+
+	objectName := "test.png"
+	testURL := "azuretest://azure.example.com/test-container"
+	p := &objectstore.GoCloudObjectParams{Ctx: ctx, Mux: mux, BucketURL: testURL, ObjectName: objectName}
+	object, err := objectstore.NewGoCloudObject(p)
+	require.NotNil(t, object)
+	require.NoError(t, err)
+
+	// copy data
+	n, err := object.Consume(ctx, strings.NewReader(test.ObjectContent), deadline)
+	require.NoError(t, err)
+	require.Equal(t, test.ObjectSize, n, "Uploaded file mismatch")
+
+	bucket, err := mux.OpenBucket(ctx, testURL)
+	require.NoError(t, err)
+
+	// Verify the data was copied correctly.
+	received, err := bucket.ReadAll(ctx, objectName)
+	require.NoError(t, err)
+	require.Equal(t, []byte(test.ObjectContent), received)
+
+	cancel()
+
+	testhelper.Retry(t, 5*time.Second, func() error {
+		exists, err := bucket.Exists(ctx, objectName)
+		require.NoError(t, err)
+
+		if exists {
+			return fmt.Errorf("file %s is still present", objectName)
+		} else {
+			return nil
+		}
+	})
+}