forked from TrueCloudLab/frostfs-s3-gw
86 lines
2.7 KiB
Go
86 lines
2.7 KiB
Go
|
package handler
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"net/http"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/nspcc-dev/neofs-s3-gw/api"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestConditionalHead(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
tc := prepareHandlerContext(t)
|
||
|
|
|
||
|
|
bktName := "bucket-for-conditional"
|
||
|
|
createTestBucket(ctx, t, tc, bktName)
|
||
|
|
bktInfo, err := tc.Layer().GetBucketInfo(ctx, bktName)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
objName := "object"
|
||
|
|
createTestObject(ctx, t, tc, bktInfo, objName)
|
||
|
|
|
||
|
|
w, r := prepareTestRequest(t, bktName, objName, nil)
|
||
|
|
tc.Handler().HeadObjectHandler(w, r)
|
||
|
|
assertStatus(t, w, http.StatusOK)
|
||
|
|
|
||
|
|
etag := w.Result().Header.Get(api.ETag)
|
||
|
|
lastModified := w.Result().Header.Get(api.LastModified)
|
||
|
|
_ = lastModified
|
||
|
|
|
||
|
|
w, r = prepareTestRequest(t, bktName, objName, nil)
|
||
|
|
r.Header.Set(api.IfMatch, etag)
|
||
|
|
tc.Handler().HeadObjectHandler(w, r)
|
||
|
|
assertStatus(t, w, http.StatusOK)
|
||
|
|
|
||
|
|
w, r = prepareTestRequest(t, bktName, objName, nil)
|
||
|
|
r.Header.Set(api.IfMatch, "etag")
|
||
|
|
tc.Handler().HeadObjectHandler(w, r)
|
||
|
|
assertStatus(t, w, http.StatusPreconditionFailed)
|
||
|
|
|
||
|
|
w, r = prepareTestRequest(t, bktName, objName, nil)
|
||
|
|
r.Header.Set(api.IfUnmodifiedSince, time.Now().UTC().Format(http.TimeFormat))
|
||
|
|
tc.Handler().HeadObjectHandler(w, r)
|
||
|
|
assertStatus(t, w, http.StatusOK)
|
||
|
|
|
||
|
|
var zeroTime time.Time
|
||
|
|
w, r = prepareTestRequest(t, bktName, objName, nil)
|
||
|
|
r.Header.Set(api.IfUnmodifiedSince, zeroTime.UTC().Format(http.TimeFormat))
|
||
|
|
tc.Handler().HeadObjectHandler(w, r)
|
||
|
|
assertStatus(t, w, http.StatusPreconditionFailed)
|
||
|
|
|
||
|
|
w, r = prepareTestRequest(t, bktName, objName, nil)
|
||
|
|
r.Header.Set(api.IfMatch, etag)
|
||
|
|
r.Header.Set(api.IfUnmodifiedSince, zeroTime.UTC().Format(http.TimeFormat))
|
||
|
|
tc.Handler().HeadObjectHandler(w, r)
|
||
|
|
assertStatus(t, w, http.StatusOK)
|
||
|
|
|
||
|
|
w, r = prepareTestRequest(t, bktName, objName, nil)
|
||
|
|
r.Header.Set(api.IfNoneMatch, etag)
|
||
|
|
tc.Handler().HeadObjectHandler(w, r)
|
||
|
|
assertStatus(t, w, http.StatusNotModified)
|
||
|
|
|
||
|
|
w, r = prepareTestRequest(t, bktName, objName, nil)
|
||
|
|
r.Header.Set(api.IfNoneMatch, "etag")
|
||
|
|
tc.Handler().HeadObjectHandler(w, r)
|
||
|
|
assertStatus(t, w, http.StatusOK)
|
||
|
|
|
||
|
|
w, r = prepareTestRequest(t, bktName, objName, nil)
|
||
|
|
r.Header.Set(api.IfModifiedSince, zeroTime.UTC().Format(http.TimeFormat))
|
||
|
|
tc.Handler().HeadObjectHandler(w, r)
|
||
|
|
assertStatus(t, w, http.StatusOK)
|
||
|
|
|
||
|
|
w, r = prepareTestRequest(t, bktName, objName, nil)
|
||
|
|
r.Header.Set(api.IfModifiedSince, time.Now().Add(time.Minute).UTC().Format(http.TimeFormat))
|
||
|
|
tc.Handler().HeadObjectHandler(w, r)
|
||
|
|
assertStatus(t, w, http.StatusNotModified)
|
||
|
|
|
||
|
|
w, r = prepareTestRequest(t, bktName, objName, nil)
|
||
|
|
r.Header.Set(api.IfNoneMatch, etag)
|
||
|
|
r.Header.Set(api.IfModifiedSince, zeroTime.UTC().Format(http.TimeFormat))
|
||
|
|
tc.Handler().HeadObjectHandler(w, r)
|
||
|
|
assertStatus(t, w, http.StatusNotModified)
|
||
|
|
}
|