archive/pillar
0
0
Fork 0
Code Issues 42 Pull Requests Projects Releases Wiki Activity

Modernised ZencoderNotificationTest

Browse Source
This commit is contained in:
Sybren A. Stüvel
2018-01-26 12:15:42 +01:00
parent fd95135f66
commit 66ac8c6587
1 changed files with 24 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
tests/test_api/test_encoding.py
View File

@@ -5,17 +5,19 @@ from pillar.tests import AbstractPillarTest
class ZencoderNotificationTest(AbstractPillarTest): class ZencoderNotificationTest(AbstractPillarTest):
def setUp(self, **kwargs):
super().setUp(**kwargs)
self.enter_app_context()
self.secret = self.app.config['ZENCODER_NOTIFICATIONS_SECRET']
def test_missing_secret(self): def test_missing_secret(self):
with self.app.test_request_context(): self.post('/api/encoding/zencoder/notifications',
resp = self.client.post('/api/encoding/zencoder/notifications') expected_status=401)
self.assertEqual(401, resp.status_code)
def test_wrong_secret(self): def test_wrong_secret(self):
with self.app.test_request_context(): self.post('/api/encoding/zencoder/notifications',
resp = self.client.post('/api/encoding/zencoder/notifications', headers={'X-Zencoder-Notification-Secret': 'koro'},
headers={'X-Zencoder-Notification-Secret': 'koro'}) expected_status=401)
self.assertEqual(401, resp.status_code)
def test_good_secret_existing_file(self): def test_good_secret_existing_file(self):
self.ensure_file_exists(file_overrides={ self.ensure_file_exists(file_overrides={
@@ -24,10 +26,8 @@ class ZencoderNotificationTest(AbstractPillarTest):
'status': 'processing'} 'status': 'processing'}
}) })
with self.app.test_request_context(): self.post('/api/encoding/zencoder/notifications',
secret = self.app.config['ZENCODER_NOTIFICATIONS_SECRET'] json={'job': {'id': 'koro-007',
resp = self.client.post('/api/encoding/zencoder/notifications',
data=json.dumps({'job': {'id': 'koro-007',
'state': 'done'}, 'state': 'done'},
'outputs': [{ 'outputs': [{
'format': 'jpg', 'format': 'jpg',
@@ -38,9 +38,8 @@ class ZencoderNotificationTest(AbstractPillarTest):
}], }],
'input': { 'input': {
'duration_in_ms': 5000, 'duration_in_ms': 5000,
}}), }},
headers={'X-Zencoder-Notification-Secret': secret, headers={'X-Zencoder-Notification-Secret': self.secret},
'Content-Type': 'application/json'}) expected_status=204)
# TODO: check that the file in MongoDB is actually updated properly. # TODO: check that the file in MongoDB is actually updated properly.
self.assertEqual(204, resp.status_code)