Class TestCancelBuildInProgressFactory

tests/test_unit_command.cc:395–456

    //

    class TestCancelBuildInProgressFactory : public TestUnitCommandNoTarget {
    public:
        Unit test_factory_;
        bool factory_built_ = false;

        TestCancelBuildInProgressFactory() {
            test_unit_type_ = UNIT_TYPEID::TERRAN_SCV;
            test_ability_ = ABILITY_ID::CANCEL;
        }

        void SetTestTime() override {
            wait_game_loops_ = 100;
        }

        void OnStep() override {
            const ObservationInterface* obs = agent_->Observation();
            ActionInterface* act = agent_->Actions();

            if (obs->GetGameLoop() < order_on_game_loop_) {
                return;
            }

            const Units& units = obs->GetUnits(Unit::Alliance::Self);
            for (const Unit& unit : units) {
                if (unit.unit_type == test_unit_type_) {
                    test_unit_ = unit;
                }
            }

            if (!factory_built_) {
                act->UnitCommand(test_unit_, ABILITY_ID::BUILD_FACTORY, origin_pt_);
                factory_built_ = true;
            }

            for (const Unit& unit : units) {
                if (unit.unit_type == UNIT_TYPEID::TERRAN_FACTORY) {
                    test_factory_ = unit;
                }
            }

            if (obs->GetGameLoop() < order_on_game_loop_ + 10) {
                return;
            }

            if (!orders_verified_ && factory_built_) {
                VerifyUnitOrders(test_unit_, ABILITY_ID::BUILD_FACTORY);
            }

            if (!ability_command_sent_) {
                VerifyUnitExistsAndComplete(UNIT_TYPEID::TERRAN_FACTORY, false);
                act->UnitCommand(test_factory_, test_ability_);
                ability_command_sent_ = true;
            }

        }

        void OnTestFinish() override {
            VerifyUnitIdleAfterOrder(test_unit_type_);