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

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

            if (message.isSuccess()) {
                log.debug("Witness unregistration successful");
            } else {

        lenient().when(registerResponse.getRegistrationId()).thenReturn("test-reg-123");

        // Setup successful unregistration
        WitnessUnregisterResponse unregisterResponse = mock(WitnessUnregisterResponse.class);
        lenient().when(unregisterResponse.isSuccess()).thenReturn(true);

        WitnessRpcClient mockRpc = mock(WitnessRpcClient.class);

        }
        
        @Override
        public void decode_out(NdrBuffer buffer) throws NdrException {
            response = new WitnessUnregisterResponse();
            response.setReturnCode(buffer.dec_ndr_long());
        }
        
        public WitnessUnregisterResponse getResponse() { return response; }
    }
    
    private static class WitnessHeartbeatStub extends DcerpcMessage {

                WitnessUnregisterRequest request = new WitnessUnregisterRequest();
                request.setRegistrationId(registration.getRegistrationId());

                WitnessUnregisterResponse response = rpcClient.unregister(request);

                if (response != null && response.isSuccess()) {
                    // Stop async notification monitoring