def setUp(self):
self.handler = DigestHandler(mock.MagicMock(), mock.MagicMock())
self.handler._start = time.time()
self.handler.application.cache.get_av = mock.MagicMock()
self.handler.finish = mock.MagicMock()
self.handler.send_error = mock.MagicMock()
# Mock out zmq so we don't fail if we try to report stats during the
# test.
self.real_zmq = base.zmq
base.zmq = mock.MagicMock()
class TestDigestHandler(unittest.TestCase):
def setUp(self):
self.handler = DigestHandler(mock.MagicMock(), mock.MagicMock())
self.handler._start = time.time()
self.handler.application.cache.get_av = mock.MagicMock()
self.handler.finish = mock.MagicMock()
self.handler.send_error = mock.MagicMock()
# Mock out zmq so we don't fail if we try to report stats during the
# test.
self.real_zmq = base.zmq
base.zmq = mock.MagicMock()
def tearDown(self):
base.zmq = self.real_zmq
del self.real_zmq
def test_cache_success_output(self):
self.handler.application.cache.get_av.return_value = DigestAuthVector("ha1_test", None, None, True)
self.handler.get("private_id")
self.handler.finish.assert_called_once_with({"digest_ha1": "ha1_test"})
def test_backend_success_output(self):
self.handler.application.cache.get_av.return_value = None
self.handler.application.backend.get_av.return_value = DigestAuthVector("ha1_test", None, None, True)
self.handler.get("private_id")
self.handler.finish.assert_called_once_with({"digest_ha1": "ha1_test"})
def test_failure_output(self):
self.handler.application.cache.get_av.return_value = None
self.handler.application.backend.get_av.return_value = None
self.handler.get("private_id")
self.handler.send_error.assert_called_once_with(404)