elements: Experiment: add missing enrollment_strategy argument

This is necessary to enable scripts to use full-mesh or manual
enrollment strategies.
Fixes #54.

1 files changed, 5 insertions, 2 deletions

diff --git a/rumba/prototypes/irati.py b/rumba/prototypes/irati.py
index 1967278..ec3e53d 100644
--- a/rumba/prototypes/irati.py
+++ b/rumba/prototypes/irati.py
@@ -67,7 +67,8 @@ class Experiment(mod.Experiment):
                  git_branch='arcfire',
                  installpath=None,
                  varpath=None,
-                 app_mappings=None):
+                 app_mappings=None,
+                 enrollment_strategy='minimal'):
         """
         Initializes the experiment class.
 
@@ -81,13 +82,15 @@ class Experiment(mod.Experiment):
                              all application which will register to
                              any given dif.
         :type app_mappings: `List[(str, str)]`
+        :param enrollment_strategy: Can be 'full-mesh', 'minimal' or 'manual'.
         """
         mod.Experiment.__init__(self,
                                 testbed,
                                 nodes,
                                 git_repo,
                                 git_branch,
-                                prototype_logs=['/tmp/ipcmstart.log'])
+                                prototype_logs=['/tmp/ipcmstart.log'],
+                                enrollment_strategy=enrollment_strategy)
         if installpath is None:
             installpath = '/usr'
         if varpath is None: