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.
60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
package server
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/cbeuw/connutil"
|
|
)
|
|
|
|
func TestFirstBuffedConn_Read(t *testing.T) {
|
|
mockConn, writingEnd := connutil.AsyncPipe()
|
|
|
|
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.Write(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 := connutil.Discard()
|
|
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")
|
|
}
|
|
}
|