#106 added few more tests

bomzheg · Nov 6, 2024 · 16888aa · 16888aa
1 parent abdf96c
commit 16888aa
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests/unit/domain/conditions_test.py b/tests/unit/domain/conditions_test.py
@@ -25,3 +25,10 @@ def test_create_empty_condition():
 def test_create_only_bonus_condition():
     with pytest.raises(exceptions.LevelError):
         scn.Conditions([action.KeyBonusCondition({keys.BonusKey(text="SH123", bonus_minutes=1)})])
+
+def test_conditions_get_keys():
+    c = scn.Conditions([
+        action.KeyWinCondition({keys.SHKey("SH123"), keys.SHKey("SH321")}),
+        action.KeyWinCondition({keys.SHKey("СХ123")})
+    ])
+    assert c.get_keys() == {keys.SHKey("SH123"), keys.SHKey("SH321"), keys.SHKey("СХ123")}