diff --git a/internal/caches/reader_file_test.go b/internal/caches/reader_file_test.go index 6a4f77a..f557442 100644 --- a/internal/caches/reader_file_test.go +++ b/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) { diff --git a/internal/stats/traffic_stat_manager_test.go b/internal/stats/traffic_stat_manager_test.go index efd4bf2..680e4ce 100644 --- a/internal/stats/traffic_stat_manager_test.go +++ b/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) } } diff --git a/internal/utils/http.go b/internal/utils/http.go index a835cee..d581952 100644 --- a/internal/utils/http.go +++ b/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 } diff --git a/internal/waf/checkpoints/request_upload_test.go b/internal/waf/checkpoints/request_upload_test.go index 0bc03a5..d3fa899 100644 --- a/internal/waf/checkpoints/request_upload_test.go +++ b/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 {