byte[] buffer = new byte[originalContext.size()];
            int encodedSize = originalContext.encode(buffer, 0);

            PreauthIntegrityNegotiateContext decodedContext = new PreauthIntegrityNegotiateContext();
            int decodedSize = decodedContext.decode(buffer, 0, encodedSize);

            // Assert
            assertEquals(encodedSize, decodedSize);

        private byte[] data;
        private int encodedSize;
        private boolean throwOnEncode = false;
        private boolean throwOnSize = false;
        private int sizeCallCount = 0;
        private int encodeCallCount = 0;

        public TestCreateContextRequest(byte[] name) {
            this.name = name;
            this.encodedSize = name != null ? name.length + 16 : 16; // 16 bytes for header
        }

            context = new PreauthIntegrityNegotiateContext(mockConfig, hashAlgos, salt);
            int encodedSize = context.encode(buffer, 0);

            PreauthIntegrityNegotiateContext decoded = new PreauthIntegrityNegotiateContext();
            int decodedSize = decoded.decode(buffer, 0, encodedSize);

            assertEquals(encodedSize, decodedSize);
            assertArrayEquals(hashAlgos, decoded.getHashAlgos());

    @Test
    @DisplayName("Test context encoding")
    public void testEncoding() {
        byte[] buffer = new byte[256];
        int encodedSize = context.encode(buffer, 0);

        assertTrue(encodedSize > 8); // Minimum size: count + padding + flags + algorithms
        assertEquals(context.size(), encodedSize);
    }

    @Test
    @DisplayName("Test context decoding")
    public void testDecoding() throws SMBProtocolDecodingException {

    @ParameterizedTest
    @ValueSource(ints = { 0, 1, 10, 100, 1000 })
    void testFileInformationEncodeDifferentSizes(int encodeSize) {
        when(mockFileInfo.encode(any(byte[].class), anyInt())).thenReturn(encodeSize);

        trans2SetFileInfo = new Trans2SetFileInformation(config, TEST_FID, mockFileInfo);

        byte[] buffer = new byte[2048];