src/emuvim/test/unittests/test_sonata_dummy_gatekeeper.py

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