def test_not_ok(self):
auth_mixin = AuthMixin()
auth_mixin.request = RequestMock(headers={})
with self.assertRaises(ServiceAuthError) as cm:
auth_mixin.get_access_token(require_auth=True)
self.assertEqual(
'HTTP 401: Authorization header missing (Authorization header is expected)',
f'{cm.exception}')
auth_mixin.request = RequestMock(
headers={'Authorization': 'Beerer my_t0k3n'})
with self.assertRaises(ServiceAuthError) as cm:
auth_mixin.get_access_token()
self.assertEqual(
'HTTP 401: Invalid header (Authorization header must start with "Bearer")',
f'{cm.exception}')
auth_mixin.request = RequestMock(headers={'Authorization': 'Bearer'})
with self.assertRaises(ServiceAuthError) as cm:
auth_mixin.get_access_token()
self.assertEqual('HTTP 401: Invalid header (Bearer token not found)',
f'{cm.exception}')
auth_mixin.request = RequestMock(
headers={'Authorization': 'Bearer my t0k3n'})
with self.assertRaises(ServiceAuthError) as cm:
auth_mixin.get_access_token()
self.assertEqual(
'HTTP 401: Invalid header (Authorization header must be Bearer token)',
f'{cm.exception}')
def test_ok_no_auth(self):
auth_mixin = AuthMixin()
auth_mixin.request = RequestMock(headers={})
self.assertEqual(None, auth_mixin.get_access_token())
def test_ok(self):
auth_mixin = AuthMixin()
auth_mixin.request = RequestMock(
headers={'Authorization': 'Bearer my_t0k3n'})
self.assertEqual('my_t0k3n', auth_mixin.get_access_token())