如何添加虚拟方法

在课堂内部，我使用VK API。在测试过程中，我在API中使用MagicMock来模拟授权此类使用：如何添加虚拟方法

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'

如何嘲笑这东西是否正确？

谢谢！

来源

2016-01-22 paus

你为什么要嘲笑它，如果你希望它返回用户返回无？ –

添加users.get属性的正确方法是什么？我试图添加：mock_vk_values = [{'uid'：user.vk_uid}] \ attrs = {'users.get.return_value'：mock_vk_values} \ vk.API = MagicMock（name ='api'，** attrs）但它没有帮助。 – paus

我添加了一些逻辑来开始我试图使用。 – paus

试试这个：

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)

来源

2016-01-22 09:35:42

在我的情况下，尝试执行vk.API.users.get（）返回：。它应该返回ID列表。 – paus

对不起，虽然API的实例在类的内部创建 - 它可以正常工作。 – paus

如何添加虚拟方法

回答

相关问题