[lnst] MessageDispatcher: use connection_mapping

[Jiří Pírko]

commit 85385ca00c3ae27569994650c0a96ecf253562af
Author: Ondrej Lichtner 
Date:   Tue Sep 30 13:25:32 2014 +0200

MessageDispatcher: use connection_mapping

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 
    Signed-off-by: Jiri Pirko

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: