Mirror/mochi
1
0
Fork 0
mirror of https://github.com/sot-tech/mochi.git synced 2026-05-09 21:59:09 -07:00
Code Issues Packages Projects Releases Wiki Activity

udp: Add constructor for ConnectionIDGenerator, return buffer to pool on socket timeout

Browse Source
This commit is contained in:
Justin Li
2015-02-22 16:58:43 -05:00
parent 7512f50731
commit 3d28f281fb
3 changed files with 17 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
udp/connection_test.go
View File

@@ -11,15 +11,13 @@ import (
)
func TestInitReturnsNoError(t *testing.T) {
gen := &ConnectionIDGenerator{}
if err := gen.Init(); err != nil {
if _, err := NewConnectionIDGenerator(); err != nil {
t.Error("Init returned", err)
}
}
func testGenerateConnectionID(t *testing.T, ip net.IP) {
gen := &ConnectionIDGenerator{}
gen.Init()
gen, _ := NewConnectionIDGenerator()
id1 := gen.Generate(ip)
id2 := gen.Generate(ip)
@@ -46,8 +44,7 @@ func TestGenerateConnectionIDIPv6(t *testing.T) {
}
func TestMatchesWorksWithPreviousIV(t *testing.T) {
gen := &ConnectionIDGenerator{}
gen.Init()
gen, _ := NewConnectionIDGenerator()
ip := net.ParseIP("192.168.1.123").To4()
id1 := gen.Generate(ip)