func TestWriteChan(t *testing.T) {
buf := bytes.NewBuffer(nil)
reader := NewReader(buf)
wchan := NewChan(10)
msgs := [1000][]byte{}
go wchan.WriteTo(buf)
r := rand.New(rand.NewSource(time.Now().UnixNano()))
for i := range msgs {
msgs[i] = randbuf.RandBuf(r, r.Intn(1000))
select {
case err := <-wchan.ErrChan:
if err != nil {
t.Fatal("unexpected error", err)
}
case wchan.MsgChan <- msgs[i]:
}
}
// tell chan we're done.
close(wchan.MsgChan)
// wait for writing to end
<-wchan.CloseChan
defer wchan.Close()
for i := 0; ; i++ {
msg2 := make([]byte, 1000)
n, err := reader.ReadMsg(msg2)
if err != nil {
if err == io.EOF {
if i < len(msg2) {
t.Error("failed to read all messages", len(msgs), i)
}
break
}
t.Error("unexpected error", err)
}
msg1 := msgs[i]
msg2 = msg2[:n]
if !bytes.Equal(msg1, msg2) {
t.Fatal("message retrieved not equal\n", msg1, "\n\n", msg2)
}
}
if err := reader.Close(); err != nil {
t.Error(err)
}
}
func TestReadChan(t *testing.T) {
buf := bytes.NewBuffer(nil)
writer := NewWriter(buf)
p := &sync.Pool{New: func() interface{} { return make([]byte, 1000) }}
rchan := NewChan(10, p)
msgs := [1000][]byte{}
r := rand.New(rand.NewSource(time.Now().UnixNano()))
for i := range msgs {
msgs[i] = randbuf.RandBuf(r, r.Intn(1000))
err := writer.WriteMsg(msgs[i])
if err != nil {
t.Fatal(err)
}
}
if err := writer.Close(); err != nil {
t.Fatal(err)
}
go rchan.ReadFrom(buf, 1000)
defer rchan.Close()
Loop:
for i := 0; ; i++ {
select {
case err := <-rchan.ErrChan:
if err != nil {
t.Fatal("unexpected error", err)
}
case msg2, ok := <-rchan.MsgChan:
if !ok {
if i < len(msg2) {
t.Error("failed to read all messages", len(msgs), i)
}
break Loop
}
msg1 := msgs[i]
if !bytes.Equal(msg1, msg2) {
t.Fatal("message retrieved not equal\n", msg1, "\n\n", msg2)
}
}
}
}
func TestReaderWriter(t *testing.T) {
buf := bytes.NewBuffer(nil)
writer := NewWriter(buf)
reader := NewReader(buf)
msgs := [1000][]byte{}
r := rand.New(rand.NewSource(time.Now().UnixNano()))
for i := range msgs {
msgs[i] = randbuf.RandBuf(r, r.Intn(1000))
err := writer.WriteMsg(msgs[i])
if err != nil {
t.Fatal(err)
}
}
if err := writer.Close(); err != nil {
t.Fatal(err)
}
for i := 0; ; i++ {
msg2 := make([]byte, 1000)
n, err := reader.ReadMsg(msg2)
if err != nil {
if err == io.EOF {
if i < len(msg2) {
t.Error("failed to read all messages", len(msgs), i)
}
break
}
t.Error("unexpected error", err)
}
msg1 := msgs[i]
msg2 = msg2[:n]
if !bytes.Equal(msg1, msg2) {
t.Fatal("message retrieved not equal\n", msg1, "\n\n", msg2)
}
}
if err := reader.Close(); err != nil {
t.Error(err)
}
}
func SubtestReadWrite(t *testing.T, writer WriteCloser, reader ReadCloser) {
msgs := [1000][]byte{}
r := rand.New(rand.NewSource(time.Now().UnixNano()))
for i := range msgs {
msgs[i] = randbuf.RandBuf(r, r.Intn(1000))
n, err := writer.Write(msgs[i])
if err != nil {
t.Fatal(err)
}
if n != len(msgs[i]) {
t.Fatal("wrong length:", n, len(msgs[i]))
}
}
if err := writer.Close(); err != nil {
t.Fatal(err)
}
for i := 0; ; i++ {
msg2 := make([]byte, 1000)
n, err := reader.Read(msg2)
if err != nil {
if err == io.EOF {
if i < len(msg2) {
t.Error("failed to read all messages", len(msgs), i)
}
break
}
t.Error("unexpected error", err)
}
msg1 := msgs[i]
msg2 = msg2[:n]
if !bytes.Equal(msg1, msg2) {
t.Fatal("message retrieved not equal\n", msg1, "\n\n", msg2)
}
}
if err := reader.Close(); err != nil {
t.Error(err)
}
}
func SubtestReadWriteMsgSync(t *testing.T, writer WriteCloser, reader ReadCloser) {
msgs := [1000][]byte{}
r := rand.New(rand.NewSource(time.Now().UnixNano()))
for i := range msgs {
msgs[i] = randbuf.RandBuf(r, r.Intn(1000)+4)
NBO.PutUint32(msgs[i][:4], uint32(i))
}
var wg1 sync.WaitGroup
var wg2 sync.WaitGroup
errs := make(chan error, 10000)
for i := range msgs {
wg1.Add(1)
go func(i int) {
defer wg1.Done()
err := writer.WriteMsg(msgs[i])
if err != nil {
errs <- err
}
}(i)
}
wg1.Wait()
if err := writer.Close(); err != nil {
t.Fatal(err)
}
for i := 0; i < len(msgs)+1; i++ {
wg2.Add(1)
go func(i int) {
defer wg2.Done()
msg2, err := reader.ReadMsg()
if err != nil {
if err == io.EOF {
if i < len(msg2) {
errs <- fmt.Errorf("failed to read all messages", len(msgs), i)
}
return
}
errs <- fmt.Errorf("unexpected error", err)
}
mi := NBO.Uint32(msg2[:4])
msg1 := msgs[mi]
if !bytes.Equal(msg1, msg2) {
errs <- fmt.Errorf("message retrieved not equal\n", msg1, "\n\n", msg2)
}
}(i)
}
wg2.Wait()
close(errs)
if err := reader.Close(); err != nil {
t.Error(err)
}
for e := range errs {
t.Error(e)
}
}
