assertEquals(24, resumeKey.length, "Resume key should be 24 bytes");

        // Verify all bytes are 0xAA
        for (int i = 0; i < 24; i++) {
            assertEquals((byte) 0xAA, resumeKey[i], "Resume key byte " + i + " should be 0xAA");
        }
    }

    @Test
    @DisplayName("Test decode throws exception when length is too short")

    }

    @Test
    @DisplayName("readUShort correctly interprets little endian")
    void testReadUShort() {
        byte[] a = new byte[] { (byte) 0xAA, (byte) 0xBB };
        int result = NtlmMessage.readUShort(a, 0);
        assertEquals(0xBBAA, result, "UShort should be little‑endian");
    }

    @Test
    @DisplayName("readUShort throws on too short array")

            uuid.time_hi_and_version = (short) 0xDEF0;
            uuid.clock_seq_hi_and_reserved = (byte) 0x11;
            uuid.clock_seq_low = (byte) 0x22;
            uuid.node = new byte[] { (byte) 0xAA, (byte) 0xBB, (byte) 0xCC, (byte) 0xDD, (byte) 0xEE, (byte) 0xFF };

            // Mock the derive method to return a buffer for node encoding
            when(mockNdrBuffer.derive(anyInt())).thenReturn(mockNdrBuffer);

        byte[] header = buildHeader(NtStatus.NT_STATUS_INVALID_PARAMETER);
        byte[] body = buildErrorBody(2, 3, new byte[] { (byte) 0xAA, (byte) 0xBB, (byte) 0xCC });
        byte[] packet = new byte[header.length + body.length];
        System.arraycopy(header, 0, packet, 0, header.length);
        System.arraycopy(body, 0, packet, header.length, body.length);

    }

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

        key.encode(buffer, 2);

        }

        @Test
        @DisplayName("Should not modify buffer during write")
        void testWriteDoesNotModifyBuffer() {
            byte[] buffer = new byte[64];
            Arrays.fill(buffer, (byte) 0xAA);
            byte[] originalBuffer = buffer.clone();

            notification.writeBytesWireFormat(buffer, 0);

            assertArrayEquals(originalBuffer, buffer);
        }
    }

    @Nested

    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
    @DisplayName("Constructor Tests")
    class ConstructorTests {

        @Test

                    (byte) 0xAA, (byte) 0xBB // Salt
            };

            // Act
            int decodedSize = context.decode(buffer, 3, buffer.length - 3);

            // Assert
            assertEquals(8, decodedSize);
            assertArrayEquals(new int[] { 1 }, context.getHashAlgos());
            assertArrayEquals(new byte[] { (byte) 0xAA, (byte) 0xBB }, context.getSalt());
        }
    }

            Security.addProvider(new BouncyCastleProvider());
        }
    }

    @BeforeEach
    void setup() {
        sessionKey = new byte[16];
        Arrays.fill(sessionKey, (byte) 0xAA);

        preauthIntegrityHash = new byte[64];
        Arrays.fill(preauthIntegrityHash, (byte) 0xBB);
    }

    @Nested
    @DisplayName("Constructor Tests")
    class ConstructorTests {

        @Test

        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(