Cloak/internal/server/websocketAux_test.go

package server

import (
	"bytes"
	"github.com/cbeuw/connutil"
	"testing"
)

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")
	}
}