test: change test for ubuntu testing

GianmarcoBasile · Jan 28, 2024 · a5a1dfb · a5a1dfb
1 parent da49beb
commit a5a1dfb
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_main.py b/tests/test_main.py
@@ -1,8 +1,11 @@
 import unittest
+from unittest.mock import patch
 from main import bot_instance as bot
 from bot import Bot
 
 
 class TestMain(unittest.TestCase):
     def test_main(self):
+        with patch("src.main.API_KEY") as mock_api_key:
+            mock_api_key.return_value = "test_api_key"
         self.assertTrue(isinstance(bot, Bot))