1.4.5.2

EdgeNode/internal/nodes/http_request_test.go

@@ -0,0 +1,71 @@
+package nodes
+
+import (
+	"github.com/TeaOSLab/EdgeCommon/pkg/serverconfigs"
+	"github.com/iwind/TeaGo/assert"
+	"net/http"
+	"runtime"
+	"testing"
+)
+
+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))
+}