Refactor so submind may be defined after test case import

NeonGeckoCom · Nov 30, 2023 · c47e438 · c47e438
1 parent 76ee630
commit c47e438
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/neon_minerva/tests/chatbot_v1_test_base.py b/neon_minerva/tests/chatbot_v1_test_base.py
@@ -29,16 +29,20 @@
 
 
 class TestSubmind(unittest.TestCase):
-    # Determine submind to test
-    submind_entrypoint = getenv("TEST_BOT_ENTRYPOINT")
-    bot_class = load_chatbot(submind_entrypoint)
     # Initialize a server for testing
     server = MachKlatServer()
     sleep(1)
     socket = start_socket("0.0.0.0")
-    # Initialize a submind instance
-    submind = bot_class(socket, "Private", "testrunner", "testpassword",
-                        on_server=False)
+    submind = None
+
+    @classmethod
+    def setUpClass(cls) -> None:
+        # Determine submind to test
+        submind_entrypoint = getenv("TEST_BOT_ENTRYPOINT")
+        bot_class = load_chatbot(submind_entrypoint)
+        # Initialize a submind instance
+        cls.submind = bot_class(cls.socket, "Private", "testrunner",
+                                "testpassword", on_server=False)
 
     @classmethod
     def tearDownClass(cls) -> None: