diff --git a/tests/test_dashboard_upload.py b/tests/test_dashboard_upload.py index 2caafef551e5598433832eed6e20e33bc6621ff0..d7369193a17ea39ab8158e312fb598e547f0f4a1 100644 --- a/tests/test_dashboard_upload.py +++ b/tests/test_dashboard_upload.py @@ -4,19 +4,25 @@ from dashboards.upload import load_config_from_env from unittest.mock import patch import os + def test_load_config_from_env(): # Case 1: Test if function raises exception for missing GRAFANA_API_KEY with pytest.raises(ValueError) as e: - load_config_from_env(env_path=".env") - assert str(e.value) == "GRAFANA_API_KEY is None or not defined in the .env file" + load_config_from_env(env_path="") + assert str( + e.value) == "GRAFANA_API_KEY is None or not defined in the .env file" # Case 2: Test if function raises exception for missing GRAFANA_SERVER with patch.dict(os.environ, {"GRAFANA_API_KEY": "api_key"}): with pytest.raises(ValueError) as e: - load_config_from_env(env_path=".env") - assert str(e.value) == "GRAFANA_SERVER is None or not defined in the .env file" + load_config_from_env(env_path="") + assert str( + e.value) == "GRAFANA_SERVER is None or not defined in the .env file" # Case 3: Test if function returns expected values when both variables are defined - with patch.dict(os.environ, {"GRAFANA_API_KEY": "api_key", "GRAFANA_SERVER": "server_url"}): + with patch.dict( + os.environ, {"GRAFANA_API_KEY": "api_key", + "GRAFANA_SERVER": "server_url"} + ): result = load_config_from_env(env_path=".env") - assert result == ("server_url", "api_key") \ No newline at end of file + assert result == ("server_url", "api_key")