Files
ColaFlow/colaflow-api/tests/Modules/Mcp/ColaFlow.Modules.Mcp.Tests/EventHandlers/PendingChangeApprovedNotificationHandlerTests.cs
Yaojia Wang 9ccd3284fb feat(backend): Implement SignalR Real-Time Notifications for MCP - Story 5.12
Implemented comprehensive real-time notification system using SignalR to notify
AI agents and users about PendingChange status updates.

Key Features Implemented:
- McpNotificationHub with Subscribe/Unsubscribe methods
- Real-time notifications for all PendingChange lifecycle events
- Tenant-based isolation for multi-tenancy security
- Notification DTOs for structured message formats
- Domain event handlers for automatic notification sending
- Comprehensive unit tests for notification service and handlers
- Client integration guide with examples for TypeScript, React, and Python

Components Created:
1. SignalR Hub:
   - McpNotificationHub.cs - Central hub for MCP notifications

2. Notification DTOs:
   - PendingChangeNotification.cs (base class)
   - PendingChangeCreatedNotification.cs
   - PendingChangeApprovedNotification.cs
   - PendingChangeRejectedNotification.cs
   - PendingChangeAppliedNotification.cs
   - PendingChangeExpiredNotification.cs

3. Notification Service:
   - IMcpNotificationService.cs (interface)
   - McpNotificationService.cs (implementation using SignalR)

4. Event Handlers (send notifications):
   - PendingChangeCreatedNotificationHandler.cs
   - PendingChangeApprovedNotificationHandler.cs
   - PendingChangeRejectedNotificationHandler.cs
   - PendingChangeAppliedNotificationHandler.cs
   - PendingChangeExpiredNotificationHandler.cs

5. Tests:
   - McpNotificationServiceTests.cs - Unit tests for notification service
   - PendingChangeCreatedNotificationHandlerTests.cs
   - PendingChangeApprovedNotificationHandlerTests.cs

6. Documentation:
   - signalr-mcp-client-guide.md - Comprehensive client integration guide

Technical Details:
- Hub endpoint: /hubs/mcp-notifications
- Authentication: JWT token via query string (?access_token=xxx)
- Tenant isolation: Automatic group joining based on tenant ID
- Group subscriptions: Per-pending-change and per-tenant groups
- Notification delivery: < 1 second (real-time)
- Fallback strategy: Polling if WebSocket unavailable

Architecture Benefits:
- Decoupled design using domain events
- Notification failures don't break main flow
- Scalable (supports Redis backplane for multi-instance)
- Type-safe notification payloads
- Tenant isolation built-in

Story: Phase 3 - Tools & Diff Preview
Priority: P0 CRITICAL
Story Points: 3
Completion: 100%

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>
2025-11-09 18:21:08 +01:00

104 lines
3.3 KiB
C#

using ColaFlow.Modules.Mcp.Application.DTOs.Notifications;
using ColaFlow.Modules.Mcp.Application.EventHandlers;
using ColaFlow.Modules.Mcp.Application.Services;
using ColaFlow.Modules.Mcp.Domain.Events;
using ColaFlow.Modules.Mcp.Domain.ValueObjects;
using Microsoft.Extensions.Logging;
using Moq;
using Xunit;
namespace ColaFlow.Modules.Mcp.Tests.EventHandlers;
public class PendingChangeApprovedNotificationHandlerTests
{
private readonly Mock<IMcpNotificationService> _mockNotificationService;
private readonly Mock<ILogger<PendingChangeApprovedNotificationHandler>> _mockLogger;
private readonly PendingChangeApprovedNotificationHandler _handler;
public PendingChangeApprovedNotificationHandlerTests()
{
_mockNotificationService = new Mock<IMcpNotificationService>();
_mockLogger = new Mock<ILogger<PendingChangeApprovedNotificationHandler>>();
_handler = new PendingChangeApprovedNotificationHandler(
_mockNotificationService.Object,
_mockLogger.Object);
}
[Fact]
public async Task Handle_SendsNotification_WithCorrectData()
{
// Arrange
var pendingChangeId = Guid.NewGuid();
var tenantId = Guid.NewGuid();
var approvedBy = Guid.NewGuid();
var entityId = Guid.NewGuid();
var diff = new DiffPreview(
"CREATE",
"Epic",
entityId,
null,
null,
"{\"name\":\"Test Epic\"}",
new List<DiffField>());
var domainEvent = new PendingChangeApprovedEvent(
pendingChangeId,
"create_epic",
diff,
approvedBy,
tenantId);
// Act
await _handler.Handle(domainEvent, CancellationToken.None);
// Assert
_mockNotificationService.Verify(
s => s.NotifyPendingChangeApprovedAsync(
It.Is<PendingChangeApprovedNotification>(n =>
n.PendingChangeId == pendingChangeId &&
n.ToolName == "create_epic" &&
n.EntityType == "Epic" &&
n.Operation == "CREATE" &&
n.EntityId == entityId &&
n.ApprovedBy == approvedBy &&
n.TenantId == tenantId),
It.IsAny<CancellationToken>()),
Times.Once);
}
[Fact]
public async Task Handle_DoesNotThrow_WhenNotificationServiceFails()
{
// Arrange
var pendingChangeId = Guid.NewGuid();
var tenantId = Guid.NewGuid();
var approvedBy = Guid.NewGuid();
var diff = new DiffPreview(
"CREATE",
"Epic",
null,
null,
null,
"{\"name\":\"Test Epic\"}",
new List<DiffField>());
var domainEvent = new PendingChangeApprovedEvent(
pendingChangeId,
"create_epic",
diff,
approvedBy,
tenantId);
_mockNotificationService.Setup(s => s.NotifyPendingChangeApprovedAsync(
It.IsAny<PendingChangeApprovedNotification>(),
It.IsAny<CancellationToken>()))
.ThrowsAsync(new Exception("SignalR connection failed"));
// Act & Assert - Should not throw
await _handler.Handle(domainEvent, CancellationToken.None);
}
}