client_internal_test.go

package pilosa

import (
	"bytes"
	"io/ioutil"
	"net/http"
	"reflect"
	"testing"
	"testing/iotest"
)

func TestNewClientFromAddresses(t *testing.T) {
	cli, err := NewClientFromAddresses([]string{":10101", "node0.pilosa.com:10101", "node2.pilosa.com"}, &ClientOptions{})

	if err != nil {
		t.Fatalf("Creating client from addresses: %v", err)
	}
	expectedHosts := []URI{
		{scheme: "http", port: 10101, host: "localhost"},
		{scheme: "http", port: 10101, host: "node0.pilosa.com"},
		{scheme: "http", port: 10101, host: "node2.pilosa.com"},
	}
	actualHosts := cli.cluster.Hosts()
	if !reflect.DeepEqual(actualHosts, expectedHosts) {
		t.Fatalf("Unexpected hosts in client's cluster, got: %v, expected: %v", actualHosts, expectedHosts)
	}

	cli, err = NewClientFromAddresses([]string{"://"}, &ClientOptions{})
	if err == nil {
		t.Fatalf("Did not get expected error when creating client: %v", cli.cluster.Hosts())
	}

	cli, err = NewClientFromAddresses([]string{}, &ClientOptions{})
	if err != nil {
		t.Fatalf("Got error when creating empty client from addresses: %v", err)
	}

	cli, err = NewClientFromAddresses(nil, &ClientOptions{})
	if err != nil {
		t.Fatalf("Got error when creating empty client from addresses: %v", err)
	}
}

func TestMakeRequestData(t *testing.T) {
	q := make([]byte, 2<<30)
	q[0] = 'a'
	p := PQLBaseQuery{
		pql: string(q),
	}
	uri, err := NewURIFromAddress("localhost:10101")
	if err != nil {
		t.Fatal(err)
	}
	cli := NewClientWithURI(uri)
	resp, err := cli.Query(&p, nil)
	if err == nil {
		t.Fatalf("expected err with too large query, but got %v", resp)
	}
}

func TestAnyError(t *testing.T) {
	err := anyError(
		&http.Response{StatusCode: 400,
			Body: ioutil.NopCloser(iotest.TimeoutReader(bytes.NewBuffer([]byte("asdf"))))},
		nil)
	if err == nil {
		t.Fatalf("should have gotten an error")
	}

	err = anyError(
		&http.Response{StatusCode: 400,
			Body: ioutil.NopCloser(bytes.NewBuffer([]byte("index already exists\n")))},
		nil)
	if err != ErrorIndexExists {
		t.Fatalf("should have gotten ErrorIndexExists, but got %v", err)
	}

}