github.com/glide-im/glide@v1.6.0/pkg/conn/ws_server_test.go (about)

     1  package conn
     2  
     3  import (
     4  	"github.com/gorilla/websocket"
     5  	"testing"
     6  	"time"
     7  )
     8  
     9  func TestConnect(t *testing.T) {
    10  
    11  	dialer := websocket.Dialer{
    12  		ReadBufferSize:  1024,
    13  		WriteBufferSize: 1024,
    14  	}
    15  
    16  	conn, _, err := dialer.Dial("ws://127.0.0.1:8080/ws", nil)
    17  	if err != nil {
    18  		t.Error(err)
    19  	}
    20  	go func() {
    21  		for {
    22  			messageType, p, e := conn.ReadMessage()
    23  			if e != nil {
    24  				break
    25  			}
    26  			t.Log(messageType, string(p))
    27  		}
    28  
    29  	}()
    30  
    31  	order := 0
    32  	for true {
    33  		time.Sleep(time.Second * 3)
    34  		msg := struct {
    35  			Order int
    36  			Msg   string
    37  		}{Order: order, Msg: "msg here"}
    38  		order++
    39  		e := conn.WriteJSON(msg)
    40  		if e != nil {
    41  			t.Error(e)
    42  		}
    43  	}
    44  
    45  	time.Sleep(time.Hour)
    46  }