client_test.go

package client

import (
	"bytes"
	"fmt"
	"net/http"
	"net/http/httptest"
	"net/url"
	"testing"

	"github.com/convox/rack/test"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func testClient(t *testing.T, serverUrl string) *Client {
	u, _ := url.Parse(serverUrl)

	client := New(u.Host, "test", "test")

	require.NotNil(t, client, "client should not be nil")

	return client
}

func testServer(t *testing.T, stubs ...test.Http) *httptest.Server {
	stubs = append(stubs, test.Http{Method: "GET", Path: "/system", Code: 200, Response: System{
		Version: "test",
	}})

	return test.Server(t, stubs...)
}

type ErrorReader struct {
	Error string
}

func (er ErrorReader) Read(buf []byte) (int, error) {
	return 0, fmt.Errorf(er.Error)
}

func (er ErrorReader) Close() error {
	return nil
}

func TestClientErrorReading(t *testing.T) {
	er := ErrorReader{Error: "error reading"}
	res := &http.Response{StatusCode: 400, Body: er}
	err := responseError(res)

	assert.NotNil(t, err, "err is not nil")
	assert.Equal(t, "error reading response body: error reading", err.Error(), "err text is valid")
}

func TestClientNonJson(t *testing.T) {
	ts := testServer(t,
		test.Http{Method: "GET", Path: "/", Code: 503, Response: "not-json"},
	)

	defer ts.Close()

	var err Error

	testClient(t, ts.URL).Get("/", &err)
}

func TestClientGetErrors(t *testing.T) {
	client := New("", "", "")

	err := client.Get("", nil)

	assert.NotNil(t, err)
	assert.Equal(t, "Get https://: http: no Host in request URL", err.Error())

	err = client.Get("/%", nil)

	assert.NotNil(t, err)
	assert.Equal(t, "parse https:///%: invalid URL escape \"%\"", err.Error())
}

func TestClientGet(t *testing.T) {
	ts := testServer(t,
		test.Http{Method: "GET", Path: "/", Code: 200, Response: "this is data"},
	)
	defer ts.Close()

	client := testClient(t, ts.URL)

	w := bytes.NewBuffer([]byte{})

	client.Get("/", w)

	assert.Equal(t, "\"this is data\"", w.String())
}