class PubSubHubbubTestCase(RockPackTestCase):
@patch('rockpack.mainsite.requests.post')
def setUp(self, requests_post):
super(PubSubHubbubTestCase, self).setUp()
self.ctx = self.app.test_request_context()
self.ctx.push()
self.channel = Channel(title='', description='', cover='', owner=UserData.test_user_a.id).save()
assert Channel.query.get(self.channel.id), self.channel.id
self.channel.add_meta('en-us')
self.subid = subscribe('test', uuid.uuid4().hex, self.channel.id).id
def tearDown(self):
self.ctx.pop()
super(PubSubHubbubTestCase, self).tearDown()
def test_verify(self):
subs = Subscription.query.get(self.subid)
self.assertFalse(bool(subs.verified))
with self.app.test_client() as client:
r = client.get(
'/ws/pubsubhubbub/callback',
query_string={
'id': subs.id,
'hub.topic': subs.topic,
'hub.challenge': 123,
'hub.verify_token': subs.verify_token,
'hub.mode': 'subscribe',
'hub.lease_seconds': 60,
})
self.assertEquals(r.status_code, 200)
self.assertTrue(subs.verified)
def _post_entries(self, subs, data):
sig = hmac.new(subs.secret, data, hashlib.sha1)
with self.app.test_client() as client:
r = client.post(
'/ws/pubsubhubbub/callback',
query_string={'id': subs.id},
content_type='application/atom+xml',
data=data,
headers={'X-Hub-Signature': 'sha1=' + sig.hexdigest()})
self.assertEquals(r.status_code, 204)
def test_post(self):
entry = PUSH_ENTRY_XML % dict(videoid='xyzzy')
data = PUSH_ATOM_XML % dict(entry=entry)
subs = Subscription.query.get(self.subid)
self._post_entries(subs, data)
video = subs.channel.video_instances[0].video_rel
self.assertEquals(video.source_videoid, 'xyzzy')
self.assertEquals(video.duration, 248)
self.assertIn('BQ', [rst.country for rst in video.restrictions])
def test_post_update(self):
# Double-check that subsequent updates and repeated videos
# are handled correctly.
ids = map(str, range(3))
entry = PUSH_ENTRY_XML % dict(videoid=ids[0])
for id in ids:
entry += PUSH_ENTRY_XML % dict(videoid=id)
data = PUSH_ATOM_XML % dict(entry=entry)
subs = Subscription.query.get(self.subid)
self._post_entries(subs, data)
channel_videos = [
v.video_rel.source_videoid for v in subs.channel.video_instances]
self.assertEquals(sorted(channel_videos), ids)
