참고소스 수정본

참고/guardrails-main/tests/unit_tests/cli/test_configure.py

@@ -0,0 +1,102 @@
+from unittest.mock import call, patch
+
+import pytest
+from tests.unit_tests.mocks.mock_file import MockFile
+
+
+@pytest.mark.parametrize(
+    "expected_token, enable_metrics, clear_token",
+    [
+        ("mock_token", True, False),
+        ("mock_token", False, False),
+        ("", True, True),
+        ("", False, True),
+    ],
+)
+def test_configure(mocker, runner, expected_token, enable_metrics, clear_token):
+    mock_save_configuration_file = mocker.patch(
+        "guardrails.cli.configure.save_configuration_file"
+    )
+    mock_logger_info = mocker.patch("guardrails.cli.configure.logger.info")
+    mock_get_auth = mocker.patch("guardrails.cli.configure.get_auth")
+
+    CLI_COMMAND = ["configure"]
+    CLI_COMMAND_ARGS = []
+    CLI_COMMAND_INPUTS = ["mock_token", "mock_input"]
+
+    # Patch sys.stdin with a StringIO object
+    from guardrails.cli.guardrails import guardrails
+
+    if enable_metrics:
+        CLI_COMMAND_ARGS.append("y")
+    else:
+        CLI_COMMAND_ARGS.append("n")
+
+    if clear_token:
+        CLI_COMMAND.append("--clear-token")
+
+    with patch("typer.prompt", side_effect=CLI_COMMAND_INPUTS):
+        result = runner.invoke(
+            guardrails,
+            CLI_COMMAND,
+            input="".join([f"{arg}\n" for arg in CLI_COMMAND_ARGS]),
+        )
+
+    assert result.exit_code == 0
+
+    expected_calls = [call("Configuration saved.")]
+
+    if clear_token:
+        expected_calls.append(call("No token provided. Skipping authentication."))
+        assert mock_get_auth.call_count == 0
+    else:
+        expected_calls.append(call("Validating credentials..."))
+        assert mock_get_auth.call_count == 1
+
+    assert mock_logger_info.call_count == 2
+    mock_logger_info.assert_has_calls(expected_calls)
+    mock_save_configuration_file.assert_called_once_with(
+        expected_token, enable_metrics, True
+    )
+
+
+def test_save_configuration_file(mocker):
+    expanduser_mock = mocker.patch("guardrails.cli.configure.expanduser")
+    expanduser_mock.return_value = "/Home"
+
+    rcexpanduser_mock = mocker.patch("guardrails.classes.rc.expanduser")
+    rcexpanduser_mock.return_value = "/Home"
+
+    import os
+
+    join_spy = mocker.spy(os.path, "join")
+
+    mock_file = MockFile()
+    mock_open = mocker.patch("guardrails.cli.configure.open")
+    mock_open.return_value = mock_file
+
+    mock_uuid = mocker.patch("guardrails.cli.configure.uuid.uuid4")
+    mock_uuid.return_value = "f49354e0-80c7-4591-81db-cc2f945e5f1e"
+
+    writelines_spy = mocker.spy(mock_file, "writelines")
+    close_spy = mocker.spy(mock_file, "close")
+
+    from guardrails.cli.configure import save_configuration_file
+
+    save_configuration_file("token", True)
+
+    assert expanduser_mock.called is True
+    assert rcexpanduser_mock.called is True
+    join_spy.assert_called_with("/Home", ".guardrailsrc")
+    assert join_spy.call_count == 2
+
+    assert mock_open.call_count == 1
+    writelines_spy.assert_called_once_with(
+        [
+            f"id=f49354e0-80c7-4591-81db-cc2f945e5f1e{os.linesep}",
+            f"token=token{os.linesep}",
+            "enable_metrics=true\n",
+            "use_remote_inferencing=true",
+        ]
+    )
+    assert close_spy.call_count == 1