assertTrue(mockService.hasRegistration(reg1));
        assertTrue(mockService.hasRegistration(reg2));
        assertTrue(mockService.hasRegistration(reg3));

        // Unregister one
        assertTrue(mockService.unregisterWitness(reg2));
        assertEquals(2, mockService.getRegistrationCount());
        assertFalse(mockService.hasRegistration(reg2));

        // Clean up remaining

        // Count how many registrations this affects
        int affectedRegistrations = 0;
        for (MockRegistration reg : registrations.values()) {
            if (reg.shareName.equalsIgnoreCase(resourceName) || reg.serverAddress.equals(resourceName)) {
                affectedRegistrations++;
            }
        }

        WitnessRegisterResponse mockResponse = mock(WitnessRegisterResponse.class);
        lenient().when(mockResponse.isSuccess()).thenReturn(true);
        lenient().when(mockResponse.getRegistrationId()).thenReturn("test-reg-123");

        WitnessRpcClient mockRpc = mock(WitnessRpcClient.class);
        lenient().when(mockRpc.register(any(WitnessRegisterRequest.class))).thenReturn(mockResponse);

    }

    /**
     * Generates a unique registration ID.
     *
     * @return a unique registration identifier
     */
    private String generateRegistrationId() {
        return "REG-" + System.currentTimeMillis() + "-" + Integer.toHexString(System.identityHashCode(this));
    }

    /**
     * Gets the next sequence number for this registration.
     *
     * @return the next sequence number

        this.lastHeartbeat = registrationTime;
        this.flags = WITNESS_REGISTER_IP_NOTIFICATION;
    }
    
    private String generateRegistrationId() {
        return "REG-" + System.currentTimeMillis() + "-" + 
               Integer.toHexString(System.identityHashCode(this));
    }
    
    public long getNextSequenceNumber() {
        return sequenceNumber.incrementAndGet();
    }