[PATCH 2/3] MessageDispatcher: use connection_mapping

[Ondrej Lichtner]

From: Ondrej Lichtner

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 
---
 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: