assertEquals(1 + expectedString.length + 1, used, "Expected exactly one command byte plus OEM string and NUL terminator");
        assertEquals(0x04, buffer[0] & 0xFF, "First byte must be SMB_COM_QUERY_INFORMATION type 0x04");
        // Verify that the string part ends with the null terminator
        assertArrayEquals(expectedString, subArray(buffer, 1, expectedString.length));

        doAnswer(inv -> {
            byte[] b = inv.getArgument(0);
            int off = inv.getArgument(1);
            b[off] = 0x12;
            b[off + 1] = 0x34;
            return 2;
        }).when(raf).read(any(byte[].class), anyInt(), eq(2));
        assertEquals(0x1234, raf.readUnsignedShort());
        assertEquals((short) 0x1234, raf.readShort());

  public void testCollisionsDueToIncorrectSignExtension() {
    byte[] col1 = new byte[] {0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, (byte) 0x80};
    byte[] col2 = new byte[] {0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, (byte) 0x81};
    byte[] col3 = new byte[] {0x00, 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, (byte) 0xff};

    ImmutableSet<HashCode> hashCodes =
        ImmutableSet.of(

    void constructorSetsCorrectCommand() {
        SmbComLogoffAndX msg = new SmbComLogoffAndX(null);
        assertNotNull(msg, "Message should not be null");
        // SMB_COM_LOGOFF_ANDX = 0x74
        assertEquals((byte) 0x74, msg.command, "Command should be SMB_COM_LOGOFF_ANDX");
    }

    @Test
    @DisplayName("constructor with mock andx properly sets andx field")
    void constructorWithMockAndx() {

    private byte[] testResponseBuffer;
    private SmbNegotiation negotiation;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
        testRequestBuffer = new byte[] { 0x01, 0x02, 0x03, 0x04 };
        testResponseBuffer = new byte[] { 0x05, 0x06, 0x07, 0x08 };
        negotiation = new SmbNegotiation(mockRequest, mockResponse, testRequestBuffer, testResponseBuffer);
    }

    @Test

        byte[] data = { 0x01, 0x02, 0x03, 0x04, 0x05 };
        ByteEncodable encodable = new ByteEncodable(data, 1, 3);

        // Verify size
        assertEquals(3, encodable.size(), "Size should be equal to the specified length");
    }

    @Test
    void testEncodeBasic() {
        // Test basic encoding
        byte[] data = { 0x01, 0x02, 0x03, 0x04, 0x05 };

 */
class SMBUtilTest {

    @Test
    void testWriteInt2() {
        byte[] dst = new byte[10];

        // Test with simple value
        SMBUtil.writeInt2(0x1234L, dst, 0);
        assertEquals((byte) 0x34, dst[0]);
        assertEquals((byte) 0x12, dst[1]);

        // Test with max value for 2 bytes
        SMBUtil.writeInt2(0xFFFFL, dst, 2);
        assertEquals((byte) 0xFF, dst[2]);

                Arguments.of("too short header", new byte[] { 0x04, 0x01, 0x00 }, IllegalArgumentException.class),

                // Wrong TOK_ID
                Arguments.of("wrong TOK_ID", new byte[] { 0x00, 0x02, 0x00, 0x01 }, IllegalArgumentException.class),

                // OID length claims more than available
                Arguments.of("oid length exceeds buffer", new byte[] { 0x04, 0x01, 0x00, 0x10, 0x06 }, IllegalArgumentException.class),

        void testConstructorWithConfigHashAlgosAndSalt() {
            int[] hashAlgos = { PreauthIntegrityNegotiateContext.HASH_ALGO_SHA512 };
            byte[] salt = { 0x01, 0x02, 0x03, 0x04 };

            PreauthIntegrityNegotiateContext context = new PreauthIntegrityNegotiateContext(mockConfig, hashAlgos, salt);

            assertNotNull(context);

    @Override
    protected int writeBytesWireFormat(final byte[] dst, int dstIndex) {
        final int start = dstIndex;

        dst[dstIndex] = (byte) 0x04;
        dstIndex++;
        dstIndex += writeString(this.oldFileName, dst, dstIndex);
        dst[dstIndex++] = (byte) 0x04;
        if (this.isUseUnicode()) {
            dst[dstIndex++] = (byte) '\0';
        }
        dstIndex += writeString(this.newFileName, dst, dstIndex);