model: Perform split between different flows

This performs a split between enrollment, data transfer and management
flows.

1 files changed, 21 insertions, 1 deletions

diff --git a/rumba/prototypes/ouroboros.py b/rumba/prototypes/ouroboros.py
index afc80d3..49323c5 100644
--- a/rumba/prototypes/ouroboros.py
+++ b/rumba/prototypes/ouroboros.py
@@ -124,7 +124,23 @@ class Experiment(mod.Experiment):
                 ssh.execute_commands(self.testbed,
                                      e['enrollee'].node.ssh_config,
                                      cmds, time_out=None)
-                time.sleep(2)
+
+    def setup_flows(self, flows):
+        for el in flows:
+            for e in el:
+                ipcp = e['src']
+                if flows == self.mgmt_flows:
+                    comp = 'mgmt'
+                elif flows == self.dt_flows:
+                    comp = 'dt'
+                else:
+                    assert(False)
+
+                cmd = "irm i conn n " + ipcp.name + " comp " + \
+                      comp + " dst " + e['dst'].name
+                ssh.execute_command(self.testbed,
+                                    ipcp.node.ssh_config,
+                                    cmd, time_out=None)
 
     def install_prototype(self):
         logger.info("Installing Ouroboros...")
@@ -138,4 +154,8 @@ class Experiment(mod.Experiment):
         self.create_ipcps()
         logger.info("Enrolling IPCPs...")
         self.enroll_ipcps()
+        logger.info("Setting up management flows...")
+        self.setup_flows(self.mgmt_flows)
+        logger.info("Setting up data transfer flows...")
+        self.setup_flows(self.dt_flows)
         logger.info("All done, have fun!")