}

    @Override
    public byte[] getExpected(byte[] bytes) {
      if (initialBytes == null) {
        return checkNotNull(bytes);
      } else {
        byte[] result = new byte[initialBytes.length + bytes.length];
        System.arraycopy(initialBytes, 0, result, 0, initialBytes.length);
        System.arraycopy(bytes, 0, result, initialBytes.length, bytes.length);
        return result;
      }
    }

        private int dataWireFormatReturn = 0;
        private byte[] setupBuffer;
        private byte[] parametersBuffer;
        private byte[] dataBuffer;

        public TestSmbComTransaction(Configuration config, byte command, byte subCommand) {
            super(config, command, subCommand);
        }

        @Override
        protected int writeSetupWireFormat(byte[] dst, int dstIndex) {

        dst[di] = (byte) (i & 0xFF);
        di++;
        dst[di++] = (byte) (i >> 8 & 0xFF);
        dst[di++] = (byte) (i >> 16 & 0xFF);
        dst[di] = (byte) (i >> 24 & 0xFF);
        return 4;
    }

    /*
     * Decode integers
     */

    /**
     * Decodes a 16-bit unsigned integer from big-endian byte order.
     *

    private Configuration mockConfig;
    private CIFSContext mockContext;
    private SecureRandom mockRandom;
    private Smb2NegotiateRequest request;
    private byte[] testMachineId = new byte[16];
    private byte[] testSalt = new byte[32];

    @BeforeEach
    void setUp() {
        mockConfig = mock(Configuration.class);
        mockContext = mock(CIFSContext.class);
        mockRandom = mock(SecureRandom.class);

        // Test with file ID having maximum byte values
        byte[] maxFileId = new byte[16];
        Arrays.fill(maxFileId, (byte) 0xFF);
        Smb2FlushRequest maxRequest = new Smb2FlushRequest(mockConfig, maxFileId);

        byte[] buffer = new byte[256];
        int written = maxRequest.writeBytesWireFormat(buffer, 0);

        assertEquals(24, written);

        byte[] copiedFileId = new byte[16];

        // Verify content
        assertArrayEquals(new byte[] { (byte) 0xFF, 0x01, 0x02, 0x03, 0x00 }, dest, "Encoded bytes should be placed at the correct offset");
    }

    @Test
    void testEncodeZeroLength() {
        // Test encoding with zero length
        byte[] data = { 0x01, 0x02, 0x03 };
        ByteEncodable encodable = new ByteEncodable(data, 0, 0);
        byte[] dest = new byte[3];

     * The header format is:
     * - Byte 0: Message type (0x00 for SESSION_MESSAGE)
     * - Byte 1: High bit of 17-bit length field
     * - Bytes 2-3: Lower 16 bits of length field
     */
    private static byte[] messageHeader(int length) {
        byte[] h = new byte[4];
        h[0] = (byte) SessionServicePacket.SESSION_MESSAGE;
        h[1] = (byte) ((length >> 16) & 0x01);
        h[2] = (byte) ((length >> 8) & 0xFF);

    }

    @Test
    void testReadInt4() {
        byte[] src = new byte[] { (byte) 0x78, (byte) 0x56, (byte) 0x34, (byte) 0x12, // 0x12345678
                (byte) 0xFF, (byte) 0xFF, (byte) 0xFF, (byte) 0xFF, // 0xFFFFFFFF
                (byte) 0x00, (byte) 0x00, (byte) 0x00, (byte) 0x00, // 0x00000000
                (byte) 0xEF, (byte) 0xCD, (byte) 0xAB, (byte) 0x89 // 0x89ABCDEF
        };

	}
	return
}

// MarshalMsg implements msgp.Marshaler
func (z BatchJobSize) MarshalMsg(b []byte) (o []byte, err error) {
	o = msgp.Require(b, z.Msgsize())
	o = msgp.AppendInt64(o, int64(z))
	return
}

// UnmarshalMsg implements msgp.Unmarshaler
func (z *BatchJobSize) UnmarshalMsg(bts []byte) (o []byte, err error) {
	{
		var zb0001 int64
		zb0001, bts, err = msgp.ReadInt64Bytes(bts)
		if err != nil {

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

        // When
        req.encode(buffer, 0);