contentful_test.go

package contentful

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"log"
	"net/http"
	"net/http/httptest"
	"net/url"
	"strconv"
	"strings"
	"testing"
	"time"

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

var (
	server         *httptest.Server
	cma            *Client
	c              *Client
	CMAToken       = "b4c0n73n7fu1"
	CDAToken       = "cda-token"
	CPAToken       = "cpa-token"
	spaceID        = "id1"
	organizationID = "org-id"
)

func readTestData(fileName string) string {
	path := "testdata/" + fileName
	content, err := ioutil.ReadFile(path)
	if err != nil {
		log.Fatal(err)
		return ""
	}

	return string(content)
}

func checkHeaders(req *http.Request, assert *assert.Assertions) {
	assert.Equal("Bearer "+CMAToken, req.Header.Get("Authorization"))
	assert.Equal("application/vnd.contentful.management.v1+json", req.Header.Get("Content-Type"))
}

func spaceFromTestData(fileName string) (*Space, error) {
	content := readTestData(fileName)

	var space Space
	err := json.NewDecoder(strings.NewReader(content)).Decode(&space)
	if err != nil {
		return nil, err
	}

	return &space, nil
}

func webhookFromTestData(fileName string) (*Webhook, error) {
	content := readTestData(fileName)

	var webhook Webhook
	err := json.NewDecoder(strings.NewReader(content)).Decode(&webhook)
	if err != nil {
		return nil, err
	}

	return &webhook, nil
}

func contentTypeFromTestData(fileName string) (*ContentType, error) {
	content := readTestData(fileName)

	var ct ContentType
	err := json.NewDecoder(strings.NewReader(content)).Decode(&ct)
	if err != nil {
		return nil, err
	}

	return &ct, nil
}

func localeFromTestData(fileName string) (*Locale, error) {
	content := readTestData(fileName)

	var locale Locale
	err := json.NewDecoder(strings.NewReader(content)).Decode(&locale)
	if err != nil {
		return nil, err
	}

	return &locale, nil
}

func setup() {
	handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		fixture := strings.Replace(r.URL.Path, "/", "-", -1)
		fixture = strings.TrimLeft(fixture, "-")
		var path string

		if e := r.URL.Query().Get("error"); e != "" {
			path = "testdata/error-" + e + ".json"
		} else {
			if r.Method == "GET" {
				path = "testdata/" + fixture + ".json"
			}

			if r.Method == "POST" {
				path = "testdata/" + fixture + "-new.json"
			}

			if r.Method == "PUT" {
				path = "testdata/" + fixture + "-updated.json"
			}
		}

		file, err := ioutil.ReadFile(path)
		if err != nil {
			fmt.Fprintln(w, err)
			return
		}

		fmt.Fprintln(w, string(file))
		return
	})

	server = httptest.NewServer(handler)

	c = NewCMA(CMAToken)
	c.BaseURL = server.URL
}

func teardown() {
	server.Close()
	c = nil
}

func TestContentfulNewCMA(t *testing.T) {
	assert := assert.New(t)

	cma := NewCMA(CMAToken)
	assert.IsType(Client{}, *cma)
	assert.Equal("https://api.contentful.com", cma.BaseURL)
	assert.Equal("CMA", cma.api)
	assert.Equal(CMAToken, cma.token)
	assert.Equal(fmt.Sprintf("Bearer %s", CMAToken), cma.Headers["Authorization"])
	assert.Equal("application/vnd.contentful.management.v1+json", cma.Headers["Content-Type"])
	assert.Equal(fmt.Sprintf("sdk contentful-go/%s", Version), cma.Headers["X-Contentful-User-Agent"])
}

func TestContentfulNewCDA(t *testing.T) {
	assert := assert.New(t)

	cda := NewCDA(CDAToken)
	assert.IsType(Client{}, *cda)
	assert.Equal("https://cdn.contentful.com", cda.BaseURL)
	assert.Equal("CDA", cda.api)
	assert.Equal(CDAToken, cda.token)
	assert.Equal(fmt.Sprintf("Bearer %s", CDAToken), cda.Headers["Authorization"])
	assert.Equal("application/vnd.contentful.delivery.v1+json", cda.Headers["Content-Type"])
	assert.Equal(fmt.Sprintf("sdk contentful-go/%s", Version), cda.Headers["X-Contentful-User-Agent"])
}

func TestContentfulNewCPA(t *testing.T) {
	assert := assert.New(t)

	cpa := NewCPA(CPAToken)
	assert.IsType(Client{}, *cpa)
	assert.Equal("https://preview.contentful.com", cpa.BaseURL)
	assert.Equal("CPA", cpa.api)
	assert.Equal(CPAToken, cpa.token)
}

func TestContentfulSetOrganization(t *testing.T) {
	assert := assert.New(t)

	cma := NewCMA(CMAToken)
	cma.SetOrganization(organizationID)
	assert.Equal(organizationID, cma.Headers["X-Contentful-Organization"])
}

func TestContentfulSetClient(t *testing.T) {
	assert := assert.New(t)

	newClient := &http.Client{}
	cma := NewCMA(CMAToken)
	cma.SetHTTPClient(newClient)
	assert.Equal(newClient, cma.client)
}

func TestNewRequest(t *testing.T) {
	setup()
	defer teardown()

	assert := assert.New(t)

	method := "GET"
	path := "/some/path"
	query := url.Values{}
	query.Add("foo", "bar")
	query.Add("faz", "zoo")

	expectedURL, _ := url.Parse(c.BaseURL)
	expectedURL.Path = path
	expectedURL.RawQuery = query.Encode()

	req, err := c.newRequest(method, path, query, nil)
	assert.Nil(err)
	assert.Equal(req.Header.Get("Authorization"), "Bearer "+CMAToken)
	assert.Equal(req.Header.Get("Content-Type"), "application/vnd.contentful.management.v1+json")
	assert.Equal(req.Method, method)
	assert.Equal(req.URL.String(), expectedURL.String())

	method = "POST"
	type RequestBody struct {
		Name string `json:"name"`
		Age  int    `json:"age"`
	}
	bodyData := RequestBody{
		Name: "test",
		Age:  10,
	}
	body, _ := json.Marshal(bodyData)
	req, err = c.newRequest(method, path, query, bytes.NewReader(body))
	assert.Nil(err)
	assert.Equal(req.Header.Get("Authorization"), "Bearer "+CMAToken)
	assert.Equal(req.Header.Get("Content-Type"), "application/vnd.contentful.management.v1+json")
	assert.Equal(req.Method, method)
	assert.Equal(req.URL.String(), expectedURL.String())
	defer req.Body.Close()
	var requestBody RequestBody
	err = json.NewDecoder(req.Body).Decode(&requestBody)
	assert.Nil(err)
	assert.Equal(requestBody, bodyData)
}

func TestHandleError(t *testing.T) {
	setup()
	defer teardown()

	assert := assert.New(t)

	method := "GET"
	path := "/some/path"
	requestID := "request-id"
	query := url.Values{}
	errResponse := ErrorResponse{
		Sys: &Sys{
			ID:   "AccessTokenInvalid",
			Type: "Error",
		},
		Message:   "Access token is invalid",
		RequestID: requestID,
	}

	marshaled, _ := json.Marshal(errResponse)
	errResponseReader := bytes.NewReader(marshaled)
	errResponseReadCloser := ioutil.NopCloser(errResponseReader)

	req, _ := c.newRequest(method, path, query, nil)
	responseHeaders := http.Header{}
	responseHeaders.Add("X-Contentful-Request-Id", requestID)
	res := &http.Response{
		Header:     responseHeaders,
		StatusCode: http.StatusUnauthorized,
		Body:       errResponseReadCloser,
		Request:    req,
	}

	err := c.handleError(req, res)
	assert.IsType(AccessTokenInvalidError{}, err)
	assert.Equal(req, err.(AccessTokenInvalidError).APIError.req)
	assert.Equal(res, err.(AccessTokenInvalidError).APIError.res)
	assert.Equal(&errResponse, err.(AccessTokenInvalidError).APIError.err)
}

func TestBackoffForPerSecondLimiting(t *testing.T) {
	var err error
	assert := assert.New(t)
	rateLimited := true
	waitSeconds := 2

	handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		if rateLimited == true {
			w.Header().Set("X-Contentful-Request-Id", "request-id")
			w.Header().Set("Content-Type", "application/vnd.contentful.management.v1+json")
			w.Header().Set("X-Contentful-Ratelimit-Hour-Limit", "36000")
			w.Header().Set("X-Contentful-Ratelimit-Hour-Remaining", "35883")
			w.Header().Set("X-Contentful-Ratelimit-Reset", strconv.Itoa(waitSeconds))
			w.Header().Set("X-Contentful-Ratelimit-Second-Limit", "10")
			w.Header().Set("X-Contentful-Ratelimit-Second-Remaining", "0")
			w.WriteHeader(429)

			w.Write([]byte(readTestData("error-ratelimit.json")))
		} else {
			w.Write([]byte(readTestData("space-1.json")))
		}
	})

	// test server
	server := httptest.NewServer(handler)
	defer server.Close()

	// cma client
	cma = NewCMA(CMAToken)
	cma.BaseURL = server.URL

	go func() {
		time.Sleep(time.Second * time.Duration(waitSeconds))
		rateLimited = false
	}()

	space, err := cma.Spaces.Get("id1")
	assert.Nil(err)
	assert.Equal(space.Name, "Contentful Example API")
	assert.Equal(space.Sys.ID, "id1")
}