57 lines
1.9 KiB
Python
57 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
import unittest
|
|
|
|
import mock
|
|
|
|
from ..exceptions import TIF, TIFException
|
|
|
|
|
|
TESTING_MODULE = 'sqrl.exceptions'
|
|
|
|
|
|
class TestTIF(unittest.TestCase):
|
|
def test_as_hex_string(self):
|
|
self.assertEqual(TIF(0x1).as_hex_string(), '1')
|
|
self.assertEqual(TIF(0x4).as_hex_string(), '4')
|
|
self.assertEqual(TIF(0x88).as_hex_string(), '88')
|
|
self.assertEqual(TIF(0x84).as_hex_string(), '84')
|
|
|
|
def test_breakdown(self):
|
|
self.assertDictEqual(TIF(0x34).breakdown(), {
|
|
'id_match': False,
|
|
'previous_id_match': False,
|
|
'ip_match': True,
|
|
'sqrl_disabled': False,
|
|
'not_supported': True,
|
|
'transient_failure': True,
|
|
'command_failed': False,
|
|
'client_failure': False,
|
|
'bad_id_association': False,
|
|
})
|
|
|
|
def test_update(self):
|
|
tif = TIF(0x3).update(TIF(0x40))
|
|
|
|
self.assertIsInstance(tif, TIF)
|
|
self.assertEqual(tif, 0x43)
|
|
|
|
def test_properties(self):
|
|
self.assertTrue(TIF(TIF.ID_MATCH).is_id_match)
|
|
self.assertTrue(TIF(TIF.PREVIOUS_ID_MATCH).is_previous_id_match)
|
|
self.assertTrue(TIF(TIF.IP_MATCH).is_ip_match)
|
|
self.assertTrue(TIF(TIF.SQRL_DISABLED).is_sqrl_disabled)
|
|
self.assertTrue(TIF(TIF.NOT_SUPPORTED).is_not_supported)
|
|
self.assertTrue(TIF(TIF.TRANSIENT_FAILURE).is_transient_failure)
|
|
self.assertTrue(TIF(TIF.COMMAND_FAILED).is_command_failed)
|
|
self.assertTrue(TIF(TIF.CLIENT_FAILURE).is_client_failure)
|
|
self.assertTrue(TIF(TIF.BAD_ID_ASSOCIATION).is_bad_id_association)
|
|
|
|
|
|
class TestTIFException(unittest.TestCase):
|
|
@mock.patch(TESTING_MODULE + '.TIF')
|
|
def test_init(self, mock_tif):
|
|
e = TIFException(mock.sentinel.tif)
|
|
|
|
self.assertEqual(e.tif, mock_tif.return_value)
|
|
mock_tif.assert_called_once_with(mock.sentinel.tif)
|