byte testInfoType = (byte) 0x01;

        request.setInfoType(testInfoType);

        // Verify info type was set
        Field infoTypeField;
        try {
            infoTypeField = Smb2QueryInfoRequest.class.getDeclaredField("infoType");
            infoTypeField.setAccessible(true);
            byte actualInfoType = (byte) infoTypeField.get(request);

    private static Stream<Arguments> provideFidTestCases() {
        return Stream.of(Arguments.of(0x0000, (byte) 0x00, (byte) 0x00), Arguments.of(0x00FF, (byte) 0xFF, (byte) 0x00),
                Arguments.of(0xFF00, (byte) 0x00, (byte) 0xFF), Arguments.of(0xFFFF, (byte) 0xFF, (byte) 0xFF),
                Arguments.of(0x1234, (byte) 0x34, (byte) 0x12));
    }

    @Test
    @DisplayName("readSetupWireFormat should return 0")

    }

    @Test
    @DisplayName("Should handle AES encryption operations")
    void testAESOperations() throws Exception {
        // Given
        byte[] key = new byte[16]; // 128-bit key
        byte[] iv = new byte[16]; // 128-bit IV
        byte[] plaintext = "This is test data for AES encryption test".getBytes();

        new SecureRandom().nextBytes(key);
        new SecureRandom().nextBytes(iv);

    public int getContextType() {
        return CONTEXT_ID;
    }

    /**
     * Get the context data
     *
     * @return encoded context data
     */
    public byte[] getData() {
        // Encode the context data
        byte[] data = new byte[20]; // 2 + 2 + 4 + 4 + 4 + 4

        int idx = 0;
        SMBUtil.writeInt2(transformCount, data, idx);
        idx += 2;
        SMBUtil.writeInt2(reserved1, data, idx);

        void writeSetupWireFormatReturnsZero(int index) {
            TransWaitNamedPipeResponse resp = new TransWaitNamedPipeResponse();
            assertEquals(0, resp.writeSetupWireFormat(new byte[10], index), "writeSetupWireFormat should always return 0");
        }

        @ParameterizedTest
        @ValueSource(ints = { 0, 5, -5, Integer.MAX_VALUE })

		q[j-1] = 1 // difference is only found on the last byte
		for i := range b {
			idx := Index(b[i:], q[:j])
			if idx != -1 {
				t.Fatalf("Index(b[%d:], q[:%d])=%d, want -1\n", i, j, idx)
			}
		}
		q[j-1] = 0
	}

	// Test differing alignments and sizes of q which always end on a page boundary.
	q[len(q)-1] = 1 // difference is only found on the last byte
	for j := 0; j < len(q); j++ {
		for i := range b {

    }

    /**
     * The NTLMSSP "preamble".
     */
    protected static final byte[] NTLMSSP_SIGNATURE =
            { (byte) 'N', (byte) 'T', (byte) 'L', (byte) 'M', (byte) 'S', (byte) 'S', (byte) 'P', (byte) 0 };

    /**
     * NTLM version
     */
    protected static final byte[] NTLMSSP_VERSION = { 6, 1, 0, 0, 0, 0, 0, 15 };

    /**
     * NTLMSSP Type 1 message identifier.
     */

    void testWithLargeBuffers() {
        // Test with large byte arrays
        byte[] largeRequestBuffer = new byte[1024];
        byte[] largeResponseBuffer = new byte[2048];

        // Fill with some test data
        for (int i = 0; i < largeRequestBuffer.length; i++) {
            largeRequestBuffer[i] = (byte) (i % 256);
        }
        for (int i = 0; i < largeResponseBuffer.length; i++) {

            // Arrange
            context = new PreauthIntegrityNegotiateContext();
            byte[] buffer = new byte[] { (byte) 0xFF, (byte) 0xFF, (byte) 0xFF, // Padding
                    0x01, 0x00, // 1 hash algo
                    0x02, 0x00, // 2 bytes salt
                    0x01, 0x00, // SHA512
                    (byte) 0xAA, (byte) 0xBB // Salt
            };

            // Act

        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;
        Type2Message type2Message = new Type2Message(type2Flags, serverChallenge, domain);
        byte[] type2Token = type2Message.toByteArray();