1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
/*
* BattleFake.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 "BattleFake.h"
namespace test
{
namespace battle
{
void UnitFake::addNewBonus(const std::shared_ptr<Bonus> & b)
{
bonusFake.addNewBonus(b);
}
void UnitFake::makeAlive()
{
EXPECT_CALL(*this, alive()).Times(AtLeast(1)).WillRepeatedly(Return(true));
}
void UnitFake::makeDead()
{
EXPECT_CALL(*this, alive()).Times(AtLeast(1)).WillRepeatedly(Return(false));
EXPECT_CALL(*this, isGhost()).Times(AtLeast(1)).WillRepeatedly(Return(false));
}
void UnitFake::redirectBonusesToFake()
{
ON_CALL(*this, getAllBonuses(_, _, _)).WillByDefault(Invoke(&bonusFake, &BonusBearerMock::getAllBonuses));
ON_CALL(*this, getTreeVersion()).WillByDefault(Invoke(&bonusFake, &BonusBearerMock::getTreeVersion));
}
void UnitFake::expectAnyBonusSystemCall()
{
EXPECT_CALL(*this, getAllBonuses(_, _, _)).Times(AtLeast(0));
EXPECT_CALL(*this, getTreeVersion()).Times(AtLeast(0));
}
UnitFake & UnitsFake::add(BattleSide side)
{
auto * unit = new UnitFake();
ON_CALL(*unit, unitSide()).WillByDefault(Return(side));
unit->redirectBonusesToFake();
allUnits.emplace_back(unit);
return *allUnits.back().get();
}
Units UnitsFake::getUnitsIf(UnitFilter predicate) const
{
Units ret;
for(auto & unit : allUnits)
{
if(predicate(unit.get()))
ret.push_back(unit.get());
}
return ret;
}
void UnitsFake::setDefaultBonusExpectations()
{
for(auto & unit : allUnits)
{
unit->expectAnyBonusSystemCall();
}
}
BattleFake::~BattleFake() = default;
#if SCRIPTING_ENABLED
BattleFake::BattleFake(std::shared_ptr<scripting::PoolMock> pool_):
pool(pool_)
{
}
#else
BattleFake::BattleFake() = default;
#endif
void BattleFake::setUp()
{
}
void BattleFake::setupEmptyBattlefield()
{
EXPECT_CALL(*this, getDefendedTown()).WillRepeatedly(Return(nullptr));
EXPECT_CALL(*this, getAllObstacles()).WillRepeatedly(Return(IBattleInfo::ObstacleCList()));
EXPECT_CALL(*this, getBattlefieldType()).WillRepeatedly(Return(BattleField(0)));
}
#if SCRIPTING_ENABLED
scripting::Pool * BattleFake::getContextPool() const
{
return pool.get();
}
#endif
}
}
|
