func run(t *testing.T, size byte, endianess binary.ByteOrder) {
message := "13 bytes long"
l, err := net.Listen("tcp", ":0")
if err != nil {
t.Fatal(err)
}
defer l.Close()
port := l.Addr().(*net.TCPAddr).Port
go func() {
conn, err := net.Dial("tcp", ":"+strconv.Itoa(port))
if err != nil {
t.Fatal(err)
}
framed, err := framing.NewConn(conn, size, endianess)
if err != nil {
t.Fatal(err)
}
defer framed.Close()
for i := 1; i <= 2; i++ {
if _, err := fmt.Fprintf(framed, message); err != nil {
t.Fatal(err)
}
}
}()
conn, err := l.Accept()
if err != nil {
t.Fatal(err)
}
framed, err := framing.NewConn(conn, size, endianess)
if err != nil {
t.Fatal(err)
}
defer framed.Close()
buf, err := framed.ReadFrame()
if err != nil {
t.Fatal(err)
}
if msg := string(buf[:]); msg != message {
t.Fatalf("Unexpected message:\nGot:\t\t%s\nExpected:\t%s\n", msg, message)
}
fixed := make([]byte, 20) // More than 13
n, err := framed.Read(fixed)
if err != nil {
t.Fatal(err)
}
if n != 13 {
t.Fatal("Frame is not of correct size")
}
if msg := string(fixed[:13]); msg != message {
t.Fatalf("Unexpected message:\nGot:\t\t%s\nExpected:\t%s\n", msg, message)
}
}
func TestWriteError(t *testing.T) {
conn := &fake{
ReadWriter: bytes.NewBuffer([]byte{}),
err: errors.New("WriteError"),
}
framed, _ := framing.NewConn(conn, 1, nil)
_, err := framed.Write([]byte{0})
if err.Error() != "WriteError" {
t.Fatal(err)
}
}
func TestInnerReadFrameError(t *testing.T) {
conn := &fake{
ReadWriter: bytes.NewBuffer([]byte{4}),
err: errors.New("InnerReadFrameError"),
}
framed, _ := framing.NewConn(conn, 1, nil)
_, err := framed.ReadFrame()
if err.Error() != "InnerReadFrameError" {
t.Fatal(err)
}
}
func packetTooLarge(t *testing.T, size byte, max uint) {
conn, err := framing.NewConn(nil, size, nil)
if err != nil {
t.Fatal(err)
}
b := make([]byte, max+1)
_, err = conn.Write(b)
if err != framing.ErrFrameTooLarge {
t.Fail()
}
}
func TestMessageSmallerThanBuffer(t *testing.T) {
conn := &fake{ReadWriter: bytes.NewBuffer([]byte{4, 0, 1, 2, 3})}
framed, _ := framing.NewConn(&fake{ReadWriter: conn}, 1, nil)
first_b := make([]byte, 6)
first_n, first_err := framed.Read(first_b)
if first_n != 4 || first_err != nil {
t.Fatalf("first_n = %d, first_err = %#v", first_n, first_err)
}
if !bytes.Equal(first_b, []byte{0, 1, 2, 3, 0, 0}) {
t.Fatal(first_b)
}
}
func TestMessageBiggerThanBuffer(t *testing.T) {
conn := &fake{ReadWriter: bytes.NewBuffer([]byte{
6, 0, 1, 2, 3, 4, 5,
// [message 1 ]
// ^ first read ends
4, 7, 8, 9, 10,
// [message 2 ]
// ^ second read ends
// ^ third read ends
})}
framed, _ := framing.NewConn(&fake{ReadWriter: conn}, 1, nil)
first_b := make([]byte, 4)
first_n, first_err := framed.Read(first_b)
if first_n != 4 || first_err != nil {
t.Fatal(first_n, first_err)
}
if !bytes.Equal(first_b, []byte{0, 1, 2, 3}) {
t.Fatal(first_b)
}
second_b := make([]byte, 3)
second_n, second_err := framed.Read(second_b)
if second_n != 2 || second_err != nil {
t.Fatal(second_n, second_err)
}
if !bytes.Equal(second_b, []byte{4, 5, 0}) {
t.Fatal(second_b)
}
third_b := make([]byte, 5)
third_n, third_err := framed.Read(third_b)
if third_n != 4 || third_err != nil {
t.Fatal(third_n, third_err)
}
if !bytes.Equal(third_b, []byte{7, 8, 9, 10, 0}) {
t.Fatal(third_b)
}
fourth_b := make([]byte, 1)
fourth_n, fourth_err := framed.Read(fourth_b)
if fourth_n != 0 || fourth_err != io.EOF {
t.Fatal(fourth_n, fourth_err)
}
}
func TestInvalidFrameSiz(t *testing.T) {
_, err := framing.NewConn(nil, 3, nil)
if err != framing.ErrPrefixLength {
t.Fail()
}
}