Question

我有一个python项目，它使用GitPython对远程Git存储库执行克隆和拉取函数。

举个简单的例子：

import git
from git import Git
from git import Repo


def clone_and_checkout(full_dir, git_url, repo_ver):

    repo = Repo.clone_from(
        url=git_url,
        to_path=full_dir
    )

    # Trigger re-create if repository is bare
    if repo.bare:
        raise git.exc.InvalidGitRepositoryError

    # Set origin and pull
    origin = repo.remotes.origin
    origin.pull()

    # Check out desired version of repository
    g = Git(full_dir)
    g.checkout(repo_ver)

我希望能够为此功能编写单元测试，但显然这需要与当前的外部系统联系。

我很好奇是否有人有过模拟这种外部交互的经历，其方式类似于使用Mock来模拟HTTP调用。我希望能够以一种可以在测试时模拟的方式执行这些任务，而无需调用实际的Git遥控器。

我该如何为此编写测试？

编辑：为了更清楚我要问的是什么，我应该提到我是Mock的新手并且正在努力理解如何模拟这些类的实例而不是类本身。我的问题应该更好地表达 - 有些内容与“如何使用Mock设置特定于实例的属性如裸？”

我已经了解了很多关于Mock的知识，并且已经想出如何做到这一点，所以我将回答我自己的问题。

Answer 1

这似乎是对Mock不完全理解或使用Patch方法的常见结果。

要做的第一件事是阅读Mock文档中的“where to patch”部分。有了这些信息，您应该能够使用补丁函数来模拟上面函数中使用的GitPython对象。这些装饰器将出现在您的单元测试功能上方。

@mock.patch('gitter.Repo')
@mock.patch('gitter.Git')

为了为这些模拟对象之一的实例提供返回值，您可以使用PropertyMock。以下是利用此功能的单元测试的完整示例：

import gitter  # file containing our clone function
import mock
import unittest


class test_gitter(unittest.TestCase):

    @mock.patch('gitter.Repo')
    @mock.patch('gitter.Git')
    def runTest(self, mock_git, mock_repo):

        # Set the "bare" attribute of the Repo instance to be False
        p = mock.PropertyMock(return_value=False)
        type(mock_repo.clone_from.return_value).bare = p

        gitter.clone_and_checkout(
            '/tmp/docker',
            'git@github.com:docker/docker.git',
            'master'
        )
        mock_git.checkout.called_once_with('master')

如何为GitPython克隆/拉取函数编写单元测试？

1 个答案: