114 lines
2.5 KiB
Go
114 lines
2.5 KiB
Go
package model
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestSuppressor(t *testing.T) {
|
|
s := suppressor{threshold: 10000}
|
|
t0 := time.Now()
|
|
|
|
t1 := t0
|
|
sup, prev := s.suppress("foo", 10000, t1)
|
|
if sup {
|
|
t.Fatal("Never suppress first change")
|
|
}
|
|
if prev {
|
|
t.Fatal("Incorrect prev status")
|
|
}
|
|
|
|
// bw is 10000 / 10 = 1000
|
|
t1 = t0.Add(10 * time.Second)
|
|
if bw := s.changes["foo"].bandwidth(t1); bw != 1000 {
|
|
t.Error("Incorrect bw %d", bw)
|
|
}
|
|
sup, prev = s.suppress("foo", 10000, t1)
|
|
if sup {
|
|
t.Fatal("Should still be fine")
|
|
}
|
|
if prev {
|
|
t.Fatal("Incorrect prev status")
|
|
}
|
|
|
|
// bw is (10000 + 10000) / 11 = 1818
|
|
t1 = t0.Add(11 * time.Second)
|
|
if bw := s.changes["foo"].bandwidth(t1); bw != 1818 {
|
|
t.Error("Incorrect bw %d", bw)
|
|
}
|
|
sup, prev = s.suppress("foo", 100500, t1)
|
|
if sup {
|
|
t.Fatal("Should still be fine")
|
|
}
|
|
if prev {
|
|
t.Fatal("Incorrect prev status")
|
|
}
|
|
|
|
// bw is (10000 + 10000 + 100500) / 12 = 10041
|
|
t1 = t0.Add(12 * time.Second)
|
|
if bw := s.changes["foo"].bandwidth(t1); bw != 10041 {
|
|
t.Error("Incorrect bw %d", bw)
|
|
}
|
|
sup, prev = s.suppress("foo", 10000000, t1) // value will be ignored
|
|
if !sup {
|
|
t.Fatal("Should be over threshold")
|
|
}
|
|
if prev {
|
|
t.Fatal("Incorrect prev status")
|
|
}
|
|
|
|
// bw is (10000 + 10000 + 100500) / 15 = 8033
|
|
t1 = t0.Add(15 * time.Second)
|
|
if bw := s.changes["foo"].bandwidth(t1); bw != 8033 {
|
|
t.Error("Incorrect bw %d", bw)
|
|
}
|
|
sup, prev = s.suppress("foo", 10000000, t1)
|
|
if sup {
|
|
t.Fatal("Should be Ok")
|
|
}
|
|
if !prev {
|
|
t.Fatal("Incorrect prev status")
|
|
}
|
|
}
|
|
|
|
func TestHistory(t *testing.T) {
|
|
h := changeHistory{}
|
|
|
|
t0 := time.Now()
|
|
h.append(40, t0)
|
|
|
|
if l := len(h.changes); l != 1 {
|
|
t.Errorf("Incorrect history length %d", l)
|
|
}
|
|
if s := h.changes[0].size; s != 40 {
|
|
t.Errorf("Incorrect first record size %d", s)
|
|
}
|
|
|
|
for i := 1; i < MAX_CHANGE_HISTORY; i++ {
|
|
h.append(int64(40+i), t0.Add(time.Duration(i)*time.Second))
|
|
}
|
|
|
|
if l := len(h.changes); l != MAX_CHANGE_HISTORY {
|
|
t.Errorf("Incorrect history length %d", l)
|
|
}
|
|
if s := h.changes[0].size; s != 40 {
|
|
t.Errorf("Incorrect first record size %d", s)
|
|
}
|
|
if s := h.changes[MAX_CHANGE_HISTORY-1].size; s != 40+MAX_CHANGE_HISTORY-1 {
|
|
t.Errorf("Incorrect last record size %d", s)
|
|
}
|
|
|
|
h.append(999, t0.Add(time.Duration(999)*time.Second))
|
|
|
|
if l := len(h.changes); l != MAX_CHANGE_HISTORY {
|
|
t.Errorf("Incorrect history length %d", l)
|
|
}
|
|
if s := h.changes[0].size; s != 41 {
|
|
t.Errorf("Incorrect first record size %d", s)
|
|
}
|
|
if s := h.changes[MAX_CHANGE_HISTORY-1].size; s != 999 {
|
|
t.Errorf("Incorrect last record size %d", s)
|
|
}
|
|
|
|
}
|