*/
    public CompletableFuture<WitnessRegistration> registerForNotifications(String shareName, InetAddress serverAddress,
            WitnessNotificationListener listener) {

        return CompletableFuture.supplyAsync(() -> {
            try {
                // Create registration
                WitnessRegistration registration =

        TestNotificationListener listener = new TestNotificationListener();

        CompletableFuture<WitnessRegistration> future = client.registerForNotifications("\\\\server\\share", serverAddress, listener);

        WitnessRegistration registration = future.get(5, TimeUnit.SECONDS);

        assertNotNull(registration);

    
    public interface WitnessNotificationListener {
        void onWitnessNotification(WitnessNotification notification);
        void onRegistrationFailed(WitnessRegistration registration, Exception error);
        void onRegistrationExpired(WitnessRegistration registration);
    }
    
    public WitnessClient(InetAddress witnessServer, CIFSContext context) {
        this.witnessServer = witnessServer;

        // 1. Service discovery and registration
        assertEquals(0, mockService.getRegistrationCount());

        String regId =
                mockService.registerWitness("\\\\cluster\\share", "192.168.1.100", WitnessRegistration.WITNESS_REGISTER_IP_NOTIFICATION);

        assertEquals(1, mockService.getRegistrationCount());
        assertTrue(mockService.hasRegistration(regId));

        // 2. Heartbeat processing