* 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");

        assertNull(config.getMachineId());
    }

    @Test
    @DisplayName("Test getBatchLimit method")
    void testGetBatchLimit() {
        // Test default batch limit
        assertEquals(0, config.getBatchLimit("TreeConnectAndX.QueryInformation"));

        // Test unspecified batch limit
        assertEquals(1, config.getBatchLimit("UnknownCommand"));

        // Test caching behavior

    }

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

        assertNull(testBlock.getAndx());
    }

    @Test
    @DisplayName("Test getBatchLimit returns 0 by default")
    void testGetBatchLimit() {
        testBlock = new TestAndXServerMessageBlock(mockConfig);

        assertEquals(0, testBlock.getBatchLimit(mockConfig, (byte) 0x25));
    }

    @Test
    @DisplayName("Test encode without andx command")

        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;

        assertSame(next, block.andx, "andx should reference provided SMB");
    }

    @Test
    @DisplayName("getBatchLimit defaults to 0")
    void testGetBatchLimitDefault() {
        DummyAndXBlock block = new DummyAndXBlock();
        assertEquals(0, block.getBatchLimit((byte) 0x25));
    }

    @Test
    @DisplayName("encode without andx succeeds and signs when digest present")

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