test_posters.py 文件源码

def test_poster_with_tags_facebook(self, facebook_mock, phantom_driver):
        mock_body = open(os.path.join(os.path.dirname(__file__), 'data', 'crs_page.html')).read()

        posters_register._posters = []
        posters_register.register_poster(facebook_status, 'facebook_done')

        m = mock.Mock()

        m.put_photo.side_effect = mock.Mock(id='1')

        facebook_mock.return_value = m

        phantom_driver.return_value = MockPhantomJS(mock_body)

        self.assertEqual(len(Log.objects.all()), 0)

        meteo_fake_status = '?? Emergenza: A+ , A- , O+ , O- , B- , AB- , AB+\n'
        meteo_upload_fake = os.path.join(os.path.dirname(__file__), 'data', 'meteo_fake_upload.png')

        facebook_status(meteo_fake_status, meteo_upload_fake)

        self.assertEqual(m.put_wall_post.call_count, 0)
        self.assertEqual(m.put_photo.call_count, 1)
        self.assertEqual(
            m.put_photo.call_args[1]['tags'],
            '[{"tag_uid": "andrea.stagi"}, {"tag_uid": "patrick.arminio"}]'
        )