64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
import unittest
|
||
|
|
from collections import OrderedDict
|
||
|
|
|
||
|
|
import mock
|
||
|
|
from django.test.utils import override_settings
|
||
|
|
|
||
|
|
from ..crypto import HMAC
|
||
|
|
from ..response import SQRLHttpResponse
|
||
|
|
from ..utils import Encoder
|
||
|
|
|
||
|
|
|
||
|
|
TESTING_MODULE = 'sqrl.response'
|
||
|
|
|
||
|
|
|
||
|
|
class TestSQRLHttpResponse(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
super(TestSQRLHttpResponse, self).setUp()
|
||
|
|
self.data = OrderedDict([
|
||
|
|
('ver', 1),
|
||
|
|
('nut', b'nonce'),
|
||
|
|
('tif', '8'),
|
||
|
|
('qry', '/sqrl/auth/?nut=nonce'),
|
||
|
|
('sfn', 'Test Server'),
|
||
|
|
])
|
||
|
|
self.nut = mock.MagicMock(session_key='session')
|
||
|
|
|
||
|
|
@override_settings(DEBUG=False)
|
||
|
|
def test_response(self):
|
||
|
|
response = SQRLHttpResponse(self.nut, self.data)
|
||
|
|
|
||
|
|
expected_data = self.data.copy()
|
||
|
|
expected_data['mac'] = HMAC(self.nut, self.data).sign_data()
|
||
|
|
|
||
|
|
self.assertEqual(
|
||
|
|
response.content,
|
||
|
|
Encoder.base64_dumps(expected_data).encode('ascii')
|
||
|
|
)
|
||
|
|
self.assertEqual(response['Content-Length'], str(len(response.content)))
|
||
|
|
self.assertEqual(response['Content-Type'], 'application/sqrl')
|
||
|
|
|
||
|
|
@override_settings(DEBUG=False)
|
||
|
|
def test_response_without_nut(self):
|
||
|
|
response = SQRLHttpResponse(None, self.data)
|
||
|
|
|
||
|
|
self.assertEqual(
|
||
|
|
response.content,
|
||
|
|
Encoder.base64_dumps(self.data).encode('ascii')
|
||
|
|
)
|
||
|
|
self.assertEqual(response['Content-Length'], str(len(response.content)))
|
||
|
|
self.assertEqual(response['Content-Type'], 'application/sqrl')
|
||
|
|
|
||
|
|
@override_settings(DEBUG=True)
|
||
|
|
@mock.patch(TESTING_MODULE + '.log')
|
||
|
|
def test_response_debug(self, mock_log):
|
||
|
|
response = SQRLHttpResponse(self.nut, self.data)
|
||
|
|
|
||
|
|
self.assertEqual(response['X-SQRL-ver'], '1')
|
||
|
|
self.assertEqual(response['X-SQRL-nut'], 'bm9uY2U')
|
||
|
|
self.assertEqual(response['X-SQRL-tif'], '8')
|
||
|
|
self.assertEqual(response['X-SQRL-qry'], '/sqrl/auth/?nut=nonce')
|
||
|
|
self.assertEqual(response['X-SQRL-sfn'], 'Test Server')
|
||
|
|
self.assertIn('X-SQRL-mac', response)
|