Function TestWebSocketMultiUserMessaging

websocket_test.go:1575–1831

(t *testing.T)

}

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

	const numUsers = 20
	const msgsPerUser = 100

	// Phase 1: Create users and get tokens concurrently
	t.Log("Phase 1: Creating users...")
	tokens := make([]string, numUsers)
	var signupWg sync.WaitGroup
	var signupErrors int64
	signupStart := time.Now()

	const signupBatch = 20
	for batchStart := 0; batchStart < numUsers; batchStart += signupBatch {
		batchEnd := batchStart + signupBatch
		if batchEnd > numUsers {
			batchEnd = numUsers
		}
		for i := batchStart; i < batchEnd; i++ {
			signupWg.Add(1)
			go func(idx int) {
				defer signupWg.Done()
				email := fmt.Sprintf("user%d@stress.test", idx)
				token, err := signUpUser(email, "tester123")
				if err != nil {
					atomic.AddInt64(&signupErrors, 1)
					return
				}
				tokens[idx] = token
			}(i)
		}
		signupWg.Wait()
	}
	signupDuration := time.Since(signupStart)

	var validUsers int
	for _, tok := range tokens {
		if tok != "" {
			validUsers++
		}
	}
	t.Logf("  created %d/%d users in %v (%d signup errors)", validUsers, numUsers, signupDuration, signupErrors)
	if validUsers < numUsers/2 {
		t.Fatalf("too few users created: %d/%d", validUsers, numUsers)
	}

	// Phase 2: Create topics — each user gets an "inbox" topic, set permissions to allow all
	t.Log("Phase 2: Creating per-user topics...")
	setupWs := dialWS(t, tokens[0])
	topicNames := make([]string, numUsers)
	for i := 0; i < numUsers; i++ {
		if tokens[i] == "" {
			continue
		}
		topicNames[i] = fmt.Sprintf("inbox-%d-%d", time.Now().UnixNano(), i)
		_ = websocket.JSON.Send(setupWs, wsPayload{Id: nextReqId(), Method: "create-topicName", Payload: map[string]interface{}{"name": topicNames[i]}})
		tryRecvJSON(setupWs, 5*time.Second) // create response