如何添加虚拟方法

2024-03-28 21:31:02 发布

您现在位置:Python中文网/ 问答频道 /正文

在类内部,我使用vkapi。在测试期间,我使用MagicMock在该类使用的API中模拟授权:

    vk.Session = MagicMock(name='session', return_value=None)
    vk.API = MagicMock(name='api')
    mock_vk_values = [{'uid': user.vk_uid}]
    vk.API.users.get = MagicMock(name='uid', return_value=mock_vk_values)

但是,在类内部,我使用此API获取用户的uid:

 class VKAuth(object):
      def __init__(self, access_token, user):
          ...
          self.session = vk.Session(access_token=access_token)
          self.api = vk.API(self.session)
          ...

      def authenticate(self):
          try:
              vk_uid = self.api.users.get()[0]['uid']

在这个地方会出现一个错误:

*** AttributeError: 'NoneType' object has no attribute 'users'

怎么嘲笑这些东西?你知道吗

谢谢你!你知道吗


Tags: nameselftokenapiuidreturnaccessvalue
1条回答
网友
1楼 · 发布于 2024-03-28 21:31:02

试试这个: 你知道吗

vk.Session = MagicMock(name='session', return_value=None)
mock_vk_values = [{'uid': user.vk_uid}]
# create an explicit mock for the users attribute
users_mock = MagicMock(name='users')
users_mock.get = MagicMock(name='uid', return_value=mock_vk_values)
# create a mock for the api
api_mock = MagicMock(name='api', users=users_mock)
# this is where a lot of people get mocking wrong -
# mistaking the mock of a constructor with the object returned/created
vk.API = MagicMock(return_value=api_mock)

相关问题 更多 >