<T extends SmbSession> T unwrap(Class<T> type);

    /**
     * Returns the CIFS context that this session is attached to.
     *
     * @return the context this session is attached to
     */
    CIFSContext getContext();

public interface SmbTransport extends AutoCloseable {

    /**
     * Gets the CIFS context associated with this transport.
     *
     * @return the context this transport is attached to
     */
    CIFSContext getContext();

    /**
     * Unwraps the transport to the specified type.
     *
     * @param <T> the type to unwrap to
     * @param type the class of the transport type to unwrap

        // Arrange
        when(smbTransport.getContext()).thenReturn(cifsContext);

        // Act
        CIFSContext result = smbTransport.getContext();

        // Assert
        assertNotNull(result, "Context should not be null");
        assertEquals(cifsContext, result, "Returned context should be the mocked context");
        verify(smbTransport).getContext(); // Verify that the method was called
    }

    @Test

      const = 0;
  virtual absl::Status SetInput(size_t index,
                                ImmediateExecutionTensorHandle* input) = 0;

  virtual ImmediateExecutionContext* GetContext() const = 0;

  // Following two methods are used to support custom device.
  // Return true if the inputs contain custom device tensor handle. It means
  // that the argument need to be handled by a custom device.

    }

    @Override
    public int getInflightRequests() {
        return delegate.getInflightRequests();
    }

    @Override
    public CIFSContext getContext() {
        return delegate.getContext();
    }

    @Override
    @SuppressWarnings("unchecked")
    public <T extends SmbTransport> T unwrap(Class<T> type) {
        if (type.isAssignableFrom(this.getClass())) {

     */
    public Category getCategory() {
        return category;
    }

    /**
     * Gets the error context
     *
     * @return the context map
     */
    public Map<String, Object> getContext() {
        return new HashMap<>(context);
    }

    /**
     * Gets a specific context value
     *
     * @param key the context key
     * @return the context value or null
     */

        this.hashesExternal = true;
    }

    /**
     * Get the CIFS context associated with this authentication.
     *
     * @return the CIFS context
     */
    protected CIFSContext getContext() {
        return this.context;
    }

    @Override
    public NtlmPasswordAuthentication clone() {
        final NtlmPasswordAuthentication cloned = new NtlmPasswordAuthentication();

    private SmbTransportImpl transport;

    @BeforeEach
    void setUp() throws Exception {
        MockitoAnnotations.openMocks(this);
        when(session.getConfig()).thenReturn(config);
        when(session.getContext()).thenReturn(context);
        when(session.getTransport()).thenReturn(transport);
        when(session.acquire()).thenReturn(session);
        when(config.isTraceResourceUsage()).thenReturn(false);

            // Check if we can use directory leasing
            DirectoryLeaseManager dirManager = getDirectoryLeaseManager(th);
            if (dirManager != null && smbFile.getContext().getConfig().isUseDirectoryLeasing()) {
                return listFilesWithDirectoryLeasing(smbFile, dirManager);
            } else {
                // Fall back to regular directory listing

        // Assert
        assertTrue(conn instanceof SmbFile);
        SmbFile file = (SmbFile) conn;
        assertNotNull(file.getContext(), "Context must be non-null");
        assertSame(SingletonContext.getInstance(), file.getContext(), "Should use SingletonContext when none provided");
    }

    @Test
    @DisplayName("parseURL: exact 'smb://' sets default port")