diff --git a/internal/stats/http_request_stat_manager_test.go b/internal/stats/http_request_stat_manager_test.go index f83c3e7..d0b1391 100644 --- a/internal/stats/http_request_stat_manager_test.go +++ b/internal/stats/http_request_stat_manager_test.go @@ -14,9 +14,9 @@ func TestHTTPRequestStatManager_Loop_Region(t *testing.T) { iplibrary.SharedLibrary = library manager := NewHTTPRequestStatManager() - manager.AddRemoteAddr(11, "202.196.0.20") - manager.AddRemoteAddr(11, "202.196.0.20") // 重复添加一个测试相加 - manager.AddRemoteAddr(11, "8.8.8.8") + manager.AddRemoteAddr(11, "202.196.0.20", 0, false) + manager.AddRemoteAddr(11, "202.196.0.20", 0, false) // 重复添加一个测试相加 + manager.AddRemoteAddr(11, "8.8.8.8", 0, false) /**for i := 0; i < 100; i++ { manager.AddRemoteAddr(11, strconv.Itoa(rands.Int(10, 250))+"."+strconv.Itoa(rands.Int(10, 250))+"."+strconv.Itoa(rands.Int(10, 250))+".8") diff --git a/internal/stats/traffic_stat_manager_test.go b/internal/stats/traffic_stat_manager_test.go index 680e4ce..4dadec3 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, 0, 0) + manager.Add(1, "goedge.cn", 1, 0, 0, 0, 0, 0, false, 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, 0, 0) + manager.Add(1, "goedge.cn", 1, 0, 0, 0, 0, 0, false, 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, 0, 0) + manager.Add(1, "goedge.cn", 1024, 1, 0, 0, 0, 0, false, 0) } }