prototype: rlite: implement register_ipcps()

1 files changed, 15 insertions, 1 deletions

diff --git a/rumba/prototypes/rlite.py b/rumba/prototypes/rlite.py
index 4ab4f1f..e1e3729 100644
--- a/rumba/prototypes/rlite.py
+++ b/rumba/prototypes/rlite.py
@@ -56,8 +56,8 @@ class Experiment(mod.Experiment):
     def create_ipcps(self):
         for node in self.nodes:
             cmds = []
-            for ipcp in node.ipcps:
 
+            for ipcp in node.ipcps:
                 # Generate the command to create the IPCP
                 if type(ipcp.dif) is mod.NormalDIF:
                     ipcp_type = 'normal'
@@ -82,9 +82,23 @@ class Experiment(mod.Experiment):
             ssh.execute_commands(self.testbed, node.ssh_config, cmds,
                                  time_out = None)
 
+    def register_ipcps(self):
+        for node in self.nodes:
+            cmds = []
+
+            for ipcp in node.ipcps:
+                for lower in ipcp.registrations:
+                    cmds.append("rlite-ctl ipcp-register %s %s" \
+                                % (ipcp.name, lower.name))
+
+            ssh.execute_commands(self.testbed, node.ssh_config, cmds,
+                                 time_out = None)
+
     def run_prototype(self):
         print("rlite: setting up")
         self.init()
         print("rlite: software initialized on all nodes")
         self.create_ipcps()
         print("rlite: IPCPs created on all nodes")
+        self.register_ipcps()
+        print("rlite: IPCPs registered to their lower DIFs on all nodes")