packet_test.go

package minq

import (
	"encoding/hex"
	"fmt"
	"testing"
)

var kTestpacketHeader = packetHeader{
	0,
	0x0123456789abcdef,
	0xdeadbeef,
	0xff000001,
}

// Packet header tests.
func packetHeaderEDE(t *testing.T, p *packetHeader) {
	var p2 packetHeader
	res, err := encode(p)
	assertNotError(t, err, "Could not encode")

	fmt.Println("Result = ", hex.EncodeToString(res))

	_, err = decode(&p2, res)
	assertNotError(t, err, "Could not decode")

	res2, err := encode(&p2)
	assertNotError(t, err, "Could not re-encode")
	fmt.Println("Result2 = ", hex.EncodeToString(res2))
	assertByteEquals(t, res, res2)
}

func TestLongHeader(t *testing.T) {
	p := kTestpacketHeader

	p.setLongHeaderType(packetTypeInitial)

	packetHeaderEDE(t, &p)
}

func TestShortHeader(t *testing.T) {
	p := kTestpacketHeader

	p.Type = 0x1e

	packetHeaderEDE(t, &p)
}

/*
* TODO(ekr@rtfm.com): Rewrite this code and merge it into
* connection.go
// Mock for connection state
type ConnectionStateMock struct {
	aead aeadFNV
}

func (c *ConnectionStateMock) established() bool    { return false }
func (c *ConnectionStateMock) zeroRttAllowed() bool { return false }
func (c *ConnectionStateMock) expandPacketNumber(pn uint64) uint64 {
	return pn
}

func TestEDEPacket(t *testing.T) {
	var c ConnectionStateMock

	p := Packet{
		kTestpacketHeader,
		[]byte{'a', 'b', 'c', 'd', 'e', 'f', 'g'},
	}

	encoded, err := encodePacket(&c, &c.aead, &p)
	assertNotError(t, err, "Could not encode packet")

	p2, err := decodePacket(&c, &c.aead, encoded)
	assertNotError(t, err, "Could not decode packet")

	encoded2, err := encodePacket(&c, &c.aead, p2)
	assertNotError(t, err, "Could not re-encode packet")

	assertByteEquals(t, encoded, encoded2)
}
*/