import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;

import jcifs.CIFSException;
import jcifs.Configuration;
import jcifs.SmbConstants;
import jcifs.internal.smb2.io.Smb2WriteRequest;
import jcifs.internal.smb2.io.Smb2WriteResponse;

@ExtendWith(MockitoExtension.class)
class SmbFileOutputStreamTest {

    @Mock
    private SmbFile mockFile;

    @Mock

import jcifs.internal.util.SMBUtil;

/**
 * Test class for Smb2WriteRequest functionality
 */
@DisplayName("Smb2WriteRequest Tests")
class Smb2WriteRequestTest {

    @Mock
    private Configuration mockConfig;

    @Mock
    private CIFSContext mockContext;

    private byte[] testFileId;
    private Smb2WriteRequest request;

    @BeforeEach
    void setUp() {

import jcifs.internal.smb2.info.Smb2QueryInfoRequest;
import jcifs.internal.smb2.info.Smb2SetInfoRequest;
import jcifs.internal.smb2.io.Smb2FlushRequest;
import jcifs.internal.smb2.io.Smb2ReadRequest;
import jcifs.internal.smb2.io.Smb2WriteRequest;
import jcifs.internal.smb2.ioctl.Smb2IoctlRequest;
import jcifs.internal.smb2.lock.Smb2Lock;
import jcifs.internal.smb2.lock.Smb2LockRequest;

/**
 * Test class for RequestWithFileId interface.

        
        if (request instanceof Smb2ReadRequest) {
            handleRdmaRead((Smb2ReadRequest) request);
        } else if (request instanceof Smb2WriteRequest) {
            handleRdmaWrite((Smb2WriteRequest) request);
        } else {
            // Fall back to send/receive for non-data operations
            sendWithRdmaSendReceive(request);
        }
    }
    

        
        if (message instanceof Smb2ReadRequest) {
            affinityKey = Arrays.hashCode(((Smb2ReadRequest)message).getFileId());
        } else if (message instanceof Smb2WriteRequest) {
            affinityKey = Arrays.hashCode(((Smb2WriteRequest)message).getFileId());
        }
        
        if (affinityKey != 0) {
            // Select channel based on affinity key

                w = len > blockSize ? blockSize : len;

                if (this.smb2) {
                    final Smb2WriteRequest wr = new Smb2WriteRequest(th.getConfig(), fh.getFileId());
                    wr.setOffset(this.fp);
                    wr.setData(b, off, w);

                    final Smb2WriteResponse resp = th.send(wr, RequestParam.NO_RETRY);

            int w;
            do {
                w = len > this.writeSize ? this.writeSize : len;
                long cnt;

                if (th.isSMB2()) {
                    final Smb2WriteRequest request = new Smb2WriteRequest(th.getConfig(), fh.getFileId());
                    request.setOffset(this.fp);
                    request.setRemainingBytes(len - w - off);
                    request.setData(b, off, w);

import jcifs.internal.SmbNegotiationResponse;
import jcifs.internal.smb2.ServerMessageBlock2Response;
import jcifs.internal.smb2.Smb2Constants;
import jcifs.internal.smb2.io.Smb2ReadResponse;
import jcifs.internal.smb2.io.Smb2WriteRequest;
import jcifs.internal.util.SMBUtil;
import jcifs.util.Hexdump;
import jcifs.util.transport.Response;

/**
 * SMB2 Negotiate Protocol response message.
 *