response_test.go

package req

import (
	"github.com/imroc/req/v3/internal/header"
	"github.com/imroc/req/v3/internal/tests"
	"net/http"
	"testing"
)

type User struct {
	Name string `json:"name" xml:"name"`
}

type Message struct {
	Message string `json:"message"`
}

func TestUnmarshalJson(t *testing.T) {
	var user User
	resp, err := tc().R().Get("/json")
	assertSuccess(t, resp, err)
	err = resp.UnmarshalJson(&user)
	tests.AssertNoError(t, err)
	tests.AssertEqual(t, "roc", user.Name)
}

func TestUnmarshalXml(t *testing.T) {
	var user User
	resp, err := tc().R().Get("/xml")
	assertSuccess(t, resp, err)
	err = resp.UnmarshalXml(&user)
	tests.AssertNoError(t, err)
	tests.AssertEqual(t, "roc", user.Name)
}

func TestUnmarshal(t *testing.T) {
	var user User
	resp, err := tc().R().Get("/xml")
	assertSuccess(t, resp, err)
	err = resp.Unmarshal(&user)
	tests.AssertNoError(t, err)
	tests.AssertEqual(t, "roc", user.Name)
}

func TestResponseResult(t *testing.T) {
	resp, _ := tc().R().SetResult(&User{}).Get("/json")
	user, ok := resp.Result().(*User)
	if !ok {
		t.Fatal("Response.Result() should return *User")
	}
	tests.AssertEqual(t, "roc", user.Name)

	tests.AssertEqual(t, true, resp.TotalTime() > 0)
	tests.AssertEqual(t, false, resp.ReceivedAt().IsZero())
}

func TestResponseError(t *testing.T) {
	resp, _ := tc().R().SetError(&Message{}).Get("/json?error=yes")
	msg, ok := resp.Error().(*Message)
	if !ok {
		t.Fatal("Response.Error() should return *Message")
	}
	tests.AssertEqual(t, "not allowed", msg.Message)
}

func TestResponseWrap(t *testing.T) {
	resp, err := tc().R().Get("/json")
	assertSuccess(t, resp, err)
	tests.AssertEqual(t, true, resp.GetStatusCode() == http.StatusOK)
	tests.AssertEqual(t, true, resp.GetStatus() == "200 OK")
	tests.AssertEqual(t, true, resp.GetHeader(header.ContentType) == header.JsonContentType)
	tests.AssertEqual(t, true, len(resp.GetHeaderValues(header.ContentType)) == 1)
}