src/emuvim/test/test_sonata_dummy_gatekeeper.py

   1 import time
   2 import requests
   3 import subprocess
   4 import os
   5 from emuvim.test.base import SimpleTestTopology
   6 from emuvim.api.sonata import SonataDummyGatekeeperEndpoint
   7
   8
   9
  10 class testSonataDummyGatekeeper(SimpleTestTopology):
  11
  12     def testAPI(self):
  13         # create network
  14         self.createNet(nswitches=0, ndatacenter=2, nhosts=2, ndockers=0)
  15         # setup links
  16         self.net.addLink(self.dc[0], self.h[0])
  17         self.net.addLink(self.dc[0], self.dc[1])
  18         self.net.addLink(self.h[1], self.dc[1])
  19         # connect dummy GK to data centers
  20         sdkg1 = SonataDummyGatekeeperEndpoint("0.0.0.0", 5000)
  21         sdkg1.connectDatacenter(self.dc[0])
  22         sdkg1.connectDatacenter(self.dc[1])
  23         # run the dummy gatekeeper (in another thread, don't block)
  24         sdkg1.start()
  25         # start Mininet network
  26         self.startNet()
  27         time.sleep(1)
  28
  29         # download example from GitHub
  30         print "downloading latest son-demo.son from GitHub"
  31         subprocess.call(
  32             ["wget",
  33              "http://github.com/sonata-nfv/son-schema/blob/master/package-descriptor/examples/sonata-demo.son?raw=true",
  34              "-O",
  35              "son-demo.son"]
  36         )
  37
  38         print "starting tests"
  39         # board package
  40         files = {"package": open("son-demo.son", "rb")}
  41         r = requests.post("http://127.0.0.1:5000/packages", files=files)
  42         self.assertEqual(r.status_code, 200)
  43         self.assertTrue(r.json().get("service_uuid") is not None)
  44         os.remove("son-demo.son")
  45
  46         # instantiate service
  47         service_uuid = r.json().get("service_uuid")
  48         r2 = requests.post("http://127.0.0.1:5000/instantiations", json={"service_uuid": service_uuid})
  49         self.assertEqual(r2.status_code, 200)
  50
  51         # give the emulator some time to instantiate everything
  52         time.sleep(2)
  53
  54         # check get request APIs
  55         r3 = requests.get("http://127.0.0.1:5000/packages")
  56         self.assertEqual(len(r3.json().get("service_uuid_list")), 1)
  57         r4 = requests.get("http://127.0.0.1:5000/instantiations")
  58         self.assertEqual(len(r4.json().get("service_instance_list")), 1)
  59
  60         # check number of running nodes
  61         self.assertTrue(len(self.getDockernetContainers()) == 3)
  62         self.assertTrue(len(self.net.hosts) == 5)
  63         self.assertTrue(len(self.net.switches) == 2)
  64         # check compute list result
  65         self.assertTrue(len(self.dc[0].listCompute()) == 3)
  66         # check connectivity by using ping
  67         for vnf in self.dc[0].listCompute():
  68             self.assertTrue(self.net.ping([self.h[0], vnf]) <= 0.0)
  69         # stop Mininet network
  70         self.stopNet()
  71
  72