// Testing invalid protocol ID during decode instead
        byte[] invalidBuffer = new byte[52];
        // Write invalid protocol ID
        invalidBuffer[0] = 0x00;
        invalidBuffer[1] = 0x00;
        invalidBuffer[2] = 0x00;
        invalidBuffer[3] = 0x00;

        assertThrows(IllegalArgumentException.class, () -> {
            Smb2TransformHeader.decode(invalidBuffer, 0);
        });
    }

    @Test

 */
internal object DnsRecordCodec {
  private const val SERVFAIL = 2
  private const val NXDOMAIN = 3
  const val TYPE_A = 0x0001
  const val TYPE_AAAA = 0x001c
  private const val TYPE_PTR = 0x000c
  private val ASCII = Charsets.US_ASCII

  fun encodeQuery(
    host: String,
    type: Int,
  ): ByteString =
    Buffer()
      .apply {

    // type/class
    static final int NB_IN = 0x00200001;
    static final int NBSTAT_IN = 0x00210001;
    static final int NB = 0x0020;
    static final int NBSTAT = 0x0021;
    static final int IN = 0x0001;
    static final int A = 0x0001;
    static final int NS = 0x0002;
    static final int NULL = 0x000a;

    static final int HEADER_LENGTH = 12;

    // header field offsets
    static final int OPCODE_OFFSET = 2;

    // type/class
    static final int NB_IN = 0x00200001;
    static final int NBSTAT_IN = 0x00210001;
    static final int NB = 0x0020;
    static final int NBSTAT = 0x0021;
    static final int IN = 0x0001;
    static final int A = 0x0001;
    static final int NS = 0x0002;
    static final int NULL = 0x000a;

    static final int HEADER_LENGTH = 12;

    // header field offsets
    static final int OPCODE_OFFSET = 2;

        assertEquals(0x01, dst[dstIndex + 6]);
    }

    @ParameterizedTest
    @DisplayName("Test writeSetupWireFormat with various FID values")
    @ValueSource(ints = { 0x0000, 0x0001, 0x7FFF, 0xFFFF, 0x1234, 0xABCD })
    void testWriteSetupWireFormatWithVariousFids(int fid) {
        notifyChange = new NtTransNotifyChange(mockConfig, fid, FILE_NOTIFY_CHANGE_FILE_NAME, false);

        assertEquals(0x00, dst[dstIndex + 2]);
        assertEquals(0x00, dst[dstIndex + 3]);

        // Verify security information (4 bytes)
        assertEquals(securityInfo, SMBUtil.readInt4(dst, dstIndex + 4));
    }

    @ParameterizedTest
    @DisplayName("Test writeParametersWireFormat with various FID values")
    @ValueSource(ints = { 0x0000, 0x0001, 0x7FFF, 0xFFFF, 0x1234, 0xABCD })

        assertEquals(0x2000, Smb2CreateRequest.FILE_OPEN_BY_FILE_ID);
        assertEquals(0x4000, Smb2CreateRequest.FILE_OPEN_FOR_BACKUP_INTENT);
        assertEquals(0x8000, Smb2CreateRequest.FILE_NO_COMPRESSION);
        assertEquals(0x10000, Smb2CreateRequest.FILE_OPEN_REQUIRING_OPLOCK);
        assertEquals(0x20000, Smb2CreateRequest.FILE_DISALLOW_EXCLUSIVE);

        byte[] wireData = new byte[] { 0x10, 0x00, // StructureSize (must be 16)
                0x01, // ShareType (DISK)
                0x00, // Reserved
                0x01, 0x00, 0x00, 0x00, // ShareFlags
                0x08, 0x00, 0x00, 0x00, // Capabilities
                (byte) 0xFF, 0x01, 0x1F, 0x00 // MaximalAccess
        };

        // When

                // For domain: Type 0x0002 (NetBIOS Domain Name)
                // For server: Type 0x0001 (NetBIOS Computer Name)

                // Verify domain part
                assertEquals(2, Type2Message.readUShort(targetInfo, 0)); // Type 0x0002

        byte[] encryptionKey = "12345678".getBytes();
        ByteBuffer buffer = ByteBuffer.allocate(response.byteCount);
        buffer.put(encryptionKey);
        buffer.put(domainNameBytes);
        buffer.putShort((short) 0x0000); // Null terminator

        byte[] byteData = buffer.array();

        // Call the method
        int bytesRead = response.readBytesWireFormat(byteData, 0);

        // Assertions