Class TestCancelTrainingQueue

tests/test_unit_command.cc:463–512

    //

    class TestCancelTrainingQueue : public TestUnitCommandNoTarget {
    public:
        bool marine_trained_ = false;

        TestCancelTrainingQueue() {
            test_unit_type_ = UNIT_TYPEID::TERRAN_BARRACKS;
            test_ability_ = ABILITY_ID::CANCEL_LAST;
        }

        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 (!marine_trained_) {
                act->UnitCommand(test_unit_, ABILITY_ID::TRAIN_MARINE);
                marine_trained_ = true;
            }

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

            if (!orders_verified_ && marine_trained_) {
                VerifyUnitOrders(test_unit_, ABILITY_ID::TRAIN_MARINE);
            }

            IssueUnitCommand(act);
        }

        void OnTestFinish() override {
            VerifyUnitIdleAfterOrder(test_unit_type_);
            VerifyUnitDoesNotExist(UNIT_TYPEID::TERRAN_MARINE);
            KillAllUnits();
        }
    };


    //