From f8c6ca2797eefec6a013c802d5185842893b9cbb Mon Sep 17 00:00:00 2001
From: Avalon Hope <96471977+avalonhope@users.noreply.github.com>
Date: Wed, 22 Jun 2022 09:38:57 +0100
Subject: [PATCH] Unit test for search_script_attribute

---
 evennia/utils/tests/test_search.py | 19 ++++++++++++++++++-
 1 file changed, 18 insertions(+), 1 deletion(-)

diff --git a/evennia/utils/tests/test_search.py b/evennia/utils/tests/test_search.py
index ae9b9a9694..f9ab2e2e4a 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_tag
+from evennia.utils.search import search_script_attribute, search _script_tag
 
 class TestSearch(EvenniaTest):
 
@@ -33,4 +33,21 @@ class TestSearch(EvenniaTest):
         script.tags.add("a-tag", category="a-category")
         found = search_script_tag("wrong-tag", category="a-category")
         self.assertEqual(len(found), 0, errors)
+        
+    def test_search_script_attribute(self):
+        """Check that a script can be found by its attributes."""
+        script, errors = DefaultScript.create("a-script")
+        script.db.an_attribute = "some value"
+        found = search_script_attribute(key="an_attribute", value="some value")
+        self.assertEqual(len(found), 1, errors)
+        self.assertEqual(script.key, found[0].key, errors)
+        
+    def test_search_script_attribute_wrong(self):
+        """Check that a script can be found by its attributes."""
+        script, errors = DefaultScript.create("a-script")
+        script.db.an_attribute = "some value"
+        found = search_script_attribute(key="an_attribute", value="wrong value")
+        self.assertEqual(len(found), 1, errors)
+        self.assertEqual(script.key, found[0].key, errors)
+