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() {

        
        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

        
        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);
        }
    }
    

import jcifs.internal.smb2.info.Smb2QueryInfoRequest;
import jcifs.internal.smb2.info.Smb2SetInfoRequest;
import jcifs.internal.smb2.io.Smb2ReadRequest;
import jcifs.internal.smb2.io.Smb2WriteRequest;

/**
 * Load balancer for SMB3 Multi-Channel connections
 */
public class ChannelLoadBalancer {

    private final ChannelManager manager;
    private LoadBalancingStrategy strategy;

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