EdgeNode/internal/nodes/http_request_test.go

package nodes

import (
	"net/http"
	"runtime"
	"testing"

	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
	"github.com/iwind/TeaGo/assert"
)

func TestHTTPRequest_RedirectToHTTPS(t *testing.T) {
	var a = assert.NewAssertion(t)
	{
		rawReq, err := http.NewRequest(http.MethodGet, "/", nil)
		if err != nil {
			t.Fatal(err)
		}
		var req = &HTTPRequest{
			RawReq:    rawReq,
			RawWriter: NewEmptyResponseWriter(nil),
			ReqServer: &serverconfigs.ServerConfig{
				IsOn: true,
				Web: &serverconfigs.HTTPWebConfig{
					IsOn:            true,
					RedirectToHttps: &serverconfigs.HTTPRedirectToHTTPSConfig{},
				},
			},
		}
		req.init()
		req.Do()

		a.IsBool(req.web.RedirectToHttps.IsOn == false)
	}
	{
		rawReq, err := http.NewRequest(http.MethodGet, "/", nil)
		if err != nil {
			t.Fatal(err)
		}
		var req = &HTTPRequest{
			RawReq:    rawReq,
			RawWriter: NewEmptyResponseWriter(nil),
			ReqServer: &serverconfigs.ServerConfig{
				IsOn: true,
				Web: &serverconfigs.HTTPWebConfig{
					IsOn: true,
					RedirectToHttps: &serverconfigs.HTTPRedirectToHTTPSConfig{
						IsOn: true,
					},
				},
			},
		}
		req.init()
		req.Do()
		a.IsBool(req.web.RedirectToHttps.IsOn == true)
	}
}

func TestHTTPRequest_Memory(t *testing.T) {
	var stat1 = &runtime.MemStats{}
	runtime.ReadMemStats(stat1)

	var requests = []*HTTPRequest{}
	for i := 0; i < 1_000_000; i++ {
		requests = append(requests, &HTTPRequest{})
	}

	var stat2 = &runtime.MemStats{}
	runtime.ReadMemStats(stat2)
	t.Log((stat2.HeapInuse-stat1.HeapInuse)/1024/1024, "MB,")
	t.Log(len(requests))
}