// When: Get challenge
        byte[] key = poolSpy.getChallenge(ctx, address);

        // Then: Should return server key
        assertArrayEquals(expectedKey, key, "Should return correct server key");
        verify(internal).ensureConnected();
    }

    @Test
    @DisplayName("Should wrap IOException in SmbException for getChallenge")
    void testGetChallengeIOException() throws Exception {

        // Final cleanup
        cleanup();
        return inUse;
    }

    @Override
    public byte[] getChallenge(final CIFSContext tf, final Address dc) throws SmbException {
        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()

            if (log.isTraceEnabled()) {
                log.trace(msg2.toString());
                log.trace(Hexdump.toHexString(token));
            }

            this.serverChallenge = msg2.getChallenge();

            if (this.requireKeyExchange) {
                if (this.transportContext.getConfig().isEnforceSpnegoIntegrity() && (!msg2.getFlag(NtlmFlags.NTLMSSP_NEGOTIATE_KEY_EXCH)

        lenient().when(configuration.getDefaultDomain()).thenReturn("TEST_DOMAIN");
        lenient().when(nameServiceClient.getByName(anyString(), anyBoolean())).thenReturn(mock(Address.class));
        lenient().when(transportPool.getChallenge(any(), any())).thenReturn(new byte[8]);
    }

                    dc = chal.dc;
                    challenge = chal.challenge;
                } else {
                    dc = UniAddress.getByName(domainController, true);
                    challenge = SmbSession.getChallenge(dc);
                }

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

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

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

                    dc = UniAddress.getByName(server, possibleWorkgroup);
                }

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