def test_follow_congresspeople(self, logging_mock):
profiles = pd.DataFrame([
['DepEduardoCunha', 'DepEduardoCunha2'],
['DepRodrigomaia', None],
[None, None]
], columns=['twitter_profile', 'secondary_twitter_profile'])
self.subject._profiles = profiles
calls = [
mock.call.CreateFriendship(screen_name='DepEduardoCunha'),
mock.call.CreateFriendship(screen_name='DepEduardoCunha2'),
mock.call.CreateFriendship(screen_name='DepRodrigomaia'),
]
self.subject.follow_congresspeople()
self.api.assert_has_calls(calls, any_order=True)
self.assertEqual(3, self.api.CreateFriendship.call_count)
self.api.CreateFriendship.side_effect = TwitterError('Not found')
self.subject.follow_congresspeople()
logging_mock.warning.assert_called()
self.assertEqual(3, logging_mock.warning.call_count)
评论列表
文章目录