vcmi/test/battle/CHealthTest.cpp

/*
 * CHealthTest.cpp, part of VCMI engine
 *
 * Authors: listed in file AUTHORS in main folder
 *
 * License: GNU General Public License v2.0 or later
 * Full text of license available in license.txt file, in main folder
 *
 */

#include "StdInc.h"
#include <boost/test/unit_test.hpp>
#include "../../lib/CStack.h"

static const int32_t UNIT_HEALTH = 123;
static const int32_t UNIT_AMOUNT = 300;

class CUnitHealthInfoMock : public IUnitHealthInfo
{
public:
	CUnitHealthInfoMock():
		maxHealth(UNIT_HEALTH),
		baseAmount(UNIT_AMOUNT),
		health(this)
	{
		health.init();
	}

	int32_t maxHealth;
	int32_t baseAmount;

	CHealth health;

	int32_t unitMaxHealth() const override
	{
		return maxHealth;
	};

	int32_t unitBaseAmount() const override
	{
		return baseAmount;
	};
};

static void checkTotal(const CHealth & health, const CUnitHealthInfoMock & mock)
{
	BOOST_CHECK_EQUAL(health.total(), mock.maxHealth * mock.baseAmount);
}

static void checkEmptyHealth(const CHealth & health, const CUnitHealthInfoMock & mock)
{
	checkTotal(health, mock);
	BOOST_CHECK_EQUAL(health.getCount(), 0);
	BOOST_CHECK_EQUAL(health.getFirstHPleft(), 0);
	BOOST_CHECK_EQUAL(health.getResurrected(), 0);
	BOOST_CHECK_EQUAL(health.available(), 0);
}

static void checkFullHealth(const CHealth & health, const CUnitHealthInfoMock & mock)
{
	checkTotal(health, mock);
	BOOST_CHECK_EQUAL(health.getCount(), mock.baseAmount);
	BOOST_CHECK_EQUAL(health.getFirstHPleft(), mock.maxHealth);
	BOOST_CHECK_EQUAL(health.getResurrected(), 0);
	BOOST_CHECK_EQUAL(health.available(), mock.maxHealth * mock.baseAmount);
}

static void checkDamage(CHealth & health, const int32_t initialDamage, const int32_t expectedDamage)
{
	int32_t damage = initialDamage;
	health.damage(damage);
	BOOST_CHECK_EQUAL(damage, expectedDamage);
}

static void checkNormalDamage(CHealth & health, const int32_t initialDamage)
{
	checkDamage(health, initialDamage, initialDamage);
}

static void checkNoDamage(CHealth & health, const int32_t initialDamage)
{
	checkDamage(health, initialDamage, 0);
}

static void checkHeal(CHealth & health, EHealLevel level, EHealPower power, const int32_t initialHeal, const int32_t expectedHeal)
{
	int32_t heal = initialHeal;
	health.heal(heal, level, power);
	BOOST_CHECK_EQUAL(heal, expectedHeal);
}

BOOST_AUTO_TEST_SUITE(CHealthTest_Suite)

BOOST_AUTO_TEST_CASE(empty)
{
	CUnitHealthInfoMock uhi;
	CHealth health(&uhi);
	checkEmptyHealth(health, uhi);

	health.init();
	checkFullHealth(health, uhi);

	health.reset();
	checkEmptyHealth(health, uhi);
}

BOOST_FIXTURE_TEST_CASE(damage, CUnitHealthInfoMock)
{
	checkNormalDamage(health, 0);
	checkFullHealth(health, *this);

	checkNormalDamage(health, maxHealth - 1);
	BOOST_CHECK_EQUAL(health.getCount(), UNIT_AMOUNT);
	BOOST_CHECK_EQUAL(health.getFirstHPleft(), 1);
	BOOST_CHECK_EQUAL(health.getResurrected(), 0);

	checkNormalDamage(health, 1);
	BOOST_CHECK_EQUAL(health.getCount(), UNIT_AMOUNT - 1);
	BOOST_CHECK_EQUAL(health.getFirstHPleft(), UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getResurrected(), 0);

	checkNormalDamage(health, UNIT_HEALTH * (UNIT_AMOUNT - 1));
	checkEmptyHealth(health, *this);

	checkNoDamage(health, 1337);
	checkEmptyHealth(health, *this);
}

BOOST_FIXTURE_TEST_CASE(heal, CUnitHealthInfoMock)
{
	checkNormalDamage(health, 99);
	BOOST_CHECK_EQUAL(health.getCount(), UNIT_AMOUNT);
	BOOST_CHECK_EQUAL(health.getFirstHPleft(), UNIT_HEALTH-99);
	BOOST_CHECK_EQUAL(health.getResurrected(), 0);

	checkHeal(health, EHealLevel::HEAL, EHealPower::PERMANENT, 9, 9);
	BOOST_CHECK_EQUAL(health.getCount(), UNIT_AMOUNT);
	BOOST_CHECK_EQUAL(health.getFirstHPleft(), UNIT_HEALTH-90);
	BOOST_CHECK_EQUAL(health.getResurrected(), 0);

	checkHeal(health, EHealLevel::RESURRECT, EHealPower::ONE_BATTLE, 40, 40);
	BOOST_CHECK_EQUAL(health.getCount(), UNIT_AMOUNT);
	BOOST_CHECK_EQUAL(health.getFirstHPleft(), UNIT_HEALTH-50);
	BOOST_CHECK_EQUAL(health.getResurrected(), 0);

	checkHeal(health, EHealLevel::OVERHEAL, EHealPower::PERMANENT, 50, 50);
	checkFullHealth(health, *this);
}

BOOST_FIXTURE_TEST_CASE(resurrectOneBattle, CUnitHealthInfoMock)
{
	checkNormalDamage(health, UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getCount(), UNIT_AMOUNT - 1);
	BOOST_CHECK_EQUAL(health.getFirstHPleft(), UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getResurrected(), 0);

	checkHeal(health, EHealLevel::RESURRECT, EHealPower::ONE_BATTLE, UNIT_HEALTH, UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getCount(), UNIT_AMOUNT);
	BOOST_CHECK_EQUAL(health.getFirstHPleft(), UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getResurrected(), 1);

	checkNormalDamage(health, UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getCount(), UNIT_AMOUNT - 1);
	BOOST_CHECK_EQUAL(health.getFirstHPleft(), UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getResurrected(), 0);

	health.init();

	checkNormalDamage(health, UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getCount(), UNIT_AMOUNT - 1);
	BOOST_CHECK_EQUAL(health.getFirstHPleft(), UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getResurrected(), 0);

	health.takeResurrected();
	BOOST_CHECK_EQUAL(health.getCount(), UNIT_AMOUNT - 1);
	BOOST_CHECK_EQUAL(health.getFirstHPleft(), UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getResurrected(), 0);

	health.init();

	checkNormalDamage(health, UNIT_HEALTH * UNIT_AMOUNT);
	checkEmptyHealth(health, *this);

	checkHeal(health, EHealLevel::RESURRECT, EHealPower::ONE_BATTLE, UNIT_HEALTH * UNIT_AMOUNT, UNIT_HEALTH * UNIT_AMOUNT);
	BOOST_CHECK_EQUAL(health.getCount(), UNIT_AMOUNT);
	BOOST_CHECK_EQUAL(health.getFirstHPleft(), UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getResurrected(), UNIT_AMOUNT);

	health.takeResurrected();
	checkEmptyHealth(health, *this);
}

BOOST_FIXTURE_TEST_CASE(resurrectPermanent, CUnitHealthInfoMock)
{
	checkNormalDamage(health, UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getCount(), UNIT_AMOUNT - 1);
	BOOST_CHECK_EQUAL(health.getFirstHPleft(), UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getResurrected(), 0);

	checkHeal(health, EHealLevel::RESURRECT, EHealPower::PERMANENT, UNIT_HEALTH, UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getCount(), UNIT_AMOUNT);
	BOOST_CHECK_EQUAL(health.getFirstHPleft(), UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getResurrected(), 0);

	checkNormalDamage(health, UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getCount(), UNIT_AMOUNT - 1);
	BOOST_CHECK_EQUAL(health.getFirstHPleft(), UNIT_HEALTH);
	BOOST_CHECK_EQUAL(health.getResurrected(), 0);

	health.init();

	checkNormalDamage(health, UNIT_HEALTH * UNIT_AMOUNT);
	checkEmptyHealth(health, *this);

	checkHeal(health, EHealLevel::RESURRECT, EHealPower::PERMANENT, UNIT_HEALTH * UNIT_AMOUNT, UNIT_HEALTH * UNIT_AMOUNT);
	checkFullHealth(health, *this);

	health.takeResurrected();
	checkFullHealth(health, *this);
}

BOOST_FIXTURE_TEST_CASE(singleUnitStack, CUnitHealthInfoMock)
{
	//related to issue 2612

	//one Titan
	baseAmount = 1;
	maxHealth = 300;

	health.init();

	checkDamage(health, 1000, 300);
	checkEmptyHealth(health, *this);

	checkHeal(health, EHealLevel::RESURRECT, EHealPower::PERMANENT, 300, 300);
	checkFullHealth(health, *this);
}

BOOST_AUTO_TEST_SUITE_END()