mirror of https://github.com/cbeuw/Cloak
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
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")
|
|
}
|
|
} |