35 lines
710 B
Go
35 lines
710 B
Go
package gateway_test
|
|
|
|
import (
|
|
"fmt"
|
|
"go_dreamfactory/pb"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/gorilla/websocket"
|
|
"google.golang.org/protobuf/proto"
|
|
)
|
|
|
|
func Test_WebSocket(t *testing.T) {
|
|
url := "ws://localhost:7891/gateway" //服务器地址
|
|
ws, _, err := websocket.DefaultDialer.Dial(url, nil)
|
|
if err != nil {
|
|
fmt.Printf("err:%v", err)
|
|
return
|
|
}
|
|
loginreq := &pb.UserLoginReq{
|
|
Name: "liwei",
|
|
}
|
|
logindata, _ := proto.Marshal(loginreq)
|
|
message := &pb.Message{
|
|
Head: &pb.MessageHead{ServiceMethod: "Login"},
|
|
Data: logindata,
|
|
}
|
|
data, _ := proto.Marshal(message)
|
|
err = ws.WriteMessage(websocket.BinaryMessage, data)
|
|
if err != nil {
|
|
fmt.Printf("err:%v", err)
|
|
}
|
|
time.Sleep(time.Second * 2)
|
|
}
|