github.com/wangyougui/gf/v2@v2.6.5/net/ghttp/ghttp_z_unit_test.go

// Copyright GoFrame Author(https://goframe.org). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://github.com/wangyougui/gf.

package ghttp_test

import (
	"context"
	"fmt"
	"net/http"
	"net/url"
	"runtime"
	"testing"
	"time"

	"github.com/wangyougui/gf/v2/encoding/gurl"
	"github.com/wangyougui/gf/v2/frame/g"
	"github.com/wangyougui/gf/v2/internal/httputil"
	"github.com/wangyougui/gf/v2/net/ghttp"
	"github.com/wangyougui/gf/v2/os/genv"
	"github.com/wangyougui/gf/v2/test/gtest"
	"github.com/wangyougui/gf/v2/util/guid"
)

var (
	ctx = context.TODO()
)

func init() {
	genv.Set("UNDER_TEST", "1")
}

func Test_GetUrl(t *testing.T) {
	s := g.Server(guid.S())
	s.BindHandler("/url", func(r *ghttp.Request) {
		r.Response.Write(r.GetUrl())
	})
	s.SetDumpRouterMap(false)
	s.Start()
	defer s.Shutdown()

	time.Sleep(100 * time.Millisecond)
	gtest.C(t, func(t *gtest.T) {
		prefix := fmt.Sprintf("http://127.0.0.1:%d", s.GetListenedPort())
		client := g.Client()
		client.SetBrowserMode(true)
		client.SetPrefix(prefix)

		t.Assert(client.GetContent(ctx, "/url"), prefix+"/url")
	})
}

func Test_XUrlPath(t *testing.T) {
	s := g.Server(guid.S())
	s.BindHandler("/test1", func(r *ghttp.Request) {
		r.Response.Write(`test1`)
	})
	s.BindHandler("/test2", func(r *ghttp.Request) {
		r.Response.Write(`test2`)
	})
	s.SetHandler(func(w http.ResponseWriter, r *http.Request) {
		r.Header.Set(ghttp.HeaderXUrlPath, "/test2")
		s.ServeHTTP(w, r)
	})
	s.SetDumpRouterMap(false)
	s.Start()
	defer s.Shutdown()

	time.Sleep(100 * time.Millisecond)
	gtest.C(t, func(t *gtest.T) {
		c := g.Client()
		c.SetPrefix(fmt.Sprintf("http://127.0.0.1:%d", s.GetListenedPort()))

		t.Assert(c.GetContent(ctx, "/"), "test2")
		t.Assert(c.GetContent(ctx, "/test/test"), "test2")
		t.Assert(c.GetContent(ctx, "/test1"), "test2")
	})
}

func Test_GetListenedAddress(t *testing.T) {
	s := g.Server(guid.S())
	s.BindHandler("/", func(r *ghttp.Request) {
		r.Response.Write(`test`)
	})
	s.SetDumpRouterMap(false)
	s.Start()
	defer s.Shutdown()

	time.Sleep(100 * time.Millisecond)
	gtest.C(t, func(t *gtest.T) {
		c := g.Client()
		c.SetPrefix(fmt.Sprintf("http://127.0.0.1:%d", s.GetListenedPort()))
		t.Assert(c.GetContent(ctx, "/"), "test")
	})
	gtest.C(t, func(t *gtest.T) {
		t.Assert(fmt.Sprintf(`:%d`, s.GetListenedPort()), s.GetListenedAddress())
	})
}

func Test_GetListenedAddressWithHost(t *testing.T) {
	s := g.Server(guid.S())
	s.BindHandler("/", func(r *ghttp.Request) {
		r.Response.Write(`test`)
	})
	s.SetAddr("127.0.0.1:0")
	s.SetDumpRouterMap(false)
	s.Start()
	defer s.Shutdown()

	time.Sleep(100 * time.Millisecond)
	gtest.C(t, func(t *gtest.T) {
		c := g.Client()
		c.SetPrefix(fmt.Sprintf("http://127.0.0.1:%d", s.GetListenedPort()))
		t.Assert(c.GetContent(ctx, "/"), "test")
	})
	gtest.C(t, func(t *gtest.T) {
		t.Assert(fmt.Sprintf(`127.0.0.1:%d`, s.GetListenedPort()), s.GetListenedAddress())
	})
}

func Test_RoutePathParams(t *testing.T) {
	s := g.Server(guid.S())
	s.BindHandler("/:param", func(r *ghttp.Request) {
		r.Response.Write(r.Get("param"), ",", r.Get("c"))
	})
	s.SetAddr("127.0.0.1:0")
	s.SetDumpRouterMap(false)
	s.Start()
	defer s.Shutdown()

	time.Sleep(100 * time.Millisecond)
	gtest.C(t, func(t *gtest.T) {
		c := g.Client()
		param := "net/http/get"
		c.SetPrefix(fmt.Sprintf("http://127.0.0.1:%d", s.GetListenedPort()))
		t.Assert(c.GetContent(
			ctx,
			"/"+gurl.Encode(param)+"?a=1&b=2&c="+gurl.Encode(param)),
			"net/http/get,net/http/get",
		)
	})
}

func Test_BuildParams(t *testing.T) {
	// normal && special cases
	params := map[string]string{
		"val":   "12345678",
		"code1": "x&a=1", // for fix
		"code2": "x&a=111",
		"id":    "1+- ", // for fix
		"f":     "1#a=+- ",
		"v":     "",
		"n":     "null",
	}

	gtest.C(t, func(t *gtest.T) {
		res1 := httputil.BuildParams(params)
		vs, _ := url.ParseQuery(res1)
		t.Assert(len(params), len(vs))
		for k := range vs {
			vv := vs.Get(k)
			_, ok := params[k]
			// check no additional param
			t.Assert(ok, true)
			// check equal
			t.AssertEQ(params[k], vv)
		}
	})
}

func Test_ServerSignal(t *testing.T) {
	if runtime.GOOS == "windows" {
		t.Log("skip windows")
		return
	}
	s := g.Server(guid.S())
	s.BindHandler("/", func(r *ghttp.Request) {
		r.Response.Write("hello world")
	})
	gtest.Assert(s.Start(), nil)
	g.Wait()
	time.Sleep(100 * time.Millisecond)
	gtest.C(t, func(t *gtest.T) {
		t.AssertEQ(s.Shutdown(), nil)
	})
}