forked from TrueCloudLab/frostfs-s3-gw
72 lines
2.2 KiB
Go
72 lines
2.2 KiB
Go
|
package handler
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"context"
|
||
|
|
"net/http"
|
||
|
|
"net/url"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/nspcc-dev/neofs-s3-gw/api"
|
||
|
|
"github.com/nspcc-dev/neofs-s3-gw/api/layer"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestGetObjectPartsAttributes(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
hc := prepareHandlerContext(t)
|
||
|
|
|
||
|
|
bktName := "bucket-get-attributes"
|
||
|
|
objName, objMultipartName := "object", "object-multipart"
|
||
|
|
|
||
|
|
createTestBucket(ctx, t, hc, bktName)
|
||
|
|
|
||
|
|
body := bytes.NewReader([]byte("content"))
|
||
|
|
w, r := prepareTestPayloadRequest(bktName, objName, body)
|
||
|
|
hc.Handler().PutObjectHandler(w, r)
|
||
|
|
assertStatus(t, w, http.StatusOK)
|
||
|
|
|
||
|
|
w, r = prepareTestRequest(t, bktName, objName, nil)
|
||
|
|
r.Header.Set(api.AmzObjectAttributes, objectParts)
|
||
|
|
hc.Handler().GetObjectAttributesHandler(w, r)
|
||
|
|
result := &GetObjectAttributesResponse{}
|
||
|
|
parseTestResponse(t, w, result)
|
||
|
|
require.Nil(t, result.ObjectParts)
|
||
|
|
|
||
|
|
w, r = prepareTestRequest(t, bktName, objMultipartName, nil)
|
||
|
|
hc.Handler().CreateMultipartUploadHandler(w, r)
|
||
|
|
multipartUpload := &InitiateMultipartUploadResponse{}
|
||
|
|
parseTestResponse(t, w, multipartUpload)
|
||
|
|
|
||
|
|
body2 := bytes.NewReader([]byte("content2"))
|
||
|
|
w, r = prepareTestPayloadRequest(bktName, objMultipartName, body2)
|
||
|
|
query := make(url.Values)
|
||
|
|
query.Add(uploadIDHeaderName, multipartUpload.UploadID)
|
||
|
|
query.Add(partNumberHeaderName, "1")
|
||
|
|
r.URL.RawQuery = query.Encode()
|
||
|
|
hc.Handler().UploadPartHandler(w, r)
|
||
|
|
assertStatus(t, w, http.StatusOK)
|
||
|
|
etag := w.Result().Header.Get(api.ETag)
|
||
|
|
|
||
|
|
completeUpload := &CompleteMultipartUpload{
|
||
|
|
Parts: []*layer.CompletedPart{{
|
||
|
|
ETag: etag,
|
||
|
|
PartNumber: 1,
|
||
|
|
}},
|
||
|
|
}
|
||
|
|
w, r = prepareTestRequest(t, bktName, objMultipartName, completeUpload)
|
||
|
|
query = make(url.Values)
|
||
|
|
query.Add(uploadIDHeaderName, multipartUpload.UploadID)
|
||
|
|
r.URL.RawQuery = query.Encode()
|
||
|
|
hc.Handler().CompleteMultipartUploadHandler(w, r)
|
||
|
|
assertStatus(t, w, http.StatusOK)
|
||
|
|
|
||
|
|
w, r = prepareTestRequest(t, bktName, objMultipartName, nil)
|
||
|
|
r.Header.Set(api.AmzObjectAttributes, objectParts)
|
||
|
|
hc.Handler().GetObjectAttributesHandler(w, r)
|
||
|
|
result = &GetObjectAttributesResponse{}
|
||
|
|
parseTestResponse(t, w, result)
|
||
|
|
require.NotNil(t, result.ObjectParts)
|
||
|
|
require.Equal(t, 1, result.ObjectParts.PartsCount)
|
||
|
|
}
|