blob: f8fe978210634dcb9a9e6d4705ac2b9c0cb2e4f3 [file] [log] [blame]
Chetan Gaonkercb122cc2016-05-10 10:58:34 -07001#!/usr/bin/env python
A.R Karthick95d044e2016-06-10 18:44:36 -07002#
Chetan Gaonkercfcce782016-05-10 10:10:42 -07003# Copyright 2016-present Ciena Corporation
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
A.R Karthick95d044e2016-06-10 18:44:36 -07008#
Chetan Gaonkercfcce782016-05-10 10:10:42 -07009# http://www.apache.org/licenses/LICENSE-2.0
A.R Karthick95d044e2016-06-10 18:44:36 -070010#
Chetan Gaonkercfcce782016-05-10 10:10:42 -070011# 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#
Chetan Gaonker93e302d2016-04-05 10:51:07 -070017from argparse import ArgumentParser
A R Karthick14118c62016-07-27 14:54:04 -070018import os,sys,time,socket,errno
ChetanGaonkereadad482016-08-26 01:21:47 -070019import shutil, platform, re
Chetan Gaonker4d842ad2016-04-26 10:04:24 -070020utils_dir = os.path.join( os.path.dirname(os.path.realpath(__file__)), '../utils')
Chetan Gaonker7142a342016-04-07 14:53:12 -070021sys.path.append(utils_dir)
ChetanGaonker68a047f2016-10-12 10:31:48 -070022sys.path.insert(1, '/usr/local/lib/python2.7/dist-packages')
A R Karthick946141b2017-01-24 16:37:47 -080023from OnosCtrl import OnosCtrl, get_mac
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -070024from OltConfig import OltConfig
A R Karthick946141b2017-01-24 16:37:47 -080025from OnosFlowCtrl import OnosFlowCtrl
A R Karthick1f03e912016-05-18 11:39:22 -070026from threadPool import ThreadPool
Chetan Gaonker3533faa2016-04-25 17:50:14 -070027from CordContainer import *
A R Karthicke99ab5c2016-09-30 13:59:57 -070028from CordTestServer import cord_test_server_start,cord_test_server_stop,cord_test_server_shutdown,CORD_TEST_HOST,CORD_TEST_PORT
A R Karthick07608ef2016-08-23 16:51:19 -070029from TestManifest import TestManifest
ChetanGaonkereadad482016-08-26 01:21:47 -070030from docker import Client
31from docker.utils import kwargs_from_env
A R Karthickea8bfce2016-10-13 16:32:07 -070032from Xos import XosServiceProfile
A R Karthick07608ef2016-08-23 16:51:19 -070033try:
34 from Fabric import FabricMAAS
35except:
36 FabricMAAS = None
Chetan Gaonker93e302d2016-04-05 10:51:07 -070037
Chetan Gaonker93e302d2016-04-05 10:51:07 -070038class CordTester(Container):
Chetan Gaonker93e302d2016-04-05 10:51:07 -070039 sandbox = '/root/test'
Chetan Gaonker7142a342016-04-07 14:53:12 -070040 sandbox_setup = '/root/test/src/test/setup'
Chetan Gaonker4d842ad2016-04-26 10:04:24 -070041 tester_base = os.path.dirname(os.path.realpath(__file__))
42 tester_paths = os.path.realpath(__file__).split(os.path.sep)
A R Karthickb7e80902016-05-17 09:38:31 -070043 tester_path_index = tester_paths.index('src') - 1
Chetan Gaonker7142a342016-04-07 14:53:12 -070044 sandbox_host = os.path.sep.join(tester_paths[:tester_path_index+1])
Chetan Gaonker93e302d2016-04-05 10:51:07 -070045
46 host_guest_map = ( (sandbox_host, sandbox),
Chetan Gaonker85b7bd52016-04-20 10:29:12 -070047 ('/lib/modules', '/lib/modules'),
48 ('/var/run/docker.sock', '/var/run/docker.sock')
Chetan Gaonker93e302d2016-04-05 10:51:07 -070049 )
50 basename = 'cord-tester'
A R Karthick36cfcef2016-08-18 15:20:07 -070051 switch_on_olt = False
Chetan Gaonker503032a2016-05-12 12:06:29 -070052 IMAGE = 'cord-test/nose'
ChetanGaonker2c0e9bb2016-09-21 13:38:37 -070053 ALL_TESTS = ('tls', 'dhcp', 'dhcprelay','igmp', 'subscriber',
54 'cordSubscriber', 'vrouter', 'flows', 'proxyarp', 'acl', 'xos', 'fabric',
Chetan Gaonkerefb55282017-01-27 23:07:41 +000055 'cbench', 'cluster', 'netCondition', 'cordvtn', 'iperf', 'mini', 'vsg')
Chetan Gaonker93e302d2016-04-05 10:51:07 -070056
A R Karthicka013a272016-08-16 16:40:19 -070057 def __init__(self, tests, instance = 0, num_instances = 1, ctlr_ip = None,
A R Karthick07608ef2016-08-23 16:51:19 -070058 name = '', image = IMAGE, prefix = '', tag = 'candidate',
A R Karthick85eb1862017-01-23 16:10:57 -080059 env = None, rm = False, update = False, network = None):
A R Karthick1f03e912016-05-18 11:39:22 -070060 self.tests = tests
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -070061 self.ctlr_ip = ctlr_ip
Chetan Gaonker93e302d2016-04-05 10:51:07 -070062 self.rm = rm
A R Karthicka013a272016-08-16 16:40:19 -070063 self.name = name or self.get_name()
A R Karthick07608ef2016-08-23 16:51:19 -070064 super(CordTester, self).__init__(self.name, image = image, prefix = prefix, tag = tag)
Chetan Gaonker93e302d2016-04-05 10:51:07 -070065 host_config = self.create_host_config(host_guest_map = self.host_guest_map, privileged = True)
66 volumes = []
Chetan Gaonkerb84835f2016-04-19 15:12:10 -070067 for _, g in self.host_guest_map:
Chetan Gaonker93e302d2016-04-05 10:51:07 -070068 volumes.append(g)
Chetan Gaonker85b7bd52016-04-20 10:29:12 -070069 if update is True or not self.img_exists():
A R Karthick07608ef2016-08-23 16:51:19 -070070 self.build_image(self.image_name)
A R Karthicka013a272016-08-16 16:40:19 -070071 self.create = True
72 #check if are trying to run tests on existing container
73 if not name or not self.exists():
74 ##Remove test container if any
75 self.remove_container(self.name, force=True)
76 else:
77 self.create = False
A R Karthick078e63a2016-07-28 13:59:31 -070078 self.olt = False
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -070079 if env is not None and env.has_key('OLT_CONFIG'):
80 self.olt = True
A R Karthick078e63a2016-07-28 13:59:31 -070081 olt_conf_file = os.path.join(self.tester_base, 'olt_config.json')
82 olt_config = OltConfig(olt_conf_file)
83 self.port_map, _ = olt_config.olt_port_map()
84 #Try using the host interface in olt conf to setup the switch
A.R Karthick88e80b92016-12-05 20:23:45 -080085 self.switches = self.port_map['switches']
A R Karthick1f03e912016-05-18 11:39:22 -070086 if env is not None:
A.R Karthick88e80b92016-12-05 20:23:45 -080087 env['TEST_SWITCH'] = self.switches[0]
88 env['TEST_SWITCHES'] = ','.join(self.switches)
A R Karthick1f03e912016-05-18 11:39:22 -070089 env['TEST_HOST'] = self.name
90 env['TEST_INSTANCE'] = instance
91 env['TEST_INSTANCES'] = num_instances
A R Karthicka013a272016-08-16 16:40:19 -070092 if self.create:
93 print('Starting test container %s, image %s, tag %s' %(self.name, self.image, self.tag))
94 self.start(rm = False, volumes = volumes, environment = env,
95 host_config = host_config, tty = True)
A R Karthick85eb1862017-01-23 16:10:57 -080096 if network is not None:
97 Container.connect_to_network(self.name, network)
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -070098
A.R Karthicke4631062016-11-03 14:28:19 -070099 def execute_switch(self, cmd, shell = False):
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700100 if self.olt:
101 return os.system(cmd)
102 return self.execute(cmd, shell = shell)
103
A R Karthick946141b2017-01-24 16:37:47 -0800104 def test_flow(self, switch):
105 if not self.olt:
106 return False
107 egress = 1
108 ingress = 2
109 egress_map = { 'ether': '00:00:00:00:00:03', 'ip': '192.168.30.1' }
110 ingress_map = { 'ether': '00:00:00:00:00:04', 'ip': '192.168.40.1' }
111 device_id = 'of:{}'.format(get_mac(switch))
A R Karthick973010f2017-02-06 16:41:51 -0800112 ctlr = self.ctlr_ip.split(',')[0]
A R Karthick946141b2017-01-24 16:37:47 -0800113 flow = OnosFlowCtrl(deviceId = device_id,
114 egressPort = egress,
115 ingressPort = ingress,
116 ethType = '0x800',
117 ipSrc = ('IPV4_SRC', ingress_map['ip']+'/32'),
118 ipDst = ('IPV4_DST', egress_map['ip']+'/32'),
A R Karthick973010f2017-02-06 16:41:51 -0800119 controller = ctlr
A R Karthick946141b2017-01-24 16:37:47 -0800120 )
121 result = flow.addFlow()
122 if result != True:
123 return result
124 time.sleep(1)
125 #find and remove the flow
126 flow_id = flow.findFlow(device_id, IN_PORT = ('port', ingress),
127 ETH_TYPE = ('ethType','0x800'), IPV4_SRC = ('ip', ingress_map['ip']+'/32'),
128 IPV4_DST = ('ip', egress_map['ip']+'/32'))
129 result = False
130 if flow_id:
131 result = True
132 flow.removeFlow(device_id, flow_id)
133 return result
134
135 def ctlr_switch_availability(self, switch):
136 '''Test Add and verify flows with IPv4 selectors'''
137 if not self.olt:
138 return False
139 device_id = 'of:{}'.format(get_mac(switch))
A R Karthick973010f2017-02-06 16:41:51 -0800140 ctlr = self.ctlr_ip.split(',')[0]
141 devices = OnosCtrl.get_devices(controller = ctlr)
A R Karthick946141b2017-01-24 16:37:47 -0800142 if devices:
143 device = filter(lambda d: d['id'] == device_id, devices)
144 return True
145 return False
146
A R Karthick078e63a2016-07-28 13:59:31 -0700147 def start_switch(self, boot_delay = 2):
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700148 """Start OVS"""
149 ##Determine if OVS has to be started locally or not
150 s_file,s_sandbox = ('of-bridge-local.sh',self.tester_base) if self.olt else ('of-bridge.sh',self.sandbox_setup)
A.R Karthick88e80b92016-12-05 20:23:45 -0800151 ovs_cmd = os.path.join(s_sandbox, s_file)
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700152 if self.olt:
A R Karthick36cfcef2016-08-18 15:20:07 -0700153 if CordTester.switch_on_olt is True:
154 return
155 CordTester.switch_on_olt = True
A.R Karthick88e80b92016-12-05 20:23:45 -0800156 ovs_cmd += ' {} {}'.format(len(self.switches), self.ctlr_ip)
157 print('Starting OVS on the host with %d switches for controller: %s' %(len(self.switches), self.ctlr_ip))
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700158 else:
A.R Karthick88e80b92016-12-05 20:23:45 -0800159 ovs_cmd += ' {}'.format(self.switches[0])
160 print('Starting OVS on test container %s for controller: %s' %(self.name, self.ctlr_ip))
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700161 self.execute_switch(ovs_cmd)
A R Karthick946141b2017-01-24 16:37:47 -0800162 time.sleep(5)
163 ## Wait for the controller to see the switch
A.R Karthick88e80b92016-12-05 20:23:45 -0800164 for switch in self.switches:
165 status = 1
166 tries = 0
A R Karthick946141b2017-01-24 16:37:47 -0800167 result = self.ctlr_switch_availability(switch) and self.test_flow(switch)
168 if result == True:
169 status = 0
A.R Karthick88e80b92016-12-05 20:23:45 -0800170 while status != 0 and tries < 500:
171 cmd = 'sudo ovs-ofctl dump-flows {0} | grep \"type=0x8942\"'.format(switch)
172 status = self.execute_switch(cmd, shell = True)
173 tries += 1
A R Karthick946141b2017-01-24 16:37:47 -0800174 if status != 0 and tries > 100:
175 if self.ctlr_switch_availability(switch):
176 status = 0
A.R Karthick88e80b92016-12-05 20:23:45 -0800177 if tries % 10 == 0:
178 print('Waiting for test switch %s to be connected to ONOS controller ...' %switch)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700179
A.R Karthick88e80b92016-12-05 20:23:45 -0800180 if status != 0:
181 print('Test Switch %s not connected to ONOS container.'
182 'Please remove ONOS container and restart the test' %switch)
183 if self.rm:
184 self.kill()
185 sys.exit(1)
186 else:
187 print('Test Switch %s connected to ONOS container.' %switch)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700188
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700189 if boot_delay:
190 time.sleep(boot_delay)
191
A R Karthick1f03e912016-05-18 11:39:22 -0700192 def setup_intfs(self, port_num = 0):
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700193 tester_intf_subnet = '192.168.100'
194 res = 0
A.R Karthick88e80b92016-12-05 20:23:45 -0800195 switches = self.port_map['switches']
Chetan Gaonker5209fe82016-04-19 10:09:53 -0700196 start_vlan = self.port_map['start_vlan']
A R Karthick07769362016-07-28 17:36:15 -0700197 start_vlan += port_num
198 uplink = self.port_map['uplink']
199 wan = self.port_map['wan']
A.R Karthick88e80b92016-12-05 20:23:45 -0800200 port_list = self.port_map['switch_port_list'] + self.port_map['switch_relay_port_list']
A R Karthick52414732017-01-31 09:59:47 -0800201 print('Provisioning the ports for the test container\n')
A.R Karthick88e80b92016-12-05 20:23:45 -0800202 for host_intf, ports in port_list:
203 uplink = self.port_map[host_intf]['uplink']
204 for port in ports:
205 guest_if = port
A R Karthickca11f5c2017-01-11 18:01:50 -0800206 local_if = port #'{0}_{1}'.format(guest_if, port_num+1)
A.R Karthick88e80b92016-12-05 20:23:45 -0800207 guest_ip = '{0}.{1}/24'.format(tester_intf_subnet, port_num+1)
208 ##Use pipeworks to configure container interfaces on host/bridge interfaces
209 pipework_cmd = 'pipework {0} -i {1} -l {2} {3} {4}'.format(host_intf, guest_if,
A R Karthick07769362016-07-28 17:36:15 -0700210 local_if, self.name, guest_ip)
A.R Karthick88e80b92016-12-05 20:23:45 -0800211 #if the wan interface is specified for uplink, then use it instead
212 if wan and port == self.port_map[uplink]:
213 pipework_cmd = 'pipework {0} -i {1} -l {2} {3} {4}'.format(wan, guest_if,
214 local_if, self.name, guest_ip)
215 else:
216 if start_vlan != 0:
217 pipework_cmd += ' @{}'.format(start_vlan)
218 start_vlan += 1
219 #print('Running PIPEWORK cmd: %s' %pipework_cmd)
220 res += os.system(pipework_cmd)
221 port_num += 1
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700222
A R Karthick1f03e912016-05-18 11:39:22 -0700223 return res, port_num
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700224
225 @classmethod
A.R Karthick88e80b92016-12-05 20:23:45 -0800226 def get_intf_type(cls, intf):
227 intf_type = 0
228 if os.path.isdir('/sys/class/net/{}/bridge'.format(intf)):
229 intf_type = 1 ##linux bridge
230 else:
231 cmd = 'ovs-vsctl list-br | grep -q "^{0}$"'.format(intf)
232 res = os.system(cmd)
233 if res == 0: ##ovs bridge
234 intf_type = 2
235
236 return intf_type
237
238 @classmethod
A R Karthickb50f5592016-07-26 12:19:29 -0700239 def cleanup_intfs(cls):
240 olt_conf_file = os.path.join(cls.tester_base, 'olt_config.json')
241 olt_config = OltConfig(olt_conf_file)
A R Karthickb03cecd2016-07-27 10:27:55 -0700242 port_map, _ = olt_config.olt_port_map()
A R Karthickb50f5592016-07-26 12:19:29 -0700243 port_num = 0
A R Karthickb50f5592016-07-26 12:19:29 -0700244 start_vlan = port_map['start_vlan']
A R Karthick07769362016-07-28 17:36:15 -0700245 wan = port_map['wan']
A R Karthickb50f5592016-07-26 12:19:29 -0700246 cmds = ()
247 res = 0
A.R Karthick88e80b92016-12-05 20:23:45 -0800248 port_list = port_map['switch_port_list'] + port_map['switch_relay_port_list']
249 for intf_host, ports in port_list:
250 intf_type = cls.get_intf_type(intf_host)
251 for port in ports:
A R Karthickca11f5c2017-01-11 18:01:50 -0800252 local_if = port #'{0}_{1}'.format(port, port_num+1)
A.R Karthick88e80b92016-12-05 20:23:45 -0800253 if intf_type == 0:
254 if start_vlan != 0:
255 cmds = ('ip link del {}.{}'.format(intf_host, start_vlan),)
256 start_vlan += 1
A R Karthickb50f5592016-07-26 12:19:29 -0700257 else:
A.R Karthick88e80b92016-12-05 20:23:45 -0800258 if intf_type == 1:
259 cmds = ('brctl delif {} {}'.format(intf_host, local_if),
260 'ip link del {}'.format(local_if))
261 else:
262 cmds = ('ovs-vsctl del-port {} {}'.format(intf_host, local_if),
263 'ip link del {}'.format(local_if))
A R Karthickb50f5592016-07-26 12:19:29 -0700264
A.R Karthick88e80b92016-12-05 20:23:45 -0800265 for cmd in cmds:
266 res += os.system(cmd)
267 port_num += 1
A R Karthickb50f5592016-07-26 12:19:29 -0700268
269 @classmethod
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700270 def get_name(cls):
271 cnt_name = '/{0}'.format(cls.basename)
272 cnt_name_len = len(cnt_name)
273 names = list(flatten(n['Names'] for n in cls.dckr.containers(all=True)))
274 test_names = filter(lambda n: n.startswith(cnt_name), names)
275 last_cnt_number = 0
276 if test_names:
277 last_cnt_name = reduce(lambda n1, n2: n1 if int(n1[cnt_name_len:]) > \
278 int(n2[cnt_name_len:]) else n2,
279 test_names)
280 last_cnt_number = int(last_cnt_name[cnt_name_len:])
281 test_cnt_name = cls.basename + str(last_cnt_number+1)
282 return test_cnt_name
283
284 @classmethod
285 def build_image(cls, image):
286 print('Building test container docker image %s' %image)
Chetan Gaonkerb6064fa2016-05-02 16:29:57 -0700287 ovs_version = '2.5.0'
288 image_format = (ovs_version,)*4
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700289 dockerfile = '''
290FROM ubuntu:14.04
291MAINTAINER chetan@ciena.com
A R Karthickc762df42016-05-25 10:09:21 -0700292RUN apt-get update && \
293 apt-get install -y git git-core autoconf automake autotools-dev pkg-config \
294 make gcc g++ libtool libc6-dev cmake libpcap-dev libxerces-c2-dev \
295 unzip libpcre3-dev flex bison libboost-dev \
296 python python-pip python-setuptools python-scapy tcpdump doxygen doxypy wget \
297 openvswitch-common openvswitch-switch \
A R Karthick07608ef2016-08-23 16:51:19 -0700298 python-twisted python-sqlite sqlite3 python-pexpect telnet arping isc-dhcp-server \
Chetan Gaonker53f16382017-02-20 20:31:22 +0000299 python-paramiko python-maas-client python-keystoneclient python-neutronclient \
300 python-glanceclient
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700301RUN easy_install nose
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700302RUN mkdir -p /root/ovs
303WORKDIR /root
Chetan Gaonkerb6064fa2016-05-02 16:29:57 -0700304RUN wget http://openvswitch.org/releases/openvswitch-{}.tar.gz -O /root/ovs/openvswitch-{}.tar.gz && \
305(cd /root/ovs && tar zxpvf openvswitch-{}.tar.gz && \
306 cd openvswitch-{} && \
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700307 ./configure --prefix=/usr --sysconfdir=/etc --localstatedir=/var --disable-ssl && make && make install)
308RUN service openvswitch-switch restart || /bin/true
A.R Karthickec5b72a2016-11-03 09:53:07 -0700309RUN pip install scapy==2.3.2 scapy-ssl_tls==1.2.2 monotonic configObj docker-py pyyaml nsenter pyroute2 netaddr python-daemon
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700310RUN mv /usr/sbin/tcpdump /sbin/
311RUN ln -sf /sbin/tcpdump /usr/sbin/tcpdump
A R Karthickf4999472016-07-01 16:42:13 -0700312RUN mv /usr/sbin/dhcpd /sbin/
313RUN ln -sf /sbin/dhcpd /usr/sbin/dhcpd
A R Karthickb7e80902016-05-17 09:38:31 -0700314WORKDIR /root
315RUN wget -nc http://de.archive.ubuntu.com/ubuntu/pool/main/b/bison/bison_2.5.dfsg-2.1_amd64.deb \
316 http://de.archive.ubuntu.com/ubuntu/pool/main/b/bison/libbison-dev_2.5.dfsg-2.1_amd64.deb
317RUN sudo dpkg -i bison_2.5.dfsg-2.1_amd64.deb libbison-dev_2.5.dfsg-2.1_amd64.deb
318RUN rm bison_2.5.dfsg-2.1_amd64.deb libbison-dev_2.5.dfsg-2.1_amd64.deb
319RUN wget -nc http://www.nbee.org/download/nbeesrc-jan-10-2013.zip && \
320 unzip nbeesrc-jan-10-2013.zip && \
321 cd nbeesrc-jan-10-2013/src && cmake . && make && \
322 cp ../bin/libn*.so /usr/local/lib && ldconfig && \
323 cp -R ../include/* /usr/include/
324WORKDIR /root
325RUN git clone https://github.com/CPqD/ofsoftswitch13.git && \
326 cd ofsoftswitch13 && \
A R Karthickb7e80902016-05-17 09:38:31 -0700327 ./boot.sh && \
328 ./configure --prefix=/usr --sysconfdir=/etc --localstatedir=/var --disable-ssl && \
329 make && make install
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700330CMD ["/bin/bash"]
Chetan Gaonkerb6064fa2016-05-02 16:29:57 -0700331'''.format(*image_format)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700332 super(CordTester, cls).build_image(dockerfile, image)
333 print('Done building docker image %s' %image)
334
A R Karthick1f03e912016-05-18 11:39:22 -0700335 def run_tests(self):
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700336 '''Run the list of tests'''
A R Karthick9a5edc42016-08-24 19:10:22 -0700337 res = 0
Thangavelu K Sef6f0a52016-12-14 19:57:05 +0000338 print('Modifying scapy tool files before running a test: %s' %self.tests)
339 self.modify_scapy_files_for_specific_tests()
A R Karthick1f03e912016-05-18 11:39:22 -0700340 print('Running tests: %s' %self.tests)
341 for t in self.tests:
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700342 test = t.split(':')[0]
A R Karthick24f1de62016-05-12 15:16:38 -0700343 test_file = '{}Test.py'.format(test)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700344 if t.find(':') >= 0:
A R Karthick24f1de62016-05-12 15:16:38 -0700345 test_case = '{0}:{1}'.format(test_file, t.split(':')[1])
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700346 else:
347 test_case = test_file
Chetan Gaonker7142a342016-04-07 14:53:12 -0700348 cmd = 'nosetests -v {0}/src/test/{1}/{2}'.format(self.sandbox, test, test_case)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700349 status = self.execute(cmd, shell = True)
A R Karthick9a5edc42016-08-24 19:10:22 -0700350 if status > 255:
351 status = 1
352 res |= status
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700353 print('Test %s %s' %(test_case, 'Success' if status == 0 else 'Failure'))
354 print('Done running tests')
355 if self.rm:
356 print('Removing test container %s' %self.name)
357 self.kill(remove=True)
358
A R Karthick9a5edc42016-08-24 19:10:22 -0700359 return res
360
Thangavelu K Sef6f0a52016-12-14 19:57:05 +0000361 def modify_scapy_files_for_specific_tests(self):
362 name = self.name
363 container_cmd_exec = Container(name = name, image = 'cord-test/nose')
364 tty = False
365 dckr = Client()
366 cmd = 'cp test/src/test/scapy/fields.py /usr/local/lib/python2.7/dist-packages/scapy/fields.py '
367 i = container_cmd_exec.execute(cmd = cmd, tty= tty, stream = True)
368
Chetan Gaonkerfb3cb5e2016-05-06 11:55:44 -0700369 @classmethod
370 def list_tests(cls, tests):
371 print('Listing test cases')
372 for test in tests:
A R Karthick24f1de62016-05-12 15:16:38 -0700373 test_file = '{}Test.py'.format(test)
Chetan Gaonkerfb3cb5e2016-05-06 11:55:44 -0700374 cmd = 'nosetests -v --collect-only {0}/../{1}/{2}'.format(cls.tester_base, test, test_file)
375 os.system(cmd)
376
A R Karthicka013a272016-08-16 16:40:19 -0700377
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700378##default onos/radius/test container images and names
379onos_image_default='onosproject/onos:latest'
A R Karthick07608ef2016-08-23 16:51:19 -0700380nose_image_default= '{}:candidate'.format(CordTester.IMAGE)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700381test_type_default='dhcp'
A.R Karthick95d044e2016-06-10 18:44:36 -0700382onos_app_version = '2.0-SNAPSHOT'
Chetan Gaonker4d842ad2016-04-26 10:04:24 -0700383cord_tester_base = os.path.dirname(os.path.realpath(__file__))
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700384onos_app_file = os.path.abspath('{0}/../apps/ciena-cordigmp-'.format(cord_tester_base) + onos_app_version + '.oar')
A R Karthick81acbff2016-06-17 14:45:16 -0700385cord_test_server_address = '{}:{}'.format(CORD_TEST_HOST, CORD_TEST_PORT)
A R Karthick07608ef2016-08-23 16:51:19 -0700386identity_file_default = '/etc/maas/ansible/id_rsa'
A R Karthicke14fc022016-12-08 14:50:29 -0800387onos_log_level = 'INFO'
A R Karthick07608ef2016-08-23 16:51:19 -0700388
389##sets up the ssh key file for the test container
390def set_ssh_key_file(identity_file):
391 ssh_key_file = None
392 if os.access(identity_file, os.F_OK):
393 ##copy it to setup directory
394 identity_dest = os.path.join(CordTester.tester_base, 'id_rsa')
395 if os.path.abspath(identity_file) != identity_dest:
396 try:
397 shutil.copy(identity_file, identity_dest)
398 ssh_key_file = os.path.join(CordTester.sandbox_setup, 'id_rsa')
399 except: pass
400
401 return ssh_key_file
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700402
403def runTest(args):
Chetan Gaonker823cdc52016-05-09 15:51:23 -0700404 #Start the cord test tcp server
A.R Karthickb17e2022017-01-27 11:29:26 -0800405 test_manifest = TestManifest(args = args)
406 test_server_params = test_manifest.server.split(':')
A R Karthick81acbff2016-06-17 14:45:16 -0700407 test_host = test_server_params[0]
408 test_port = CORD_TEST_PORT
409 if len(test_server_params) > 1:
410 test_port = int(test_server_params[1])
A R Karthick81acbff2016-06-17 14:45:16 -0700411
A R Karthick1f03e912016-05-18 11:39:22 -0700412 test_containers = []
413 #These tests end up restarting ONOS/quagga/radius
A R Karthick4e0c0912016-08-17 16:57:42 -0700414 tests_exempt = ('vrouter', 'cordSubscriber', 'proxyarp', 'dhcprelay')
Chetan Gaonker503032a2016-05-12 12:06:29 -0700415 if args.test_type.lower() == 'all':
416 tests = CordTester.ALL_TESTS
Chetan Gaonker503032a2016-05-12 12:06:29 -0700417 args.quagga = True
418 else:
A R Karthickacae3b42016-05-12 15:27:24 -0700419 tests = args.test_type.split('-')
Chetan Gaonker503032a2016-05-12 12:06:29 -0700420
A R Karthick1f03e912016-05-18 11:39:22 -0700421 tests_parallel = [ t for t in tests if t.split(':')[0] not in tests_exempt ]
422 tests_not_parallel = [ t for t in tests if t.split(':')[0] in tests_exempt ]
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700423 onos_cnt = {'tag':'latest'}
A R Karthick07608ef2016-08-23 16:51:19 -0700424 nose_cnt = {'image': CordTester.IMAGE, 'tag': 'candidate'}
Chetan Gaonker503032a2016-05-12 12:06:29 -0700425 update_map = { 'quagga' : False, 'test' : False, 'radius' : False }
426 update_map[args.update.lower()] = True
A.R Karthick95d044e2016-06-10 18:44:36 -0700427
Chetan Gaonker503032a2016-05-12 12:06:29 -0700428 if args.update.lower() == 'all':
429 for c in update_map.keys():
430 update_map[c] = True
A.R Karthick95d044e2016-06-10 18:44:36 -0700431
A R Karthick07608ef2016-08-23 16:51:19 -0700432 use_manifest = False
433 if args.manifest:
434 if os.access(args.manifest, os.F_OK):
435 ##copy it to setup directory
A.R Karthick8b2f1e32017-02-09 15:10:07 -0800436 dest = os.path.join(CordTester.tester_base,
437 os.path.basename(args.manifest))
A R Karthick07608ef2016-08-23 16:51:19 -0700438 if os.path.abspath(args.manifest) != dest:
439 try:
440 shutil.copy(args.manifest, dest)
441 except: pass
A R Karthick65d950d2016-12-19 19:41:55 -0800442 test_manifest = TestManifest(manifest = dest)
A R Karthick07608ef2016-08-23 16:51:19 -0700443 use_manifest = True
444 else:
445 print('Unable to access test manifest: %s' %args.manifest)
Chetan Gaonkerfb3cb5e2016-05-06 11:55:44 -0700446
A R Karthick65d950d2016-12-19 19:41:55 -0800447 onos_ip = test_manifest.onos_ip
448 radius_ip = test_manifest.radius_ip
449 head_node = test_manifest.head_node
A R Karthick5af23712017-01-20 09:49:24 -0800450 iterations = test_manifest.iterations
A.R Karthick263d3fc2017-01-27 12:52:53 -0800451 onos_cord_loc = test_manifest.onos_cord
A.R Karthickf184b342017-01-27 19:30:50 -0800452 service_profile = test_manifest.service_profile
453 synchronizer = test_manifest.synchronizer
454 onos_cord = None
A R Karthick973010f2017-02-06 16:41:51 -0800455 Onos.update_data_dir(test_manifest.karaf_version)
456
A.R Karthick263d3fc2017-01-27 12:52:53 -0800457 if onos_cord_loc:
458 if onos_cord_loc.find(os.path.sep) < 0:
459 onos_cord_loc = os.path.join(os.getenv('HOME'), onos_cord_loc)
A.R Karthickf184b342017-01-27 19:30:50 -0800460 if not os.access(onos_cord_loc, os.F_OK):
461 print('ONOS cord config location %s is not accessible' %onos_cord_loc)
462 sys.exit(1)
A.R Karthick263d3fc2017-01-27 12:52:53 -0800463 if not onos_ip:
464 ##Unexpected case. Specify the external controller ip when running on cord node
465 print('Specify ONOS ip using \"-e\" option when running the cord-tester on cord node')
466 sys.exit(1)
A.R Karthickf184b342017-01-27 19:30:50 -0800467 if not service_profile:
468 print('Specify service profile location for the ONOS cord instance. Eg: $HOME/service-profile/cord-pod')
469 sys.exit(1)
470 if not synchronizer:
471 print('Specify synchronizer to use for the ONOS cord instance. Eg: vtn, fabric, cord')
472 sys.exit(1)
473 if not os.access(service_profile, os.F_OK):
474 print('Service profile location for ONOS cord instance does not exist')
475 sys.exit(1)
476 onos_cord = OnosCord(onos_ip, onos_cord_loc, service_profile, synchronizer)
A.R Karthick263d3fc2017-01-27 12:52:53 -0800477
478 try:
479 test_server = cord_test_server_start(daemonize = False, cord_test_host = test_host, cord_test_port = test_port,
480 onos_cord = onos_cord)
481 except:
482 ##Most likely a server instance is already running (daemonized earlier)
483 test_server = None
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700484
A R Karthick65d950d2016-12-19 19:41:55 -0800485 Container.IMAGE_PREFIX = test_manifest.image_prefix
486 Onos.MAX_INSTANCES = test_manifest.onos_instances
A R Karthickc69d73e2017-01-20 11:44:34 -0800487 Onos.JVM_HEAP_SIZE = test_manifest.jvm_heap_size
A R Karthick65d950d2016-12-19 19:41:55 -0800488 cluster_mode = True if test_manifest.onos_instances > 1 else False
489 async_mode = cluster_mode and test_manifest.async_mode
490 existing_list = [ c['Names'][0][1:] for c in Container.dckr.containers() if c['Image'] == test_manifest.onos_image ]
491 setup_cluster = False if len(existing_list) == test_manifest.onos_instances else True
A.R Karthickc4e474d2016-12-12 15:24:57 -0800492 onos_ips = []
A R Karthick65d950d2016-12-19 19:41:55 -0800493 if cluster_mode is True and len(existing_list) > 1:
494 ##don't setup cluster config again
495 cluster_mode = False
A R Karthick07608ef2016-08-23 16:51:19 -0700496 if onos_ip is None:
A R Karthick65d950d2016-12-19 19:41:55 -0800497 image_names = test_manifest.onos_image.rsplit(':', 1)
A R Karthick07608ef2016-08-23 16:51:19 -0700498 onos_cnt['image'] = image_names[0]
499 if len(image_names) > 1:
500 if image_names[1].find('/') < 0:
501 onos_cnt['tag'] = image_names[1]
502 else:
503 #tag cannot have slashes
A R Karthick65d950d2016-12-19 19:41:55 -0800504 onos_cnt['image'] = test_manifest.onos_image
A R Karthick07608ef2016-08-23 16:51:19 -0700505
506 Onos.IMAGE = onos_cnt['image']
A R Karthick65d950d2016-12-19 19:41:55 -0800507 Onos.PREFIX = test_manifest.image_prefix
A R Karthick07608ef2016-08-23 16:51:19 -0700508 Onos.TAG = onos_cnt['tag']
A R Karthick65d950d2016-12-19 19:41:55 -0800509 data_volume = '{}-data'.format(Onos.NAME) if test_manifest.shared_volume else None
A R Karthick07608ef2016-08-23 16:51:19 -0700510 onos = Onos(image = Onos.IMAGE,
A.R Karthickc4e474d2016-12-12 15:24:57 -0800511 tag = Onos.TAG, boot_delay = 60, cluster = cluster_mode,
A R Karthick85eb1862017-01-23 16:10:57 -0800512 data_volume = data_volume, async = async_mode, network = test_manifest.docker_network)
A.R Karthickc4e474d2016-12-12 15:24:57 -0800513 if onos.running:
A R Karthick65d950d2016-12-19 19:41:55 -0800514 onos_ips.append(onos.ipaddr)
A.R Karthickc4e474d2016-12-12 15:24:57 -0800515 else:
516 onos_ips.append(onos_ip)
517
A R Karthick65d950d2016-12-19 19:41:55 -0800518 num_onos_instances = test_manifest.onos_instances
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -0700519 if num_onos_instances > 1 and onos is not None:
520 onos_instances = []
521 onos_instances.append(onos)
522 for i in range(1, num_onos_instances):
523 name = '{}-{}'.format(Onos.NAME, i+1)
A R Karthick65d950d2016-12-19 19:41:55 -0800524 data_volume = '{}-data'.format(name) if test_manifest.shared_volume else None
A.R Karthickc4e474d2016-12-12 15:24:57 -0800525 quagga_config = Onos.get_quagga_config(i)
A R Karthickec2db322016-11-17 15:06:01 -0800526 onos = Onos(name = name, image = Onos.IMAGE, tag = Onos.TAG, boot_delay = 60, cluster = cluster_mode,
A R Karthick3b811152016-12-15 10:24:24 -0800527 data_volume = data_volume, async = async_mode,
A R Karthick85eb1862017-01-23 16:10:57 -0800528 quagga_config = quagga_config, network = test_manifest.docker_network)
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -0700529 onos_instances.append(onos)
A.R Karthickc4e474d2016-12-12 15:24:57 -0800530 if onos.running:
531 onos_ips.append(onos.ipaddr)
A R Karthick65d950d2016-12-19 19:41:55 -0800532 if async_mode is True and cluster_mode is True:
A.R Karthickc4e474d2016-12-12 15:24:57 -0800533 Onos.start_cluster_async(onos_instances)
534 if not onos_ips:
535 for onos in onos_instances:
536 onos_ips.append(onos.ipaddr)
A R Karthick65d950d2016-12-19 19:41:55 -0800537 if cluster_mode is True:
538 try:
539 for ip in onos_ips:
540 print('Installing cord tester ONOS app %s in ONOS instance %s' %(args.app,ip))
541 OnosCtrl.install_app(args.app, onos_ip = ip)
542 except: pass
A R Karthick1ef70552016-11-17 17:33:36 -0800543 if setup_cluster is True:
544 Onos.setup_cluster(onos_instances)
545 else:
546 print('ONOS instances already running. Skipping ONOS form cluster for %d instances' %num_onos_instances)
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -0700547 ctlr_addr = ','.join(onos_ips)
Chetan Gaonker5209fe82016-04-19 10:09:53 -0700548
A R Karthick65d950d2016-12-19 19:41:55 -0800549 print('Controller IP %s, Test type %s' %(onos_ips, args.test_type))
550 if onos_ip is not None:
A R Karthickbd9b8a32016-07-21 09:56:45 -0700551 print('Installing ONOS cord apps')
A R Karthick07608ef2016-08-23 16:51:19 -0700552 try:
553 Onos.install_cord_apps(onos_ip = onos_ip)
554 except: pass
A R Karthickeaf1c4e2016-07-19 12:22:35 -0700555
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -0700556 if not cluster_mode:
557 print('Installing cord tester ONOS app %s' %args.app)
558 try:
559 for ip in onos_ips:
A R Karthick65d950d2016-12-19 19:41:55 -0800560 OnosCtrl.install_app(args.app, onos_ip = ip)
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -0700561 except: pass
A R Karthickeaf1c4e2016-07-19 12:22:35 -0700562
563 if radius_ip is None:
A R Karthicka661b552016-05-25 10:18:50 -0700564 ##Start Radius container
A R Karthick85eb1862017-01-23 16:10:57 -0800565 radius = Radius(prefix = Container.IMAGE_PREFIX, update = update_map['radius'],
566 network = test_manifest.docker_network)
A R Karthick75844572017-01-23 16:57:44 -0800567 radius_ip = radius.ip(network = test_manifest.docker_network)
A.R Karthick95d044e2016-06-10 18:44:36 -0700568
A R Karthickeaf1c4e2016-07-19 12:22:35 -0700569 print('Radius server running with IP %s' %radius_ip)
A.R Karthick95d044e2016-06-10 18:44:36 -0700570
Chetan Gaonkerb84835f2016-04-19 15:12:10 -0700571 if args.quagga == True:
572 #Start quagga. Builds container if required
A R Karthick85eb1862017-01-23 16:10:57 -0800573 quagga = Quagga(prefix = Container.IMAGE_PREFIX, update = update_map['quagga'],
574 network = test_manifest.docker_network)
A R Karthick81acbff2016-06-17 14:45:16 -0700575
A R Karthick07608ef2016-08-23 16:51:19 -0700576 try:
577 maas_api_key = FabricMAAS.get_api_key()
578 except:
579 maas_api_key = 'UNKNOWN'
580
581 ssh_key_file = set_ssh_key_file(args.identity_file)
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -0700582 test_cnt_env = { 'ONOS_CONTROLLER_IP' : ctlr_addr,
Chetan Gaonkerc170f3f2016-04-19 17:24:45 -0700583 'ONOS_AAA_IP' : radius_ip if radius_ip is not None else '',
A R Karthick8d03cc52016-06-28 14:51:59 -0700584 'QUAGGA_IP': test_host,
A R Karthick81acbff2016-06-17 14:45:16 -0700585 'CORD_TEST_HOST' : test_host,
586 'CORD_TEST_PORT' : test_port,
A R Karthick65d950d2016-12-19 19:41:55 -0800587 'ONOS_RESTART' : 0 if test_manifest.olt and args.test_controller else 1,
588 'LOG_LEVEL': test_manifest.log_level,
A R Karthick07608ef2016-08-23 16:51:19 -0700589 'HEAD_NODE': head_node if head_node else CORD_TEST_HOST,
A R Karthick973010f2017-02-06 16:41:51 -0800590 'MAAS_API_KEY': maas_api_key,
591 'KARAF_VERSION' : test_manifest.karaf_version
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700592 }
A R Karthick07608ef2016-08-23 16:51:19 -0700593
594 if ssh_key_file:
595 test_cnt_env['SSH_KEY_FILE'] = ssh_key_file
596
A R Karthick65d950d2016-12-19 19:41:55 -0800597 if test_manifest.olt:
Chetan Gaonker7142a342016-04-07 14:53:12 -0700598 olt_conf_test_loc = os.path.join(CordTester.sandbox_setup, 'olt_config.json')
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700599 test_cnt_env['OLT_CONFIG'] = olt_conf_test_loc
600
A.R Karthick8b2f1e32017-02-09 15:10:07 -0800601 if use_manifest:
602 test_cnt_env['MANIFEST'] = os.path.join(CordTester.sandbox_setup,
603 os.path.basename(args.manifest))
604
A R Karthick5af23712017-01-20 09:49:24 -0800605 if iterations is not None:
606 test_cnt_env['ITERATIONS'] = iterations
607
A R Karthicka013a272016-08-16 16:40:19 -0700608 if args.num_containers > 1 and args.container:
609 print('Cannot specify number of containers with container option')
610 sys.exit(1)
611 if args.container:
612 args.keep = True
A R Karthick1f03e912016-05-18 11:39:22 -0700613 port_num = 0
614 num_tests = len(tests_parallel)
A R Karthick2b93d6a2016-09-06 15:19:09 -0700615 if num_tests > 0 and num_tests < args.num_containers:
616 tests_parallel *= args.num_containers/num_tests
617 num_tests = len(tests_parallel)
A R Karthick1f03e912016-05-18 11:39:22 -0700618 tests_per_container = max(1, num_tests/args.num_containers)
619 test_slice_start = 0
620 test_slice_end = test_slice_start + tests_per_container
621 num_test_containers = min(num_tests, args.num_containers)
622 if tests_parallel:
623 print('Running %s tests across %d containers in parallel' %(tests_parallel, num_test_containers))
624 for container in range(num_test_containers):
625 test_cnt = CordTester(tests_parallel[test_slice_start:test_slice_end],
626 instance = container, num_instances = num_test_containers,
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -0700627 ctlr_ip = ctlr_addr,
A R Karthick07608ef2016-08-23 16:51:19 -0700628 name = args.container,
629 image = nose_cnt['image'],
630 prefix = Container.IMAGE_PREFIX,
631 tag = nose_cnt['tag'],
A R Karthick1f03e912016-05-18 11:39:22 -0700632 env = test_cnt_env,
633 rm = False if args.keep else True,
A R Karthick85eb1862017-01-23 16:10:57 -0800634 update = update_map['test'],
635 network = test_manifest.docker_network)
A R Karthick1f03e912016-05-18 11:39:22 -0700636 test_slice_start = test_slice_end
637 test_slice_end = test_slice_start + tests_per_container
638 update_map['test'] = False
639 test_containers.append(test_cnt)
A R Karthicka013a272016-08-16 16:40:19 -0700640 if not test_cnt.create:
641 continue
A R Karthick65d950d2016-12-19 19:41:55 -0800642 if test_cnt.create and (test_manifest.start_switch or not test_manifest.olt):
A R Karthick07608ef2016-08-23 16:51:19 -0700643 if not args.no_switch:
644 test_cnt.start_switch()
A R Karthicka013a272016-08-16 16:40:19 -0700645 if test_cnt.create and test_cnt.olt:
A R Karthick1f03e912016-05-18 11:39:22 -0700646 _, port_num = test_cnt.setup_intfs(port_num = port_num)
647
A R Karthick9a5edc42016-08-24 19:10:22 -0700648 status = 0
649 if len(test_containers) > 1:
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -0700650 if True:
651 status = test_containers[0].run_tests()
652 else:
653 thread_pool = ThreadPool(len(test_containers), queue_size = 1, wait_timeout=1)
654 for test_cnt in test_containers:
655 thread_pool.addTask(test_cnt.run_tests)
656 thread_pool.cleanUpThreads()
A R Karthick9a5edc42016-08-24 19:10:22 -0700657 else:
A R Karthickcee37412016-08-29 10:10:56 -0700658 if test_containers:
659 status = test_containers[0].run_tests()
A R Karthick1f03e912016-05-18 11:39:22 -0700660
661 ##Run the linear tests
662 if tests_not_parallel:
663 test_cnt = CordTester(tests_not_parallel,
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -0700664 ctlr_ip = ctlr_addr,
A R Karthick07608ef2016-08-23 16:51:19 -0700665 name = args.container,
666 image = nose_cnt['image'],
667 prefix = Container.IMAGE_PREFIX,
668 tag = nose_cnt['tag'],
A R Karthick1f03e912016-05-18 11:39:22 -0700669 env = test_cnt_env,
670 rm = False if args.keep else True,
A R Karthick85eb1862017-01-23 16:10:57 -0800671 update = update_map['test'],
672 network = test_manifest.docker_network)
A R Karthick65d950d2016-12-19 19:41:55 -0800673 if test_cnt.create and (test_manifest.start_switch or not test_manifest.olt):
A R Karthick36cfcef2016-08-18 15:20:07 -0700674 #For non parallel tests, we just restart the switch also for OLT's
675 CordTester.switch_on_olt = False
A R Karthick07608ef2016-08-23 16:51:19 -0700676 if not args.no_switch:
677 test_cnt.start_switch()
A R Karthicka013a272016-08-16 16:40:19 -0700678 if test_cnt.create and test_cnt.olt:
A R Karthick1f03e912016-05-18 11:39:22 -0700679 test_cnt.setup_intfs(port_num = port_num)
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -0700680 test_cnt.run_tests()
A R Karthick1f03e912016-05-18 11:39:22 -0700681
A R Karthick81acbff2016-06-17 14:45:16 -0700682 if test_server:
A.R Karthickf184b342017-01-27 19:30:50 -0800683 if onos_cord:
684 onos_cord.restore()
A R Karthick81acbff2016-06-17 14:45:16 -0700685 cord_test_server_stop(test_server)
686
A R Karthick9a5edc42016-08-24 19:10:22 -0700687 return status
688
A R Karthick81acbff2016-06-17 14:45:16 -0700689##Starts onos/radius/quagga containers as appropriate
690def setupCordTester(args):
691 onos_cnt = {'tag':'latest'}
A R Karthick07608ef2016-08-23 16:51:19 -0700692 nose_cnt = {'image': CordTester.IMAGE, 'tag': 'candidate'}
A R Karthick92a0e5a2016-06-22 17:11:05 -0700693 update_map = { 'quagga' : False, 'radius' : False, 'test': False }
A R Karthick81acbff2016-06-17 14:45:16 -0700694 update_map[args.update.lower()] = True
A R Karthick65d950d2016-12-19 19:41:55 -0800695 test_manifest = TestManifest(args = args)
696
A R Karthick81acbff2016-06-17 14:45:16 -0700697 if args.update.lower() == 'all':
698 for c in update_map.keys():
699 update_map[c] = True
700
A R Karthick07608ef2016-08-23 16:51:19 -0700701 use_manifest = False
702 if args.manifest:
703 if os.access(args.manifest, os.F_OK):
704 ##copy it to setup directory
A.R Karthick8b2f1e32017-02-09 15:10:07 -0800705 dest = os.path.join(CordTester.tester_base,
706 os.path.basename(args.manifest))
A R Karthick07608ef2016-08-23 16:51:19 -0700707 if os.path.abspath(args.manifest) != dest:
708 try:
709 shutil.copy(args.manifest, dest)
710 except: pass
A R Karthick65d950d2016-12-19 19:41:55 -0800711 test_manifest = TestManifest(manifest = dest)
A R Karthick07608ef2016-08-23 16:51:19 -0700712 use_manifest = True
713
A.R Karthickf184b342017-01-27 19:30:50 -0800714 onos_ip = test_manifest.onos_ip
715 radius_ip = test_manifest.radius_ip
716 head_node = test_manifest.head_node
717 iterations = test_manifest.iterations
718 service_profile = test_manifest.service_profile
719 synchronizer = test_manifest.synchronizer
720 onos_cord = None
A.R Karthick263d3fc2017-01-27 12:52:53 -0800721 onos_cord_loc = test_manifest.onos_cord
A R Karthick973010f2017-02-06 16:41:51 -0800722 Onos.update_data_dir(test_manifest.karaf_version)
723
A.R Karthick263d3fc2017-01-27 12:52:53 -0800724 if onos_cord_loc:
725 if onos_cord_loc.find(os.path.sep) < 0:
726 onos_cord_loc = os.path.join(os.getenv('HOME'), onos_cord_loc)
727 if not os.access(onos_cord_loc, os.F_OK):
728 print('ONOS cord config location %s is not accessible' %onos_cord_loc)
729 sys.exit(1)
A.R Karthick263d3fc2017-01-27 12:52:53 -0800730 if not onos_ip:
A R Karthickd44cea12016-07-20 12:16:41 -0700731 ##Unexpected case. Specify the external controller ip when running on cord node
732 print('Specify ONOS ip using \"-e\" option when running the cord-tester on cord node')
733 sys.exit(1)
A.R Karthickf184b342017-01-27 19:30:50 -0800734 if not service_profile:
735 print('Specify service profile location for the ONOS cord instance. Eg: $HOME/service-profile/cord-pod')
736 sys.exit(1)
737 if not synchronizer:
738 print('Specify synchronizer to use for the ONOS cord instance. Eg: vtn, fabric, cord')
739 sys.exit(1)
740 if not os.access(service_profile, os.F_OK):
741 print('Service profile location for ONOS cord instance does not exist')
742 sys.exit(1)
743 onos_cord = OnosCord(onos_ip, onos_cord_loc, service_profile, synchronizer)
A R Karthickd44cea12016-07-20 12:16:41 -0700744
A R Karthick65d950d2016-12-19 19:41:55 -0800745 Container.IMAGE_PREFIX = test_manifest.image_prefix
A R Karthick81acbff2016-06-17 14:45:16 -0700746 #don't spawn onos if the user had started it externally
A R Karthick65d950d2016-12-19 19:41:55 -0800747 image_names = test_manifest.onos_image.rsplit(':', 1)
A R Karthick07608ef2016-08-23 16:51:19 -0700748 onos_cnt['image'] = image_names[0]
749 if len(image_names) > 1:
750 if image_names[1].find('/') < 0:
751 onos_cnt['tag'] = image_names[1]
752 else:
753 #tag cannot have slashes
A R Karthick65d950d2016-12-19 19:41:55 -0800754 onos_cnt['image'] = test_manifest.onos_image
A R Karthick81acbff2016-06-17 14:45:16 -0700755
A R Karthick07608ef2016-08-23 16:51:19 -0700756 Onos.IMAGE = onos_cnt['image']
A R Karthick65d950d2016-12-19 19:41:55 -0800757 Onos.PREFIX = test_manifest.image_prefix
A R Karthick07608ef2016-08-23 16:51:19 -0700758 Onos.TAG = onos_cnt['tag']
A R Karthick65d950d2016-12-19 19:41:55 -0800759 Onos.MAX_INSTANCES = test_manifest.onos_instances
A R Karthickc69d73e2017-01-20 11:44:34 -0800760 Onos.JVM_HEAP_SIZE = test_manifest.jvm_heap_size
A R Karthick65d950d2016-12-19 19:41:55 -0800761 cluster_mode = True if test_manifest.onos_instances > 1 else False
762 async_mode = cluster_mode and test_manifest.async_mode
763 existing_list = [ c['Names'][0][1:] for c in Container.dckr.containers() if c['Image'] == test_manifest.onos_image ]
764 setup_cluster = False if len(existing_list) == test_manifest.onos_instances else True
A R Karthickc41c2422016-12-09 10:59:19 -0800765 #cleanup existing volumes before forming a new cluster
766 if setup_cluster is True:
767 print('Cleaning up existing cluster volumes')
768 data_dir = os.path.join(Onos.setup_dir, 'cord-onos*-data')
769 try:
770 os.system('rm -rf {}'.format(data_dir))
771 except: pass
772
A R Karthick2b93d6a2016-09-06 15:19:09 -0700773 onos = None
A.R Karthickc4e474d2016-12-12 15:24:57 -0800774 onos_ips = []
A R Karthick81acbff2016-06-17 14:45:16 -0700775 if onos_ip is None:
A R Karthick65d950d2016-12-19 19:41:55 -0800776 data_volume = '{}-data'.format(Onos.NAME) if test_manifest.shared_volume else None
A R Karthickec2db322016-11-17 15:06:01 -0800777 onos = Onos(image = Onos.IMAGE, tag = Onos.TAG, boot_delay = 60, cluster = cluster_mode,
A R Karthick85eb1862017-01-23 16:10:57 -0800778 data_volume = data_volume, async = async_mode, network = test_manifest.docker_network)
A.R Karthickc4e474d2016-12-12 15:24:57 -0800779 if onos.running:
A R Karthick65d950d2016-12-19 19:41:55 -0800780 onos_ips.append(onos.ipaddr)
A.R Karthickc4e474d2016-12-12 15:24:57 -0800781 else:
782 onos_ips.append(onos_ip)
A R Karthick81acbff2016-06-17 14:45:16 -0700783
A R Karthick65d950d2016-12-19 19:41:55 -0800784 num_onos_instances = test_manifest.onos_instances
A R Karthick2b93d6a2016-09-06 15:19:09 -0700785 if num_onos_instances > 1 and onos is not None:
786 onos_instances = []
787 onos_instances.append(onos)
788 for i in range(1, num_onos_instances):
789 name = '{}-{}'.format(Onos.NAME, i+1)
A R Karthick65d950d2016-12-19 19:41:55 -0800790 data_volume = '{}-data'.format(name) if test_manifest.shared_volume else None
A.R Karthickc4e474d2016-12-12 15:24:57 -0800791 quagga_config = Onos.get_quagga_config(i)
A R Karthickec2db322016-11-17 15:06:01 -0800792 onos = Onos(name = name, image = Onos.IMAGE, tag = Onos.TAG, boot_delay = 60, cluster = cluster_mode,
A R Karthick3b811152016-12-15 10:24:24 -0800793 data_volume = data_volume, async = async_mode,
A R Karthick85eb1862017-01-23 16:10:57 -0800794 quagga_config = quagga_config, network = test_manifest.docker_network)
A R Karthick2b93d6a2016-09-06 15:19:09 -0700795 onos_instances.append(onos)
A.R Karthickc4e474d2016-12-12 15:24:57 -0800796 if onos.running:
797 onos_ips.append(onos.ipaddr)
798 if async_mode is True:
799 Onos.start_cluster_async(onos_instances)
800 if not onos_ips:
801 for onos in onos_instances:
802 onos_ips.append(onos.ipaddr)
A R Karthick51e6fd82016-11-22 14:39:19 -0800803 if setup_cluster is True:
804 Onos.setup_cluster(onos_instances)
A R Karthick2b93d6a2016-09-06 15:19:09 -0700805
806 ctlr_addr = ','.join(onos_ips)
807 print('Onos IP %s' %ctlr_addr)
A R Karthick65d950d2016-12-19 19:41:55 -0800808 if onos_ip is not None:
A R Karthickbd9b8a32016-07-21 09:56:45 -0700809 print('Installing ONOS cord apps')
A R Karthick07608ef2016-08-23 16:51:19 -0700810 try:
811 Onos.install_cord_apps(onos_ip = onos_ip)
812 except: pass
A R Karthickbd9b8a32016-07-21 09:56:45 -0700813
A R Karthickedab01c2016-09-08 14:05:44 -0700814 print('Installing cord tester ONOS app %s' %args.app)
A R Karthick07608ef2016-08-23 16:51:19 -0700815 try:
A R Karthick2b93d6a2016-09-06 15:19:09 -0700816 for ip in onos_ips:
817 OnosCtrl.install_app(args.app, onos_ip = ip)
A R Karthick07608ef2016-08-23 16:51:19 -0700818 except: pass
A R Karthickeaf1c4e2016-07-19 12:22:35 -0700819
A R Karthick81acbff2016-06-17 14:45:16 -0700820 ##Start Radius container if not started
821 if radius_ip is None:
A R Karthick85eb1862017-01-23 16:10:57 -0800822 radius = Radius(prefix = Container.IMAGE_PREFIX, update = update_map['radius'],
823 network = test_manifest.docker_network)
A R Karthick75844572017-01-23 16:57:44 -0800824 radius_ip = radius.ip(network = test_manifest.docker_network)
A R Karthick81acbff2016-06-17 14:45:16 -0700825
826 print('Radius server running with IP %s' %radius_ip)
A R Karthick81acbff2016-06-17 14:45:16 -0700827
828 if args.quagga == True:
829 #Start quagga. Builds container if required
A R Karthick85eb1862017-01-23 16:10:57 -0800830 quagga = Quagga(prefix = Container.IMAGE_PREFIX, update = update_map['quagga'],
831 network = test_manifest.docker_network)
A R Karthick8d03cc52016-06-28 14:51:59 -0700832 print('Quagga started')
A R Karthick81acbff2016-06-17 14:45:16 -0700833
A.R Karthickb17e2022017-01-27 11:29:26 -0800834 params = test_manifest.server.split(':')
A R Karthick81acbff2016-06-17 14:45:16 -0700835 ip = params[0]
836 port = CORD_TEST_PORT
837 if len(params) > 1:
838 port = int(params[1])
A R Karthick92a0e5a2016-06-22 17:11:05 -0700839
A R Karthick07608ef2016-08-23 16:51:19 -0700840 try:
841 maas_api_key = FabricMAAS.get_api_key()
842 except:
843 maas_api_key = 'UNKNOWN'
844
845 ssh_key_file = set_ssh_key_file(args.identity_file)
846
A R Karthick92a0e5a2016-06-22 17:11:05 -0700847 #provision the test container
848 if not args.dont_provision:
A R Karthick2b93d6a2016-09-06 15:19:09 -0700849 test_cnt_env = { 'ONOS_CONTROLLER_IP' : ctlr_addr,
A R Karthick92a0e5a2016-06-22 17:11:05 -0700850 'ONOS_AAA_IP' : radius_ip,
A R Karthick8d03cc52016-06-28 14:51:59 -0700851 'QUAGGA_IP': ip,
A R Karthick92a0e5a2016-06-22 17:11:05 -0700852 'CORD_TEST_HOST' : ip,
853 'CORD_TEST_PORT' : port,
A R Karthick65d950d2016-12-19 19:41:55 -0800854 'ONOS_RESTART' : 0 if test_manifest.olt and args.test_controller else 1,
855 'LOG_LEVEL': test_manifest.log_level,
A R Karthick07608ef2016-08-23 16:51:19 -0700856 'HEAD_NODE': head_node if head_node else CORD_TEST_HOST,
A R Karthick973010f2017-02-06 16:41:51 -0800857 'MAAS_API_KEY': maas_api_key,
858 'KARAF_VERSION' : test_manifest.karaf_version
A R Karthick92a0e5a2016-06-22 17:11:05 -0700859 }
A R Karthick07608ef2016-08-23 16:51:19 -0700860
861 if ssh_key_file:
862 test_cnt_env['SSH_KEY_FILE'] = ssh_key_file
A R Karthick65d950d2016-12-19 19:41:55 -0800863 if test_manifest.olt:
A R Karthick92a0e5a2016-06-22 17:11:05 -0700864 olt_conf_test_loc = os.path.join(CordTester.sandbox_setup, 'olt_config.json')
865 test_cnt_env['OLT_CONFIG'] = olt_conf_test_loc
A R Karthick5af23712017-01-20 09:49:24 -0800866 if test_manifest.iterations is not None:
867 test_cnt_env['ITERATIONS'] = iterations
A.R Karthick8b2f1e32017-02-09 15:10:07 -0800868 if use_manifest:
869 test_cnt_env['MANIFEST'] = os.path.join(CordTester.sandbox_setup,
870 os.path.basename(args.manifest))
A R Karthick92a0e5a2016-06-22 17:11:05 -0700871 test_cnt = CordTester((),
A R Karthick2b93d6a2016-09-06 15:19:09 -0700872 ctlr_ip = ctlr_addr,
A R Karthick92a0e5a2016-06-22 17:11:05 -0700873 image = nose_cnt['image'],
A R Karthick07608ef2016-08-23 16:51:19 -0700874 prefix = Container.IMAGE_PREFIX,
A R Karthick92a0e5a2016-06-22 17:11:05 -0700875 tag = nose_cnt['tag'],
876 env = test_cnt_env,
877 rm = False,
A R Karthick85eb1862017-01-23 16:10:57 -0800878 update = update_map['test'],
879 network = test_manifest.docker_network)
A R Karthick92a0e5a2016-06-22 17:11:05 -0700880
A R Karthick65d950d2016-12-19 19:41:55 -0800881 if test_manifest.start_switch or not test_manifest.olt:
A R Karthick92a0e5a2016-06-22 17:11:05 -0700882 test_cnt.start_switch()
883 if test_cnt.olt:
884 test_cnt.setup_intfs(port_num = 0)
885 print('Test container %s started and provisioned to run tests using nosetests' %(test_cnt.name))
886
887 #Finally start the test server and daemonize
A R Karthick14118c62016-07-27 14:54:04 -0700888 try:
A R Karthickbd82f362016-11-10 15:08:52 -0800889 cord_test_server_start(daemonize = not args.foreground, cord_test_host = ip, cord_test_port = port,
890 onos_cord = onos_cord, foreground = args.foreground)
A R Karthick14118c62016-07-27 14:54:04 -0700891 except socket.error, e:
892 #the test agent address could be remote or already running. Exit gracefully
893 sys.exit(0)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700894
A R Karthick9a5edc42016-08-24 19:10:22 -0700895 return 0
896
Chetan Gaonker503032a2016-05-12 12:06:29 -0700897def cleanupTests(args):
A R Karthick757eb4d2017-01-09 14:51:16 -0800898 if args.manifest and os.access(args.manifest, os.F_OK):
899 manifest = TestManifest(manifest = args.manifest)
900 args.prefix = manifest.image_prefix
901 args.olt = manifest.olt
902 args.onos = manifest.onos_image
903 args.server = manifest.server
A.R Karthickb17e2022017-01-27 11:29:26 -0800904 args.onos_ip = manifest.onos_ip
905 args.radius_ip = manifest.radius_ip
906 args.onos_cord = manifest.onos_cord
A.R Karthickf184b342017-01-27 19:30:50 -0800907 args.service_profile = manifest.service_profile
908 args.synchronizer = manifest.synchronizer
A.R Karthickb17e2022017-01-27 11:29:26 -0800909 else:
910 args.onos_ip = None
911 args.radius_ip = None
912 if args.test_controller:
913 ips = args.test_controller.split('/')
914 args.onos_ip = ips[0]
915 if len(ips) > 1:
916 args.radius_ip = ips[1]
A R Karthick757eb4d2017-01-09 14:51:16 -0800917
A R Karthick2b93d6a2016-09-06 15:19:09 -0700918 image_name = args.onos
A R Karthick07608ef2016-08-23 16:51:19 -0700919 prefix = args.prefix
920 if prefix:
921 prefix += '/'
922 test_container = '{}{}:candidate'.format(prefix, CordTester.IMAGE)
Chetan Gaonker503032a2016-05-12 12:06:29 -0700923 print('Cleaning up Test containers ...')
924 Container.cleanup(test_container)
A R Karthickb50f5592016-07-26 12:19:29 -0700925 if args.olt:
926 print('Cleaning up test container OLT configuration')
927 CordTester.cleanup_intfs()
A R Karthick2b93d6a2016-09-06 15:19:09 -0700928
929 onos_list = [ c['Names'][0][1:] for c in Container.dckr.containers() if c['Image'] == image_name ]
930 if len(onos_list) > 1:
931 for onos in onos_list:
932 Container.dckr.kill(onos)
933 Container.dckr.remove_container(onos, force=True)
A R Karthickec2db322016-11-17 15:06:01 -0800934 for index in range(len(onos_list)):
935 volume = '{}-data'.format(Onos.NAME) if index == 0 else '{}-{}-data'.format(Onos.NAME, index+1)
936 Onos.remove_data_map(volume, Onos.guest_data_dir)
A R Karthick9d48c652016-09-15 09:16:36 -0700937 Onos.cleanup_runtime()
A R Karthickec2db322016-11-17 15:06:01 -0800938
A R Karthick394976f2017-01-31 14:25:16 -0800939 radius_container = '{}{}:candidate'.format(prefix, Radius.IMAGE)
940 quagga_container = '{}{}:candidate'.format(prefix, Quagga.IMAGE)
941 Container.cleanup(radius_container)
942 Container.cleanup(quagga_container)
A.R Karthickb17e2022017-01-27 11:29:26 -0800943 if args.onos_cord:
A.R Karthickf184b342017-01-27 19:30:50 -0800944 #try restoring the onos cord instance
945 try:
946 onos_cord = OnosCord(args.onos_ip, args.onos_cord, args.service_profile, args.synchronizer, start = False)
947 onos_cord.restore(force = True)
948 except Exception as e:
949 print(e)
A.R Karthickb17e2022017-01-27 11:29:26 -0800950
A.R Karthick842f0122016-09-28 14:48:47 -0700951 if args.xos:
952 ##cleanup XOS images
953 xos_images = ( '{}:{}'.format(XosServer.IMAGE,XosServer.TAG),
954 '{}:{}'.format(XosSynchronizerOpenstack.IMAGE,
955 XosSynchronizerOpenstack.TAG),
956 '{}:{}'.format(XosSynchronizerOnboarding.IMAGE,
957 XosSynchronizerOnboarding.TAG),
958 '{}:{}'.format(XosSynchronizerOpenvpn.IMAGE,
959 XosSynchronizerOpenvpn.TAG),
960 '{}:{}'.format(XosPostgresql.IMAGE,
961 XosPostgresql.TAG),
962 '{}:{}'.format(XosSyndicateMs.IMAGE,
963 XosSyndicateMs.TAG),
964 )
965 for img in xos_images:
966 print('Cleaning up XOS image: %s' %img)
967 Container.cleanup(img)
968
A R Karthicke99ab5c2016-09-30 13:59:57 -0700969 server_params = args.server.split(':')
970 server_host = server_params[0]
971 server_port = CORD_TEST_PORT
972 if len(server_params) > 1:
973 server_port = int(server_params[1])
974 cord_test_server_shutdown(server_host, server_port)
A R Karthick9a5edc42016-08-24 19:10:22 -0700975 return 0
Chetan Gaonker503032a2016-05-12 12:06:29 -0700976
977def listTests(args):
978 if args.test == 'all':
979 tests = CordTester.ALL_TESTS
980 else:
A R Karthickacae3b42016-05-12 15:27:24 -0700981 tests = args.test.split('-')
Chetan Gaonker503032a2016-05-12 12:06:29 -0700982 CordTester.list_tests(tests)
A R Karthickcee37412016-08-29 10:10:56 -0700983 return 0
ChetanGaonkereadad482016-08-26 01:21:47 -0700984
985def getMetrics(args):
A R Karthickcee37412016-08-29 10:10:56 -0700986 try:
987 detail = c.inspect_container(args.container)
988 except:
989 print('Unknown container %s' %args.container)
990 return 0
991 user_hz = os.sysconf(os.sysconf_names['SC_CLK_TCK'])
ChetanGaonkereadad482016-08-26 01:21:47 -0700992 state = detail["State"]
993 if bool(state["Paused"]):
994 print("Container is in Paused State")
995 elif bool(state["Running"]):
996 print("Container is in Running State")
997 elif int(state["ExitCode"]) == 0:
998 print("Container is in Stopped State")
999 else:
1000 print("Container is in Crashed State")
1001
A R Karthickcee37412016-08-29 10:10:56 -07001002 print("Ip Address of the container: " +detail['NetworkSettings']['IPAddress'])
ChetanGaonkereadad482016-08-26 01:21:47 -07001003
1004 if bool(detail["State"]["Running"]):
1005 container_id = detail['Id']
1006 cpu_usage = {}
A R Karthickcee37412016-08-29 10:10:56 -07001007 cur_usage = 0
1008 last_usage = 0
1009 for i in range(2):
1010 with open('/sys/fs/cgroup/cpuacct/docker/' + container_id + '/cpuacct.stat', 'r') as f:
1011 for line in f:
1012 m = re.search(r"(system|user)\s+(\d+)", line)
1013 if m:
1014 cpu_usage[m.group(1)] = int(m.group(2))
1015 cpu = cpu_usage["system"] + cpu_usage["user"]
1016 last_usage = cur_usage
1017 cur_usage = cpu
1018 time.sleep(1)
1019 cpu_percent = (cur_usage - last_usage)*100.0/user_hz
1020 print("CPU Usage: %.2f %%" %(cpu_percent))
ChetanGaonkereadad482016-08-26 01:21:47 -07001021 else:
1022 print(0)
1023
1024 if bool(detail["State"]["Running"]):
1025 container_id = detail['Id']
A R Karthickcee37412016-08-29 10:10:56 -07001026 print("Docker Port Info:")
ChetanGaonkereadad482016-08-26 01:21:47 -07001027 cmd = "sudo docker port {}".format(container_id)
1028 os.system(cmd)
1029
1030 if bool(detail["State"]["Running"]):
1031 container_id = detail['Id']
1032 with open('/sys/fs/cgroup/memory/docker/' + container_id + '/memory.stat', 'r') as f:
1033 for line in f:
1034 m = re.search(r"total_rss\s+(\d+)", line)
1035 if m:
A R Karthickcee37412016-08-29 10:10:56 -07001036 mem = int(m.group(1))
1037 print("Memory: %s KB "%(mem/1024.0))
ChetanGaonkereadad482016-08-26 01:21:47 -07001038 o = re.search(r"usage\s+(\d+)", line)
1039 if o:
A R Karthickcee37412016-08-29 10:10:56 -07001040 print("Usage: %s "%(o.group(1)))
ChetanGaonkereadad482016-08-26 01:21:47 -07001041 p = re.search(r"max_usage\s+(\d+)", line)
1042 if p:
A R Karthickcee37412016-08-29 10:10:56 -07001043 print("Max Usage: %s "%(p.group(1)))
ChetanGaonkereadad482016-08-26 01:21:47 -07001044
1045 if bool(detail["State"]["Running"]):
1046 container_id = detail['Id']
1047 with open('/sys/fs/cgroup/cpuacct/docker/' + container_id + '/cpuacct.stat', 'r') as f:
1048 for line in f:
1049 m = re.search(r"user\s+(\d+)", line)
1050 if m:
A R Karthickcee37412016-08-29 10:10:56 -07001051 user_ticks = int(m.group(1))
1052 print("Time spent by running processes: %.2f ms"%(user_ticks*1000.0/user_hz))
1053 print("List Networks:")
ChetanGaonkereadad482016-08-26 01:21:47 -07001054 cmd = "docker network ls"
1055 os.system(cmd)
A R Karthick9a5edc42016-08-24 19:10:22 -07001056 return 0
Chetan Gaonker503032a2016-05-12 12:06:29 -07001057
1058def buildImages(args):
A R Karthick07608ef2016-08-23 16:51:19 -07001059 tag = 'candidate'
1060 prefix = args.prefix
1061 if prefix:
1062 prefix += '/'
Chetan Gaonker503032a2016-05-12 12:06:29 -07001063 if args.image == 'all' or args.image == 'quagga':
A R Karthick07608ef2016-08-23 16:51:19 -07001064 image_name = '{}{}:{}'.format(prefix, Quagga.IMAGE, tag)
1065 Quagga.build_image(image_name)
A.R Karthick95d044e2016-06-10 18:44:36 -07001066
Chetan Gaonker503032a2016-05-12 12:06:29 -07001067 if args.image == 'all' or args.image == 'radius':
A R Karthick07608ef2016-08-23 16:51:19 -07001068 image_name = '{}{}:{}'.format(prefix, Radius.IMAGE, tag)
1069 Radius.build_image(image_name)
Chetan Gaonker503032a2016-05-12 12:06:29 -07001070
1071 if args.image == 'all' or args.image == 'test':
A R Karthick07608ef2016-08-23 16:51:19 -07001072 image_name = '{}{}:{}'.format(prefix, CordTester.IMAGE, tag)
1073 CordTester.build_image(image_name)
Chetan Gaonker503032a2016-05-12 12:06:29 -07001074
A R Karthick9a5edc42016-08-24 19:10:22 -07001075 return 0
1076
A R Karthickbec27762016-07-28 10:59:34 -07001077def startImages(args):
A R Karthickbec27762016-07-28 10:59:34 -07001078 ##starts the latest ONOS image
A R Karthick07608ef2016-08-23 16:51:19 -07001079 onos_cnt = {'tag': 'latest'}
1080 image_names = args.onos.rsplit(':', 1)
1081 onos_cnt['image'] = image_names[0]
1082 if len(image_names) > 1:
1083 if image_names[1].find('/') < 0:
1084 onos_cnt['tag'] = image_names[1]
1085 else:
1086 #tag cannot have slashes
1087 onos_cnt['image'] = args.onos
1088
A R Karthickbec27762016-07-28 10:59:34 -07001089 if args.image == 'all' or args.image == 'onos':
A R Karthick07608ef2016-08-23 16:51:19 -07001090 onos = Onos(image = onos_cnt['image'], tag = onos_cnt['tag'])
A R Karthickbec27762016-07-28 10:59:34 -07001091 print('ONOS started with ip %s' %(onos.ip()))
1092
1093 if args.image == 'all' or args.image == 'quagga':
A R Karthick07608ef2016-08-23 16:51:19 -07001094 quagga = Quagga(prefix = args.prefix)
A R Karthickbec27762016-07-28 10:59:34 -07001095 print('Quagga started with ip %s' %(quagga.ip()))
1096
1097 if args.image == 'all' or args.image == 'radius':
A R Karthick07608ef2016-08-23 16:51:19 -07001098 radius = Radius(prefix = args.prefix)
A R Karthickbec27762016-07-28 10:59:34 -07001099 print('Radius started with ip %s' %(radius.ip()))
1100
A R Karthick9a5edc42016-08-24 19:10:22 -07001101 return 0
1102
A R Karthickea8bfce2016-10-13 16:32:07 -07001103def xosCommand(args):
1104 update = False
1105 profile = args.profile
1106 if args.command == 'update':
1107 update = True
1108 xos = XosServiceProfile(profile = profile, update = update)
1109 if args.command == 'build':
1110 xos.build_images(force = True)
1111 if args.command == 'start':
1112 xos.start_services()
1113 if args.command == 'stop':
1114 xos.stop_services(rm = True)
ChetanGaonker2c0e9bb2016-09-21 13:38:37 -07001115 return 0
1116
Chetan Gaonker93e302d2016-04-05 10:51:07 -07001117if __name__ == '__main__':
Chetan Gaonker678743f2016-04-26 09:54:31 -07001118 parser = ArgumentParser(description='Cord Tester')
Chetan Gaonker503032a2016-05-12 12:06:29 -07001119 subparser = parser.add_subparsers()
1120 parser_run = subparser.add_parser('run', help='Run cord tester')
1121 parser_run.add_argument('-t', '--test-type', default=test_type_default, help='Specify test type or test case to run')
1122 parser_run.add_argument('-o', '--onos', default=onos_image_default, type=str, help='ONOS container image')
Chetan Gaonker503032a2016-05-12 12:06:29 -07001123 parser_run.add_argument('-q', '--quagga',action='store_true',help='Provision quagga container for vrouter')
1124 parser_run.add_argument('-a', '--app', default=onos_app_file, type=str, help='Cord ONOS app filename')
A R Karthick07608ef2016-08-23 16:51:19 -07001125 parser_run.add_argument('-l', '--olt', action='store_true', help='Use OLT config')
Chetan Gaonker503032a2016-05-12 12:06:29 -07001126 parser_run.add_argument('-e', '--test-controller', default='', type=str, help='External test controller ip for Onos and/or radius server. '
Chetan Gaonker5209fe82016-04-19 10:09:53 -07001127 'Eg: 10.0.0.2/10.0.0.3 to specify ONOS and Radius ip to connect')
A R Karthick81acbff2016-06-17 14:45:16 -07001128 parser_run.add_argument('-r', '--server', default=cord_test_server_address, type=str,
1129 help='ip:port address to connect for cord test server for container requests')
Chetan Gaonker503032a2016-05-12 12:06:29 -07001130 parser_run.add_argument('-k', '--keep', action='store_true', help='Keep test container after tests')
1131 parser_run.add_argument('-s', '--start-switch', action='store_true', help='Start OVS when running under OLT config')
1132 parser_run.add_argument('-u', '--update', default='none', choices=['test','quagga','radius', 'all'], type=str, help='Update cord tester container images. '
1133 'Eg: --update=quagga to rebuild quagga image.'
1134 ' --update=radius to rebuild radius server image.'
1135 ' --update=test to rebuild cord test image.(Default)'
1136 ' --update=all to rebuild all cord tester images.')
A R Karthick1f03e912016-05-18 11:39:22 -07001137 parser_run.add_argument('-n', '--num-containers', default=1, type=int,
1138 help='Specify number of test containers to spawn for tests')
A R Karthicka013a272016-08-16 16:40:19 -07001139 parser_run.add_argument('-c', '--container', default='', type=str, help='Test container name for running tests')
A R Karthick07608ef2016-08-23 16:51:19 -07001140 parser_run.add_argument('-m', '--manifest', default='', type=str, help='Provide test configuration manifest')
1141 parser_run.add_argument('-p', '--prefix', default='', type=str, help='Provide container image prefix')
1142 parser_run.add_argument('-d', '--no-switch', action='store_true', help='Dont start test switch.')
1143 parser_run.add_argument('-i', '--identity-file', default=identity_file_default,
1144 type=str, help='ssh identity file to access compute nodes from test container')
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -07001145 parser_run.add_argument('-j', '--onos-instances', default=1, type=int,
1146 help='Specify number to test onos instances to form cluster')
A R Karthick09dbc6d2016-11-22 10:37:42 -08001147 parser_run.add_argument('-v', '--shared-volume', action='store_true', help='Start ONOS cluster instances with shared volume')
A.R Karthickc4e474d2016-12-12 15:24:57 -08001148 parser_run.add_argument('-async', '--async-mode', action='store_true',
1149 help='Start ONOS cluster instances in async mode')
A R Karthicke14fc022016-12-08 14:50:29 -08001150 parser_run.add_argument('-log', '--log-level', default=onos_log_level,
1151 choices=['DEBUG','TRACE','ERROR','WARN','INFO'],
1152 type=str,
1153 help='Specify the log level for the test cases')
A R Karthickc69d73e2017-01-20 11:44:34 -08001154 parser_run.add_argument('-jvm-heap-size', '--jvm-heap-size', default='', type=str, help='ONOS JVM heap size')
A R Karthick44a95602017-01-23 16:17:16 -08001155 parser_run.add_argument('-network', '--network', default='', type=str, help='Docker network to attach')
A.R Karthickb17e2022017-01-27 11:29:26 -08001156 parser_run.add_argument('-onos-cord', '--onos-cord', default='', type=str,
1157 help='Specify config location for ONOS cord when running on podd')
A.R Karthickf184b342017-01-27 19:30:50 -08001158 parser_run.add_argument('-service-profile', '--service-profile', default='', type=str,
1159 help='Specify config location for ONOS cord service profile when running on podd.'
1160 'Eg: $HOME/service-profile/cord-pod')
1161 parser_run.add_argument('-synchronizer', '--synchronizer', default='', type=str,
1162 help='Specify the synchronizer to use for ONOS cord instance when running on podd.'
1163 'Eg: vtn,fabric,cord')
A.R Karthickdda22062017-02-09 14:39:20 -08001164 parser_run.add_argument('-karaf', '--karaf', default='3.0.8', type=str, help='Karaf version for ONOS')
Chetan Gaonker503032a2016-05-12 12:06:29 -07001165 parser_run.set_defaults(func=runTest)
1166
A R Karthick81acbff2016-06-17 14:45:16 -07001167 parser_setup = subparser.add_parser('setup', help='Setup cord tester environment')
1168 parser_setup.add_argument('-o', '--onos', default=onos_image_default, type=str, help='ONOS container image')
1169 parser_setup.add_argument('-r', '--server', default=cord_test_server_address, type=str,
1170 help='ip:port address for cord test server to listen for container restart requests')
1171 parser_setup.add_argument('-q', '--quagga',action='store_true',help='Provision quagga container for vrouter')
1172 parser_setup.add_argument('-a', '--app', default=onos_app_file, type=str, help='Cord ONOS app filename')
1173 parser_setup.add_argument('-e', '--test-controller', default='', type=str, help='External test controller ip for Onos and/or radius server. '
1174 'Eg: 10.0.0.2/10.0.0.3 to specify ONOS and Radius ip to connect')
1175 parser_setup.add_argument('-u', '--update', default='none', choices=['quagga','radius', 'all'], type=str, help='Update cord tester container images. '
1176 'Eg: --update=quagga to rebuild quagga image.'
1177 ' --update=radius to rebuild radius server image.'
1178 ' --update=all to rebuild all cord tester images.')
A R Karthick92a0e5a2016-06-22 17:11:05 -07001179 parser_setup.add_argument('-d', '--dont-provision', action='store_true', help='Dont start test container.')
A R Karthick07608ef2016-08-23 16:51:19 -07001180 parser_setup.add_argument('-l', '--olt', action='store_true', help='Use OLT config')
A R Karthicke14fc022016-12-08 14:50:29 -08001181 parser_setup.add_argument('-log', '--log-level', default=onos_log_level, type=str,
1182 choices=['DEBUG','TRACE','ERROR','WARN','INFO'],
1183 help='Specify the log level for the test cases')
A R Karthick92a0e5a2016-06-22 17:11:05 -07001184 parser_setup.add_argument('-s', '--start-switch', action='store_true', help='Start OVS when running under OLT config')
A.R Karthickb17e2022017-01-27 11:29:26 -08001185 parser_setup.add_argument('-onos-cord', '--onos-cord', default='', type=str,
1186 help='Specify config location for ONOS cord when running on podd')
A.R Karthickf184b342017-01-27 19:30:50 -08001187 parser_setup.add_argument('-service-profile', '--service-profile', default='', type=str,
1188 help='Specify config location for ONOS cord service profile when running on podd.'
1189 'Eg: $HOME/service-profile/cord-pod')
1190 parser_setup.add_argument('-synchronizer', '--synchronizer', default='', type=str,
1191 help='Specify the synchronizer to use for ONOS cord instance when running on podd.'
1192 'Eg: vtn,fabric,cord')
A R Karthick07608ef2016-08-23 16:51:19 -07001193 parser_setup.add_argument('-m', '--manifest', default='', type=str, help='Provide test configuration manifest')
1194 parser_setup.add_argument('-p', '--prefix', default='', type=str, help='Provide container image prefix')
1195 parser_setup.add_argument('-i', '--identity-file', default=identity_file_default,
1196 type=str, help='ssh identity file to access compute nodes from test container')
A R Karthick2b93d6a2016-09-06 15:19:09 -07001197 parser_setup.add_argument('-n', '--onos-instances', default=1, type=int,
A R Karthickbd82f362016-11-10 15:08:52 -08001198 help='Specify number of test onos instances to spawn')
A R Karthick09dbc6d2016-11-22 10:37:42 -08001199 parser_setup.add_argument('-v', '--shared-volume', action='store_true',
1200 help='Start ONOS cluster instances with shared volume')
A.R Karthickc4e474d2016-12-12 15:24:57 -08001201 parser_setup.add_argument('-async', '--async-mode', action='store_true',
1202 help='Start ONOS cluster instances in async mode')
A R Karthickbd82f362016-11-10 15:08:52 -08001203 parser_setup.add_argument('-f', '--foreground', action='store_true', help='Run in foreground')
A R Karthickc69d73e2017-01-20 11:44:34 -08001204 parser_setup.add_argument('-jvm-heap-size', '--jvm-heap-size', default='', type=str, help='ONOS JVM heap size')
A R Karthick44a95602017-01-23 16:17:16 -08001205 parser_setup.add_argument('-network', '--network', default='', type=str, help='Docker network to attach')
A.R Karthickdda22062017-02-09 14:39:20 -08001206 parser_setup.add_argument('-karaf', '--karaf', default='3.0.8', type=str, help='Karaf version for ONOS')
A R Karthick81acbff2016-06-17 14:45:16 -07001207 parser_setup.set_defaults(func=setupCordTester)
1208
ChetanGaonker2c0e9bb2016-09-21 13:38:37 -07001209 parser_xos = subparser.add_parser('xos', help='Building xos into cord tester environment')
A R Karthickea8bfce2016-10-13 16:32:07 -07001210 parser_xos.add_argument('command', choices=['build', 'update', 'start', 'stop'])
1211 parser_xos.add_argument('-p', '--profile', default='cord-pod', type=str, help='Provide service profile')
1212 parser_xos.set_defaults(func=xosCommand)
ChetanGaonker2c0e9bb2016-09-21 13:38:37 -07001213
Chetan Gaonker503032a2016-05-12 12:06:29 -07001214 parser_list = subparser.add_parser('list', help='List test cases')
1215 parser_list.add_argument('-t', '--test', default='all', help='Specify test type to list test cases. '
1216 'Eg: -t tls to list tls test cases.'
1217 ' -t tls-dhcp-vrouter to list tls,dhcp and vrouter test cases.'
1218 ' -t all to list all test cases.')
1219 parser_list.set_defaults(func=listTests)
1220
1221 parser_build = subparser.add_parser('build', help='Build cord test container images')
ChetanGaonker2c0e9bb2016-09-21 13:38:37 -07001222 parser_build.add_argument('image', choices=['quagga', 'radius', 'test','all'])
A.R Karthick842f0122016-09-28 14:48:47 -07001223 parser_build.add_argument('-p', '--prefix', default='', type=str, help='Provide container image prefix')
Chetan Gaonker503032a2016-05-12 12:06:29 -07001224 parser_build.set_defaults(func=buildImages)
1225
ChetanGaonkereadad482016-08-26 01:21:47 -07001226 parser_metrics = subparser.add_parser('metrics', help='Info of container')
1227 parser_metrics.add_argument("container", help="Container name")
1228 parser_metrics.set_defaults(func=getMetrics)
1229
A R Karthickbec27762016-07-28 10:59:34 -07001230 parser_start = subparser.add_parser('start', help='Start cord tester containers')
A R Karthick07608ef2016-08-23 16:51:19 -07001231 parser_start.add_argument('-p', '--prefix', default='', type=str, help='Provide container image prefix')
1232 parser_start.add_argument('-o', '--onos', default=onos_image_default, type=str, help='ONOS container image')
A R Karthickbec27762016-07-28 10:59:34 -07001233 parser_start.add_argument('image', choices=['onos', 'quagga', 'radius', 'all'])
1234 parser_start.set_defaults(func=startImages)
1235
Chetan Gaonker503032a2016-05-12 12:06:29 -07001236 parser_cleanup = subparser.add_parser('cleanup', help='Cleanup test containers')
A R Karthick07608ef2016-08-23 16:51:19 -07001237 parser_cleanup.add_argument('-p', '--prefix', default='', type=str, help='Provide container image prefix')
1238 parser_cleanup.add_argument('-l', '--olt', action = 'store_true', help = 'Cleanup OLT config')
A R Karthick2b93d6a2016-09-06 15:19:09 -07001239 parser_cleanup.add_argument('-o', '--onos', default=onos_image_default, type=str,
1240 help='ONOS container image to cleanup')
A.R Karthick842f0122016-09-28 14:48:47 -07001241 parser_cleanup.add_argument('-x', '--xos', action='store_true',
1242 help='Cleanup XOS containers')
A R Karthicke99ab5c2016-09-30 13:59:57 -07001243 parser_cleanup.add_argument('-r', '--server', default=cord_test_server_address, type=str,
1244 help='ip:port address for cord test server to cleanup')
A.R Karthickb17e2022017-01-27 11:29:26 -08001245 parser_cleanup.add_argument('-e', '--test-controller', default='', type=str,
1246 help='External test controller ip for Onos and/or radius server. '
1247 'Eg: 10.0.0.2/10.0.0.3 to specify ONOS and Radius ip')
1248 parser_cleanup.add_argument('-onos-cord', '--onos-cord', default='', type=str,
1249 help='Specify config location for ONOS cord instance when running on podd to restore')
A.R Karthickf184b342017-01-27 19:30:50 -08001250 parser_cleanup.add_argument('-service-profile', '--service-profile', default='', type=str,
1251 help='Specify config location for ONOS cord service profile when running on podd.'
1252 'Eg: $HOME/service-profile/cord-pod')
1253 parser_cleanup.add_argument('-synchronizer', '--synchronizer', default='', type=str,
1254 help='Specify the synchronizer to use for ONOS cord instance when running on podd.'
1255 'Eg: vtn,fabric,cord')
A R Karthick757eb4d2017-01-09 14:51:16 -08001256 parser_cleanup.add_argument('-m', '--manifest', default='', type=str, help='Provide test manifest')
Chetan Gaonker503032a2016-05-12 12:06:29 -07001257 parser_cleanup.set_defaults(func=cleanupTests)
1258
ChetanGaonkereadad482016-08-26 01:21:47 -07001259 c = Client(**(kwargs_from_env()))
1260
Chetan Gaonker93e302d2016-04-05 10:51:07 -07001261 args = parser.parse_args()
A R Karthick9a5edc42016-08-24 19:10:22 -07001262 res = args.func(args)
1263 sys.exit(res)