assertSame(transportPool, context.getTransportPool(), "Should return same instance on multiple calls");
    }

    @Test
    @DisplayName("getConfig should return the provided configuration")
    void testGetConfig() {
        // When
        Configuration config = context.getConfig();

        // Then
        assertSame(mockConfig, config, "Should return the same configuration instance");
    }

    @Test

    void testConstructor() {
        // Given & When
        Smb2FlushResponse flushResponse = new Smb2FlushResponse(mockConfig);

        // Then
        assertNotNull(flushResponse);
        // Cannot test getConfig() as it's protected
    }

    @Nested
    @DisplayName("writeBytesWireFormat tests")
    class WriteBytesWireFormatTests {

        @Test
        @DisplayName("Should return 0 for write bytes")

    }

    private void setupMockConfig() {
        when(mockConfig.getTransactionBufferSize()).thenReturn(65536);
    }

    private void setupMockContext() {
        when(mockContext.getConfig()).thenReturn(mockConfig);
    }

    @Test
    @DisplayName("Test constructor with config and control code")
    void testConstructorWithConfigAndControlCode() {
        setupMockConfig();

    @DisplayName("getters and basic state (happy path)")
    void testGettersAndState() {
        SmbSessionImpl session = newSession();

        // Basic getters
        assertSame(configuration, session.getConfig(), "getConfig delegates to transportContext");
        assertEquals("EXAMPLE", session.getTargetDomain());
        assertEquals("server.example", session.getTargetHost());

            }
        } else {
            final Trans2FindFirst2 req = new Trans2FindFirst2(tf.getConfig(), "\\", "*", SmbConstants.ATTR_DIRECTORY,
                    tf.getConfig().getListCount(), tf.getConfig().getListSize());
            final Trans2FindFirst2Response resp = new Trans2FindFirst2Response(tf.getConfig());
            try {
                send(req, resp);
            } catch (final SmbException e) {

        return new URL(null, spec, new Handler());
    }

    @Test
    @DisplayName("getConfig returns context config")
    void getConfig_returnsConfig() {
        SmbTreeConnection c = newConn();
        assertSame(config, c.getConfig());
    }

    @Test
    @DisplayName("acquire calls tree.acquire on first usage, release calls tree.release at zero")

            when(mockLocator.getUNCPath()).thenReturn("\\localhost\share\newdir");
            when(mockLocator.getShare()).thenReturn("share");
            // Mock tree handle's getConfig() to return our mock config
            when(mockTreeHandle.getConfig()).thenReturn(mockConfig);
        }

        @Test
        void testMkdir() throws SmbException, CIFSException {
            // Arrange

    void setUp() {
        MockitoAnnotations.openMocks(this);
        when(mockConfig.getMaximumBufferSize()).thenReturn(65536);
        when(mockConfig.getListSize()).thenReturn(65536);
        when(mockContext.getConfig()).thenReturn(mockConfig);
    }

    @Test
    @DisplayName("Test constructor with configuration only")
    void testConstructorWithConfig() {
        request = new Smb2QueryDirectoryRequest(mockConfig);

        // make sure that validation is performed in any case
        final Smb2SigningDigest dgst = getDigest();
        if (dgst != null && !isAsync() && (getConfig().isRequireSecureNegotiate() || getErrorCode() == NtStatus.NT_STATUS_SUCCESS)) {
            // MID checking not required here as we only read responses for which we're waiting

        this.expiration = exp;
    }

    /**
     * Gets the configuration object for this message block
     * @return the config
     */
    protected final Configuration getConfig() {
        return this.config;
    }

    /**
     *
     */
    @Override
    public void reset() {