TestNotificationListener listener = new TestNotificationListener();

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

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

        assertNotNull(registration);

        
        // Registration monitoring
        scheduler.scheduleAtFixedRate(this::monitorRegistrations, 10, 10, TimeUnit.SECONDS);
    }
    
    public CompletableFuture<WitnessRegistration> registerForNotifications(
            String shareName, 
            InetAddress serverAddress,
            WitnessNotificationListener listener) {
        
        return CompletableFuture.supplyAsync(() -> {
            try {

     * @param serverAddress the server address
     * @param listener the notification listener
     * @return a future that completes with the registration
     */
    public CompletableFuture<WitnessRegistration> registerForNotifications(String shareName, InetAddress serverAddress,
            WitnessNotificationListener listener) {

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