#!/usr/bin/env/python

from rumba.model import *

# import testbed plugins
import rumba.testbeds.emulab as emulab
import rumba.testbeds.jfed as jfed
import rumba.testbeds.faketestbed as fake
import rumba.testbeds.qemu as qemu

# import prototype plugins
import rumba.prototypes.ouroboros as our
import rumba.prototypes.rlite as rl
import rumba.prototypes.irati as irati

shim700 = ShimEthDIF("shim700", link_speed=100))
shim400 = ShimEthDIF("shim400", link_speed=100))
shim500 = ShimEthDIF("shim500", link_speed=100))
shim600 = ShimEthDIF("shim600", link_speed=1))
shim300 = ShimEthDIF("shim300", link_speed=100))

n1 = NormalDIF("n1")

c = Node("c",
    difs=[400, 500, 700, n1],
    dif_registrations={n1: [400, 500, 700]})

b = Node("b",
    difs=[300, 400, n1],
    dif_registrations={n1: [300, 400]})

a = Node("a",
    difs=[300, 600, 700, n1],
    dif_registrations={n1: [300, 600, 700]})

d = Node("d",
    difs=[500, 600, n1],
    dif_registrations={n1: [500, 600]})

tb = qemu.Testbed(exp_name = "resilient-square",
    bzimage = "/home/mcapitani/RINA/checkouts/rlite/demo/buildroot/bzImage",
    initramfs = "/home/mcapitani/RINA/checkouts/rlite/demo/buildroot/rootfs.cpio")

exp = rl.Experiment(tb, nodes = [a, b, c, d])

print(exp)

exp.run()