diff --git a/tests/unit/domain/conditions_test.py b/tests/unit/domain/conditions_test.py index 7df306fa..0ea91b38 100644 --- 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")}