[BattleCollector] Refactor test