1 files changed, 43 insertions, 0 deletions
diff --git a/autogpts/autogpt/tests/unit/test_git_commands.py b/autogpts/autogpt/tests/unit/test_git_commands.py
new file mode 100644
index 000000000..42bcd7b99
--- /dev/null
+++ b/autogpts/autogpt/tests/unit/test_git_commands.py
@@ -0,0 +1,43 @@
+import pytest
+from git.exc import GitCommandError
+from git.repo.base import Repo
+
+from autogpt.agents.agent import Agent
+from autogpt.agents.utils.exceptions import CommandExecutionError
+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, agent: Agent):
+    mock_clone_from.return_value = None
+
+    repo = "github.com/Significant-Gravitas/Auto-GPT.git"
+    scheme = "https://"
+    url = scheme + repo
+    clone_path = workspace.get_path("auto-gpt-repo")
+
+    expected_output = f"Cloned {url} to {clone_path}"
+
+    clone_result = clone_repository(url=url, clone_path=clone_path, agent=agent)
+
+    assert clone_result == expected_output
+    mock_clone_from.assert_called_once_with(
+        url=f"{scheme}{agent.legacy_config.github_username}:{agent.legacy_config.github_api_key}@{repo}",  # noqa: E501
+        to_path=clone_path,
+    )
+
+
+def test_clone_repository_error(workspace, mock_clone_from, agent: Agent):
+    url = "https://github.com/this-repository/does-not-exist.git"
+    clone_path = workspace.get_path("does-not-exist")
+
+    mock_clone_from.side_effect = GitCommandError(
+        "clone", "fatal: repository not found", ""
+    )
+
+    with pytest.raises(CommandExecutionError):
+        clone_repository(url=url, clone_path=clone_path, agent=agent)