TEST_F(CampaignTest, testAircraftHandling)
{
const vec2_t destination = { 10, 10 };
base_t* base;
aircraft_t* aircraft;
aircraft_t* newAircraft;
aircraft_t* aircraftTemplate;
int firstIdx;
int initialCount;
int count;
int newFound;
base = CreateBase("unittestaircraft", destination);
ASSERT_TRUE(nullptr != base);
/** @todo we should not assume that initial base has aircraft. It's a campaign parameter */
aircraft = AIR_GetFirstFromBase(base);
ASSERT_TRUE(nullptr != aircraft);
/* aircraft should have a template */
aircraftTemplate = aircraft->tpl;
ASSERT_TRUE(nullptr != aircraftTemplate);
firstIdx = aircraft->idx;
initialCount = AIR_BaseCountAircraft(base);
/* test deletion (part 1) */
AIR_DeleteAircraft(aircraft);
count = AIR_BaseCountAircraft(base);
ASSERT_EQ(count, initialCount - 1);
/* test addition (part 1) */
newAircraft = AIR_NewAircraft(base, aircraftTemplate);
ASSERT_TRUE(nullptr != newAircraft);
count = AIR_BaseCountAircraft(base);
ASSERT_EQ(count, initialCount);
/* new aircraft assigned to the right base */
ASSERT_EQ(newAircraft->homebase, base);
newFound = 0;
AIR_Foreach(a) {
/* test deletion (part 2) */
ASSERT_NE(firstIdx, a->idx);
/* for test addition (part 2) */
if (a->idx == newAircraft->idx)
newFound++;
}
/* test addition (part 2) */
ASSERT_EQ(newFound, 1);
/* check if AIR_Foreach iterates through all aircraft */
AIR_Foreach(a) {
AIR_DeleteAircraft(a);
}
aircraft = AIR_GetFirstFromBase(base);
ASSERT_TRUE(nullptr == aircraft);
count = AIR_BaseCountAircraft(base);
ASSERT_EQ(count, 0);
/* cleanup for the following tests */
E_DeleteAllEmployees(nullptr);
base->founded = false;
}
/**
* @brief Initialises base.
* @note This command is executed in the init node of the base menu.
* It is called everytime the base menu pops up and sets the cvars.
* @todo integrate building status tooltips from B_CheckBuildingStatusForMenu_f into this one
*/
static void B_BaseInit_f (void)
{
int i;
base_t *base = B_GetCurrentSelectedBase();
if (!base)
return;
/* make sure the credits cvar is up-to-date */
CP_UpdateCredits(ccs.credits);
cgi->Cvar_SetValue("mn_base_num_aircraft", AIR_BaseCountAircraft(base));
/* activate or deactivate the aircraft button */
if (AIR_AircraftAllowed(base)) {
if (AIR_BaseHasAircraft(base))
cgi->UI_ExecuteConfunc("update_basebutton aircraft false \"%s\"", _("Aircraft management and crew equipment"));
else
cgi->UI_ExecuteConfunc("update_basebutton aircraft true \"%s\"", _("Buy or produce at least one aircraft first."));
} else {
cgi->UI_ExecuteConfunc("update_basebutton aircraft true \"%s\"", _("No Hangar functional in base."));
}
if (BS_BuySellAllowed(base))
cgi->UI_ExecuteConfunc("update_basebutton buysell false \"%s\"", _("Buy/Sell equipment, aircraft and UGV"));
else
cgi->UI_ExecuteConfunc("update_basebutton buysell true \"%s\"", va(_("No %s functional in base."), _("Storage")));
if (B_GetCount() > 1)
cgi->UI_ExecuteConfunc("update_basebutton transfer false \"%s\"", _("Transfer equipment, vehicles, aliens and employees to other bases"));
else
cgi->UI_ExecuteConfunc("update_basebutton transfer true \"%s\"", _("Build at least a second base to transfer equipment or personnel"));
if (RS_ResearchAllowed(base))
cgi->UI_ExecuteConfunc("update_basebutton research false \"%s\"", _("Research new technology"));
else
cgi->UI_ExecuteConfunc("update_basebutton research true \"%s\"", va(_("No %s functional in base."), _("Laboratory")));
if (PR_ProductionAllowed(base))
cgi->UI_ExecuteConfunc("update_basebutton production false \"%s\"", _("Produce equipment, aircraft and UGV"));
else
cgi->UI_ExecuteConfunc("update_basebutton production true \"%s\"", va(_("No %s functional in base."), _("Workshop")));
if (E_HireAllowed(base))
cgi->UI_ExecuteConfunc("update_basebutton hire false \"%s\"", _("Hire or dismiss employees"));
else
cgi->UI_ExecuteConfunc("update_basebutton hire true \"%s\"", va(_("No %s functional in base."), _("Living Quarters")));
if (AC_ContainmentAllowed(base))
cgi->UI_ExecuteConfunc("update_basebutton containment false \"%s\"", _("Manage captured aliens"));
else
cgi->UI_ExecuteConfunc("update_basebutton containment true \"%s\"", va(_("No %s functional in base."), _("Containment")));
if (HOS_HospitalAllowed(base))
cgi->UI_ExecuteConfunc("update_basebutton hospital false \"%s\"", _("Treat wounded soldiers and perform implant surgery"));
else
cgi->UI_ExecuteConfunc("update_basebutton hospital true \"%s\"", va(_("No %s functional in base."), _("Hospital")));
/*
* Gather data on current/max space for living quarters, storage, lab and workshop
* clear_bld_space ensures 0/0 data for facilities which may not exist in base
*/
cgi->UI_ExecuteConfunc("clear_bld_space");
for (i = 0; i < ccs.numBuildingTemplates; i++) {
const building_t* b = &ccs.buildingTemplates[i];
const baseCapacities_t capType = B_GetCapacityFromBuildingType(b->buildingType);
capacities_t cap;
if (capType == MAX_CAP)
continue;
/* Check if building matches one of our four types */
if (b->buildingType != B_QUARTERS && b->buildingType != B_STORAGE && b->buildingType != B_WORKSHOP && b->buildingType != B_LAB && b->buildingType != B_ANTIMATTER)
continue;
/* only show already researched buildings */
if (!RS_IsResearched_ptr(b->tech))
continue;
cap = *CAP_Get(base, capType);
assert(b->tpl);
const int count = B_GetNumberOfBuildingsInBaseByTemplate(base, b->tpl);
if (count < 1)
continue;
cgi->UI_ExecuteConfunc("show_bld_space \"%s\" \"%s\" %i %i %i %i", _(b->name), b->id, cap.cur, cap.max, count, b->tpl->maxCount);
}
/*
* Get the number of different employees in the base
* @todo: Get the number of injured soldiers if hospital exists
*/
cgi->UI_ExecuteConfunc("current_employees %i %i %i %i", E_CountHired(base, EMPL_SOLDIER), E_CountHired(base, EMPL_PILOT), E_CountHired(base, EMPL_SCIENTIST), E_CountHired(base, EMPL_WORKER));
/*
* List the first five aircraft in the base if they exist
*/
cgi->UI_ExecuteConfunc("clear_aircraft");
if (AIR_AircraftAllowed(base)) {
if (AIR_BaseHasAircraft(base)) {
i = 0;
AIR_ForeachFromBase(aircraft, base) {
if (i > 5)
break;
/*
* UI node should use global IDX to identify aircraft but it uses order of aircraft in base (i)
* See @todo in cp_aircraft_callbacks.c in AIR_AircraftSelect()
*/
cgi->UI_ExecuteConfunc("show_aircraft %i \"%s\" \"%s\" \"%s\" %i", i, aircraft->name, aircraft->id, AIR_AircraftStatusToName(aircraft), AIR_IsAircraftInBase(aircraft));
i++;
}
}
}
/* Get the research item closest to completion in the base if it exists */
cgi->UI_ExecuteConfunc("clear_research");
if (RS_ResearchAllowed(base)) {
const technology_t *closestTech = NULL;
double finished = -1;
for (i = 0; i < ccs.numTechnologies; i++) {
const technology_t *tech = RS_GetTechByIDX(i);
if (!tech)
continue;
if (tech->base != base)
continue;
if (tech->statusResearch == RS_RUNNING) {
const double percent = (1 - tech->time / tech->overallTime) * 100;
if (percent > finished) {
finished = percent;
closestTech = tech;
}
}
}
if (closestTech != NULL)
cgi->UI_ExecuteConfunc("show_research \"%s\" %i %3.0f", closestTech->name, closestTech->scientists, finished);
}
/* Get the production item closest to completion in the base if it exists */
cgi->UI_ExecuteConfunc("clear_production");
if (PR_ProductionAllowed(base)) {
const production_queue_t *queue = PR_GetProductionForBase(base);
if (queue->numItems > 0) {
const production_t *production = &queue->items[0];
cgi->UI_ExecuteConfunc("show_production \"%s\" %3.0f", PR_GetName(&production->data), PR_GetProgress(production) * 100);
}
}
}
static void testAircraftHandling (void)
{
const vec2_t destination = { 10, 10 };
campaign_t *campaign;
base_t *base;
aircraft_t *aircraft;
aircraft_t *newAircraft;
aircraft_t *aircraftTemplate;
int firstIdx;
int initialCount;
int count;
int newFound;
ResetCampaignData();
campaign = GetCampaign();
base = CreateBase("unittestaircraft", destination);
CU_ASSERT_PTR_NOT_NULL_FATAL(base);
/** @todo we should not assume that initial base has aircraft. It's a campaign parameter */
aircraft = AIR_GetFirstFromBase(base);
CU_ASSERT_PTR_NOT_NULL_FATAL(aircraft);
/* aircraft should have a template */
aircraftTemplate = aircraft->tpl;
CU_ASSERT_PTR_NOT_NULL_FATAL(aircraftTemplate);
firstIdx = aircraft->idx;
initialCount = AIR_BaseCountAircraft(base);
/* test deletion (part 1) */
AIR_DeleteAircraft(aircraft);
count = AIR_BaseCountAircraft(base);
CU_ASSERT_EQUAL(count, initialCount - 1);
/* test addition (part 1) */
newAircraft = AIR_NewAircraft(base, aircraftTemplate);
CU_ASSERT_PTR_NOT_NULL_FATAL(newAircraft);
count = AIR_BaseCountAircraft(base);
CU_ASSERT_EQUAL(count, initialCount);
/* new aircraft assigned to the right base */
CU_ASSERT_EQUAL(newAircraft->homebase, base);
newFound = 0;
AIR_Foreach(aircraft) {
/* test deletion (part 2) */
CU_ASSERT_NOT_EQUAL(firstIdx, aircraft->idx);
/* for test addition (part 2) */
if (aircraft->idx == newAircraft->idx)
newFound++;
}
/* test addition (part 2) */
CU_ASSERT_EQUAL(newFound, 1);
/* check if AIR_Foreach iterates through all aircraft */
AIR_Foreach(aircraft) {
AIR_DeleteAircraft(aircraft);
}
aircraft = AIR_GetFirstFromBase(base);
CU_ASSERT_PTR_NULL_FATAL(aircraft);
count = AIR_BaseCountAircraft(base);
CU_ASSERT_EQUAL(count, 0);
/* cleanup for the following tests */
E_DeleteAllEmployees(NULL);
base->founded = qfalse;
}
