svc to s3

treeverse · Dec 26, 2024 · 8b4d12f · 8b4d12f
1 parent 0b2e2b4
commit 8b4d12f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/esti/multipart_test.go b/esti/multipart_test.go
@@ -52,7 +52,7 @@ func TestMultipartUpload(t *testing.T) {
 		partsConcat = append(partsConcat, parts[i]...)
 	}
 
-	completedParts := uploadMultipartParts(t, ctx, logger, resp, parts, 0)
+	completedParts := uploadMultipartParts(t, ctx, svc, logger, resp, parts, 0)
 
 	if isBlockstoreType(block.BlockstoreTypeS3) == nil {
 		// Object should have Last-Modified time at around time of MPU creation.  Ensure
@@ -166,7 +166,7 @@ func reverse(s string) string {
 	return string(runes)
 }
 
-func uploadMultipartParts(t *testing.T, ctx context.Context, logger logging.Logger, resp *s3.CreateMultipartUploadOutput, parts [][]byte, firstIndex int) []types.CompletedPart {
+func uploadMultipartParts(t *testing.T, ctx context.Context, client *s3.Client, logger logging.Logger, resp *s3.CreateMultipartUploadOutput, parts [][]byte, firstIndex int) []types.CompletedPart {
 	count := len(parts)
 	completedParts := make([]types.CompletedPart, count)
 	errs := make([]error, count)
@@ -176,7 +176,7 @@ func uploadMultipartParts(t *testing.T, ctx context.Context, logger logging.Logg
 		go func(i int) {
 			defer wg.Done()
 			partNumber := firstIndex + i + 1
-			completedParts[i], errs[i] = uploadMultipartPart(ctx, logger, svc, resp, parts[i], partNumber)
+			completedParts[i], errs[i] = uploadMultipartPart(ctx, logger, client, resp, parts[i], partNumber)
 		}(i)
 	}
 	wg.Wait()

diff --git a/esti/s3_gateway_test.go b/esti/s3_gateway_test.go
@@ -231,7 +231,7 @@ func TestMultipartUploadIfNoneMatch(t *testing.T) {
 
 		parts := make([][]byte, multipartNumberOfParts)
 
-		completedParts := uploadMultipartParts(t, ctx, logger, resp, parts, 0)
+		completedParts := uploadMultipartParts(t, ctx, s3Client, logger, resp, parts, 0)
 
 		if isBlockstoreType(block.BlockstoreTypeS3) == nil {
 			// Object should have Last-Modified time at around time of MPU creation.  Ensure