VOL-276: Fix for Failing OMCI & OFAGENT utests
Addressed review comments
Change-Id: I4d08c842e0a4dadba7aa09420490f1c29310fa31
diff --git a/tests/utests/ofagent/test_of_protocol_handler.py b/tests/utests/ofagent/test_of_protocol_handler.py
index d72b0b2..a2eb74e 100644
--- a/tests/utests/ofagent/test_of_protocol_handler.py
+++ b/tests/utests/ofagent/test_of_protocol_handler.py
@@ -35,7 +35,7 @@
with self.assertRaises(Exception) as context:
of_proto_handler.handle_flow_mod_request(generic_obj)
print context.exception
- self.assertTrue('\'function\' object has no attribute \'send\'' in context.exception)
+ self.assertTrue('\'function\' object has no attribute \'send\'' in str(context.exception))
def test_handle_flow_mod_request_role_master(self):
generic_obj = self.gen_generic_obj()
@@ -53,7 +53,7 @@
of_proto_handler.handle_role_request(req)
self.assertEqual(of_proto_handler.role,req.role)
print context.exception
- self.assertTrue('\'function\' object has no attribute \'send\'' in context.exception)
+ self.assertTrue('\'function\' object has no attribute \'generation_is_defined\'' in str(context.exception))
def test_forward_packet_in_role_none(self):
packet_in = self.gen_packet_in()
@@ -71,7 +71,7 @@
with self.assertRaises(Exception) as context:
of_proto_handler.forward_packet_in(packet_in)
print context.exception
- self.assertTrue('\'function\' object has no attribute \'send\'' in context.exception)
+ self.assertTrue('\'function\' object has no attribute \'send\'' in str(context.exception))
if __name__ == '__main__':
- main()
\ No newline at end of file
+ main()