Create IncomingMsgChan only on ConnectTo

topfreegames · Jun 26, 2018 · d9f9046 · d9f9046
1 parent 9765843
commit d9f9046
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/client/client.go b/client/client.go
@@ -104,7 +104,6 @@ func New(logLevel logrus.Level, requestTimeout ...time.Duration) *Client {
 		packetEncoder:   codec.NewPomeloPacketEncoder(),
 		packetDecoder:   codec.NewPomeloPacketDecoder(),
 		packetChan:      make(chan *packet.Packet, 10),
-		IncomingMsgChan: make(chan *message.Message, 10),
 		pendingRequests: make(map[uint]*pendingRequest),
 		requestTimeout:  reqTimeout,
 		// 30 here is the limit of inflight messages

diff --git a/client/client_test.go b/client/client_test.go
@@ -37,6 +37,8 @@ func TestSendRequestShouldTimeout(t *testing.T) {
 
 	mockConn.EXPECT().Write(pkt)
 
+	c.IncomingMsgChan = make(chan *message.Message, 10)
+
 	c.nextID = 0
 	c.SendRequest(route, data)