@Test
public void testWitnessNotification() {
    WitnessNotification notification = new WitnessNotification(
        WitnessEventType.CLIENT_MOVE, "TestResource");
    
    notification.addNewIPAddress(InetAddress.getByName("192.168.1.101"));
    
    assertEquals(WitnessEventType.CLIENT_MOVE, notification.getEventType());
    assertEquals("TestResource", notification.getResourceName());

        String regId = mockService.registerWitness("TestResource", "192.168.1.100", 1);

        // Send notification affecting the registration
        mockService.sendNotification(WitnessEventType.CLIENT_MOVE, "TestResource");

        // Clean up
        mockService.unregisterWitness(regId);
    }

    @Test
    void testMockServiceHeartbeat() {

        // Create and process a notification
        WitnessNotification notification = new WitnessNotification(WitnessEventType.CLIENT_MOVE, "\\\\server\\share");
        notification.addNewIPAddress(InetAddress.getByName("192.168.1.101"));

        client.processNotification(notification);

        // Verify notification was delivered

            return WitnessEventType.RESOURCE_CHANGE;
        case WitnessAsyncNotifyMessage.WitnessNotificationMessage.WITNESS_CLIENT_MOVE:
            return WitnessEventType.CLIENT_MOVE;
        case WitnessAsyncNotifyMessage.WitnessNotificationMessage.WITNESS_SHARE_MOVE:
            return WitnessEventType.SHARE_MOVE;
        case WitnessAsyncNotifyMessage.WitnessNotificationMessage.WITNESS_IP_CHANGE: