commit 7ecabd21e1787ba69b4455e7e06de5563b524c3f
author sumithdev09 <sumith.hy398@gmail.com> Fri Jul 12 11:10:53 2013 -0400
committer sumithdev09 <sumith.hy398@gmail.com> Fri Jul 12 11:10:53 2013 -0400
tree 63b1eafc635f5e54095872e0ca907237786a0a43
parent 6d2e396f5561091c2ee34a903efab955dec0eeb9

Update controller.py

Bugfix:We cannot poll the OFPT_HELLO message using the poll function if exp_msg=None.

src/python/oftest/controller.py

diff --git a/src/python/oftest/controller.py b/src/python/oftest/controller.py
index bee9ab9..c59cfe8 100644
--- a/src/python/oftest/controller.py
+++ b/src/python/oftest/controller.py
@@ -569,7 +569,7 @@
             return
         self.handlers[msg_type] = handler
 
-    def poll(self, exp_msg=None, timeout=-1):
+    def poll(self, exp_msg=-1, timeout=-1):
         """
         Wait for the next OF message received from the switch.
 
@@ -588,11 +588,11 @@
         """
 
         exp_msg_str = "unspecified"
-        if exp_msg:
+        if exp_msg!=-1:
             exp_msg_str = ofp.ofp_type_map.get(exp_msg, "unknown (%d)" % 
                                                exp_msg)
 
-        if exp_msg is not None:
+        if exp_msg!=-1:
             self.logger.debug("Poll for %s", exp_msg_str)
         else:
             self.logger.debug("Poll for any OF message")
@@ -600,7 +600,7 @@
         # Take the packet from the queue
         def grab():
             if len(self.packets) > 0:
-                if exp_msg is None:
+                if exp_msg==-1:
                     self.logger.debug("Looking for any packet")
                     (msg, pkt) = self.packets.pop(0)
                     return (msg, pkt)