make pytest and linter happy

QuantumChemist · Feb 9, 2025 · 20660b9 · 20660b9
1 parent 050291b
commit 20660b9
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 5 deletions.
diff --git a/.github/workflows/openglados.yml b/.github/workflows/openglados.yml
@@ -44,5 +44,6 @@ jobs:
         BOT_TOKEN: ${{ secrets.BOT_TOKEN }}
         CHICHI: ${{ secrets.CHICHI }}
         TENOR_API_KEY: ${{ secrets.TENOR_API_KEY }}
+        PYTEST_RUNNING: "1"
       run: |
         pytest
diff --git a/bot.py b/bot.py
@@ -35,7 +35,17 @@
     replace_mentions_with_display_names,
     generate_plot,
 )
-from variables import WHITELIST_GUILDS_ID, BLACKLIST_USERS_ID  # noqa: F401
+
+# Conditional import for testing
+if os.getenv("PYTEST_RUNNING"):
+    from unittest.mock import Mock
+    import sys
+
+    sys.modules["variables"] = Mock()
+    sys.modules["variables"].WHITELIST_GUILDS_ID = []
+    sys.modules["variables"].BLACKLIST_USERS_ID = []
+else:
+    from variables import WHITELIST_GUILDS_ID, BLACKLIST_USERS_ID  # noqa: F401
 
 # Directory to save screenshots
 SCREENSHOTS_DIR = "screenshots"
@@ -55,10 +65,10 @@
 # Load environment variables from .env file
 load_dotenv()
 
-if not BLACKLIST_USERS_ID:
-    BLACKLIST_USERS_ID = []
-if not WHITELIST_GUILDS_ID:
-    WHITELIST_GUILDS_ID = []
+if not sys.modules["variables"].BLACKLIST_USERS_ID:
+    sys.modules["variables"].BLACKLIST_USERS_ID = []
+if not sys.modules["variables"].WHITELIST_GUILDS_ID:
+    sys.modules["variables"].WHITELIST_GUILDS_ID = []
 
 
 # Define your custom bot class