RPGKata/test_rpg.py at main · okiwi/RPGKata · GitHub

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


class Character:
    def __init__(self):
        self.damage_taken = 0
        self.max_health = 1000
        self.health = self.max_health

    @property
    def is_alive(self) -> bool:
        return self.health > 0

    @property
    def level(self) -> int:
        return self.damage_taken // 1000 + 1

    def attack(self, targeted_character: Character, damage: int):
        if targeted_character is self:
            print("PLEASE DO NOT HARM YOURSELF")
            return
        targeted_character.health = max(targeted_character.health - damage,0)
        targeted_character.damage_taken += damage


    def heal(self, health: int):
        if not self.is_alive: return
        self.health = min(self.health + health, self.max_health)

def test_character_exists():

    character = Character()

    assert character.health == 1000
    assert character.is_alive
    assert character.level == 1

def test_character_takes_damage():

    character1 = Character()
    character2 = Character()

    character1.attack(character2, 500)

    assert character2.health == 500
    assert character2.is_alive


def test_character_dies_after_damage():

    character1 = Character()
    character2 = Character()

    character1.attack(character2, 1000)

    assert character2.health == 0
    assert not character2.is_alive

def test_character_dead_has_health_zero():

    character1 = Character()
    character2 = Character()

    character1.attack(character2, 1500)

    assert character2.health == 0

def test_is_character_healed_correctly():

    character1 = Character()
    character2 = Character()

    character1.attack(character2, 500)
    character2.heal(1000)

    assert character2.health == 1000

def test_leveling_up():

    character1 = Character()
    character2 = Character()

    character1.attack(character2, 500)
    character2.heal(1000)
    character1.attack(character2, 500)

    assert character2.level == 2

def test_character_cant_hurt_self():
    # arrange
    character1 = Character()
    # act
    character1.attack(character1, 500)
    # assert
    assert character1.health == 1000 # you can't attack yourself

def test_character_can_not_heal_when_is_dead():
    # arrange
    character1 = Character()
    character2 = Character()
    character1.attack(character2, 1000)
    # act
    character2.heal(500)

    # assert
    assert character2.health == 0
    assert not character2.is_alive