/**
     * Check if negotiation was successful
     *
     * @return true if successful, false otherwise
     */
    public boolean isSuccess() {
        return status == 0;
    }

    /**
     * Get negotiation status
     *
     * @return status code (0 = success)
     */
    public int getStatus() {
        return status;
    }

    /**
     * Set negotiation status
     *

    }

    private int returnCode;
    private String error;

    /**
     * Checks if the unregistration was successful.
     *
     * @return true if successful
     */
    public boolean isSuccess() {
        return returnCode == 0;
    }

    /**
     * Gets a human-readable error description.
     *
     * @return the error description
     */
    public String getError() {

    private long recommendedHeartbeatInterval;
    private String error;

    /**
     * Checks if the heartbeat was successful.
     *
     * @return true if successful
     */
    public boolean isSuccess() {
        return returnCode == 0;
    }

    /**
     * Gets the sequence number.
     *
     * @return the sequence number
     */
    public long getSequenceNumber() {
        return sequenceNumber;

    private byte[] contextHandle;
    private String error;

    /**
     * Checks if the registration was successful.
     *
     * @return true if successful
     */
    public boolean isSuccess() {
        return returnCode == 0;
    }

    /**
     * Gets a human-readable error description.
     *
     * @return the error description
     */
    public String getError() {

                request.setFlags(registration.getFlags());
                
                WitnessRegisterResponse response = rpcClient.register(request);
                
                if (response.isSuccess()) {
                    registration.setState(WitnessRegistrationState.REGISTERED);
                    registrations.put(registration.getRegistrationId(), registration);

     * The response sequence number should match the request sequence number.
     *
     * @return true if the response is valid
     */
    public boolean isValidResponse() {
        return isSuccess() && responseSequenceNumber == sequenceNumber;
    }

    @Override
    protected void encodeWitnessParameters(NdrBuffer buf) throws NdrException {
        // Encode input parameters for WitnessHeartbeat

        return data;
    }

    /**
     * Check if negotiation was successful
     *
     * @return true if successful
     */
    public boolean isSuccess() {
        return status == STATUS_SUCCESS;
    }

    // Getters and setters

    /**
     * Get the minimum SMB Direct protocol version
     *
     * @return minimum protocol version
     */

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

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

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

            if (message.isSuccess()) {
                // Generate registration ID from context handle
                byte[] contextHandle = message.getContextHandle();

        if (connectResult == null) {
          connectResult = awaitTcpConnect(awaitTimeoutNanos, TimeUnit.NANOSECONDS) ?: continue
        }

        if (connectResult.isSuccess) {
          // We have a connected TCP connection. Cancel and defer the racing connects that all lost.
          cancelInFlightConnects()