def test_deregister_context_failure_no_context(self):
lst = [Response("Failed")]
self.mocked_obj.get_srx_pfe_names = MagicMock(return_value=["fpc0"])
self.mocked_obj.vty = MagicMock(side_effect=lst)
try:
context_propagation.deregister_context(device=self.mocked_obj,
protocol_id="75")
except Exception as err:
self.assertEqual(err.args[0], "Contexts deregistration failed")
def test_deregister_context_success_no_context(self):
lst = [Response("Successfully Deregistered")]
self.mocked_obj.get_srx_pfe_names = MagicMock(return_value=["fpc0"])
self.mocked_obj.vty = MagicMock(side_effect=lst)
self.assertEqual(
context_propagation.deregister_context(device=self.mocked_obj,
protocol_id="75"), True)
def test_deregister_context_exception(self):
try:
context_propagation.deregister_context(device=self.mocked_obj)
except Exception as err:
self.assertEqual(err.args[0], "protocol_id is mandatory argument")
def test_deregister_context_exception_no_device(self):
try:
context_propagation.deregister_context()
except Exception as err:
self.assertEqual(err.args[0], "device is mandatory argument")