/**
 * Represents a witness registration response as defined in MS-SWN specification.
 * Contains the result of a witness registration request.
 */
public class WitnessRegisterResponse {
    /**
     * Creates a new witness register response.
     */
    public WitnessRegisterResponse() {
        // Default constructor
    }

    private String registrationId;
    private int returnCode;
    private byte[] contextHandle;

    }

    @Test
    void testSuccessfulRegistration() throws Exception {
        // Setup mock RPC response
        WitnessRegisterResponse mockResponse = mock(WitnessRegisterResponse.class);
        lenient().when(mockResponse.isSuccess()).thenReturn(true);
        lenient().when(mockResponse.getRegistrationId()).thenReturn("test-reg-123");

            // Send the RPC request
            rpcHandle.sendrecv(message);

            // Create response from RPC message results
            WitnessRegisterResponse response = new WitnessRegisterResponse();
            response.setReturnCode(message.getReturnCode());

            if (message.isSuccess()) {
                // Generate registration ID from context handle

            // Decode WitnessRegister response
            response = new WitnessRegisterResponse();
            response.setRegistrationId(buffer.dec_ndr_string());
            response.setReturnCode(buffer.dec_ndr_long());
        }
        
        public WitnessRegisterResponse getResponse() { return response; }
    }
    
    private static class WitnessUnregisterStub extends DcerpcMessage {

                request.setShareName(shareName);
                request.setServerAddress(serverAddress.getHostAddress());
                request.setFlags(registration.getFlags());

                WitnessRegisterResponse response = rpcClient.register(request);

                if (response != null && response.isSuccess()) {
                    registration.setState(WitnessRegistrationState.REGISTERED);