}
    }

    /**
     * Performs witness unregistration.
     *
     * @param request the unregistration request
     * @return the unregistration response
     * @throws IOException if the RPC call fails
     */
    public WitnessUnregisterResponse unregister(WitnessUnregisterRequest request) throws IOException {
        if (!connected) {

        // 4. Cleanup and unregistration
        assertTrue(mockService.unregisterWitness(regId));
        assertEquals(0, mockService.getRegistrationCount());
        assertFalse(mockService.hasRegistration(regId));

        // 5. Verify heartbeat fails after unregistration
        assertFalse(mockService.processHeartbeat(regId, 4));
    }

    /**

 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */
package jcifs.internal.witness;

/**
 * Represents a witness unregistration response as defined in MS-SWN specification.
 * Contains the result of a witness unregistration request.
 */
public class WitnessUnregisterResponse {
    /**
     * Creates a new witness unregister response.
     */
    public WitnessUnregisterResponse() {

    private byte[] contextHandle;

    /**
     * Gets the registration ID to unregister.
     *
     * @return the registration ID
     */
    public String getRegistrationId() {
        return registrationId;
    }

    /**
     * Sets the registration ID to unregister.
     *
     * @param registrationId the registration ID
     */
    public void setRegistrationId(String registrationId) {

     * @param contextHandle the context handle from registration
     */
    public WitnessUnregisterMessage(byte[] contextHandle) {
        super(WITNESS_UNREGISTER);
        this.contextHandle = contextHandle != null ? contextHandle.clone() : new byte[20];
    }

    /**
     * Sets the context handle for unregistration.
     *
     * @param contextHandle the context handle from registration
     */

    }

    @Test
    @DisplayName("Test resource unregistration")
    void testResourceUnregistration() {
        TestResource resource = new TestResource("test5");
        String resourceId = resourceManager.registerResource(resource);

        resourceManager.unregisterResource(resourceId);

        // After unregistration, the resource should be removed from active tracking

        /**
         * Called when a witness registration fails.
         *
         * @param registration the failed registration
         * @param error the error that occurred
         */
        void onRegistrationFailed(WitnessRegistration registration, Exception error);

        /**
         * Called when a witness registration expires.
         *
         * @param registration the expired registration
         */

     * Enumeration of possible witness registration states.
     */
    public enum WitnessRegistrationState {
        /** Registration is in progress */
        REGISTERING,
        /** Registration is active and confirmed */
        REGISTERED,
        /** Unregistration is in progress */
        UNREGISTERING,
        /** Registration has failed */
        FAILED,
        /** Registration has expired due to timeout */
        EXPIRED

     * @return the registration count
     */
    public int getRegistrationCount() {
        return registrations.size();
    }

    /**
     * Check if a registration exists
     *
     * @param registrationId the registration ID
     * @return true if registration exists
     */
    public boolean hasRegistration(String registrationId) {
        return registrations.containsKey(registrationId);
    }

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

        assertNotNull(registration);
        assertEquals(WitnessRegistrationState.REGISTERED, registration.getState());
        assertEquals("\\\\server\\share", registration.getShareName());
        assertEquals(serverAddress, registration.getServerAddress());