From: Ondrej Lichtner <olichtne(a)redhat.com>
This patch updates both the MessageDispatcher and the
MessageDispatcherLite classes to use the _connection_mapping attribute
of the parent class.
Signed-off-by: Ondrej Lichtner <olichtne(a)redhat.com>
---
lnst/Controller/MessageDispatcherLite.py | 4 ++--
lnst/Controller/NetTestController.py | 4 ++--
2 files changed, 4 insertions(+), 4 deletions(-)
diff --git a/lnst/Controller/MessageDispatcherLite.py
b/lnst/Controller/MessageDispatcherLite.py
index 70a006a..8e26d17 100644
--- a/lnst/Controller/MessageDispatcherLite.py
+++ b/lnst/Controller/MessageDispatcherLite.py
@@ -31,11 +31,11 @@ class MessageDispatcherLite(ConnectionHandler):
def wait_for_result(self, machine_id):
wait = True
while wait:
- connected_slaves = self._connections.keys()
+ connected_slaves = self._connection_mapping.keys()
messages = self.check_connections()
- remaining_slaves = self._connections.keys()
+ remaining_slaves = self._connection_mapping.keys()
for msg in messages:
if msg[1]["type"] == "result" and msg[0] == 1:
diff --git a/lnst/Controller/NetTestController.py b/lnst/Controller/NetTestController.py
index da7df41..bdf1bd3 100644
--- a/lnst/Controller/NetTestController.py
+++ b/lnst/Controller/NetTestController.py
@@ -836,11 +836,11 @@ class MessageDispatcher(ConnectionHandler):
def wait_for_result(self, machine_id):
wait = True
while wait:
- connected_slaves = self._connections.keys()
+ connected_slaves = self._connection_mapping.keys()
messages = self.check_connections()
- remaining_slaves = self._connections.keys()
+ remaining_slaves = self._connection_mapping.keys()
for msg in messages:
if msg[1]["type"] == "result" and msg[0] ==
machine_id:
--
1.9.3