}

    @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];

    void testUUIDOperations() {
        // Given
        byte[] uuid = { 0x12, 0x34, 0x56, 0x78, (byte) 0x9A, (byte) 0xBC, (byte) 0xDE, (byte) 0xF0, 0x11, 0x22, 0x33, 0x44, 0x55, 0x66,
                0x77, (byte) 0x88 };
        byte[] buffer = new byte[16];

        // When
        System.arraycopy(uuid, 0, buffer, 0, 16);

        // Then
        assertArrayEquals(uuid, buffer);
    }

            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());
        }
    }

    @Nested

    private static final short TIME_MID = (short) 0x4455;
    private static final short TIME_HI_AND_VERSION = (short) 0x6677;
    private static final byte CLOCK_SEQ_HI_AND_RESERVED = (byte) 0x88;
    private static final byte CLOCK_SEQ_LOW = (byte) 0x99;
    private static final byte[] NODE = { (byte) 0xAA, (byte) 0xBB, (byte) 0xCC, (byte) 0xDD, (byte) 0xEE, (byte) 0xFF };

    @Nested

        Longs.toByteArray(0x1213141516171819L));
    assertByteArrayEquals(
        new byte[] {
          (byte) 0xFF, (byte) 0xEE, (byte) 0xDD, (byte) 0xCC,
          (byte) 0xBB, (byte) 0xAA, (byte) 0x99, (byte) 0x88
        },
        Longs.toByteArray(0xFFEEDDCCBBAA9988L));
  }

  public void testFromByteArray() {
    assertThat(

        int readDataAvailable = 256;
        int numberOfMessages = 5;
        int messageLength = 128;
        byte[] testData = { (byte) 0xFF, (byte) 0xEE, (byte) 0xDD, (byte) 0xCC, (byte) 0xBB, (byte) 0xAA, (byte) 0x99, (byte) 0x88 };

        // Write values to buffer at offset
        SMBUtil.writeInt4(namedPipeState, buffer, bufferIndex);
        SMBUtil.writeInt4(readDataAvailable, buffer, bufferIndex + 4);

            originalUuid.time_mid = (short) 0x4321;
            originalUuid.time_hi_and_version = (short) 0x8765;
            originalUuid.clock_seq_hi_and_reserved = (byte) 0x99;
            originalUuid.clock_seq_low = (byte) 0x88;
            originalUuid.node = new byte[] { (byte) 0x11, (byte) 0x22, (byte) 0x33, (byte) 0x44, (byte) 0x55, (byte) 0x66 };

            // Mock encoding buffer
            NdrBuffer encodeBuffer = mock(NdrBuffer.class);

        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];

        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);

0xeb, 0x83, 0x5a, 0xc6, 0x45, 0xd3, 0x17, 0x21, 0xe7, 0xa1, 0x98, 0x52, 0x88, 0xc, 0xc4, 0x4, 0xfe, 0x54, 0xbc, 0x2f, 0x2, 0x0, 0xc4, 0xd5, 0x83, 0xa4, 0x54, 0x79, 0x70, 0x65, 0x2, 0xa6, 0x44, 0x65, 0x6c, 0x4f, 0x62, 0x6a, 0x83, 0xa2, 0x49, 0x44, 0xc4, 0x10, 0x6a, 0x33, 0x51, 0xe3, 0x48, 0x0, 0x4e, 0xed, 0xab, 0x9, 0x21, 0xeb, 0x83, 0x5a, 0xc6, 0x45, 0xa5, 0x4d, 0x54, 0x69, 0x6d, 0x65, 0xd3, 0x17, 0x21, 0xe7, 0xa1, 0x98, 0x52, 0x88, 0xc, 0xa7, 0x4d, 0x65, 0x74, 0x61, 0x53, 0x79, 0x73, 0x82, 0xd9, 0x26,...