add tests

lxzan · Jan 11, 2023 · 7ad9792 · 7ad9792
1 parent 5d65be6
commit 7ad9792
Show file tree

Hide file tree

Showing 3 changed files with 44 additions and 1 deletion.
diff --git a/Makefile b/Makefile
@@ -11,5 +11,5 @@ run-testsuite-server:
 	go run github.com/lxzan/gws/examples/testsuite
 
 coverage:
-	go test -v $(go list ./... |grep -v /cmd |grep -v /vendor) -coverprofile=bin/coverage.out
+	go test -mod=readonly -covermode=count -coverprofile=./bin/coverprofile.cov -run="^Test" -coverpkg=$(go list -mod=vendor ./... | grep -v "/test" | tr '\n' ',') ./...
 
diff --git a/examples/testsuite/config/readtest.json b/examples/testsuite/config/readtest.json
@@ -195,6 +195,30 @@
             "reason": "68656c6c6f"
         }
     },
+    {
+        "title": "close with invalid code",
+        "fin": true,
+        "opcode": 8,
+        "length": 0,
+        "payload": "03f6",
+        "expected": {
+            "event": "onClose",
+            "code": 1014,
+            "reason": ""
+        }
+    },
+    {
+        "title": "close with code 0",
+        "fin": true,
+        "opcode": 8,
+        "length": 0,
+        "payload": "",
+        "expected": {
+            "event": "onClose",
+            "code": 0,
+            "reason": ""
+        }
+    },
     {
         "title": "close with invalid payload",
         "fin": true,

diff --git a/websocket_test.go b/websocket_test.go
@@ -11,6 +11,7 @@ import (
 	"net/http"
 	"sync"
 	"testing"
+	"time"
 	"unsafe"
 )
 
@@ -219,3 +220,21 @@ func TestMask(t *testing.T) {
 		}
 	}
 }
+
+func TestConn(t *testing.T) {
+	conn, _ := net.Pipe()
+	socket := &Conn{
+		conn:    conn,
+		handler: new(webSocketMocker),
+		wmu:     &sync.Mutex{},
+		wbuf:    bufio.NewWriter(bytes.NewBuffer(nil)),
+	}
+	socket.SetDeadline(time.Time{})
+	socket.SetReadDeadline(time.Time{})
+	socket.SetWriteDeadline(time.Time{})
+	socket.LocalAddr()
+	socket.RemoteAddr()
+	socket.Close(1000, []byte("goodbye"))
+	socket.Listen()
+	return
+}