}
    }

    @Test
    @DisplayName("getTransportPool should return initialized transport pool")
    void testGetTransportPool() {
        // When
        SmbTransportPool transportPool = context.getTransportPool();

        // Then
        assertNotNull(transportPool, "Transport pool should not be null");
        assertSame(transportPool, context.getTransportPool(), "Should return same instance on multiple calls");
    }

        // Setup default mock behaviors
        when(ctx.getConfig()).thenReturn(config);
        when(ctx.getNameServiceClient()).thenReturn(nameSvc);
        when(ctx.getCredentials()).thenReturn(creds);
        when(ctx.getTransportPool()).thenReturn(pool);

        // Default config values
        when(config.getLocalAddr()).thenReturn(null);
        when(config.getLocalPort()).thenReturn(0);
        when(config.getSessionLimit()).thenReturn(10);

    @Test
    void testGetTransportPool() {
        // Test getTransportPool() method
        SmbTransportPool mockSmbTransportPool = mock(SmbTransportPool.class);
        when(mockDelegate.getTransportPool()).thenReturn(mockSmbTransportPool);
        assertEquals(mockSmbTransportPool, cifsContextWrapper.getTransportPool());
        verify(mockDelegate).getTransportPool();
    }

    @Test

                } else {
                    dc = getTransportContext().getNameServiceClient().getByName(this.domainController, true);
                    challenge = getTransportContext().getTransportPool().getChallenge(getTransportContext(), dc);
                }

                ntlm = NtlmSsp.authenticate(getTransportContext(), req, resp, challenge);
                if (ntlm == null) {

            // https://lists.samba.org/archive/samba-technical/2009-August/066486.html
            // UniAddress addr = UniAddress.getByName(authDomain, true, tf);
            // SmbTransport trans = tf.getTransportPool().getSmbTransport(tf, addr, 0);
            try (SmbTransport dc = getDc(tf, authDomain)) {
                final CacheEntry<Map<String, CacheEntry<DfsReferralDataInternal>>> entry =

        return getChallenge(tf, dc, 0);
    }

    @Override
    public byte[] getChallenge(final CIFSContext tf, final Address dc, final int port) throws SmbException {
        try (SmbTransportInternal trans = tf.getTransportPool()
                .getSmbTransport(tf, dc, port, false, !tf.getCredentials().isAnonymous() && tf.getConfig().isIpcSigningEnforced())
                .unwrap(SmbTransportInternal.class)) {
            trans.ensureConnected();

                    if (log.isDebugEnabled()) {
                        log.debug("Using exclusive transport for " + this);
                    }
                    this.exclusiveTransport = this.ctx.getTransportPool()
                            .getSmbTransport(this.ctx, host, loc.getPort(), true, loc.shouldForceSigning())
                            .unwrap(SmbTransportInternal.class);

            transportContextField.set(ntlmServlet, cifsContext);
        } catch (Exception e) {
            throw new RuntimeException("Failed to inject mock CIFSContext", e);
        }

        when(cifsContext.getTransportPool()).thenReturn(transportPool);
        when(cifsContext.getNameServiceClient()).thenReturn(nameServiceClient);
        lenient().when(cifsContext.getConfig()).thenReturn(configuration);

                }

                req.getSession(); /* ensure session id is set for cluster env. */
                challenge = getTransportContext().getTransportPool().getChallenge(getTransportContext(), dc);
                if ((ntlm = NtlmSsp.authenticate(getTransportContext(), req, resp, challenge)) == null) {
                    return;
                }