cleanup and license header for examples and tests
[osm/vim-emu.git] / src / emuvim / test / unittests / test_restapi.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 """
30 Test suite to automatically test emulator REST API endpoints.
31 """
32
33 import time
34 import unittest
35 from emuvim.test.api_base import SimpleTestTopology
36 import subprocess
37 from emuvim.dcemulator.node import EmulatorCompute
38 import ast
39
40 class testRestApi( SimpleTestTopology ):
41 """
42 Tests to check the REST API endpoints of the emulator.
43 """
44
45 def testRestApi(self):
46
47 # create network
48 self.createNet(nswitches=0, ndatacenter=2, nhosts=2, ndockers=0)
49
50 # setup links
51 self.net.addLink(self.dc[0], self.h[0])
52 self.net.addLink(self.h[1], self.dc[1])
53 self.net.addLink(self.dc[0], self.dc[1])
54
55 # start api
56 self.startApi()
57
58 # start Mininet network
59 self.startNet()
60
61 print('->>>>>>> son-emu-cli compute start -d datacenter0 -n vnf1 ->>>>>>>>>>>>>>>')
62 print('->>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>')
63 subprocess.call("son-emu-cli compute start -d datacenter0 -n vnf1", shell=True)
64 print('->>>>>>> son-emu-cli compute start -d datacenter0 -n vnf2 ->>>>>>>>>>>>>>>')
65 print('->>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>')
66 subprocess.call("son-emu-cli compute start -d datacenter0 -n vnf2", shell=True)
67 print('->>>>>>> son-emu-cli compute start -d datacenter0 -n vnf3 ->>>>>>>>>>>>>>>')
68 print('->>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>')
69 subprocess.call("son-emu-cli compute start -d datacenter1 -n vnf3", shell=True)
70 subprocess.call("son-emu-cli compute list", shell=True)
71 print('->>>>>>> checking running nodes, compute list, and connectivity >>>>>>>>>>')
72
73 # check number of running nodes
74 self.assertTrue(len(self.getContainernetContainers()) == 3)
75 self.assertTrue(len(self.net.hosts) == 5)
76 self.assertTrue(len(self.net.switches) == 2)
77
78 # check compute list result
79 self.assertTrue(len(self.dc[0].listCompute()) == 2)
80 self.assertTrue(len(self.dc[1].listCompute()) == 1)
81 self.assertTrue(isinstance(self.dc[0].listCompute()[0], EmulatorCompute))
82 self.assertTrue(isinstance(self.dc[0].listCompute()[1], EmulatorCompute))
83 self.assertTrue(isinstance(self.dc[1].listCompute()[0], EmulatorCompute))
84 self.assertTrue(self.dc[0].listCompute()[1].name == "vnf1")
85 self.assertTrue(self.dc[0].listCompute()[0].name == "vnf2")
86 self.assertTrue(self.dc[1].listCompute()[0].name == "vnf3")
87
88 # check connectivity by using ping
89 self.assertTrue(self.net.ping([self.dc[0].listCompute()[1], self.dc[0].listCompute()[0]]) <= 0.0)
90 self.assertTrue(self.net.ping([self.dc[0].listCompute()[0], self.dc[1].listCompute()[0]]) <= 0.0)
91 self.assertTrue(self.net.ping([self.dc[1].listCompute()[0], self.dc[0].listCompute()[1]]) <= 0.0)
92
93 print('network add vnf1 vnf2->>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>')
94 print('->>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>')
95 subprocess.call("son-emu-cli network add -src vnf1 -dst vnf2 -b -c 10", shell=True)
96 print('network remove vnf1 vnf2->>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>')
97 print('->>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>')
98 subprocess.call("son-emu-cli network remove -src vnf1 -dst vnf2 -b", shell=True)
99
100 print('>>>>> checking --> son-emu-cli compute stop -d datacenter0 -n vnf2 ->>>>>>')
101 print('->>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>')
102 subprocess.call("son-emu-cli compute stop -d datacenter0 -n vnf2", shell=True)
103
104 # check number of running nodes
105 self.assertTrue(len(self.getContainernetContainers()) == 2)
106 self.assertTrue(len(self.net.hosts) == 4)
107 self.assertTrue(len(self.net.switches) == 2)
108 # check compute list result
109 self.assertTrue(len(self.dc[0].listCompute()) == 1)
110 self.assertTrue(len(self.dc[1].listCompute()) == 1)
111
112 print('>>>>> checking --> son-emu-cli compute list ->>>>>>>>>>>>>>>>>>>>>>>>>>>>>')
113 print('->>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>')
114 subprocess.call("son-emu-cli compute list", shell=True)
115 output = subprocess.check_output("son-emu-cli compute list", shell=True)
116
117 # check datacenter list result
118 self.assertTrue("datacenter0" in output)
119
120 print('>>>>> checking --> son-emu-cli compute status -d datacenter0 -n vnf1 ->>>>')
121 print('->>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>')
122 subprocess.call("son-emu-cli compute status -d datacenter0 -n vnf1", shell=True)
123 output = subprocess.check_output("son-emu-cli compute status -d datacenter0 -n vnf1", shell=True)
124 output= ast.literal_eval(output)
125
126 # check compute status result
127 self.assertTrue(output["name"] == "vnf1")
128 self.assertTrue(output["state"]["Running"])
129
130 print('>>>>> checking --> son-emu-cli datacenter list ->>>>>>>>>>>>>>>>>>>>>>>>>>')
131 print('->>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>')
132 subprocess.call("son-emu-cli datacenter list", shell=True)
133 output = subprocess.check_output("son-emu-cli datacenter list", shell=True)
134
135 # check datacenter list result
136
137 self.assertTrue("datacenter0" in output)
138
139 print('->>>>> checking --> son-emu-cli datacenter status -d datacenter0 ->>>>>>>>')
140 print('->>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>')
141 subprocess.call("son-emu-cli datacenter status -d datacenter0", shell=True)
142 output = subprocess.check_output("son-emu-cli datacenter status -d datacenter0", shell=True)
143
144 # check datacenter status result
145 self.assertTrue("datacenter0" in output)
146
147 self.stopNet()
148
149
150 if __name__ == '__main__':
151 unittest.main()