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")

            // Then
            assertArrayEquals(testTargetInfo, message.getTargetInformation());
        }

        @Test
        @DisplayName("setContext and getContext should work correctly")
        void testGetSetContext() {
            // Given
            CIFSContext mockContext = createMockContext();
            Type2Message message = new Type2Message(mockContext);

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

     * authentication.
     */
    public byte[] getContext() {
        return context;
    }

    /**
     * Sets the local security context.  This is used by the client
     * to negotiate local authentication.
     *
     * @param context The local security context.
     */
    public void setContext(final byte[] context) {
        this.context = context;
    }

    @Override

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
            SmbFile src = mock(SmbFile.class, RETURNS_DEEP_STUBS);
            SmbFile dest = mock(SmbFile.class);
            CIFSContext ctx = ctxWithIgnore(false);
            when(src.getContext()).thenReturn(ctx);

            // Force failure before any stream construction
            when(src.openUnshared(anyInt(), anyInt(), anyInt(), anyInt(), anyInt())).thenThrow(new SmbException("boom"));

        LocalMetadataResult result = new LocalMetadataResult(request);

        String path;

        Metadata metadata = request.getMetadata();
        String context = request.getContext();
        RemoteRepository remote = request.getRepository();

        if (remote != null) {
            path = getPathForRemoteMetadata(metadata, remote, context);
        } else {

        assertSame(dg, transport.getDigest());
    }

    @Test
    @DisplayName("Context accessor returns constructor-provided context")
    void contextAccessor() {
        assertSame(ctx, transport.getContext());
    }

    @Test
    @DisplayName("acquire returns same instance")
    void acquireReturnsSameInstance() {
        SmbTransportImpl acquired = transport.acquire();
        assertSame(transport, acquired);