* Tests the batch limit configuration for various SMB commands.
 */
public class SmbConnectionTest {

    /**
     * Test that getBatchLimit returns correct values for different commands
     */
    @Test
    @DisplayName("getBatchLimit returns correct values for different SMB commands")
    public void testBatchLimitForDifferentCommands() throws CIFSException {
        Properties props = new Properties();

        case SMB_COM_DELETE_DIRECTORY:
            return cfg.getBatchLimit("TreeConnectAndX.DeleteDirectory");
        case SMB_COM_OPEN_ANDX:
            return cfg.getBatchLimit("TreeConnectAndX.OpenAndX");
        case SMB_COM_RENAME:
            return cfg.getBatchLimit("TreeConnectAndX.Rename");
        case SMB_COM_TRANSACTION:
            return cfg.getBatchLimit("TreeConnectAndX.Transaction");

    }

    /**
     * Test getBatchLimit with SMB_COM_READ_ANDX command.
     */
    @Test
    void testGetBatchLimit_ReadAndX() {
        smbComOpenAndX = new SmbComOpenAndX(fileName, access, 0, andx);
        assertEquals(1, smbComOpenAndX.getBatchLimit(ServerMessageBlock.SMB_COM_READ_ANDX));
    }

    /**
     * Test getBatchLimit with a command other than SMB_COM_READ_ANDX.
     */
    @Test

        this.writeMode = writeMode;
    }

    @Override
    protected int getBatchLimit(final Configuration cfg, final byte cmd) {
        if (cmd == SMB_COM_READ_ANDX) {
            return cfg.getBatchLimit("WriteAndX.ReadAndX");
        }
        if (cmd == SMB_COM_CLOSE) {
            return cfg.getBatchLimit("WriteAndX.Close");
        }
        return 0;
    }

    @Override

        this.fid = fid;
        this.offset = offset;
        this.maxCount = this.minCount = maxCount;
    }

    @Override
    protected int getBatchLimit(final Configuration cfg, final byte cmd) {
        return cmd == SMB_COM_CLOSE ? cfg.getBatchLimit("ReadAndX.Close") : 0;
    }

    @Override
    protected int writeParameterWordsWireFormat(final byte[] dst, int dstIndex) {
        final int start = dstIndex;

            } else {
                openFunction = OPEN_FN_CREATE | OPEN_FN_OPEN;
            }
        } else {
            openFunction = OPEN_FN_OPEN;
        }
    }

    @Override
    int getBatchLimit(final byte command) {
        return command == SMB_COM_READ_ANDX ? BATCH_LIMIT : 0;
    }

    @Override
    int writeParameterWordsWireFormat(final byte[] dst, int dstIndex) {
        final int start = dstIndex;

            }
        } else {
            throw new SmbException("Unsupported");
        }
    }

    @Override
    protected int getBatchLimit(final Configuration cfg, final byte cmd) {
        return cmd == SMB_COM_TREE_CONNECT_ANDX ? cfg.getBatchLimit("SessionSetupAndX.TreeConnectAndX") : 0;
    }

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

            }
        } else {
            this.openFunction = OPEN_FN_OPEN;
        }
    }

    @Override
    protected int getBatchLimit(final Configuration cfg, final byte cmd) {
        return cmd == SMB_COM_READ_ANDX ? cfg.getBatchLimit("OpenAndX.ReadAndX") : 0;
    }

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

            mockConfig.getListCount();
            mockConfig.getListSize();
            mockConfig.getAttributeCacheTimeout();
            mockConfig.isIgnoreCopyToException();
            mockConfig.getBatchLimit("cmd");
            mockConfig.getNotifyBufferSize();
            mockConfig.getMaxRequestRetries();
            mockConfig.isStrictResourceLifecycle();
            mockConfig.isTraceResourceUsage();

    void setParam(final int fid, final long offset, final int maxCount) {
        this.fid = fid;
        this.offset = offset;
        this.maxCount = minCount = maxCount;
    }

    @Override
    int getBatchLimit(final byte command) {
        return command == SMB_COM_CLOSE ? BATCH_LIMIT : 0;
    }

    @Override
    int writeParameterWordsWireFormat(final byte[] dst, int dstIndex) {
        final int start = dstIndex;