feat: mock environment variables in tests and update SYMBOL_VERSION type to string

Author: ImBIOS

sources/graphics_chart_drawer_test.py

@@ -3,8 +3,27 @@
 
 import pytest
 
+# Mock environment variables before importing the modules
+os.environ["INPUT_GH_TOKEN"] = "mock_gh_token"
+os.environ["INPUT_WAKATIME_API_KEY"] = "mock_wakatime_key"
+
+
 # Import the function and constants we're testing
-from graphics_chart_drawer import GRAPH_PATH, MAX_LANGUAGES, create_loc_graph
+from graphics_chart_drawer import MAX_LANGUAGES  # noqa: E402
+from graphics_chart_drawer import GRAPH_PATH, create_loc_graph  # noqa: E402
+
+
+@pytest.fixture(autouse=True)
+def mock_environment():
+    """Fixture to ensure environment variables are set for all tests"""
+    with patch.dict(
+        os.environ,
+        {
+            "INPUT_GH_TOKEN": "mock_gh_token",
+            "INPUT_WAKATIME_API_KEY": "mock_wakatime_key",
+        },
+    ):
+        yield
 
 
 @pytest.fixture

sources/manager_environment.py

@@ -1,4 +1,4 @@
-from os import getenv, environ
+from os import environ, getenv
 
 
 class EnvironmentManager:
@@ -48,7 +48,7 @@ class EnvironmentManager:
     LOCALE = getenv("INPUT_LOCALE", "en")
     UPDATED_DATE_FORMAT = getenv("INPUT_UPDATED_DATE_FORMAT", "%d/%m/%Y %H:%M:%S")
     IGNORED_REPOS = getenv("INPUT_IGNORED_REPOS", "").replace(" ", "").split(",")
-    SYMBOL_VERSION = int(getenv("INPUT_SYMBOL_VERSION"))
+    SYMBOL_VERSION = getenv("INPUT_SYMBOL_VERSION", "1")
 
     DEBUG_LOGGING = getenv("INPUT_DEBUG_LOGGING", "0").lower() in _TRUTHY
     DEBUG_RUN = getenv("DEBUG_RUN", "False").lower() in _TRUTHY

sources/manager_environment_test.py

@@ -29,8 +29,8 @@ def test_list_variable(env_manager):
     assert isinstance(env_manager.IGNORED_REPOS, list)
 
 
-def test_integer_variable(env_manager):
-    assert isinstance(env_manager.SYMBOL_VERSION, int)
+def test_string_variable(env_manager):
+    assert isinstance(env_manager.SYMBOL_VERSION, str)
 
 
 def test_debugging_variables(env_manager):