Method start

bwapi/TestAIModule/Source/CancelTrainTest.cpp:39–129

  return true;
}
void CancelTrainTest::start()
{
  if (fail) return;
  running = true;

  int producerCount = Broodwar->self()->completedUnitCount(producerType);
  BWAssertF(producerCount>=1,{fail=true;return;});
  for (Unit u : Broodwar->self()->getUnits())
  {
    if (u->getType()==producerType)
    {
      producer = u;
      break;
    }
  }
  BWAssertF(producer!=NULL,{fail=true;return;});

  originalMineralCount = Broodwar->self()->minerals();
  originalGasCount = Broodwar->self()->gas();
  originalSupplyUsedCount = Broodwar->self()->supplyUsed();
  originalAllUnit1Count = Broodwar->self()->allUnitCount(unitType1);
  originalAllUnit2Count = Broodwar->self()->allUnitCount(unitType2);
  originalAllUnit3Count = Broodwar->self()->allUnitCount(unitType3);
  correctMineralCount = Broodwar->self()->minerals();
  correctGasCount = Broodwar->self()->gas();
  correctSupplyUsedCount = Broodwar->self()->supplyUsed();

  BWAssertF(producer->isTraining()==false,{fail=true;return;});
  BWAssertF(producer->isConstructing()==false,{fail=true;return;});
  BWAssertF(producer->isIdle()==true,{fail=true;return;});
  BWAssertF(producer->isLifted()==false,{fail=true;return;});
  BWAssertF(producer->getTrainingQueue().empty()==true,{fail=true;return;});
  BWAssertF(producer->getRemainingTrainTime() == 0,{fail=true;return;});
  BWAssertF(Broodwar->self()->allUnitCount(unitType1) == originalAllUnit1Count,{fail=true;return;});
  BWAssertF(Broodwar->self()->allUnitCount(unitType2) == originalAllUnit2Count,{fail=true;return;});
  BWAssertF(Broodwar->self()->allUnitCount(unitType3) == originalAllUnit3Count,{fail=true;return;});

  producer->train(unitType1);
  correctMineralCount = correctMineralCount - unitType1.mineralPrice();
  correctGasCount = correctGasCount - unitType1.gasPrice();
  correctSupplyUsedCount = correctSupplyUsedCount + unitType1.supplyRequired();
  correctTrainingQueue.push_back(unitType1);

  BWAssertF(verifyTrainingQueue(),{fail=true;return;});
  BWAssertF(producer->isTraining()==true,{fail=true;return;});
  BWAssertF(producer->isConstructing()==false,{fail=true;return;});
  BWAssertF(producer->isIdle()==false,{fail=true;return;});
  BWAssertF(producer->isLifted()==false,{fail=true;return;});
  BWAssertF(producer->getTrainingQueue().size()==1,{fail=true;return;});
  BWAssertF(producer->getRemainingTrainTime() == unitType1.buildTime(),{fail=true;return;});
  BWAssertF(Broodwar->self()->minerals() == correctMineralCount,{Broodwar->printf("%d != %d",Broodwar->self()->minerals(), correctMineralCount);fail=true;return;});
  BWAssertF(Broodwar->self()->gas() == correctGasCount,{fail=true;return;});
  BWAssertF(Broodwar->self()->supplyUsed() == correctSupplyUsedCount,{fail=true;return;});

  producer->train(unitType2);
  correctMineralCount = correctMineralCount - unitType2.mineralPrice();
  correctGasCount = correctGasCount - unitType2.gasPrice();
  correctTrainingQueue.push_back(unitType2);