Test: wscontroller

2020-03-27 16:01:04 +08:00
parent b79374c9cc
commit 82c62402e7
1 changed files with 8 additions and 6 deletions
@@ -3,7 +3,6 @@ package controller
 import (
 	"github.com/c-my/MahjongServer/container"
 	"github.com/gorilla/websocket"
-	"log"
 	"net/url"
 	"testing"
 )
@@ -11,11 +10,14 @@ import (
 func TestWsHandler(t *testing.T) {
 	container.GetHall().CreateRoom(100, "100r", "100")
 	u := url.URL{Scheme: "ws", Host: "127.0.0.1:1114", Path: "/100"}
-	c, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
-	if err != nil {
-		log.Fatal("dial:", err)
+	_, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
+	if err == nil {
+		t.Errorf("connected via wrong userid")
+	}
+	u = url.URL{Scheme: "ws", Host: "127.0.0.1:1114", Path: "/some_string"}
+	_, _, err = websocket.DefaultDialer.Dial(u.String(), nil)
+	if err == nil {
+		t.Errorf("connected via non-number param")
 	}
-	//time.Sleep(1)
-	defer c.Close()

 }