BWAPI
|
00001 #pragma once 00002 #include "TestCase.h" 00003 #include <BWAPI.h> 00004 00005 class CancelTrainTest : public TestCase 00006 { 00007 public: 00008 CancelTrainTest(BWAPI::UnitType unitType1,BWAPI::UnitType unitType2,BWAPI::UnitType unitType3); 00009 virtual void start(); 00010 virtual void update(); 00011 virtual void stop(); 00012 private: 00013 bool verifyTrainingQueue(); 00014 BWAPI::UnitType unitType1; 00015 BWAPI::UnitType unitType2; 00016 BWAPI::UnitType unitType3; 00017 BWAPI::UnitType producerType; 00018 BWAPI::Unit* producer; 00019 int startFrame; 00020 int nextFrame; 00021 int correctMineralCount; 00022 int correctGasCount; 00023 int correctSupplyUsedCount; 00024 int originalMineralCount; 00025 int originalGasCount; 00026 int originalSupplyUsedCount; 00027 int originalAllUnit1Count; 00028 int originalAllUnit2Count; 00029 int originalAllUnit3Count; 00030 std::list<BWAPI::UnitType> correctTrainingQueue; 00031 enum State 00032 { 00033 Start, 00034 CancelledFirstSlot, 00035 CancelledLast1, 00036 CancelledLast2 00037 } state; 00038 };