httpmock_test.go

package testutil_test

import (
	"fmt"
	"io"
	"net/http"
	"strings"
	"testing"

	"github.com/gookit/goutil/testutil"
	"github.com/gookit/goutil/testutil/assert"
	"github.com/gookit/goutil/testutil/fakeobj"
)

var testSrvAddr string

func TestMain(m *testing.M) {
	s := testutil.NewEchoServer()
	defer s.Close()

	testSrvAddr = s.HTTPHost()
	fmt.Println("server addr:", testSrvAddr)

	m.Run()
}

func TestNewHTTPRequest(t *testing.T) {
	r := testutil.NewHTTPRequest("GET", testSrvAddr+"/hello", &testutil.MD{
		Headers: map[string]string{
			"X-Test": "val",
		},
		BeforeSend: func(req *http.Request) {
			req.Header.Set("X-Test2", "val2")
		},
	})

	assert.Eq(t, "GET", r.Method)
	assert.Eq(t, testSrvAddr+"/hello", r.URL.String())
	assert.Eq(t, "val", r.Header.Get("X-Test"))
	assert.Eq(t, "val2", r.Header.Get("X-Test2"))

	assert.Panics(t, func() {
		testutil.NewHTTPRequest("invalid", "://", nil)
	})
}

func TestMockRequest(t *testing.T) {
	r := http.NewServeMux()
	r.Handle("/", http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		_, _ = w.Write([]byte("hello!"))

		if r.Body != nil {
			bs, _ := io.ReadAll(r.Body)
			_, _ = w.Write(bs)
		}
	}))

	w := testutil.MockRequest(r, "GET", "/", nil)
	assert.Eq(t, "hello!", w.Body.String())

	w = testutil.MockRequest(r, "POST", "/", &testutil.MD{BodyString: "body"})
	assert.Eq(t, "hello!body", w.Body.String())

	w = testutil.MockRequest(r, "POST", "/", &testutil.MD{Body: strings.NewReader("BODY")})
	assert.Eq(t, "hello!BODY", w.Body.String())
}

func TestNewEchoServer(t *testing.T) {
	r, err := http.Post(testSrvAddr, "text/plain", strings.NewReader("hello!"))
	assert.NoErr(t, err)

	rr := testutil.ParseRespToReply(r)
	// dump.P(rr)
	assert.Eq(t, "POST", rr.Method)
	assert.Eq(t, "text/plain", rr.ContentType())
	assert.Eq(t, "hello!", rr.Body)

	r, err = http.Post(testSrvAddr, "application/json", strings.NewReader(`{"name": "inhere", "age": 18}`))
	assert.NoErr(t, err)

	rr = testutil.ParseRespToReply(r)
	// dump.P(rr)
	assert.Eq(t, "POST", rr.Method)
	assert.Eq(t, "application/json", rr.ContentType())
	assert.Eq(t, `{"name": "inhere", "age": 18}`, rr.Body)

	r, err = http.Head(testSrvAddr + "/head")
	assert.NoErr(t, err)
	rr = testutil.ParseRespToReply(r)
	assert.Eq(t, "HEAD", rr.Method)

	rr = testutil.ParseRespToReply(&http.Response{})
	assert.Empty(t, *rr)

	rr = testutil.ParseBodyToReply(nil)
	assert.Empty(t, *rr)

	assert.Panics(t, func() {
		tr := fakeobj.NewStrReader("invalid-json")
		testutil.ParseBodyToReply(tr)
	})
}