3

我正在使用 azure-devOps python 模块来访问 azure API 以检索一些信息。我已经使用下面的代码来设置建立连接所需的身份验证。

def retrieve_projects_from_organization(self): 
   credentials = BasicAuthentication('', AZURE_PRIVATE_TOKEN)
   self.azure_connection = Connection(base_url=os.environ.get('AZURE_BASE_URL',''),creds=credentials)
   core_client = self.azure_connection.clients_v6_0.get_core_client()
   project_list = []
   projects_response = core_client.get_projects()
   while projects_response is not None:
     for project in projects_response:
        project_list.append(project)

在编写单元测试用例时,我无法模拟 Connection 类,因为它需要 BasicAuthentication 类型的对象。下面是我的单元测试用例:

def test_retrieve_projects_from_organization(mocker):
    credentials = mocker.patch('msrest.authentication. BasicAuthentication')).return_value
    connection = mocker.patch('azure.devops.connection.Connection').return_value
    core_client = mocker.patch('azure.devops.v6_0.core.CoreClient').return_value
    projects = mocker.patch('azure.devops.v6_0.core.TeamProjectReference').return_value
    connection._creds = type(credentials)
    connection.clients_v6_0.get_core_client.return_value = core_client
    core_client.get_projects.return_value = [projects]

    assert AzureDataFetcher().retrieve_projects_from_organization() == [projects]

我收到以下错误:

TypeError: super() 参数 1 必须是类型,而不是 MagicMock

是否有不同的方法来模拟此类或绕过对 API 的身份验证的方法。

4

1 回答 1

3

我认为这与您指定模拟的方式有关。我在这里遇到了同样的问题。

而不是mocker.patch('msrest.authentication.BasicAuthentication')尝试mocker.patch('your_module.BasicAuthentication')

于 2021-07-29T15:47:25.857 回答