assertEquals(exception.getMessage(), deserialized.getMessage());
        assertEquals(exception.getContext().get("key1"), deserialized.getContext().get("key1"));
        assertEquals(exception.getContext().get("key2"), deserialized.getContext().get("key2"));
    }

        }

        @Test
        @DisplayName("getContext should return CIFSContext")
        void testGetContext() throws CIFSException {
            // Given
            SmbResource resource = createMockResource();
            when(resource.getContext()).thenReturn(mockContext);

            // When
            CIFSContext context = resource.getContext();

            // Then

            switch (transport.getContext().getConfig().getLanManCompatibility()) {
            case 0:
            case 1:
            case 2:
                this.macSigningKey = new byte[40];
                auth.getUserSessionKey(transport.getContext(), serverEncryptionKey, this.macSigningKey, 0);
                System.arraycopy(auth.getUnicodeHash(transport.getContext(), serverEncryptionKey), 0, this.macSigningKey, 16, 24);

    }

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

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

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