* @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

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

        TestNotificationListener listener = new TestNotificationListener();

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

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

        assertNotNull(registration);

            return;
        }

        try {
            InetAddress serverAddress = transport.getRemoteAddress().toInetAddress();

            witnessClient.registerForNotifications(shareName, serverAddress, new WitnessNotificationHandler()).thenAccept(registration -> {
                log.info("Registered for witness notifications: share={}, id={}", shareName, registration.getRegistrationId());