// Then
        assertEquals(52, encoded);

        // Verify protocol ID (first 4 bytes) - 0xFD534D42 in little-endian
        assertEquals((byte) 0x42, buffer[0]);
        assertEquals((byte) 0x4D, buffer[1]);
        assertEquals((byte) 0x53, buffer[2]);
        assertEquals((byte) 0xFD, buffer[3]);
    }

    @Test
    @DisplayName("Should decode transform header from byte buffer")

    public static final int SUCCESS = 0;
    /** Name service error class */
    public static final int ERR_NAM_SRVC = 0x01;
    /** Session service error class */
    public static final int ERR_SSN_SRVC = 0x02;

    // name service error codes
    /** Format error in the name service */
    public static final int FMT_ERR = 0x1;
    /** Server error in the name service */
    public static final int SRV_ERR = 0x2;

    class ToStringTests {

        @Test
        @DisplayName("toString with valid data produces expected format")
        void testToStringWithValidData() {
            byte[] challenge = new byte[] { (byte) 0x01, (byte) 0x02, (byte) 0x03, (byte) 0xFF };
            UniAddress dc = mock(UniAddress.class);
            when(dc.toString()).thenReturn("SERVER123");

            NtlmChallenge nc = new NtlmChallenge(challenge, dc);

                // negative fid wraps around, lastWriteTime = 1000 -> 1 second -> int value 1
                Arguments.of(-1, 1000L, new byte[] { (byte) 0xFF, (byte) 0xFF, 0x01, 0x00, 0x00, 0x00 }));
    }

    /**
     * When lastWriteTime is zero the encoded unsigned time must be all 1s.
     */
    @Test
    @DisplayName("edge: writeParameterWordsWireFormat with lastWriteTime=0")

        assertNotNull(request);
    }

    @Test
    @DisplayName("Test all file information class constants")
    void testFileInformationClassConstants() {
        assertEquals(0x01, Smb2QueryDirectoryRequest.FILE_DIRECTORY_INFO);
        assertEquals(0x02, Smb2QueryDirectoryRequest.FILE_FULL_DIRECTORY_INFO);
        assertEquals(0x03, Smb2QueryDirectoryRequest.FILE_BOTH_DIRECTORY_INFO);

        }
    }

    @Test
    @DisplayName("Test setInfoType method")
    void testSetInfoType() {
        request = new Smb2QueryInfoRequest(mockConfig);
        byte testInfoType = (byte) 0x01;

        request.setInfoType(testInfoType);

        // Verify info type was set
        Field infoTypeField;
        try {
            infoTypeField = Smb2QueryInfoRequest.class.getDeclaredField("infoType");

    }

    @Test
    @DisplayName("parsing malformed token throws IOException")
    void testMalformedToken() {
        byte[] bad = new byte[] { 0x01, 0x02, 0x03 }; // not a valid ASN.1 tagged object
        assertThrows(IOException.class, () -> new NegTokenTarg(bad), "Malformed byte[] should cause IOException");
    }

    @Test

        @DisplayName("Should write request structure correctly")
        void testWriteBytesWireFormat() {
            // Setup
            request.setPadding((byte) 2);
            request.setReadFlags((byte) 0x01);
            request.setReadLength(4096);
            request.setOffset(8192L);
            request.setMinimumCount(512);
            request.setRemainingBytes(1024);

            byte[] buffer = new byte[256];

  const val TYPE_DATA = 0x0
  const val TYPE_HEADERS = 0x1
  const val TYPE_PRIORITY = 0x2
  const val TYPE_RST_STREAM = 0x3
  const val TYPE_SETTINGS = 0x4
  const val TYPE_PUSH_PROMISE = 0x5
  const val TYPE_PING = 0x6
  const val TYPE_GOAWAY = 0x7
  const val TYPE_WINDOW_UPDATE = 0x8
  const val TYPE_CONTINUATION = 0x9

  const val FLAG_NONE = 0x0
  const val FLAG_ACK = 0x1 // Used for settings and ping.

    private LeaseV1CreateContextRequest leaseContext;
    private Smb2LeaseKey testKey;
    private int testState;

    @BeforeEach
    void setUp() {
        byte[] keyBytes = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10 };
        testKey = new Smb2LeaseKey(keyBytes);
        testState = Smb2LeaseState.SMB2_LEASE_READ_WRITE;