2016-06-25 15:35:18 -04:00
|
|
|
package kcp_test
|
|
|
|
|
|
|
|
import (
|
|
|
|
"testing"
|
|
|
|
|
2016-08-20 14:55:45 -04:00
|
|
|
"v2ray.com/core/common/alloc"
|
|
|
|
"v2ray.com/core/testing/assert"
|
|
|
|
. "v2ray.com/core/transport/internet/kcp"
|
2016-06-25 15:35:18 -04:00
|
|
|
)
|
|
|
|
|
|
|
|
func TestRecivingWindow(t *testing.T) {
|
|
|
|
assert := assert.On(t)
|
|
|
|
|
|
|
|
window := NewReceivingWindow(3)
|
|
|
|
|
2016-06-29 04:34:34 -04:00
|
|
|
seg0 := &DataSegment{}
|
|
|
|
seg1 := &DataSegment{}
|
|
|
|
seg2 := &DataSegment{}
|
|
|
|
seg3 := &DataSegment{}
|
2016-06-25 15:35:18 -04:00
|
|
|
|
|
|
|
assert.Bool(window.Set(0, seg0)).IsTrue()
|
|
|
|
assert.Pointer(window.RemoveFirst()).Equals(seg0)
|
2016-06-27 03:03:05 -04:00
|
|
|
e := window.RemoveFirst()
|
|
|
|
if e != nil {
|
|
|
|
assert.Fail("Expecting nil.")
|
|
|
|
}
|
2016-06-25 15:35:18 -04:00
|
|
|
|
|
|
|
assert.Bool(window.Set(1, seg1)).IsTrue()
|
|
|
|
assert.Bool(window.Set(2, seg2)).IsTrue()
|
|
|
|
|
|
|
|
window.Advance()
|
|
|
|
assert.Bool(window.Set(2, seg3)).IsTrue()
|
|
|
|
|
|
|
|
assert.Pointer(window.RemoveFirst()).Equals(seg1)
|
|
|
|
assert.Pointer(window.Remove(1)).Equals(seg2)
|
|
|
|
assert.Pointer(window.Remove(2)).Equals(seg3)
|
|
|
|
}
|
2016-07-02 17:37:51 -04:00
|
|
|
|
|
|
|
func TestRecivingQueue(t *testing.T) {
|
|
|
|
assert := assert.On(t)
|
|
|
|
|
|
|
|
queue := NewReceivingQueue(2)
|
2016-07-28 10:24:15 -04:00
|
|
|
queue.Put(alloc.NewLocalBuffer(512).Clear().AppendString("abcd"))
|
|
|
|
queue.Put(alloc.NewLocalBuffer(512).Clear().AppendString("efg"))
|
2016-07-06 11:34:38 -04:00
|
|
|
assert.Bool(queue.IsFull()).IsTrue()
|
2016-07-02 17:37:51 -04:00
|
|
|
|
|
|
|
b := make([]byte, 1024)
|
2016-07-06 11:34:38 -04:00
|
|
|
nBytes := queue.Read(b)
|
2016-07-02 17:37:51 -04:00
|
|
|
assert.Int(nBytes).Equals(7)
|
|
|
|
assert.String(string(b[:nBytes])).Equals("abcdefg")
|
|
|
|
|
2016-07-28 10:24:15 -04:00
|
|
|
queue.Put(alloc.NewLocalBuffer(512).Clear().AppendString("1"))
|
2016-07-02 17:37:51 -04:00
|
|
|
queue.Close()
|
2016-07-06 11:34:38 -04:00
|
|
|
nBytes = queue.Read(b)
|
|
|
|
assert.Int(nBytes).Equals(0)
|
2016-07-02 17:37:51 -04:00
|
|
|
}
|