test send byte data

Author: szpnygo

e2e/e2e_test.go

@@ -83,6 +83,15 @@ func TestSendConnectWS(t *testing.T) {
 	sessions := session.ListSessions()
 	assert.Equal(t, 1, len(sessions))
 
+	//test send data
+	sessions[0].SendData([]byte("hello world"))
+	select {
+	case <-time.After(1 * time.Second):
+		t.Fatal("server get the data timeout")
+	case msg := <-testClient.datas:
+		assert.Equal(t, "hello world", string(msg))
+	}
+
 	uid = userLogin(t, "e2e")
 	assert.NotEmpty(t, uid)
 

session/session.go

@@ -53,6 +53,10 @@ func (sess *Session) SendMessage(in interface{}, name ...string) error {
 	return sess.agent.Send(in, name...)
 }
 
+func (sess *Session) SendData(data []byte) {
+	sess.agent.SendData(data)
+}
+
 func (sess *Session) GetLastTimeOnline() int64 {
 	return sess.agent.GetLastTimeOnline()
 }