def test_notset(self):
h = kea.CalloutHandle(kea.CalloutManager())
with self.assertRaises(TypeError) as cm:
h.getContext('foo')
self.assertEqual(
("unable to find callout context associated with the current"
" library index (-1)", ), cm.exception.args)
def test_ok(self):
h = kea.CalloutHandle(kea.CalloutManager())
foo = utils.Logger()
h.setContext('foo', foo)
self.assertEqual(3, sys.getrefcount(foo))
self.assertIsNone(h.deleteContext('foo'))
self.assertEqual(2, sys.getrefcount(foo))
def test_response4_ok(self):
h = kea.CalloutHandle(kea.CalloutManager())
x = kea.Pkt4(kea.DHCPACK, 42)
self.assertEqual(1, x.use_count)
self.assertIsNone(h.setArgument('response4', x))
self.assertEqual(2, x.use_count)
self.assertEqual(kea.DHCPACK, h.getArgument('response4').getType())
def test_lease4_ok(self):
h = kea.CalloutHandle(kea.CalloutManager())
x = kea.Lease4()
x.addr = '1.2.3.4'
self.assertEqual(1, x.use_count)
self.assertIsNone(h.setArgument('lease4', x))
self.assertEqual(2, x.use_count)
self.assertEqual('1.2.3.4', h.getArgument('lease4').addr)
def test_badarg_count(self):
h = kea.CalloutHandle(kea.CalloutManager())
self.assert_method_no_arguments(h.getStatus)
def test_badarg_type(self):
h = kea.CalloutHandle(kea.CalloutManager())
with self.assertRaises(TypeError) as cm:
h.setArgument(3, 4)
self.assertEqual(("argument 1 must be str, not int", ),
cm.exception.args)
def test_badarg_value(self):
h = kea.CalloutHandle(kea.CalloutManager())
with self.assertRaises(ValueError) as cm:
h.setArgument('foo', 3)
self.assertEqual(("unknown argument", ), cm.exception.args)
def test_ok(self):
h = kea.CalloutHandle(kea.CalloutManager())
with self.assertRaises(TypeError) as cm:
h.getArgument('query4')
self.assertEqual(("unable to find argument with name query4", ),
cm.exception.args)
def test_badarg_count(self):
h = kea.CalloutHandle(kea.CalloutManager())
self.assert_method_two_args_no_keywords(h.setArgument)
def test_badarg_type(self):
h = kea.CalloutHandle(kea.CalloutManager())
with self.assertRaises(TypeError) as cm:
h.setStatus('foo')
self.assertEqual(("an integer is required (got type str)", ),
cm.exception.args)
def test_ok(self):
h = kea.CalloutHandle(kea.CalloutManager())
self.assertIsNone(h.setStatus(kea.NEXT_STEP_SKIP))
self.assertEqual(kea.NEXT_STEP_SKIP, h.getStatus())
def test_command_ok(self):
h = kea.CalloutHandle(kea.CalloutManager())
self.assertIsNone(h.setArgument('command', {'foo': 42}))
self.assertEqual({'foo': 42}, h.getArgument('command'))
def test_badarg_count(self):
h = kea.CalloutHandle(kea.CalloutManager())
self.assert_method_one_arg_no_keywords(h.setStatus)
def test_notset(self):
h = kea.CalloutHandle(kea.CalloutManager())
self.assertIsNone(h.deleteContext('foo'))
def test_query4_bad_type(self):
h = kea.CalloutHandle(kea.CalloutManager())
with self.assertRaises(TypeError) as cm:
h.setArgument('query4', 3)
self.assertEqual(("expected Pkt4 object", ), cm.exception.args)
def test_response_ok(self):
h = kea.CalloutHandle(kea.CalloutManager())
self.assertIsNone(h.setArgument('response', [1, 'two', None]))
self.assertEqual([1, 'two', None], h.getArgument('response'))
def test_badarg_count(self):
h = kea.CalloutHandle(kea.CalloutManager())
self.assert_method_one_arg_no_keywords(h.deleteContext)
def test_response_bad_type(self):
h = kea.CalloutHandle(kea.CalloutManager())
with self.assertRaises(TypeError) as cm:
h.setArgument('response', self)
self.assertEqual(("unhandled type TestCalloutHandle_setArgument", ),
cm.exception.args)
def test_ok(self):
h = kea.CalloutHandle(kea.CalloutManager())
foo = utils.Logger()
h.setContext('foo', foo)
bar = h.getContext('foo')
self.assertIs(foo, bar)
def test_ok(self):
m = kea.CalloutManager()
h = kea.CalloutHandle(m)
self.assertEqual(2, m.use_count)
del h
self.assertEqual(1, m.use_count)
def test_ok(self):
h = kea.CalloutHandle(kea.CalloutManager())
self.assertEqual(kea.NEXT_STEP_CONTINUE, h.getStatus())
def test_badarg_type(self):
with self.assertRaises(TypeError) as cm:
kea.CalloutHandle(1)
self.assertEqual(("argument 1 must be kea.CalloutManager, not int", ),
cm.exception.args)
