Function TestWebSocketMultiTopicSubscribe

websocket_test.go:1268–1311

(t *testing.T)

}

func TestWebSocketMultiTopicSubscribe(t *testing.T) {
	ensureServer()
	token := signUpAndGetToken(t)

	const numTopics = 50

	ws := dialWS(t, token)
	defer ws.Close()

	// create N topics
	topicNames := make([]string, numTopics)
	for i := 0; i < numTopics; i++ {
		topicNames[i] = fmt.Sprintf("multi-%d-%d", time.Now().UnixNano(), i)
		sendJSON(t, ws, wsPayload{Id: nextReqId(), Method: "create-topicName", Payload: map[string]interface{}{"name": topicNames[i]}})
		recvJSON(t, ws, 5*time.Second) // create response
	}
	time.Sleep(300 * time.Millisecond)

	// subscribe to all via comma-separated list
	allTopics := ""
	for i, name := range topicNames {
		if i > 0 {
			allTopics += ","
		}
		allTopics += name
	}
	sendJSON(t, ws, wsPayload{Id: nextReqId(), Method: "subscribe", Payload: map[string]interface{}{"topicName": allTopics}})

	// should get N subscribe responses
	ackCount := 0
	for i := 0; i < numTopics; i++ {
		msg, ok := tryRecvJSON(ws, 5*time.Second)
		if !ok {
			break
		}
		if msg.Type == "response" && msg.Ok != nil && *msg.Ok && msg.Method == "subscribe" {
			ackCount++
		}
	}
	t.Logf("received %d/%d subscribe acks", ackCount, numTopics)
	if ackCount != numTopics {
		t.Errorf("expected %d acks, got %d", numTopics, ackCount)
	}
}

func TestWebSocketRapidConnectDisconnect(t *testing.T) {
	ensureServer()