package handler

import (
	"net/http"
	"testing"
	"time"

	"github.com/nspcc-dev/neofs-s3-gw/api/layer"
	"github.com/stretchr/testify/require"
)

func TestFetchRangeHeader(t *testing.T) {
	for _, tc := range []struct {
		header   string
		expected *layer.RangeParams
		fullSize uint64
		err      bool
	}{
		{header: "bytes=0-256", expected: &layer.RangeParams{Start: 0, End: 256}, err: false},
		{header: "bytes=0-0", expected: &layer.RangeParams{Start: 0, End: 0}, err: false},
		{header: "bytes=0-", expected: &layer.RangeParams{Start: 0, End: 99}, fullSize: 100, err: false},
		{header: "bytes=-10", expected: &layer.RangeParams{Start: 90, End: 99}, fullSize: 100, err: false},
		{header: "", err: false},
		{header: "bytes=-1-256", err: true},
		{header: "bytes=256-0", err: true},
		{header: "bytes=string-0", err: true},
		{header: "bytes=0-string", err: true},
		{header: "bytes:0-256", err: true},
		{header: "bytes:-", err: true},
	} {
		h := make(http.Header)
		h.Add("Range", tc.header)
		params, err := fetchRangeHeader(h, tc.fullSize)
		if tc.err {
			require.Error(t, err)
			continue
		}

		require.NoError(t, err)
		require.Equal(t, tc.expected, params)
	}
}

func newInfo(etag string, created time.Time) *layer.ObjectInfo {
	return &layer.ObjectInfo{
		HashSum: etag,
		Created: created,
	}
}

func TestPreconditions(t *testing.T) {
	today := time.Now()
	yesterday := today.Add(-24 * time.Hour)
	etag := "etag"
	etag2 := "etag2"

	for _, tc := range []struct {
		name     string
		info     *layer.ObjectInfo
		args     *getObjectArgs
		expected int
	}{
		{
			name:     "no conditions",
			info:     new(layer.ObjectInfo),
			args:     new(getObjectArgs),
			expected: http.StatusOK,
		},
		{
			name:     "IfMatch true",
			info:     newInfo(etag, today),
			args:     &getObjectArgs{IfMatch: etag},
			expected: http.StatusOK,
		},
		{
			name:     "IfMatch false",
			info:     newInfo(etag, today),
			args:     &getObjectArgs{IfMatch: etag2},
			expected: http.StatusPreconditionFailed},
		{
			name:     "IfNoneMatch true",
			info:     newInfo(etag, today),
			args:     &getObjectArgs{IfNoneMatch: etag2},
			expected: http.StatusOK},
		{
			name:     "IfNoneMatch false",
			info:     newInfo(etag, today),
			args:     &getObjectArgs{IfNoneMatch: etag},
			expected: http.StatusNotModified},
		{
			name:     "IfModifiedSince true",
			info:     newInfo(etag, today),
			args:     &getObjectArgs{IfModifiedSince: &yesterday},
			expected: http.StatusOK},
		{
			name:     "IfModifiedSince false",
			info:     newInfo(etag, yesterday),
			args:     &getObjectArgs{IfModifiedSince: &today},
			expected: http.StatusNotModified},
		{
			name:     "IfUnmodifiedSince true",
			info:     newInfo(etag, yesterday),
			args:     &getObjectArgs{IfUnmodifiedSince: &today},
			expected: http.StatusOK},
		{
			name:     "IfUnmodifiedSince false",
			info:     newInfo(etag, today),
			args:     &getObjectArgs{IfUnmodifiedSince: &yesterday},
			expected: http.StatusPreconditionFailed},

		{
			name:     "IfMatch true, IfUnmodifiedSince false",
			info:     newInfo(etag, today),
			args:     &getObjectArgs{IfMatch: etag, IfUnmodifiedSince: &yesterday},
			expected: http.StatusOK,
		},
		{
			name:     "IfMatch false, IfUnmodifiedSince true",
			info:     newInfo(etag, yesterday),
			args:     &getObjectArgs{IfMatch: etag2, IfUnmodifiedSince: &today},
			expected: http.StatusPreconditionFailed,
		},
		{
			name:     "IfNoneMatch false, IfModifiedSince true",
			info:     newInfo(etag, today),
			args:     &getObjectArgs{IfNoneMatch: etag, IfModifiedSince: &yesterday},
			expected: http.StatusNotModified,
		},
		{
			name:     "IfNoneMatch true, IfModifiedSince false",
			info:     newInfo(etag, yesterday),
			args:     &getObjectArgs{IfNoneMatch: etag2, IfModifiedSince: &today},
			expected: http.StatusNotModified,
		},
	} {
		t.Run(tc.name, func(t *testing.T) {
			actual := checkGetPreconditions(tc.info, tc.args)
			require.Equal(t, tc.expected, actual)
		})
	}
}