Python 测试补丁从未被调用

Question

我正在尝试通过 PyGithub 库 mocking 使用我的代码进行测试。 我想为组织创建一个存储库。 所以首先我需要得到它，在“组织”返回 object 时，我需要再打一个电话。

当试图断言我的第二个方法被调用时它失败了。

我是 python 的新手，我猜模拟之间缺少连接，但我不知道是什么。

class GithubService:
    def __init__(self, token: str) -> None:
        self.__github__ = Github(token)
        self.__token__ = token

    def create_repo_extra(self, repo_name, description, organization_name, team_name):
        try:
            organization = self.__github__.get_organization(organization_name)
            repo = organization.create_repo(name=repo_name,
                                            description=description,
                                            private=True,
                                            has_issues=False,
                                            has_wiki=False,
                                            has_projects=False,
                                            allow_merge_commit=False)
            # do other things with the returned repo.....
            return True
        except GithubException as ex:
            print(ex.data)
            return False

这是测试：

import unittest
from unittest.mock import patch, MagicMock, ANY

from github.Organization import Organization
from github.Repository import Repository

from src.github_service import GithubService


class TestGithubService(unittest.TestCase):

    @patch('github.Organization.Organization.create_repo',
           side_effect=MagicMock(return_value=Repository(ANY, {}, {}, True)))
    @patch('github.MainClass.Github.get_organization',
           return_value=MagicMock(return_value=Organization(ANY, {}, {}, True)))
    def test_create_repo_returns_true(self, get_organization, create_repo):

        sut = GithubService("token")
        actual = sut.create_repo_extra('repo-name', 'description', 'organization-name', 'team-name')

        get_organization.assert_called() # ok
        create_repo.assert_called()      # failed

        self.assertTrue(actual) 

Answer 1

由于您模拟了Github.get_organization ，因此您可以使用它直接返回的MagicMock而不是尝试模拟另一层。

在此，我修补相同的Github.get_organization ，但避免给它副作用或返回值，因此将它作为 arg 传递（就像你所做的那样）。

然后我创建一个方便的mock_organization ，它将是修补后的返回值Github.get_organization 。

最后，像您一样检查补丁，并通过方便的mock_organization我检查create_repo方法是否被调用。

class TestGithubService(unittest.TestCase):

    @patch("github.MainClass.Github.get_organization")
    def test_create_repo_returns_true(self, mock_get_organization):

        mock_organization = MagicMock()
        mock_get_organization.return_value = mock_organization

        sut = GithubService("token")
        actual = sut.create_repo_extra(
            "repo-name", "description", "organization-name", "team-name"
        )

        mock_get_organization.assert_called()  # ok
        mock_organization.create_repo.assert_called()  # ok

        self.assertTrue(actual)

在没有看到更多代码的情况下，我不确定为什么修补Organization不起作用，但这更简单、更清晰且同样有效。