Function TestPsyNetRPC_ConcurrentRequests

psynetrpc_test.go:192–258

(t *testing.T)

}

func TestPsyNetRPC_ConcurrentRequests(t *testing.T) {
	// Setup mock server
	mockServer := NewMockWSServer()
	defer mockServer.Close()

	// Create PsyNet instance and establish connection
	psyNet := NewPsyNet()
	rpc, err := psyNet.establishSocket(mockServer.URL(), "test-token", "test-session", "test-player")
	if err != nil {
		t.Fatalf("Failed to establish socket: %v", err)
	}

	go rpc.readMessages()
	rpc.schedulePing()
	defer rpc.Close()

	// Setup responses for multiple requests
	for i := 1; i <= 3; i++ {
		response := &PsyResponse{
			Result: json.RawMessage(fmt.Sprintf(`{"Result":{"request": %d}}`, i)),
		}
		mockServer.SetResponse(fmt.Sprintf("PsyNetMessage_X_%d", i-1), response)
	}

	// Send multiple concurrent requests
	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
	defer cancel()

	respCh1, err := rpc.sendRequestAsync(ctx, "Test/Service1", map[string]interface{}{})
	if err != nil {
		t.Fatalf("sendRequestAsync 1 failed: %v", err)
	}

	respCh2, err := rpc.sendRequestAsync(ctx, "Test/Service2", map[string]interface{}{})
	if err != nil {
		t.Fatalf("sendRequestAsync 2 failed: %v", err)
	}

	respCh3, err := rpc.sendRequestAsync(ctx, "Test/Service3", map[string]interface{}{})
	if err != nil {
		t.Fatalf("sendRequestAsync 3 failed: %v", err)
	}

	// Await all responses
	var result1, result2, result3 map[string]interface{}

	if err := rpc.awaitResponse(ctx, respCh1, &result1); err != nil {
		t.Fatalf("awaitResponse 1 failed: %v", err)
	}
	if err := rpc.awaitResponse(ctx, respCh2, &result2); err != nil {
		t.Fatalf("awaitResponse 2 failed: %v", err)
	}
	if err := rpc.awaitResponse(ctx, respCh3, &result3); err != nil {
		t.Fatalf("awaitResponse 3 failed: %v", err)
	}

	// Verify responses match expected values
	if result1["request"] != float64(1) {