Add (partial) Invulnerability

Author: miketwo

effect_catalog.py

@@ -71,6 +71,30 @@ def tick(self):
             self.amount -= 1
 
 
+class Invulnerable(Effect):
+    '''Prevents all damage. Great for testing!'''
+    registers = [Message.BEFORE_ATTACK]
+    priorities = [100]   # Called last so it can cancel all other effects.
+
+    def __init__(self, host, amount):
+        super().__init__(
+            host,
+            "Invulnerable",
+            StackType.DURATION_AND_INTENSITY,
+            EffectType.BUFF,
+            "Prevents all damage.",
+            amount,
+        )
+
+    def callback(self, message, data):
+        if message == Message.BEFORE_ATTACK:
+            attacker, target, damage_dealer = data
+            if self.host == target:
+                damage_dealer.set_damage(0, "<buff>,.-~*´¨¯¨`*·~-.¸-INVULNERABLE-,.-~*´¨¯¨`*·~-.¸</buff>")
+            else:
+                pass
+
+
 class Strength(Effect):
     registers = [Message.BEFORE_ATTACK]
 

entities.py

@@ -9,6 +9,10 @@ def modify_damage(self, change: int, context: str, *args, **kwargs):
         new_dmg = self.damage + change
         ansiprint(f"Damage modified from {self.damage} --> {new_dmg} by {context}.")
         self.damage = new_dmg
+    def set_damage(self, new_dmg: int, context: str, *args, **kwargs):
+        ansiprint(f"Damage modified from {self.damage} --> {new_dmg} by {context}.")
+        self.damage = new_dmg
+
 
 class Action:
     def __init__(self, name, action, amount):

tests/test_game.py

@@ -11,6 +11,7 @@
 from ansi_tags import ansiprint
 from definitions import CardType
 from tests.fixtures import sleepless
+import effect_interface
 
 
 def replacement_clear_screen():
@@ -97,21 +98,21 @@ def patched_input(*args, **kwargs):
 
 
 @pytest.mark.timeout(10)
-@pytest.mark.parametrize("seed", list(range(10)))
+@pytest.mark.parametrize("seed", list(range(30)))
 def test_e2e(seed, monkeypatch, sleepless):
     '''Test the game from start to finish
     Plays with (more or less) random inputs to test the game.
     Seems to find lots of bugs, but very hard to repeat.
     '''
     ansiprint(f"<red><bold>Seed for this run is: {seed}</bold></red>")
     mygame = game.Game(seed=seed)
+    effect_interface.apply_effect(mygame.player, None, "Invulnerable", 1)   # IDDQD
     with monkeypatch.context() as m:
         m.setattr('builtins.input', autoplayer(mygame))
         displayer.clear = replacement_clear_screen
 
         try:
             start = time.time()
-            assert len(mygame.bus.subscribers) == 0
             mygame.start()
         except Exception as e:
             ansiprint(f"<red><bold>Failed with seed: {seed}</bold></red>")