@Override
            protected Integer doGetBatchLimit(String cmd) {
                if ("CustomCommand".equals(cmd)) {
                    return 5;
                }
                return null;
            }
        };

        assertEquals(5, customConfig.getBatchLimit("CustomCommand"));
        assertEquals(1, customConfig.getBatchLimit("UnknownCommand"));
    }

    @Test