优化代码

internal/caches/reader_file_test.go

@@ -49,6 +49,9 @@ func TestFileReader(t *testing.T) {
 		t.Log("body:", string(buf[:n]))
 		return true, nil
 	})
+	if err != nil {
+		t.Fatal(err)
+	}
 }
 
 func TestFileReader_Range(t *testing.T) {

internal/stats/traffic_stat_manager_test.go

@@ -8,7 +8,7 @@ import (
 func TestTrafficStatManager_Add(t *testing.T) {
 	manager := NewTrafficStatManager()
 	for i := 0; i < 100; i++ {
-		manager.Add(1, "goedge.cn", 1, 0, 0, 0)
+		manager.Add(1, "goedge.cn", 1, 0, 0, 0, 0, 0)
 	}
 	t.Log(manager.itemMap)
 }
@@ -16,7 +16,7 @@ func TestTrafficStatManager_Add(t *testing.T) {
 func TestTrafficStatManager_Upload(t *testing.T) {
 	manager := NewTrafficStatManager()
 	for i := 0; i < 100; i++ {
-		manager.Add(1, "goedge.cn", 1, 0, 0, 0)
+		manager.Add(1, "goedge.cn", 1, 0, 0, 0, 0, 0)
 	}
 	err := manager.Upload()
 	if err != nil {
@@ -30,6 +30,6 @@ func BenchmarkTrafficStatManager_Add(b *testing.B) {
 
 	manager := NewTrafficStatManager()
 	for i := 0; i < b.N; i++ {
-		manager.Add(1, "goedge.cn", 1024, 1, 0, 0)
+		manager.Add(1, "goedge.cn", 1024, 1, 0, 0, 0, 0)
 	}
 }

internal/utils/http.go

@@ -16,6 +16,9 @@ var timeoutClientLocker = sync.Mutex{}
 // 导出响应
 func DumpResponse(resp *http.Response) (header []byte, body []byte, err error) {
 	header, err = httputil.DumpResponse(resp, false)
+	if err != nil {
+		return
+	}
 	body, err = ioutil.ReadAll(resp.Body)
 	return
 }

internal/waf/checkpoints/request_upload_test.go

@@ -17,46 +17,67 @@ func TestRequestUploadCheckpoint_RequestValue(t *testing.T) {
 	{
 		part, err := writer.CreateFormField("name")
 		if err == nil {
-			part.Write([]byte("lu"))
+			_, err := part.Write([]byte("lu"))
+			if err != nil {
+				t.Fatal(err)
+			}
 		}
 	}
 
 	{
 		part, err := writer.CreateFormField("age")
 		if err == nil {
-			part.Write([]byte("20"))
+			_, err = part.Write([]byte("20"))
+			if err != nil {
+				t.Fatal(err)
+			}
 		}
 	}
 
 	{
 		part, err := writer.CreateFormFile("myFile", "hello.txt")
 		if err == nil {
-			part.Write([]byte("Hello, World!"))
+			_, err = part.Write([]byte("Hello, World!"))
+			if err != nil {
+				t.Fatal(err)
+			}
 		}
 	}
 
 	{
 		part, err := writer.CreateFormFile("myFile2", "hello.PHP")
 		if err == nil {
-			part.Write([]byte("Hello, World, PHP!"))
+			_, err = part.Write([]byte("Hello, World, PHP!"))
+			if err != nil {
+				t.Fatal(err)
+			}
 		}
 	}
 
 	{
 		part, err := writer.CreateFormFile("myFile3", "hello.asp")
 		if err == nil {
-			part.Write([]byte("Hello, World, ASP Pages!"))
+			_, err = part.Write([]byte("Hello, World, ASP Pages!"))
+			if err != nil {
+				t.Fatal(err)
+			}
 		}
 	}
 
 	{
 		part, err := writer.CreateFormFile("myFile4", "hello.asp")
 		if err == nil {
-			part.Write([]byte("Hello, World, ASP Pages!"))
+			_, err = part.Write([]byte("Hello, World, ASP Pages!"))
+			if err != nil {
+				t.Fatal(err)
+			}
 		}
 	}
 
-	writer.Close()
+	err := writer.Close()
+	if err != nil {
+		t.Fatal(err)
+	}
 
 	rawReq, err := http.NewRequest(http.MethodPost, "http://teaos.cn/", body)
 	if err != nil {