andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-25 08:08:14 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix; tests

Browse Source
This commit is contained in:
Eduardo Chiarotti
2024-10-10 13:34:24 -03:00
parent 12a7cee5dd
commit 39ce2631e2
2 changed files with 6 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/crewai/cli/tools/main.py
View File

@@ -153,7 +153,7 @@ class ToolCommand(BaseCommand, PlusAPIMixin):
raise SystemExit raise SystemExit
login_response_json = login_response.json() login_response_json = login_response.json()
self._set_netrc_credentials(login_response_json["credentials"]) self._set_netrc_credentials(login_response_json["credential"])
console.print( console.print(
"Successfully authenticated to the tool repository.", style="bold green" "Successfully authenticated to the tool repository.", style="bold green"

8
tests/cli/tools/test_main.py
View File

@@ -4,6 +4,7 @@ import unittest
import unittest.mock import unittest.mock
from contextlib import contextmanager from contextlib import contextmanager
from io import StringIO from io import StringIO
from unittest import mock
from unittest.mock import MagicMock, patch from unittest.mock import MagicMock, patch
from pytest import raises from pytest import raises
@@ -69,9 +70,9 @@ def test_install_success(mock_get, mock_subprocess_run):
tool_command.install("sample-tool") tool_command.install("sample-tool")
output = fake_out.getvalue() 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( mock_subprocess_run.assert_any_call(
["uv", "add", "--source", "crewai-sample-repo", "sample-tool"], ["uv", "add", "crewai-sample-repo", "sample-tool"],
capture_output=False, capture_output=False,
text=True, text=True,
check=True, check=True,
@@ -209,7 +210,7 @@ def test_publish_success(
mock_get_project_version.assert_called_with(require=True) mock_get_project_version.assert_called_with(require=True)
mock_get_project_description.assert_called_with(require=False) mock_get_project_description.assert_called_with(require=False)
mock_subprocess_run.assert_called_with( mock_subprocess_run.assert_called_with(
["uv", "build", "-f", "sdist", "--output", unittest.mock.ANY], ["uv", "build", "--sdist", "--out-dir", unittest.mock.ANY],
check=True, check=True,
capture_output=False, capture_output=False,
) )
@@ -319,6 +320,7 @@ def test_login_success(mock_subprocess_run, mock_login):
mock_subprocess_run.return_value = MagicMock(stderr=None) mock_subprocess_run.return_value = MagicMock(stderr=None)
breakpoint()
tool_command = ToolCommand() tool_command = ToolCommand()
with patch("sys.stdout", new=StringIO()) as fake_out: with patch("sys.stdout", new=StringIO()) as fake_out: