From d7cdf02cb7a56cb86cef23146aa387d16e567555 Mon Sep 17 00:00:00 2001
From: giongto35 <giongto35@yahoo.com>
Date: Sun, 12 May 2019 17:47:39 +0800
Subject: [PATCH] Update test rejoin with overlord

---
 cmd/main_test.go | 47 ++++++++++++++++++++++++++++++++++++++++++++++-
 1 file changed, 46 insertions(+), 1 deletion(-)

diff --git a/cmd/main_test.go b/cmd/main_test.go
index 4b12f905..1d86face 100644
--- a/cmd/main_test.go
+++ b/cmd/main_test.go
@@ -459,7 +459,7 @@ func TestReconnectRoomWithOverlord(t *testing.T) {
 func TestRejoinNoOverlordMultiple(t *testing.T) {
 	/*
 		Case scenario:
-		- A server X is initialized connecting to overlord
+		- A server X without connecting to overlord
 		- Client A keeps creating a new room
 		Expected behavior:
 		- The game should running normally
@@ -495,3 +495,48 @@ func TestRejoinNoOverlordMultiple(t *testing.T) {
 	fmt.Println("Done")
 
 }
+
+func TestRejoinWithOverlordMultiple(t *testing.T) {
+	/*
+		Case scenario:
+		- A server X is initialized connecting to overlord
+		- Client A keeps creating a new room
+		Expected behavior:
+		- The game should running normally
+	*/
+
+	// Init slave server
+	o := initOverlord()
+	defer o.Close()
+
+	oconn := connectTestOverlordServer(t, o.URL)
+	// Init slave server
+	s := initServer(t, oconn)
+	defer s.Close()
+
+	fmt.Println("Num goRoutine before start: ", runtime.NumGoroutine())
+	client := initClient(t, s.URL)
+	for i := 0; i < 100; i++ {
+		fmt.Println("Sending start...")
+		// Keep starting game
+		roomID := make(chan string)
+		client.Send(cws.WSPacket{
+			ID:          "start",
+			Data:        "Contra.nes",
+			RoomID:      "",
+			PlayerIndex: 1,
+		}, func(resp cws.WSPacket) {
+			fmt.Println("RoomID:", resp.RoomID)
+			roomID <- resp.RoomID
+		})
+
+		respRoomID := <-roomID
+		if respRoomID == "" {
+			fmt.Println("The room ID should be equal to the saved room")
+			t.Fail()
+		}
+	}
+	fmt.Println("Num goRoutine should be small: ", runtime.NumGoroutine())
+	fmt.Println("Done")
+
+}