More work with cleanup of unit tests

evennia · Dec 2, 2023 · 0f39677 · 0f39677
1 parent 48fdc2c
commit 0f39677
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/evennia/contrib/rpg/llm/tests.py b/evennia/contrib/rpg/llm/tests.py
@@ -6,13 +6,13 @@
 from anything import Something
 from django.test import override_settings
 from evennia.utils.create import create_object
-from evennia.utils.test_resources import EvenniaTestCase
+from evennia.utils.test_resources import BaseEvenniaTestCase
 from mock import Mock, patch
 
 from .llm_npc import LLMNPC
 
 
-class TestLLMClient(EvenniaTestCase):
+class TestLLMClient(BaseEvenniaTestCase):
     @override_settings(LLM_PROMPT_PREFIX="You are a test bot.", TEST_ENVIRONMENT=True)
     @patch("evennia.contrib.rpg.llm.llm_npc.task.deferLater")
     def test_npc_at_talked_to(self, mock_deferLater):

diff --git a/evennia/utils/test_resources.py b/evennia/utils/test_resources.py
@@ -97,7 +97,7 @@
     BASE_GUEST_TYPECLASS="evennia.accounts.accounts.DefaultGuest",
     # a special setting boolean TEST_ENVIRONMENT is set by the test runner
     # while the test suite is running.
-    TEST_ENVIRONMENT=True
+    TEST_ENVIRONMENT=True,
 )
 
 DEFAULT_SETTINGS = {**all_from_module(settings_default), **DEFAULT_SETTING_RESETS}
@@ -548,6 +548,9 @@ class BaseEvenniaTestCase(TestCase):
 
     """
 
+    def tearDown(self) -> None:
+        flush_cache()
+
 
 class EvenniaTestCase(TestCase):
     """