request_test.go

package documentdb

import (
	"bytes"
	"encoding/json"
	"net/http"
	"testing"

	"github.com/stretchr/testify/assert"
)

type TestPartitionKey struct {
	Prop string `json:"prop"`
}

func (t *TestPartitionKey) MarshalJSON() ([]byte, error) {
	return json.Marshal(&struct {
		NewProp string `json:"newProp"`
	}{NewProp: t.Prop})
}

func TestResourceRequest(t *testing.T) {
	assert := assert.New(t)
	req := ResourceRequest("/dbs/b5NCAA==/", &http.Request{})
	assert.Equal(req.rType, "dbs")
	assert.Equal(req.rId, "b5NCAA==")
}

func TestDefaultHeaders(t *testing.T) {
	testUserAgent := "test/user agent"

	r, _ := http.NewRequest("GET", "link", &bytes.Buffer{})
	req := ResourceRequest("/dbs/b5NCAA==/", r)
	_ = req.DefaultHeaders(&Config{MasterKey: &Key{Key: "YXJpZWwNCg=="}}, testUserAgent)

	assert := assert.New(t)
	assert.NotEqual(req.Header.Get(HeaderAuth), "")
	assert.NotEqual(req.Header.Get(HeaderXDate), "")
	assert.NotEqual(req.Header.Get(HeaderVersion), "")
	assert.Equal(req.Header.Get(HeaderUserAgent), testUserAgent)
}

func TestUpsertHeaders(t *testing.T) {
	r, _ := http.NewRequest("POST", "link", &bytes.Buffer{})
	req := ResourceRequest("/dbs/b5NCAA==/", r)

	Upsert()(req)

	assert := assert.New(t)
	assert.Equal(req.Header.Get(HeaderUpsert), "true")
}

func TestPartitionKeyMarshalJSON(t *testing.T) {
	r, _ := http.NewRequest("GET", "link", &bytes.Buffer{})
	req := ResourceRequest("/dbs/b5NCAA==/", r)

	PartitionKey(&TestPartitionKey{"test"})(req)

	assert := assert.New(t)
	assert.Equal([]string{"{\"newProp\":\"test\"}"}, req.Header[HeaderPartitionKey])
}

func TestPartitionKeyAsInt(t *testing.T) {
	r, _ := http.NewRequest("GET", "link", &bytes.Buffer{})
	req := ResourceRequest("/dbs/b5NCAA==/", r)

	PartitionKey(1)(req)

	assert := assert.New(t)
	assert.Equal([]string{"[1]"}, req.Header[HeaderPartitionKey])
}

func TestPartitionKeyAsString(t *testing.T) {
	r, _ := http.NewRequest("GET", "link", &bytes.Buffer{})
	req := ResourceRequest("/dbs/b5NCAA==/", r)

	PartitionKey("1")(req)

	assert := assert.New(t)
	assert.Equal([]string{"[\"1\"]"}, req.Header[HeaderPartitionKey])
}