From a34a5d7ec2e6aa59a0a0aa0daba62f7738a1749a Mon Sep 17 00:00:00 2001
From: Sander Vrijders <sander.vrijders@ugent.be>
Date: Tue, 9 May 2017 11:50:17 +0200
Subject: rumba: model: Add StoryBoard for automated testing

This adds an initial draft of the API to automate tests on a RINA
network.
---
 examples/example.py | 20 ++++++++--------
 rumba/model.py      | 67 ++++++++++++++++++++++++++++++++++++++++++++++++++++-
 2 files changed, 76 insertions(+), 11 deletions(-)

diff --git a/examples/example.py b/examples/example.py
index 56193c2..383dbf1 100755
--- a/examples/example.py
+++ b/examples/example.py
@@ -17,7 +17,6 @@ import rumba.prototypes.irati as irati
 
 import rumba.log as log
 
-
 log.set_logging_level('DEBUG')
 
 n1 = NormalDIF("n1", policies = {"rmt.pff": "lfa",
@@ -27,19 +26,16 @@ e1 = ShimEthDIF("e1")
 
 a = Node("a",
          difs = [n1, e1],
-         dif_registrations = {n1 : [e1]},
-         registrations = {"a.crap" : [n1]},
-         bindings = {"a.crap" : "/usr/bin/crap"})
+         dif_registrations = {n1 : [e1]})
 
 b = Node("b",
          difs = [e1, n1],
-         dif_registrations = {n1 : [e1]})
+         dif_registrations = {n1 : [e1]},
+         end_user = True)
 
-tb = qemu.Testbed(exp_name = "example1",
-                  username = "root",
-                  password = "root",
-                  bzimage = '/home/vmaffione/git/rlite/demo/buildroot/bzImage',
-                  initramfs = '/home/vmaffione/git/rlite/demo/buildroot/rootfs.cpio')
+tb = jfed.Testbed(exp_name = "example1",
+                  username = "user1",
+                  cert_file = "/home/user1/cert.pem")
 
 exp = rl.Experiment(tb, nodes = [a, b])
 
@@ -48,5 +44,9 @@ print(exp)
 try:
     exp.swap_in()
     exp.bootstrap_prototype()
+    c1 = Client("rinaperf", options = "-t perf -s 1000 -c 10000")
+    s1 = Server("rinaperf", options = "-l", nodes = [a], clients = [c1])
+    sb = StoryBoard(exp, 3600, servers = [s1])
+    sb.start()
 finally:
     exp.swap_out()
diff --git a/rumba/model.py b/rumba/model.py
index 0b9fc7a..91f33a6 100644
--- a/rumba/model.py
+++ b/rumba/model.py
@@ -152,7 +152,7 @@ class SSHConfig:
 #
 class Node:
     def __init__(self, name, difs=None, dif_registrations=None,
-                 registrations=None, bindings=None):
+                 registrations=None, bindings=None, end_user=False):
         self.name = name
         if difs is None:
             difs = list()
@@ -170,6 +170,7 @@ class Node:
         self.bindings = bindings
         self.ssh_config = SSHConfig(name)
         self.ipcps = []
+        self.end_user = end_user
 
         self._validate()
 
@@ -564,3 +565,67 @@ class Experiment:
     def swap_out(self):
         # Undo the testbed (testbed-specific)
         self.testbed.swap_out(self)
+
+# Base class for client programs
+#
+# @ap: Application Process binary
+# @options: Options to pass to the binary
+#
+class Client:
+    def __init__(self, ap, options=None):
+        self.ap = ap
+        self.options = options
+
+# Base class for server programs
+#
+# @ap: Application Process binary
+# @options: Options to pass to the binary
+# @max_clients: Maximum number of clients to serve
+# @clients: Client binaries that will use this server
+# @nodes: Specific nodes to start this server on
+#
+class Server:
+    def __init__(self, ap, options=None, max_clients=None,
+                 clients=None, nodes=None):
+        self.ap = ap
+        self.options = options
+        self.max_clients = max_clients
+        if clients is None:
+            clients = list()
+        self.clients = clients
+        self.nodes = nodes
+
+    def add_client(self, client):
+        self.clients.append(client)
+
+    def del_client(self, client):
+        self.clients.remove(client)
+
+    def add_node(self, node):
+        self.nodes.append(node)
+
+    def del_node(self, node):
+        self.nodes.remove(node)
+
+# Base class for ARCFIRE storyboards
+#
+# @experiment: Experiment to use as input
+# @duration: Duration of the whole storyboard
+# @servers: App servers available in the network
+#
+class StoryBoard:
+    def __init__(self, experiment, duration, servers=None):
+        self.experiment = experiment
+        self.duration = duration
+        if servers is None:
+            servers = list()
+        self.servers = servers
+
+    def add_server(self, server):
+        self.servers.append(server)
+
+    def del_server(self, server):
+        self.servers.remove(server)
+
+    def start(self):
+        pass
-- 
cgit v1.2.3