mirror of https://github.com/cbeuw/Cloak
extract util testing function
parent
47410dbb97
commit
e33afb258a
@ -0,0 +1,58 @@
|
|||||||
|
package server
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"github.com/cbeuw/Cloak/internal/util"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestFirstBuffedConn_Read(t *testing.T) {
|
||||||
|
mockConn, writingEnd := util.GetMockConn()
|
||||||
|
|
||||||
|
expectedFirstPacket := []byte{1, 2, 3}
|
||||||
|
firstBuffedConn := &firstBuffedConn{
|
||||||
|
Conn: mockConn,
|
||||||
|
firstRead: false,
|
||||||
|
firstPacket: expectedFirstPacket,
|
||||||
|
}
|
||||||
|
|
||||||
|
buf := make([]byte, 1024)
|
||||||
|
n ,err :=firstBuffedConn.Read(buf)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !bytes.Equal(expectedFirstPacket, buf[:n]){
|
||||||
|
t.Error("first read doesn't produce given packet")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
expectedSecondPacket := []byte{4,5,6,7}
|
||||||
|
writingEnd <- expectedSecondPacket
|
||||||
|
n ,err =firstBuffedConn.Read(buf)
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !bytes.Equal(expectedSecondPacket, buf[:n]){
|
||||||
|
t.Error("second read doesn't produce subsequently written packet")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestWsAcceptor(t *testing.T){
|
||||||
|
mockConn, _ := util.GetMockConn()
|
||||||
|
expectedFirstPacket := []byte{1, 2, 3}
|
||||||
|
|
||||||
|
wsAcceptor:=newWsAcceptor(mockConn, expectedFirstPacket)
|
||||||
|
_,err := wsAcceptor.Accept()
|
||||||
|
if err != nil {
|
||||||
|
t.Error(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
_,err = wsAcceptor.Accept()
|
||||||
|
if err == nil{
|
||||||
|
t.Error("accepting second time doesn't return error")
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue