修复几处测试用例

This commit is contained in:
GoEdgeLab
2023-03-08 10:13:41 +08:00
parent 7bdea27dc5
commit 10ca7321aa

View File

@@ -10,22 +10,22 @@ import (
func TestBandwidthStatManager_Add(t *testing.T) { func TestBandwidthStatManager_Add(t *testing.T) {
var manager = stats.NewBandwidthStatManager() var manager = stats.NewBandwidthStatManager()
manager.Add(1, 1, 10) manager.Add(1, 1, 10, 10)
manager.Add(1, 1, 10) manager.Add(1, 1, 10, 10)
manager.Add(1, 1, 10) manager.Add(1, 1, 10, 10)
time.Sleep(1 * time.Second) time.Sleep(1 * time.Second)
manager.Add(1, 1, 85) manager.Add(1, 1, 85, 85)
time.Sleep(1 * time.Second) time.Sleep(1 * time.Second)
manager.Add(1, 1, 25) manager.Add(1, 1, 25, 25)
manager.Add(1, 1, 75) manager.Add(1, 1, 75, 75)
manager.Inspect() manager.Inspect()
} }
func TestBandwidthStatManager_Loop(t *testing.T) { func TestBandwidthStatManager_Loop(t *testing.T) {
var manager = stats.NewBandwidthStatManager() var manager = stats.NewBandwidthStatManager()
manager.Add(1, 1, 10) manager.Add(1, 1, 10, 10)
manager.Add(1, 1, 10) manager.Add(1, 1, 10, 10)
manager.Add(1, 1, 10) manager.Add(1, 1, 10, 10)
err := manager.Loop() err := manager.Loop()
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)