* @return A <code>byte[]</code> containing the challenge.
     */
    public byte[] getChallenge() {
        return this.challenge;
    }

    /**
     * Sets the challenge for this message.
     *
     * @param challenge
     *            The challenge from the domain controller/server.
     */
    public void setChallenge(final byte[] challenge) {
        this.challenge = challenge;
    }

    /**

            when(transportPool.getChallenge(context, address)).thenReturn(expectedChallenge);

            // When
            byte[] result = transportPool.getChallenge(context, address);

            // Then
            assertNotNull(result);
            assertArrayEquals(expectedChallenge, result);
            verify(transportPool).getChallenge(context, address);
        }

        @Test

     *
     * @return A <code>byte[]</code> containing the challenge.
     */
    public byte[] getChallenge() {
        return challenge;
    }

    /**
     * Sets the challenge for this message.
     *
     * @param challenge The challenge from the domain controller/server.
     */
    public void setChallenge(final byte[] challenge) {
        this.challenge = challenge;
    }

    /**

    class GetterSetterTests {

        @Test
        @DisplayName("setChallenge and getChallenge should work correctly")
        void testGetSetChallenge() {
            // Given
            CIFSContext mockContext = createMockContext();
            Type2Message message = new Type2Message(mockContext);

            // When
            message.setChallenge(TEST_CHALLENGE);

            // Then

     * @deprecated functionality is broken and will be removed at some point,
     *             use actual Active Directory authentication instead
     */
    @Deprecated
    byte[] getChallenge(CIFSContext tc, Address dc) throws CIFSException;

    /**
     * Get NTLM challenge from a server
     *
     * @param dc the domain controller address
     * @param port the port number

        // 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 {

                setLMResponse(clientChallenge);
                setNTResponse(ntlm2Response);

                if ((getFlags() & NTLMSSP_NEGOTIATE_SIGN) == NTLMSSP_NEGOTIATE_SIGN) {
                    final byte[] sessionNonce = new byte[16];
                    System.arraycopy(type2.getChallenge(), 0, sessionNonce, 0, 8);

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

                final byte[] ntlm2Response = NtlmUtil.getNTLM2Response(passwordHash, type2.getChallenge(), clientChallenge);

                setLMResponse(clientChallenge);
                setNTResponse(ntlm2Response);

                final byte[] sessionNonce = new byte[16];
                System.arraycopy(type2.getChallenge(), 0, sessionNonce, 0, 8);
                System.arraycopy(clientChallenge, 0, sessionNonce, 8, 8);

     * @throws SmbException if an SMB error occurs
     * @throws UnknownHostException if the host cannot be resolved
     */
    public static byte[] getChallenge(final UniAddress dc) throws SmbException, UnknownHostException {
        return getChallenge(dc, 0);
    }

    /**
     * Retrieves the NTLM challenge from the specified domain controller using the specified port.
     *