src/emuvim/test/test_sonata_dummy_gatekeeper.py

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