// 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());

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

    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() {