}

    @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);