TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-15 17:40:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

优化代码

Browse Source
This commit is contained in:
GoEdgeLab
2021-08-07 16:27:42 +08:00
parent a709b64c81
commit 1dc8c23e3b
4 changed files with 37 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
internal/caches/reader_file_test.go
View File

@@ -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) {

6
internal/stats/traffic_stat_manager_test.go
View File

@@ -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)
}
}

3
internal/utils/http.go
View File

@@ -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
}

35
internal/waf/checkpoints/request_upload_test.go
View File

@@ -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 {