def test_endpoint_register_bad_json(self):
factory = RequestFactory()
request = factory.post('/endpoint/register', {
'group': 'a',
'secret': 's',
'data': '{'
})
response = views.endpoint_register(request)
self.assertPlainText(response, "ERROR")
def test_endpoint_register_bad_data_group(self):
factory = RequestFactory()
request = factory.post('/endpoint/register', {
'group': 'a',
'secret': 's',
'data': """{"group": "a"}"""
})
response = views.endpoint_register(request)
self.assertPlainText(response, "ERROR Assert")
def test_endpoint_register_good_extended(self):
factory = RequestFactory()
request = factory.post(
'/endpoint/register', {
'group':
'a',
'secret':
's',
'data':
"""
{
"overall_bitrate": 1,
"overall_clients": 2,
"loop_bitrate": 3,
"loop_clients": 4,
"webm_high_bitrate": 5,
"webm_high_clients": 6,
"webm_low_bitrate": 7,
"webm_low_clients": 8,
"flv_high_bitrate": 9,
"flv_high_clients": 10,
"flv_low_bitrate": 11,
"flv_low_clients": 12,
"ogg_high_bitrate": 13,
"ogg_high_clients": 14,
"aac_high_bitrate": 15,
"aac_high_clients": 16,
"mp3_high_bitrate": 17,
"mp3_high_clients": 18
}
"""
})
response = views.endpoint_register(request)
self.assertPlainText(response, "OK")
endpoints = models.Endpoint.objects.all()
self.assertEqual(1, len(endpoints))
self.assertEqual(1, endpoints[0].overall_bitrate)
self.assertEqual(2, endpoints[0].overall_clients)
self.assertEqual(3, endpoints[0].loop_bitrate)
self.assertEqual(4, endpoints[0].loop_clients)
self.assertEqual(5, endpoints[0].webm_high_bitrate)
self.assertEqual(6, endpoints[0].webm_high_clients)
self.assertEqual(7, endpoints[0].webm_low_bitrate)
self.assertEqual(8, endpoints[0].webm_low_clients)
def test_endpoint_register_good(self):
factory = RequestFactory()
request = factory.post(
'/endpoint/register', {
'group':
'a',
'secret':
's',
'data':
"""
{
"overall_bitrate": 1,
"overall_clients": 2
}
"""
})
response = views.endpoint_register(request)
self.assertPlainText(response, "OK")
endpoints = models.Endpoint.objects.all()
self.assertEqual(1, len(endpoints))
self.assertEqual(1, endpoints[0].overall_bitrate)
self.assertEqual(2, endpoints[0].overall_clients)