}

    @Test
    @DisplayName("Should create lease key from byte array")
    void testLeaseKeyFromBytes() {
        byte[] testBytes = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10 };

        Smb2LeaseKey key = new Smb2LeaseKey(testBytes);

        assertArrayEquals(testBytes, key.getKey());
        assertFalse(key.isZero());
    }

    @Test

        Configuration mockConfig = mock(Configuration.class);
        byte[] expectedMachineId = new byte[32];
        expectedMachineId[0] = 0x0A;
        expectedMachineId[1] = 0x0B;
        when(mockConfig.getMachineId()).thenReturn(expectedMachineId);

        AvSingleHost avSingleHost = new AvSingleHost(mockConfig);

        assertNotNull(avSingleHost);

    private Smb2LeaseKey testKey;
    private int testState;

    @BeforeEach
    void setUp() {
        byte[] keyBytes = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10 };
        testKey = new Smb2LeaseKey(keyBytes);
        testState = Smb2LeaseState.SMB2_LEASE_READ_WRITE;
        leaseContext = new LeaseV1CreateContextRequest(testKey, testState);
    }

        assertEquals(outlen, expanded.length);

        // Test vectors from RFC 3961 Appendix A.1
        // 64-fold("012345")
        verifyNfold("012345", 8,
                new byte[] { (byte) 0xbe, (byte) 0x07, (byte) 0x26, (byte) 0x31, (byte) 0x27, (byte) 0x6b, (byte) 0x19, (byte) 0x55 });

        // 56-fold("password")
        verifyNfold("password", 7,

            when(mockNdrBuffer.dec_ndr_small()).thenReturn((int) (byte) 0x55, (int) (byte) 0x66, // clock seq bytes
                    (int) (byte) 0x07, (int) (byte) 0x08, (int) (byte) 0x09, (int) (byte) 0x0A, (int) (byte) 0x0B, (int) (byte) 0x0C // node bytes
            );
            when(mockNdrBuffer.derive(anyInt())).thenReturn(mockNdrBuffer);

            // When: Decoding the policy handle

    private byte[] testMacSigningKey;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
        testMacSigningKey = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10 };
    }

    @Test
    @DisplayName("Test constructor with MAC signing key and bypass flag")
    void testConstructorWithBypass() {

        // instead of being hashed when it exceeds the block size.

        byte[] key = { (byte) 0x0b, (byte) 0x0b, (byte) 0x0b, (byte) 0x0b, (byte) 0x0b, (byte) 0x0b, (byte) 0x0b, (byte) 0x0b, (byte) 0x0b,
                (byte) 0x0b, (byte) 0x0b, (byte) 0x0b, (byte) 0x0b, (byte) 0x0b, (byte) 0x0b, (byte) 0x0b };
        byte[] data = "Hi There".getBytes();

        // Calculate expected HMACT64 result manually

 */
@DisplayName("Type2Message Comprehensive Tests")
class Type2MessageTest {

    private static final byte[] TEST_CHALLENGE = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08 };
    private static final String TEST_TARGET = "TEST_TARGET";
    private static final String TEST_DOMAIN = "TEST_DOMAIN";
    private static final String TEST_HOSTNAME = "TEST_HOSTNAME";

    /**

    static final byte SMB_COM_CLOSE = (byte) 0x04;
    static final byte SMB_COM_DELETE = (byte) 0x06;
    static final byte SMB_COM_RENAME = (byte) 0x07;
    static final byte SMB_COM_QUERY_INFORMATION = (byte) 0x08;
    static final byte SMB_COM_WRITE = (byte) 0x0B;
    static final byte SMB_COM_CHECK_DIRECTORY = (byte) 0x10;
    static final byte SMB_COM_TRANSACTION = (byte) 0x25;

        context.initSecContext(new byte[0], 0, 0);

        // Create a mock Type 2 message (server challenge)
        byte[] serverChallenge = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08 };
        int type2Flags = NtlmFlags.NTLMSSP_NEGOTIATE_NTLM | NtlmFlags.NTLMSSP_REQUEST_TARGET | NtlmFlags.NTLMSSP_NEGOTIATE_KEY_EXCH
                | NtlmFlags.NTLMSSP_NEGOTIATE_SIGN;