From 6e39fe9b035534c52c5f53f38ba97679cb16a364 Mon Sep 17 00:00:00 2001
From: xtaci <daniel820313@gmail.com>
Date: Tue, 24 Dec 2019 17:02:15 +0800
Subject: [PATCH] add tests

---
 aio_test.go | 76 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 76 insertions(+)

diff --git a/aio_test.go b/aio_test.go
index 8bc3bb6..b592fed 100644
--- a/aio_test.go
+++ b/aio_test.go
@@ -158,6 +158,82 @@ func TestEchoHuge(t *testing.T) {
 	conn.Close()
 }
 
+func TestBufferedDone(t *testing.T) {
+	ln := echoServer(t)
+	conn, err := net.Dial("tcp", ln.Addr().String())
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	w, err := CreateWatcher(bufSize)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	fd, err := w.Watch(conn)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	err = w.Read(fd, make(chan OpResult, 1))
+	if err != ErrBufferedChan {
+		t.Fatal("misbehavior")
+	}
+	conn.Close()
+}
+
+func TestBidirectionWatcher(t *testing.T) {
+	ln := echoServer(t)
+	conn, err := net.Dial("tcp", ln.Addr().String())
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	w, err := CreateWatcher(bufSize)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	fd, err := w.Watch(conn)
+	if err != nil {
+		t.Fatal(err)
+	}
+
+	tx := []byte("hello world")
+	doneR := make(chan OpResult)
+	doneW := make(chan OpResult)
+	die := make(chan struct{})
+	go func() {
+		for {
+			select {
+			case res := <-doneW:
+				// recv
+				if res.Err != nil {
+					t.Fatal(res.Err)
+				}
+
+				t.Log("written:", res.Err, res.Size)
+				err = w.Read(fd, doneR)
+				if err != nil {
+					t.Fatal(err)
+				}
+			case res := <-doneR:
+				t.Log("read:", res.Err, res.Size)
+				close(die)
+				return
+			}
+		}
+	}()
+
+	// send
+	err = w.Write(fd, tx, doneW)
+	if err != nil {
+		t.Fatal(err)
+	}
+	<-die
+	conn.Close()
+}
+
 func BenchmarkEcho(b *testing.B) {
 	ln := echoServer(b)