Class TestTrainRoach

tests/test_unit_command.cc:1517–1563

    //

    class TestTrainRoach : public TestUnitCommandNoTarget {
    public:
        TestTrainRoach() {
            test_unit_type_ = UNIT_TYPEID::ZERG_HATCHERY;
            test_ability_ = ABILITY_ID::TRAIN_ROACH;
        }

        void SetTestTime() override {
            wait_game_loops_ = 150;
        }

        void SetOriginPoint() override {
            const GameInfo& game_info = agent_->Observation()->GetGameInfo();
            origin_pt_ = GetPointOffsetX(FindCenterOfMap(game_info), -10);
        }

        void AdditionalTestSetup() override {
            order_on_game_loop_ += 30;
            test_unit_type_ = UNIT_TYPEID::ZERG_LARVA;
        }

        void IssueUnitCommand(ActionInterface* act) override {
            if (ability_command_sent_) {
                return;
            }

            act->UnitCommand(test_unit_, test_ability_);
            act->UnitCommand(test_unit_, test_ability_);
            act->UnitCommand(test_unit_, test_ability_);
            ability_command_sent_ = true;
        }

        void OnTestFinish() override {
            const ObservationInterface* obs = agent_->Observation();
            Units roaches = obs->GetUnits(Unit::Self, [&](const Unit& unit) {
                return unit.unit_type == test_unit_type_;
            });

            if (roaches.size() != 3) {
                ReportError("The number of units trained was not correct.");
            }

            VerifyUnitExistsAndComplete(UNIT_TYPEID::ZERG_ROACH);
            VerifyUnitIdleAfterOrder(UNIT_TYPEID::ZERG_ROACH);
            KillAllUnits();
        }
    };


    //