}

    @Test
    @DisplayName("Should encode lease key to buffer")
    void testEncode() {
        byte[] testBytes = new byte[] { 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, (byte) 0x88, (byte) 0x99, (byte) 0xAA, (byte) 0xBB,
                (byte) 0xCC, (byte) 0xDD, (byte) 0xEE, (byte) 0xFF, 0x00 };

        Smb2LeaseKey key = new Smb2LeaseKey(testBytes);
        byte[] buffer = new byte[20];

            Type2Message message = new Type2Message(mockContext);
            byte[] testContext =
                    new byte[] { (byte) 0x11, (byte) 0x22, (byte) 0x33, (byte) 0x44, (byte) 0x55, (byte) 0x66, (byte) 0x77, (byte) 0x88 };

            // When
            message.setContext(testContext);

            // Then
            assertArrayEquals(testContext, message.getContext());
        }
    }

            SrvCopychunk chunk = new SrvCopychunk(999L, 888L, 777);

            // When
            int bytesWritten = chunk.encode(exactBuffer, 0);

            // Then
            assertEquals(EXPECTED_SIZE, bytesWritten);
            assertEquals(999L, SMBUtil.readInt8(exactBuffer, 0));
            assertEquals(888L, SMBUtil.readInt8(exactBuffer, 8));
            assertEquals(777, SMBUtil.readInt4(exactBuffer, 16));
        }
    }

    }

    @Test
    @DisplayName("writeSecurityBuffer copies data and sets fields")
    void testWriteSecurityBufferNonZero() {
        byte[] dest = new byte[8 + 4];
        byte[] payload = { 0x55, 0x66, 0x77 };
        NtlmMessage.writeSecurityBuffer(dest, 0, 8, payload);
        assertEquals(3, NtlmMessage.readUShort(dest, 0), "Length field must be 3");

      0x27,
      0x6,
      0x74,
      0x75,
      0x28,
      0x29,
      0x2a,
      0x7,
      0x2b,
      0x76,
      0x2c,
      0x8,
      0x9,
      0x2d,
      0x77,
      0x78,
      0x79,
      0x7a,
      0x7b,
      0x7ffe,
      0x7fc,
      0x3ffd,
      0x1ffd,
      0xffffffc,
      0xfffe6,
      0x3fffd2,
      0xfffe7,
      0xfffe8,

        buf[bodyStart + 2] = 0x02; // errorContextCount
        buf[bodyStart + 3] = 0x00; // reserved
        int bc = 4;
        SMBUtil.writeInt4(bc, buf, bodyStart + 4);
        byte[] err = new byte[] { 0x55, 0x66, 0x77, 0x01 };
        System.arraycopy(err, 0, buf, bodyStart + 8, bc);

        int decoded = resp.decode(buf, headerStart);
        assertTrue(decoded > 0);

        // When error path is taken, no blob should be parsed

        serverData.serverTimeZone = -480; // PST
        serverData.encryptionKeyLength = 8;
        serverData.encryptionKey = new byte[] { 0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, (byte) 0x88 };
        serverData.guid = new byte[16];

        // Then - verify all values are set correctly
        assertEquals((byte) 0x98, serverData.sflags);
        assertEquals(0xC853, serverData.sflags2);

        int capabilities = 0x0000000F;
        long previousSessionId = 0x123456789ABCDEF0L;
        byte[] token = { (byte) 0x11, (byte) 0x22, (byte) 0x33, (byte) 0x44, (byte) 0x55, (byte) 0x66, (byte) 0x77, (byte) 0x88 };

        Smb2SessionSetupRequest req = new Smb2SessionSetupRequest(mockContext, securityMode, capabilities, previousSessionId, token);
        req.setSessionBinding(true);
        byte[] buffer = new byte[512];