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)
}