// Arrange
        when(smbTransport.getRemoteAddress()).thenReturn(remoteAddress);

        // Act
        Address result = smbTransport.getRemoteAddress();

        // Assert
        assertNotNull(result, "Remote address should not be null");
        assertEquals(remoteAddress, result, "Returned remote address should be the mocked address");
        verify(smbTransport).getRemoteAddress(); // Verify that the method was called
    }

            }

            try {
                rdmaConnection =
                        rdmaProvider.connect(delegate.getRemoteAddress().getHostAddress(), delegate.getContext().getConfig().getRdmaPort());
                log.info("RDMA connection established to {}", delegate.getRemoteAddress());
            } catch (IOException e) {
                log.warn("Failed to establish RDMA connection: {}", e.getMessage());

            final boolean doSigning, Subject s) throws SmbException {

        String host = getTargetHost();
        if (host == null) {
            host = trans.getRemoteAddress().getHostAddress();
            try {
                host = trans.getRemoteAddress().getHostName();
            } catch (Exception e) {
                log.debug("Failed to resolve host name", e);
            }
        }

     */
    @Override
    void close();

    /**
     * Gets the address of the remote SMB server.
     *
     * @return the connected address
     */
    Address getRemoteAddress();

    /**
     * Gets the host name of the remote SMB server.
     *
     * @return the connected host name
     */
    String getRemoteHostName();

        RdmaConnection connection = provider.createConnection(remote, local);
        assertNotNull(connection, "Connection should not be null");
        assertEquals(remote, connection.getRemoteAddress(), "Remote address should match");
    }

    @Test
    public void testRegisterMemory() throws Exception {
        ByteBuffer buffer = ByteBuffer.allocateDirect(1024);

    }

    @Test
    @DisplayName("Basic getters: address, hostName, inflight, sessions")
    void basicGetters() {
        assertEquals(address, transport.getRemoteAddress());
        assertNull(transport.getRemoteHostName(), "tconHostName starts null");
        assertEquals(0, transport.getInflightRequests());
        assertEquals(0, transport.getNumSessions());
    }

    @Test

        try (RdmaConnection connection = provider.createConnection(address, null)) {
            assertNotNull(connection, "Connection should not be null");
            assertEquals(address, connection.getRemoteAddress(), "Remote address should match");
            assertNotNull(connection.getState(), "Connection state should not be null");

            System.out.println("RDMA connection created to: " + address);

    public RdmaConnectionState getState() {
        return state;
    }

    /**
     * Get remote address
     *
     * @return remote socket address
     */
    public InetSocketAddress getRemoteAddress() {
        return remoteAddress;
    }

    /**
     * Get local address
     *
     * @return local socket address
     */
    public InetSocketAddress getLocalAddress() {

                SmbSessionImpl session = th.getSession();
                SmbTransportImpl transport = session.getTransport()) {
            try {
                entries = doMsrpcShareEnum(tc, locator, transport.getRemoteAddress());
            } catch (final IOException ioe) {
                if (th.isSMB2()) {
                    throw ioe;
                }
                log.debug("doMsrpcShareEnum failed", ioe);

        log.warn("Failed to initialize witness support", e);
    }
}

private InetAddress discoverWitnessService() throws IOException {
    // Try the same server first
    InetAddress serverAddress = transport.getRemoteAddress();
    
    if (isWitnessServiceAvailable(serverAddress)) {
        return serverAddress;
    }
    
    // Query for cluster witness service via DNS
    try {