From 60acc36fa1c1fb18461e70084af6e06274e8b055 Mon Sep 17 00:00:00 2001 From: Simon Suo Date: Sat, 12 Oct 2024 12:55:42 -0700 Subject: [PATCH] wip --- tests/cli/test_deploy.py | 4 ++-- tests/cli/test_status.py | 6 +++--- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/cli/test_deploy.py b/tests/cli/test_deploy.py index 4a34ac21..31090d1f 100644 --- a/tests/cli/test_deploy.py +++ b/tests/cli/test_deploy.py @@ -9,7 +9,7 @@ def test_deploy(runner: CliRunner, data_path: Path) -> None: test_config_file = data_path / "deployment.yaml" mocked_response = mock.MagicMock(status_code=200, json=lambda: {}) - with mock.patch("llama_deploy.cli.deploy.request") as mocked_httpx: + with mock.patch("llama_deploy.cli.utils.request") as mocked_httpx: mocked_httpx.return_value = mocked_response result = runner.invoke(llamactl, ["-t", "5.0", "deploy", str(test_config_file)]) @@ -29,7 +29,7 @@ def test_deploy_failed(runner: CliRunner, data_path: Path) -> None: mocked_response = mock.MagicMock( status_code=401, json=lambda: {"detail": "Unauthorized!"} ) - with mock.patch("llama_deploy.cli.deploy.request") as mocked_httpx: + with mock.patch("llama_deploy.cli.utils.request") as mocked_httpx: mocked_httpx.return_value = mocked_response result = runner.invoke(llamactl, ["deploy", str(test_config_file)]) assert result.exit_code == 1 diff --git a/tests/cli/test_status.py b/tests/cli/test_status.py index 283ea64b..737158f4 100644 --- a/tests/cli/test_status.py +++ b/tests/cli/test_status.py @@ -14,7 +14,7 @@ def test_status_server_down(runner: CliRunner) -> None: def test_status_unhealthy(runner: CliRunner) -> None: mocked_response = mock.MagicMock(status_code=500) - with mock.patch("llama_deploy.cli.status.request") as mocked_httpx: + with mock.patch("llama_deploy.cli.utils.request") as mocked_httpx: mocked_httpx.return_value = mocked_response result = runner.invoke(llamactl, ["status"]) assert result.exit_code == 0 @@ -23,7 +23,7 @@ def test_status_unhealthy(runner: CliRunner) -> None: def test_status(runner: CliRunner) -> None: mocked_response = mock.MagicMock(status_code=200, json=lambda: {}) - with mock.patch("llama_deploy.cli.status.request") as mocked_httpx: + with mock.patch("llama_deploy.cli.utils.request") as mocked_httpx: mocked_httpx.return_value = mocked_response result = runner.invoke(llamactl, ["status"]) assert result.exit_code == 0 @@ -36,7 +36,7 @@ def test_status(runner: CliRunner) -> None: def test_status_with_deployments(runner: CliRunner) -> None: mocked_response = mock.MagicMock(status_code=200) mocked_response.json.return_value = {"deployments": ["foo", "bar"]} - with mock.patch("llama_deploy.cli.status.request") as mocked_httpx: + with mock.patch("llama_deploy.cli.utils.request") as mocked_httpx: mocked_httpx.return_value = mocked_response result = runner.invoke(llamactl, ["status"]) assert result.exit_code == 0