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
|
/*
* BattleFake.h, 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
*
*/
#pragma once
#include <vstd/RNG.h>
#include "mock_BonusBearer.h"
#include "mock_battle_IBattleState.h"
#include "mock_battle_Unit.h"
#if SCRIPTING_ENABLED
#include "mock_scripting_Pool.h"
#endif
#include "../../lib/battle/CBattleInfoCallback.h"
namespace test
{
namespace battle
{
using namespace ::testing;
using ::battle::Unit;
using ::battle::Units;
using ::battle::UnitFilter;
class UnitFake : public UnitMock
{
public:
void addNewBonus(const std::shared_ptr<Bonus> & b);
void makeAlive();
void makeDead();
void redirectBonusesToFake();
void expectAnyBonusSystemCall();
private:
BonusBearerMock bonusFake;
};
class UnitsFake
{
public:
std::vector<std::shared_ptr<UnitFake>> allUnits;
UnitFake & add(BattleSide side);
battle::Units getUnitsIf(battle::UnitFilter predicate) const;
void setDefaultBonusExpectations();
};
class BattleFake : public CBattleInfoCallback, public BattleStateMock
{
#if SCRIPTING_ENABLED
std::shared_ptr<scripting::PoolMock> pool;
#endif
public:
#if SCRIPTING_ENABLED
BattleFake(std::shared_ptr<scripting::PoolMock> pool_);
#else
BattleFake();
#endif
virtual ~BattleFake();
void setUp();
void setupEmptyBattlefield();
#if SCRIPTING_ENABLED
scripting::Pool * getContextPool() const override;
#endif
template <typename T>
void accept(T & pack)
{
pack.applyBattle(this);
}
const IBattleInfo * getBattle() const override
{
return nullptr;
}
std::optional<PlayerColor> getPlayerID() const override
{
return std::nullopt;
}
protected:
private:
};
}
}
|
