From aab79fdf6d9cb71d02c81849aa1624358b50b0b3 Mon Sep 17 00:00:00 2001 From: k-boikov <64261260+k-boikov@users.noreply.github.com> Date: Sun, 30 Apr 2023 12:41:45 +0300 Subject: [PATCH] added tests for clone_repository (#3558) Co-authored-by: Nicholas Tindle Co-authored-by: Richard Beales --- tests/integration/test_git_commands.py | 42 ++++++++++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 tests/integration/test_git_commands.py diff --git a/tests/integration/test_git_commands.py b/tests/integration/test_git_commands.py new file mode 100644 index 000000000000..d942b1b710af --- /dev/null +++ b/tests/integration/test_git_commands.py @@ -0,0 +1,42 @@ +import pytest +from git.exc import GitCommandError +from git.repo.base import Repo + +from autogpt.commands.git_operations import clone_repository + + +@pytest.fixture +def mock_clone_from(mocker): + return mocker.patch.object(Repo, "clone_from") + + +def test_clone_auto_gpt_repository(workspace, mock_clone_from, config): + mock_clone_from.return_value = None + + repo = "github.com/Significant-Gravitas/Auto-GPT.git" + scheme = "https://" + url = scheme + repo + clone_path = str(workspace.get_path("auto-gpt-repo")) + + expected_output = f"Cloned {url} to {clone_path}" + + clone_result = clone_repository(url=url, clone_path=clone_path) + + assert clone_result == expected_output + mock_clone_from.assert_called_once_with( + url=f"{scheme}{config.github_username}:{config.github_api_key}@{repo}", + to_path=clone_path, + ) + + +def test_clone_repository_error(workspace, mock_clone_from): + url = "https://github.com/this-repository/does-not-exist.git" + clone_path = str(workspace.get_path("does-not-exist")) + + mock_clone_from.side_effect = GitCommandError( + "clone", "fatal: repository not found", "" + ) + + result = clone_repository(url=url, clone_path=clone_path) + + assert "Error: " in result