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)
}