mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-11-18 02:16:10 -05:00
61 lines
1.0 KiB
Go
61 lines
1.0 KiB
Go
package collect
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/v2ray/v2ray-core/testing/unit"
|
|
)
|
|
|
|
func TestTimedQueue(t *testing.T) {
|
|
assert := unit.Assert(t)
|
|
|
|
removed := make(map[string]bool)
|
|
|
|
nowSec := time.Now().UTC().Unix()
|
|
q := NewTimedQueue(2)
|
|
|
|
go func() {
|
|
for {
|
|
entry := <-q.RemovedEntries()
|
|
removed[entry.(string)] = true
|
|
}
|
|
}()
|
|
|
|
q.Add("Value1", nowSec)
|
|
q.Add("Value2", nowSec+5)
|
|
|
|
v1, ok := removed["Value1"]
|
|
assert.Bool(ok).IsFalse()
|
|
|
|
v2, ok := removed["Value2"]
|
|
assert.Bool(ok).IsFalse()
|
|
|
|
tick := time.Tick(4 * time.Second)
|
|
<-tick
|
|
|
|
v1, ok = removed["Value1"]
|
|
assert.Bool(ok).IsTrue()
|
|
assert.Bool(v1).IsTrue()
|
|
removed["Value1"] = false
|
|
|
|
v2, ok = removed["Value2"]
|
|
assert.Bool(ok).IsFalse()
|
|
|
|
<-tick
|
|
v2, ok = removed["Value2"]
|
|
assert.Bool(ok).IsTrue()
|
|
assert.Bool(v2).IsTrue()
|
|
removed["Value2"] = false
|
|
|
|
<-tick
|
|
assert.Bool(removed["Values"]).IsFalse()
|
|
|
|
q.Add("Value1", time.Now().UTC().Unix()+10)
|
|
|
|
<-tick
|
|
v1, ok = removed["Value1"]
|
|
assert.Bool(ok).IsTrue()
|
|
assert.Bool(v1).IsFalse()
|
|
}
|