diff --git a/e2e/e2e_test.go b/e2e/e2e_test.go index 732b417..57d914a 100644 --- a/e2e/e2e_test.go +++ b/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) diff --git a/session/session.go b/session/session.go index e75473a..3ee07c0 100644 --- a/session/session.go +++ b/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() }