diff --git a/evennia/utils/tests/test_search.py b/evennia/utils/tests/test_search.py index 63515d6754..a892203e0c 100644 --- a/evennia/utils/tests/test_search.py +++ b/evennia/utils/tests/test_search.py @@ -1,6 +1,6 @@ from evennia.scripts.scripts import DefaultScript from evennia.utils.test_resources import EvenniaTest -from evennia.utils.search import search_script_attribute, search_script_tag +from evennia.utils.search import search_script_attribute, search_script_tag, search_script class TestSearch(EvenniaTest): @@ -48,3 +48,17 @@ class TestSearch(EvenniaTest): script.db.an_attribute = "some value" found = search_script_attribute(key="an_attribute", value="wrong value") self.assertEqual(len(found), 0, errors) + + def test_search_script_key(self): + """Check that a script can be found by its key value.""" + script, errors = DefaultScript.create("a-script") + found = search_script("a-script") + self.assertEqual(len(found), 1, errors) + self.assertEqual(script.key, found[0].key, errors) + + def test_search_script_wrong_key(self): + """Check that a script cannot be found by a wrong key value.""" + script, errors = DefaultScript.create("a-script") + found = search_script("wrong_key") + self.assertEqual(len(found), 0, errors) +