From 39ce2631e2e0f2daae12a25c7936672c49ea9cd2 Mon Sep 17 00:00:00 2001 From: Eduardo Chiarotti Date: Thu, 10 Oct 2024 13:34:24 -0300 Subject: [PATCH] fix; tests --- src/crewai/cli/tools/main.py | 2 +- tests/cli/tools/test_main.py | 8 +++++--- 2 files changed, 6 insertions(+), 4 deletions(-) diff --git a/src/crewai/cli/tools/main.py b/src/crewai/cli/tools/main.py index 13e45f716..08be76d16 100644 --- a/src/crewai/cli/tools/main.py +++ b/src/crewai/cli/tools/main.py @@ -153,7 +153,7 @@ class ToolCommand(BaseCommand, PlusAPIMixin): raise SystemExit login_response_json = login_response.json() - self._set_netrc_credentials(login_response_json["credentials"]) + self._set_netrc_credentials(login_response_json["credential"]) console.print( "Successfully authenticated to the tool repository.", style="bold green" diff --git a/tests/cli/tools/test_main.py b/tests/cli/tools/test_main.py index d44f74b99..a552ee300 100644 --- a/tests/cli/tools/test_main.py +++ b/tests/cli/tools/test_main.py @@ -4,6 +4,7 @@ import unittest import unittest.mock from contextlib import contextmanager from io import StringIO +from unittest import mock from unittest.mock import MagicMock, patch from pytest import raises @@ -69,9 +70,9 @@ def test_install_success(mock_get, mock_subprocess_run): tool_command.install("sample-tool") output = fake_out.getvalue() - mock_get.assert_called_once_with("sample-tool") + mock_get.assert_has_calls([mock.call("sample-tool"), mock.call().json()]) mock_subprocess_run.assert_any_call( - ["uv", "add", "--source", "crewai-sample-repo", "sample-tool"], + ["uv", "add", "crewai-sample-repo", "sample-tool"], capture_output=False, text=True, check=True, @@ -209,7 +210,7 @@ def test_publish_success( mock_get_project_version.assert_called_with(require=True) mock_get_project_description.assert_called_with(require=False) mock_subprocess_run.assert_called_with( - ["uv", "build", "-f", "sdist", "--output", unittest.mock.ANY], + ["uv", "build", "--sdist", "--out-dir", unittest.mock.ANY], check=True, capture_output=False, ) @@ -319,6 +320,7 @@ def test_login_success(mock_subprocess_run, mock_login): mock_subprocess_run.return_value = MagicMock(stderr=None) + breakpoint() tool_command = ToolCommand() with patch("sys.stdout", new=StringIO()) as fake_out: