chore(config): alert when env=prod

cal-itp · Aug 24, 2023 · de9e3a6 · de9e3a6
1 parent 8df63c5
commit de9e3a6
Show file tree

Hide file tree

Showing 3 changed files with 22 additions and 8 deletions.
diff --git a/littlepay/commands/configure.py b/littlepay/commands/configure.py
@@ -1,7 +1,7 @@
 from pathlib import Path
 
 from littlepay.commands import RESULT_FAILURE, RESULT_SUCCESS
-from littlepay.config import Config
+from littlepay.config import ENV_PROD, Config
 
 
 def configure(config_path: str | Path = None, reset: bool = False) -> int:
@@ -32,8 +32,9 @@ def configure(config_path: str | Path = None, reset: bool = False) -> int:
         credentials = None
 
     if credentials is None:
-        print(f"Active: {env}, {participant} [missing credentials]")
+        print(f"Active: {env}, {participant} [missing credentials]".strip())
         return RESULT_FAILURE
     else:
-        print(f"Active: {env}, {participant}")
+        alert = "⚠️  " if env == ENV_PROD else ""
+        print(f"Active: {alert}{env}, {participant}".strip())
         return RESULT_SUCCESS
diff --git a/tests/commands/test_configure.py b/tests/commands/test_configure.py
@@ -4,7 +4,7 @@
 
 from littlepay.commands import RESULT_FAILURE, RESULT_SUCCESS
 from littlepay.commands.configure import configure
-from littlepay.config import DEFAULT_CREDENTIALS
+from littlepay.config import DEFAULT_CREDENTIALS, ENV_PROD
 
 
 @pytest.fixture
@@ -73,7 +73,7 @@ def test_configure_participant_credentials_missing(mock_Config, capfd):
     assert "Active: env123, participant123 [missing credentials]" in capture.out
 
 
-def test_configure_participant_no_credentials(mock_Config, capfd):
+def test_configure_participant_credentials_None(mock_Config, capfd):
     mock_Config.active_env_name.return_value = "env123"
     mock_Config.active_participant_id.return_value = "participant123"
     mock_Config.active_credentials.return_value = None
@@ -88,6 +88,19 @@ def test_configure_participant_no_credentials(mock_Config, capfd):
     assert "Active: env123, participant123 [missing credentials]" in capture.out
 
 
+def test_configure_participant_credentials_prod(mock_Config, capfd):
+    mock_Config.active_env_name.return_value = ENV_PROD
+    mock_Config.active_participant_id.return_value = "participant123"
+    mock_Config.active_credentials.return_value = DEFAULT_CREDENTIALS
+
+    res = configure()
+    capture = capfd.readouterr()
+
+    assert res == RESULT_SUCCESS
+    assert "Active: ⚠️  prod, participant123" in capture.out
+    assert "[missing credentials]" not in capture.out
+
+
 def test_configure_reset(custom_config_file: Path):
     content = "Custom config content written here"
     custom_config_file.write_text(content)

diff --git a/tests/test_littlepay.py b/tests/test_littlepay.py
@@ -1,6 +1,6 @@
 import subprocess
 
-from littlepay.commands import RESULT_SUCCESS
+from littlepay.commands import RESULT_FAILURE
 
 
 def test_littlepay(capfd):
@@ -11,7 +11,7 @@ def test_littlepay(capfd):
     assert "Envs:" in capture.out
     assert "Participants:" in capture.out
     assert "Active:" in capture.out
-    assert res == RESULT_SUCCESS
+    assert res == RESULT_FAILURE
 
 
 def test_config(capfd):
@@ -22,4 +22,4 @@ def test_config(capfd):
     assert "Envs:" in capture.out
     assert "Participants:" in capture.out
     assert "Active:" in capture.out
-    assert res == RESULT_SUCCESS
+    assert res == RESULT_FAILURE