2 Copyright (c) 2015 SONATA-NFV
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
9 http://www.apache.org/licenses/LICENSE-2.0
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.
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
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).
30 Helper module that implements helpers for test implementations.
38 from emuvim
.dcemulator
.net
import DCNetwork
39 from emuvim
.api
.openstack
.openstack_api_endpoint
import OpenstackApiEndpoint
40 from mininet
.clean
import cleanup
41 from mininet
.node
import Controller
43 class ApiBaseOpenStack(unittest
.TestCase
):
45 Helper class to do basic test setups.
46 s1 -- s2 -- s3 -- ... -- sN
49 def __init__(self
, *args
, **kwargs
):
52 self
.s
= [] # list of switches
53 self
.h
= [] # list of hosts
54 self
.d
= [] # list of docker containers
55 self
.dc
= [] # list of data centers
56 self
.docker_cli
= None
57 super(ApiBaseOpenStack
, self
).__init
__(*args
, **kwargs
)
61 nswitches
=0, ndatacenter
=0, nhosts
=0, ndockers
=0,
62 autolinkswitches
=False, controller
=Controller
, **kwargs
):
64 Creates a Mininet instance and automatically adds some
67 Attention, we should always use Mininet's default controller
68 for our tests. Only use other controllers if you want to test
69 specific controller functionality.
71 self
.net
= DCNetwork(controller
=controller
, **kwargs
)
72 for i
in range(0, ndatacenter
):
73 self
.api
.append(OpenstackApiEndpoint("0.0.0.0", 15000+i
))
76 # start from s1 because ovs does not like to have dpid = 0
77 # and switch name-number is being used by mininet to set the dpid
78 for i
in range(1, nswitches
+1):
79 self
.s
.append(self
.net
.addSwitch('s%d' % i
))
80 # if specified, chain all switches
82 for i
in range(0, len(self
.s
) - 1):
83 self
.net
.addLink(self
.s
[i
], self
.s
[i
+ 1])
84 self
.net
.addLink(self
.s
[2], self
.s
[0]) # link switches s1, s2 and s3
86 # add some data centers
87 for i
in range(0, ndatacenter
):
89 self
.net
.addDatacenter(
91 metadata
={"unittest_dc": i
}))
92 self
.net
.addLink(self
.dc
[0].switch
, self
.s
[0]) # link switches dc0.s1 with s1
93 # connect data centers to the endpoint
94 for i
in range(0, ndatacenter
):
95 self
.api
[i
].connect_datacenter(self
.dc
[i
])
96 self
.api
[i
].connect_dc_network(self
.net
)
98 for i
in range(0, nhosts
):
99 self
.h
.append(self
.net
.addHost('h%d' % i
))
101 for i
in range(0, ndockers
):
102 self
.d
.append(self
.net
.addDocker('d%d' % i
, dimage
="ubuntu:trusty"))
106 i
.start(wait_for_port
=True)
110 i
.manage
.stop_floating_network()
119 def getDockerCli(self
):
121 Helper to interact with local docker instance.
123 if self
.docker_cli
is None:
124 self
.docker_cli
= docker
.Client(
125 base_url
='unix://var/run/docker.sock')
126 return self
.docker_cli
128 def getContainernetContainers(self
):
130 List the containers managed by containernet
132 return self
.getDockerCli().containers(filters
={"label": "com.containernet"})
141 print('->>>>>>> tear everything down ->>>>>>>>>>>>>>>')
142 self
.stopApi() # stop all flask threads
143 self
.stopNet() # stop some mininet and containernet stuff
145 # make sure that all pending docker containers are killed
146 with
open(os
.devnull
, 'w') as devnull
: # kill a possibly running docker process that blocks the open ports
147 subprocess
.call("kill $(netstat -npl | grep '15000' | grep -o -e'[0-9]\+/docker' | grep -o -e '[0-9]\+')",
152 with
open(os
.devnull
, 'w') as devnull
:
154 "sudo docker rm -f $(sudo docker ps --filter 'label=com.containernet' -a -q)",