BWAPI: trunk/bwapi/TestAIModule/Source/TestModule.h Source File

Go to the documentation of this file.

00001 #pragma once
00002 #include <BWAPI.h>
00003 #include "TestCase.h"
00004 class TestModule : public BWAPI::AIModule
00005 {
00006   public:
00007     TestModule();
00008     ~TestModule();
00009     virtual void onFrame();
00010     virtual void onUnitCreate(BWAPI::Unit* unit);
00011     virtual void onUnitDestroy(BWAPI::Unit* unit);
00012     void runTestCases();
00013     void addTestCase(TestCase* testCase);
00014     TestCase* getCurrentTestCase() const;
00015   private:
00016     std::list<TestCase*> testCases;
00017     TestCase* currentTestCase;
00018     int lastEndFrame;
00019 };