client_test.go

package jrpc

import (
	"encoding/json"
	"fmt"
	"io"
	"net/http"
	"net/http/httptest"
	"testing"
	"time"

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

func TestClient_Call(t *testing.T) {
	ts := testServer(t, `{"method":"test","params":[123,"abc"],"id":1}`, `{"result":"12345"}`)
	defer ts.Close()
	c := Client{API: ts.URL, Client: http.Client{}}
	resp, err := c.Call("test", 123, "abc")
	assert.NoError(t, err)
	res := ""
	err = json.Unmarshal(*resp.Result, &res)
	assert.NoError(t, err)
	assert.Equal(t, "12345", res)
	t.Logf("%v %T", res, res)
}

func TestClient_CallSingle(t *testing.T) {
	ts := testServer(t, `{"method":"test","params":"abc","id":1}`, `{"result":"12345"}`)
	defer ts.Close()
	c := Client{API: ts.URL, Client: http.Client{}}
	resp, err := c.Call("test", "abc")
	assert.NoError(t, err)
	res := ""
	err = json.Unmarshal(*resp.Result, &res)
	assert.NoError(t, err)
	assert.Equal(t, "12345", res)
	t.Logf("%v %T", res, res)
}

func TestClient_CallWithObject(t *testing.T) {
	ts := testServer(t, `{"method":"test","params":{"F1":123,"F2":"abc","F3":"2019-06-09T23:03:55Z"},"id":1}`, `{"result":"12345"}`)
	defer ts.Close()
	c := Client{API: ts.URL, Client: http.Client{}}
	obj := struct {
		F1 int
		F2 string
		F3 time.Time
	}{
		F1: 123,
		F2: "abc",
		F3: time.Date(2019, 6, 9, 23, 3, 55, 0, time.UTC),
	}

	resp, err := c.Call("test", obj)
	assert.NoError(t, err)
	res := ""
	err = json.Unmarshal(*resp.Result, &res)
	assert.NoError(t, err)
	assert.Equal(t, "12345", res)
	t.Logf("%v %T", res, res)
}

func TestClient_CallWithNoParams(t *testing.T) {
	ts := testServer(t, `{"method":"test","id":1}`, `{"result":"12345"}`)
	defer ts.Close()
	c := Client{API: ts.URL, Client: http.Client{}}
	resp, err := c.Call("test")
	assert.NoError(t, err)
	res := ""
	err = json.Unmarshal(*resp.Result, &res)
	assert.NoError(t, err)
	assert.Equal(t, "12345", res)
	t.Logf("%v %T", res, res)
}

func TestClient_CallError(t *testing.T) {
	ts := testServer(t, `{"method":"test","params":[123,"abc"],"id":1}`, `{"error":"some error"}`)
	defer ts.Close()
	c := Client{API: ts.URL, Client: http.Client{}}
	_, err := c.Call("test", 123, "abc")
	assert.EqualError(t, err, "some error")
}

func TestClient_CallBadResponse(t *testing.T) {
	ts := testServer(t, `{"method":"test","params":[123,"abc"],"id":1}`, `{"result":"12345 invalid}`)
	defer ts.Close()
	c := Client{API: ts.URL, Client: http.Client{}}
	_, err := c.Call("test", 123, "abc")
	assert.NotNil(t, err)
}

func TestClient_CallBadRemote(t *testing.T) {
	c := Client{API: "http://127.0.0.2", Client: http.Client{Timeout: 10 * time.Millisecond}}
	_, err := c.Call("test", 123)
	assert.NotNil(t, err)
}

func testServer(t *testing.T, req, resp string) *httptest.Server {
	return httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		body, err := io.ReadAll(r.Body)
		require.NoError(t, err)
		assert.Equal(t, req, string(body))
		t.Logf("req: %s", string(body))
		_, err = fmt.Fprint(w, resp)
		require.NoError(t, err)
	}))
}