75 lines
1.5 KiB
Go
75 lines
1.5 KiB
Go
|
package packer
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"net"
|
||
|
|
"testing"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestDecode1WholeBlock(t *testing.T) {
|
||
|
|
testDecodeNBlocks(t, 1)
|
||
|
|
}
|
||
|
|
|
||
|
|
func testDecodeNBlocks(t *testing.T, count int) {
|
||
|
|
wp, rp := net.Pipe()
|
||
|
|
|
||
|
|
ctx := context.Background()
|
||
|
|
decoder := NewDecoder(ctx, rp)
|
||
|
|
nAddr := 1
|
||
|
|
if count > 2 {
|
||
|
|
nAddr = count - 2
|
||
|
|
}
|
||
|
|
|
||
|
|
raw := []byte{}
|
||
|
|
for i := 0; i < count; i++ {
|
||
|
|
if i > 2 {
|
||
|
|
copied := src
|
||
|
|
src = copied
|
||
|
|
src.port += i
|
||
|
|
}
|
||
|
|
h, b, err := Encode(src, dst, domain, payload)
|
||
|
|
if nil != err {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
raw = append(raw, h...)
|
||
|
|
raw = append(raw, b...)
|
||
|
|
}
|
||
|
|
|
||
|
|
var nw int
|
||
|
|
go func() {
|
||
|
|
var err error
|
||
|
|
//fmt.Println("writers gonna write")
|
||
|
|
nw, err = wp.Write(raw)
|
||
|
|
if nil != err {
|
||
|
|
//fmt.Println("writer died")
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
// very important: don't forget to close when done!
|
||
|
|
wp.Close()
|
||
|
|
//fmt.Println("writer done wrote")
|
||
|
|
}()
|
||
|
|
|
||
|
|
th := &testHandler{
|
||
|
|
conns: map[string]*Conn{},
|
||
|
|
}
|
||
|
|
//fmt.Println("streamers gonna stream")
|
||
|
|
err := decoder.StreamDecode(th, 0)
|
||
|
|
if nil != err {
|
||
|
|
t.Fatalf("failed to decode stream: %s", err)
|
||
|
|
}
|
||
|
|
//fmt.Println("streamer done streamed")
|
||
|
|
|
||
|
|
if nAddr != len(th.conns) {
|
||
|
|
t.Fatalf("should have parsed %d connection(s)", nAddr)
|
||
|
|
}
|
||
|
|
if count != th.chunksParsed {
|
||
|
|
t.Fatalf("should have parsed %d chunk(s)", count)
|
||
|
|
}
|
||
|
|
if count*len(payload) != th.bytesRead {
|
||
|
|
t.Fatalf("should have parsed a payload of %d bytes, but saw %d\n", count*len(payload), th.bytesRead)
|
||
|
|
}
|
||
|
|
if nw != len(raw) {
|
||
|
|
t.Fatalf("should have parsed all %d bytes, not just %d\n", len(raw), nw)
|
||
|
|
}
|
||
|
|
}
|