src/emuvim/test/unittests/test_sonata_dummy_gatekeeper.py

   1 """
   2 Copyright (c) 2015 SONATA-NFV
   3 ALL RIGHTS RESERVED.
   4
   5 Licensed under the Apache License, Version 2.0 (the "License");
   6 you may not use this file except in compliance with the License.
   7 You may obtain a copy of the License at
   8
   9     http://www.apache.org/licenses/LICENSE-2.0
  10
  11 Unless required by applicable law or agreed to in writing, software
  12 distributed under the License is distributed on an "AS IS" BASIS,
  13 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 See the License for the specific language governing permissions and
  15 limitations under the License.
  16
  17 Neither the name of the SONATA-NFV [, ANY ADDITIONAL AFFILIATION]
  18 nor the names of its contributors may be used to endorse or promote
  19 products derived from this software without specific prior written
  20 permission.
  21
  22 This work has been performed in the framework of the SONATA project,
  23 funded by the European Commission under Grant number 671517 through
  24 the Horizon 2020 and 5G-PPP programmes. The authors would like to
  25 acknowledge the contributions of their colleagues of the SONATA
  26 partner consortium (www.sonata-nfv.eu).
  27 """
  28
  29 import time
  30 import requests
  31 import json
  32 import os
  33 import unittest
  34 from emuvim.test.base import SimpleTestTopology
  35 from emuvim.api.sonata import SonataDummyGatekeeperEndpoint
  36 from emuvim.api.sonata.dummygatekeeper import initialize_GK
  37 import mininet.clean
  38
  39 PACKAGE_PATH = "misc/sonata-demo-service.son"
  40
  41
  42 class testSonataDummyGatekeeper(SimpleTestTopology):
  43
  44 #    @unittest.skip("disabled")
  45     def test_GK_Api_start_service(self):
  46         # create network
  47         self.createNet(nswitches=0, ndatacenter=2, nhosts=2, ndockers=0)
  48         # setup links
  49         self.net.addLink(self.dc[0], self.h[0])
  50         self.net.addLink(self.dc[0], self.dc[1])
  51         self.net.addLink(self.h[1], self.dc[1])
  52         # connect dummy GK to data centers
  53         sdkg1 = SonataDummyGatekeeperEndpoint("0.0.0.0", 5000)
  54         sdkg1.connectDatacenter(self.dc[0])
  55         sdkg1.connectDatacenter(self.dc[1])
  56         # run the dummy gatekeeper (in another thread, don't block)
  57         sdkg1.start()
  58         # start Mininet network
  59         self.startNet()
  60         time.sleep(1)
  61
  62         print "starting tests"
  63         # board package
  64         files = {"package": open(PACKAGE_PATH, "rb")}
  65         r = requests.post("http://127.0.0.1:5000/packages", files=files)
  66         self.assertEqual(r.status_code, 201)
  67         self.assertTrue(json.loads(r.text).get("service_uuid") is not None)
  68
  69         # instantiate service
  70         self.service_uuid = json.loads(r.text).get("service_uuid")
  71         r2 = requests.post("http://127.0.0.1:5000/instantiations", data=json.dumps({"service_uuid": self.service_uuid}))
  72         self.assertEqual(r2.status_code, 201)
  73
  74         # give the emulator some time to instantiate everything
  75         time.sleep(2)
  76
  77         # check get request APIs
  78         r3 = requests.get("http://127.0.0.1:5000/packages")
  79         self.assertEqual(len(json.loads(r3.text).get("service_uuid_list")), 1)
  80         r4 = requests.get("http://127.0.0.1:5000/instantiations")
  81         self.assertEqual(len(json.loads(r4.text).get("service_instantiations_list")), 1)
  82
  83         # check number of running nodes
  84         self.assertTrue(len(self.getContainernetContainers()) == 3)
  85         self.assertTrue(len(self.net.hosts) == 5)
  86         self.assertTrue(len(self.net.switches) == 2)
  87         # check compute list result
  88         self.assertEqual(len(self.dc[0].listCompute()), 2)
  89         # check connectivity by using ping
  90         for vnf in self.dc[0].listCompute():
  91             p = self.net.ping([self.h[0], vnf])
  92             print p
  93 #            self.assertTrue(p <= 50.0)
  94         # stop Mininet network
  95         self.stopNet()
  96         initialize_GK()
  97
  98
  99     def test_GK_Api_stop_service(self):
 100         # create network
 101         self.createNet(ndatacenter=2, nhosts=2)
 102         # setup links
 103         self.net.addLink(self.dc[0], self.h[0])
 104         self.net.addLink(self.dc[0], self.dc[1])
 105         self.net.addLink(self.h[1], self.dc[1])
 106         # connect dummy GK to data centers
 107         sdkg1 = SonataDummyGatekeeperEndpoint("0.0.0.0", 5000)
 108         sdkg1.connectDatacenter(self.dc[0])
 109         sdkg1.connectDatacenter(self.dc[1])
 110         # run the dummy gatekeeper (in another thread, don't block)
 111         sdkg1.start()
 112         # start Mininet network
 113         self.startNet()
 114         time.sleep(1)
 115
 116         print "starting tests"
 117         # board package
 118         files = {"package": open(PACKAGE_PATH, "rb")}
 119         r = requests.post("http://127.0.0.1:5000/packages", files=files)
 120         self.assertEqual(r.status_code, 201)
 121         self.assertTrue(json.loads(r.text).get("service_uuid") is not None)
 122
 123         # instantiate service
 124         self.service_uuid = json.loads(r.text).get("service_uuid")
 125         r2 = requests.post("http://127.0.0.1:5000/instantiations", data=json.dumps({"service_uuid": self.service_uuid}))
 126         self.assertEqual(r2.status_code, 201)
 127
 128         # give the emulator some time to instantiate everything
 129         time.sleep(2)
 130
 131         # check get request APIs
 132         r3 = requests.get("http://127.0.0.1:5000/packages")
 133         self.assertEqual(len(json.loads(r3.text).get("service_uuid_list")), 1)
 134         r4 = requests.get("http://127.0.0.1:5000/instantiations")
 135         self.assertEqual(len(json.loads(r4.text).get("service_instantiations_list")), 1)
 136
 137         # check number of running nodes
 138         self.assertTrue(len(self.getContainernetContainers()) == 3)
 139         self.assertTrue(len(self.net.hosts) == 5)
 140         self.assertTrue(len(self.net.switches) == 2)
 141         # check compute list result
 142         self.assertEqual(len(self.dc[0].listCompute()), 2)
 143
 144         # stop the service
 145         service_instance_uuid = json.loads(r2.text).get("service_instance_uuid")
 146         self.assertTrue(service_instance_uuid is not None)
 147         requests.delete("http://127.0.0.1:5000/instantiations", data=json.dumps({"service_uuid": self.service_uuid, "service_instance_uuid":service_instance_uuid}))
 148
 149         r5 = requests.get("http://127.0.0.1:5000/instantiations")
 150         self.assertTrue(len(json.loads(r5.text).get("service_instantiations_list")), 0)     # note that there was 1 instance before
 151
 152         # stop Mininet network
 153         self.stopNet()
 154         initialize_GK()