blob: 4838e49c74b94b7d3c9fd2f00e731bdd4180528e [file] [log] [blame]
Chetan Gaonkercb122cc2016-05-10 10:58:34 -07001#!/usr/bin/env python
A R Karthick1fa66542017-08-10 14:15:29 -07002# Copyright 2017-present Open Networking Foundation
Chetan Gaonkercfcce782016-05-10 10:10:42 -07003#
4# Licensed under the Apache License, Version 2.0 (the "License");
5# you may not use this file except in compliance with the License.
6# You may obtain a copy of the License at
A.R Karthick95d044e2016-06-10 18:44:36 -07007#
Chetan Gaonkercfcce782016-05-10 10:10:42 -07008# http://www.apache.org/licenses/LICENSE-2.0
A.R Karthick95d044e2016-06-10 18:44:36 -07009#
Chetan Gaonkercfcce782016-05-10 10:10:42 -070010# Unless required by applicable law or agreed to in writing, software
11# distributed under the License is distributed on an "AS IS" BASIS,
12# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13# See the License for the specific language governing permissions and
14# limitations under the License.
15#
Chetan Gaonker93e302d2016-04-05 10:51:07 -070016from argparse import ArgumentParser
A R Karthick14118c62016-07-27 14:54:04 -070017import os,sys,time,socket,errno
ChetanGaonkereadad482016-08-26 01:21:47 -070018import shutil, platform, re
Chetan Gaonker4d842ad2016-04-26 10:04:24 -070019utils_dir = os.path.join( os.path.dirname(os.path.realpath(__file__)), '../utils')
A R Karthick03bd2812017-03-03 17:49:17 -080020cli_dir = os.path.join( os.path.dirname(os.path.realpath(__file__)), '../cli')
Chetan Gaonker7142a342016-04-07 14:53:12 -070021sys.path.append(utils_dir)
A R Karthick03bd2812017-03-03 17:49:17 -080022sys.path.append(cli_dir)
ChetanGaonker68a047f2016-10-12 10:31:48 -070023sys.path.insert(1, '/usr/local/lib/python2.7/dist-packages')
A R Karthick6708f562017-09-05 12:15:31 -070024from CordTestUtils import get_mac, getstatusoutput
A.R Karthickbe7768c2017-03-17 11:39:41 -070025from OnosCtrl import OnosCtrl
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -070026from OltConfig import OltConfig
A R Karthick946141b2017-01-24 16:37:47 -080027from OnosFlowCtrl import OnosFlowCtrl
A R Karthick1f03e912016-05-18 11:39:22 -070028from threadPool import ThreadPool
Chetan Gaonker3533faa2016-04-25 17:50:14 -070029from CordContainer import *
A R Karthicke99ab5c2016-09-30 13:59:57 -070030from 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 -070031from TestManifest import TestManifest
A.R Karthick57fa9372017-05-24 12:47:03 -070032from VolthaCtrl import VolthaService
A R Karthickefcf1ab2017-09-08 18:24:16 -070033from EapolAAA import get_radius_macs, get_radius_networks, radius_restore_users, RADIUS_USER_MAC_START, RADIUS_USER_MAC_END
A R Karthick1555c7c2017-09-07 14:59:41 -070034
A R Karthick6f2ac6f2017-07-26 12:55:24 -070035try:
36 from docker import APIClient as Client
37except:
38 from docker import Client
ChetanGaonkereadad482016-08-26 01:21:47 -070039from docker.utils import kwargs_from_env
A R Karthickea8bfce2016-10-13 16:32:07 -070040from Xos import XosServiceProfile
A R Karthick07608ef2016-08-23 16:51:19 -070041try:
42 from Fabric import FabricMAAS
43except:
44 FabricMAAS = None
Chetan Gaonker93e302d2016-04-05 10:51:07 -070045
Chetan Gaonker93e302d2016-04-05 10:51:07 -070046class CordTester(Container):
Chetan Gaonker93e302d2016-04-05 10:51:07 -070047 sandbox = '/root/test'
Chetan Gaonker7142a342016-04-07 14:53:12 -070048 sandbox_setup = '/root/test/src/test/setup'
Chetan Gaonker4d842ad2016-04-26 10:04:24 -070049 tester_base = os.path.dirname(os.path.realpath(__file__))
50 tester_paths = os.path.realpath(__file__).split(os.path.sep)
A R Karthickb7e80902016-05-17 09:38:31 -070051 tester_path_index = tester_paths.index('src') - 1
Chetan Gaonker7142a342016-04-07 14:53:12 -070052 sandbox_host = os.path.sep.join(tester_paths[:tester_path_index+1])
Chetan Gaonker93e302d2016-04-05 10:51:07 -070053
54 host_guest_map = ( (sandbox_host, sandbox),
Chetan Gaonker85b7bd52016-04-20 10:29:12 -070055 ('/lib/modules', '/lib/modules'),
56 ('/var/run/docker.sock', '/var/run/docker.sock')
Chetan Gaonker93e302d2016-04-05 10:51:07 -070057 )
58 basename = 'cord-tester'
A R Karthick36cfcef2016-08-18 15:20:07 -070059 switch_on_olt = False
A R Karthickf7a613b2017-02-24 09:36:44 -080060 IMAGE = 'cordtest/nose'
ChetanGaonker2c0e9bb2016-09-21 13:38:37 -070061 ALL_TESTS = ('tls', 'dhcp', 'dhcprelay','igmp', 'subscriber',
62 'cordSubscriber', 'vrouter', 'flows', 'proxyarp', 'acl', 'xos', 'fabric',
Chetan Gaonkerefb55282017-01-27 23:07:41 +000063 'cbench', 'cluster', 'netCondition', 'cordvtn', 'iperf', 'mini', 'vsg')
Chetan Gaonker93e302d2016-04-05 10:51:07 -070064
Thangavelu K Sa61d1da2017-09-05 05:32:21 -070065 dhcp_data_dir = os.path.join(tester_base, '..', 'setup')
66 default_config = { 'default-lease-time' : 600, 'max-lease-time' : 7200, }
67 default_options = [ ('subnet-mask', '255.255.255.0'),
Thangavelu K S905e5482017-09-12 06:48:02 -070068 ('broadcast-address', '192.168.1.255'),
69 ('domain-name-servers', '192.168.1.1'),
Thangavelu K Sa61d1da2017-09-05 05:32:21 -070070 ('domain-name', '"mydomain.cord-tester"'),
71 ]
Thangavelu K S905e5482017-09-12 06:48:02 -070072 default_subnet_config = [ ('192.168.1.2',
Thangavelu K Sa61d1da2017-09-05 05:32:21 -070073'''
Thangavelu K S905e5482017-09-12 06:48:02 -070074subnet 192.168.1.0 netmask 255.255.255.0 {
75 range 192.168.1.10 192.168.1.100;
Thangavelu K Sa61d1da2017-09-05 05:32:21 -070076}
77'''), ]
78 host_ip_map = {}
79 relay_interfaces_last = ()
80 interface_to_mac_map = {}
81 configs = {}
82
83
A R Karthicka013a272016-08-16 16:40:19 -070084 def __init__(self, tests, instance = 0, num_instances = 1, ctlr_ip = None,
A R Karthick07608ef2016-08-23 16:51:19 -070085 name = '', image = IMAGE, prefix = '', tag = 'candidate',
A R Karthick1555c7c2017-09-07 14:59:41 -070086 env = None, rm = False, update = False, network = None, radius = None):
A R Karthick1f03e912016-05-18 11:39:22 -070087 self.tests = tests
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -070088 self.ctlr_ip = ctlr_ip
A R Karthick1555c7c2017-09-07 14:59:41 -070089 #Uncomment to enable configuring radius server ports on ovs bridge
90 self.radius = radius
Chetan Gaonker93e302d2016-04-05 10:51:07 -070091 self.rm = rm
A R Karthicke4660f52017-02-23 12:08:41 -080092 self.name = name or self.get_name(num_instances)
A R Karthick07608ef2016-08-23 16:51:19 -070093 super(CordTester, self).__init__(self.name, image = image, prefix = prefix, tag = tag)
Chetan Gaonker93e302d2016-04-05 10:51:07 -070094 host_config = self.create_host_config(host_guest_map = self.host_guest_map, privileged = True)
95 volumes = []
Chetan Gaonkerb84835f2016-04-19 15:12:10 -070096 for _, g in self.host_guest_map:
Chetan Gaonker93e302d2016-04-05 10:51:07 -070097 volumes.append(g)
Chetan Gaonker85b7bd52016-04-20 10:29:12 -070098 if update is True or not self.img_exists():
A R Karthick07608ef2016-08-23 16:51:19 -070099 self.build_image(self.image_name)
A R Karthicka013a272016-08-16 16:40:19 -0700100 self.create = True
101 #check if are trying to run tests on existing container
A R Karthicke4660f52017-02-23 12:08:41 -0800102 if not self.exists():
A R Karthicka013a272016-08-16 16:40:19 -0700103 ##Remove test container if any
104 self.remove_container(self.name, force=True)
105 else:
106 self.create = False
A R Karthicke4660f52017-02-23 12:08:41 -0800107 self.rm = False
A R Karthick078e63a2016-07-28 13:59:31 -0700108 self.olt = False
A R Karthicke8bd80e2017-08-01 12:36:24 -0700109 self.switch_started = False
A.R Karthick5968e0d2017-05-16 14:50:46 -0700110 olt_config_file = 'olt_config.json'
111 if env is not None:
112 if env.has_key('OLT_CONFIG'):
113 self.olt = True
114 if env.has_key('OLT_CONFIG_FILE'):
115 olt_config_file = os.path.basename(env['OLT_CONFIG_FILE'])
A.R Karthick5968e0d2017-05-16 14:50:46 -0700116 olt_conf_file = os.path.join(self.tester_base, olt_config_file)
A R Karthick078e63a2016-07-28 13:59:31 -0700117 olt_config = OltConfig(olt_conf_file)
A R Karthicka4d958a2017-09-05 19:01:37 -0700118 self.olt_conf_file = olt_conf_file
A R Karthick078e63a2016-07-28 13:59:31 -0700119 self.port_map, _ = olt_config.olt_port_map()
A.R Karthick369f89e2017-03-02 15:22:45 -0800120 self.vcpes = olt_config.get_vcpes()
A R Karthick078e63a2016-07-28 13:59:31 -0700121 #Try using the host interface in olt conf to setup the switch
A.R Karthick88e80b92016-12-05 20:23:45 -0800122 self.switches = self.port_map['switches']
A R Karthick168e2342017-08-15 16:13:10 -0700123 voltha_network = VolthaService.get_network('voltha')
124 voltha_rest_ip = VolthaService.get_ip('chameleon')
A R Karthick1f03e912016-05-18 11:39:22 -0700125 if env is not None:
A.R Karthick88e80b92016-12-05 20:23:45 -0800126 env['TEST_SWITCH'] = self.switches[0]
127 env['TEST_SWITCHES'] = ','.join(self.switches)
A R Karthick1f03e912016-05-18 11:39:22 -0700128 env['TEST_HOST'] = self.name
129 env['TEST_INSTANCE'] = instance
130 env['TEST_INSTANCES'] = num_instances
A R Karthick168e2342017-08-15 16:13:10 -0700131 if voltha_rest_ip:
132 env['VOLTHA_HOST'] = voltha_rest_ip
A R Karthicka013a272016-08-16 16:40:19 -0700133 if self.create:
134 print('Starting test container %s, image %s, tag %s' %(self.name, self.image, self.tag))
135 self.start(rm = False, volumes = volumes, environment = env,
136 host_config = host_config, tty = True)
A R Karthick85eb1862017-01-23 16:10:57 -0800137 if network is not None:
138 Container.connect_to_network(self.name, network)
A R Karthick168e2342017-08-15 16:13:10 -0700139 if voltha_network:
140 print('Connecting container to VOLTHA container network %s' %(voltha_network))
141 Container.connect_to_network(self.name, voltha_network)
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700142
A.R Karthicke4631062016-11-03 14:28:19 -0700143 def execute_switch(self, cmd, shell = False):
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700144 if self.olt:
145 return os.system(cmd)
146 return self.execute(cmd, shell = shell)
147
A R Karthick946141b2017-01-24 16:37:47 -0800148 def test_flow(self, switch):
149 if not self.olt:
150 return False
151 egress = 1
152 ingress = 2
153 egress_map = { 'ether': '00:00:00:00:00:03', 'ip': '192.168.30.1' }
154 ingress_map = { 'ether': '00:00:00:00:00:04', 'ip': '192.168.40.1' }
155 device_id = 'of:{}'.format(get_mac(switch))
A R Karthick973010f2017-02-06 16:41:51 -0800156 ctlr = self.ctlr_ip.split(',')[0]
A R Karthick946141b2017-01-24 16:37:47 -0800157 flow = OnosFlowCtrl(deviceId = device_id,
158 egressPort = egress,
159 ingressPort = ingress,
160 ethType = '0x800',
161 ipSrc = ('IPV4_SRC', ingress_map['ip']+'/32'),
162 ipDst = ('IPV4_DST', egress_map['ip']+'/32'),
A R Karthick973010f2017-02-06 16:41:51 -0800163 controller = ctlr
A R Karthick946141b2017-01-24 16:37:47 -0800164 )
165 result = flow.addFlow()
166 if result != True:
167 return result
168 time.sleep(1)
169 #find and remove the flow
170 flow_id = flow.findFlow(device_id, IN_PORT = ('port', ingress),
171 ETH_TYPE = ('ethType','0x800'), IPV4_SRC = ('ip', ingress_map['ip']+'/32'),
172 IPV4_DST = ('ip', egress_map['ip']+'/32'))
173 result = False
174 if flow_id:
175 result = True
176 flow.removeFlow(device_id, flow_id)
177 return result
178
179 def ctlr_switch_availability(self, switch):
180 '''Test Add and verify flows with IPv4 selectors'''
181 if not self.olt:
182 return False
183 device_id = 'of:{}'.format(get_mac(switch))
A R Karthick973010f2017-02-06 16:41:51 -0800184 ctlr = self.ctlr_ip.split(',')[0]
185 devices = OnosCtrl.get_devices(controller = ctlr)
A R Karthick946141b2017-01-24 16:37:47 -0800186 if devices:
187 device = filter(lambda d: d['id'] == device_id, devices)
A R Karthicke8bd80e2017-08-01 12:36:24 -0700188 if device:
189 return True
A R Karthick946141b2017-01-24 16:37:47 -0800190 return False
191
A R Karthicke8bd80e2017-08-01 12:36:24 -0700192 def start_switch(self, manifest, boot_delay = 2):
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700193 """Start OVS"""
194 ##Determine if OVS has to be started locally or not
195 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 -0800196 ovs_cmd = os.path.join(s_sandbox, s_file)
A R Karthicke8bd80e2017-08-01 12:36:24 -0700197 switches = filter(lambda sw: sw.startswith('br-int'), self.switches)
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700198 if self.olt:
A R Karthick36cfcef2016-08-18 15:20:07 -0700199 if CordTester.switch_on_olt is True:
200 return
201 CordTester.switch_on_olt = True
A R Karthicke8bd80e2017-08-01 12:36:24 -0700202 ovs_cmd += ' {} {}'.format(len(switches), self.ctlr_ip)
203 if manifest.voltha_enable and manifest.voltha_loc and Onos.ssl_key:
204 ovs_cmd += ' {}'.format(manifest.voltha_loc)
205 print('Starting OVS on the host with %d switches for controller: %s' %(len(switches), self.ctlr_ip))
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700206 else:
A.R Karthick88e80b92016-12-05 20:23:45 -0800207 ovs_cmd += ' {}'.format(self.switches[0])
208 print('Starting OVS on test container %s for controller: %s' %(self.name, self.ctlr_ip))
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700209 self.execute_switch(ovs_cmd)
A R Karthick946141b2017-01-24 16:37:47 -0800210 time.sleep(5)
211 ## Wait for the controller to see the switch
A R Karthicke8bd80e2017-08-01 12:36:24 -0700212 for switch in switches:
A.R Karthick88e80b92016-12-05 20:23:45 -0800213 status = 1
214 tries = 0
A R Karthick946141b2017-01-24 16:37:47 -0800215 result = self.ctlr_switch_availability(switch) and self.test_flow(switch)
216 if result == True:
217 status = 0
A.R Karthick88e80b92016-12-05 20:23:45 -0800218 while status != 0 and tries < 500:
219 cmd = 'sudo ovs-ofctl dump-flows {0} | grep \"type=0x8942\"'.format(switch)
220 status = self.execute_switch(cmd, shell = True)
221 tries += 1
A R Karthick946141b2017-01-24 16:37:47 -0800222 if status != 0 and tries > 100:
223 if self.ctlr_switch_availability(switch):
224 status = 0
A.R Karthick88e80b92016-12-05 20:23:45 -0800225 if tries % 10 == 0:
226 print('Waiting for test switch %s to be connected to ONOS controller ...' %switch)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700227
A.R Karthick88e80b92016-12-05 20:23:45 -0800228 if status != 0:
229 print('Test Switch %s not connected to ONOS container.'
230 'Please remove ONOS container and restart the test' %switch)
231 if self.rm:
232 self.kill()
233 sys.exit(1)
234 else:
235 print('Test Switch %s connected to ONOS container.' %switch)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700236
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700237 if boot_delay:
238 time.sleep(boot_delay)
239
A R Karthicke8bd80e2017-08-01 12:36:24 -0700240 self.switch_started = True
241
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700242 def setup_dhcpd(self, manifest, boot_delay = 5):
A R Karthickc0a008f2017-10-04 18:35:25 -0700243 return False
A R Karthicka4d958a2017-09-05 19:01:37 -0700244 if not self.olt or not manifest.start_switch:
A R Karthick6708f562017-09-05 12:15:31 -0700245 return False
246 if self.service_running("/usr/sbin/dhcpd"):
247 print('DHCPD already running in container %s' %self.name)
248 return True
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700249 setup_for_relay = self.dhcp_relay_setup()
250 dhcp_start_status = self.dhcpd_start()
251 if setup_for_relay and dhcp_start_status:
252 return True
253 else:
254 return False
255
256 def dhcp_relay_setup(self):
A R Karthick6708f562017-09-05 12:15:31 -0700257 ctlr = self.ctlr_ip.split(',')[0]
A R Karthicka4d958a2017-09-05 19:01:37 -0700258 did = OnosCtrl.get_device_id(controller = ctlr, mfr = 'Nicira', olt_conf_file = self.olt_conf_file)
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700259 self.relay_device_id = did
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700260 if self.port_map:
A R Karthicka4d958a2017-09-05 19:01:37 -0700261 ##get the relay port for the OVS switch
262 relay_port = None
263 for host_intf, ports in self.port_map['switch_relay_port_list']:
264 if host_intf.startswith('br-int'):
265 relay_port = ports[0]
266 break
267
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700268 ##Per subscriber, we use 1 relay port
269 try:
A R Karthicka4d958a2017-09-05 19:01:37 -0700270 relay_port = self.port_map[relay_port]
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700271 except:
272 relay_port = self.port_map['uplink']
273 self.relay_interface_port = relay_port
274 self.relay_interfaces = (self.port_map[self.relay_interface_port],)
275 else:
276 print 'Setup dhcpd we must have port_map'
277 return False
278 if self.port_map:
279 ##generate a ip/mac client virtual interface config for onos
280 interface_list = []
281 for port in self.port_map['ports']:
282 port_num = self.port_map[port]
283 if port_num == self.port_map['uplink']:
284 continue
285 ip = self.get_host_ip(port_num)
286 mac = self.get_mac(port)
287 interface_list.append((port_num, ip, mac))
288
289 #configure dhcp server virtual interface on the same subnet as first client interface
290 relay_ip = self.get_host_ip(interface_list[0][0])
291 relay_mac = self.get_mac(self.port_map[self.relay_interface_port])
292 interface_list.append((self.relay_interface_port, relay_ip, relay_mac))
293 self.onos_interface_load(interface_list)
294 return True
295
A R Karthick6708f562017-09-05 12:15:31 -0700296 def onos_load_config(self, config):
297 ctlr = self.ctlr_ip.split(',')[0]
298 status, code = OnosCtrl.config(config, controller = ctlr)
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700299 if status is False:
300 log_test.info('JSON request returned status %d' %code)
301 assert_equal(status, True)
302 time.sleep(3)
303
A R Karthick6708f562017-09-05 12:15:31 -0700304 def onos_interface_load(self, interface_list):
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700305 interface_dict = { 'ports': {} }
306 for port_num, ip, mac in interface_list:
307 port_map = interface_dict['ports']
A R Karthick6708f562017-09-05 12:15:31 -0700308 port = '{}/{}'.format(self.relay_device_id, port_num)
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700309 port_map[port] = { 'interfaces': [] }
310 interface_list = port_map[port]['interfaces']
311 interface_map = { 'ips' : [ '{}/{}'.format(ip, 24) ],
312 'mac' : mac,
313 'name': 'vir-{}'.format(port_num)
314 }
315 interface_list.append(interface_map)
316
A R Karthick6708f562017-09-05 12:15:31 -0700317 self.onos_load_config(interface_dict)
318 self.configs['interface_config'] = interface_dict
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700319
A R Karthick6708f562017-09-05 12:15:31 -0700320 def get_host_ip(self, port):
321 if self.host_ip_map.has_key(port):
322 return self.host_ip_map[port]
323 self.host_ip_map[port] = '192.168.100.{}'.format(port)
324 return self.host_ip_map[port]
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700325
A R Karthick6708f562017-09-05 12:15:31 -0700326 def get_mac(self, iface):
327 if self.interface_to_mac_map.has_key(iface):
328 return self.interface_to_mac_map[iface]
329 cmd = 'docker exec %s ip link show %s | awk \'/ether/ {print $2}\'' %(self.name, iface)
330 st, mac = getstatusoutput(cmd)
331 assert st == 0, 'Cannot get MAC for interface {}'.format(iface)
332 mac = mac.strip()
333 self.interface_to_mac_map[iface] = mac
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700334 return mac
335
A R Karthick6708f562017-09-05 12:15:31 -0700336 def service_running(self, pattern):
337 st, _ = getstatusoutput('docker exec {} pgrep -f "{}"'.format(self.name, pattern))
338 return True if st == 0 else False
339
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700340 def dhcpd_conf_generate(cls, config = default_config, options = default_options,
341 subnet = default_subnet_config):
342 conf = ''
343 for k, v in config.items():
344 conf += '{} {};\n'.format(k, v)
345
346 opts = ''
347 for k, v in options:
348 opts += 'option {} {};\n'.format(k, v)
349
350 subnet_config = ''
351 for _, v in subnet:
352 subnet_config += '{}\n'.format(v)
353
354 return '{}{}{}'.format(conf, opts, subnet_config)
355
A R Karthick6708f562017-09-05 12:15:31 -0700356 def dhcpd_start(self, intf_list = None,
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700357 config = default_config, options = default_options,
358 subnet = default_subnet_config):
359 '''Start the dhcpd server by generating the conf file'''
360 if intf_list is None:
A R Karthick6708f562017-09-05 12:15:31 -0700361 intf_list = self.relay_interfaces
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700362 ##stop dhcpd if already running
A R Karthick6708f562017-09-05 12:15:31 -0700363 #self.dhcpd_stop()
364 dhcp_conf = self.dhcpd_conf_generate(config = config, options = options,
365 subnet = subnet)
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700366 ##first touch dhcpd.leases if it doesn't exist
A R Karthick6708f562017-09-05 12:15:31 -0700367 lease_file = '{}/dhcpd.leases'.format(self.dhcp_data_dir)
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700368 if os.access(lease_file, os.F_OK) is False:
369 with open(lease_file, 'w') as fd: pass
370
A R Karthick6708f562017-09-05 12:15:31 -0700371 conf_file = '{}/dhcpd.conf'.format(self.dhcp_data_dir)
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700372 with open(conf_file, 'w') as fd:
373 fd.write(dhcp_conf)
374
375 #now configure the dhcpd interfaces for various subnets
376 index = 0
377 intf_info = []
378 for ip,_ in subnet:
379 intf = intf_list[index]
A R Karthick6708f562017-09-05 12:15:31 -0700380 mac = self.get_mac(intf)
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700381 intf_info.append((ip, mac))
382 index += 1
383 cmd = 'ifconfig {} {}'.format(intf, ip)
A R Karthick6708f562017-09-05 12:15:31 -0700384 status = self.execute(cmd, shell = True)
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700385
386 intf_str = ','.join(intf_list)
387 dhcpd_cmd = '/usr/sbin/dhcpd -4 --no-pid -cf {0} -lf {1} {2}'.format('/root/test/src/test/setup/dhcpd.conf','/root/test/src/test/setup/dhcpd.leases', intf_str)
388 print('Starting DHCPD server with command: %s' %dhcpd_cmd)
A R Karthick6708f562017-09-05 12:15:31 -0700389 status = self.execute(dhcpd_cmd, shell = True)
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700390 if status > 255:
391 status = 1
392 else:
393 return False
394 time.sleep(3)
A R Karthick6708f562017-09-05 12:15:31 -0700395 self.relay_interfaces_last = self.relay_interfaces
396 self.relay_interfaces = intf_list
397 #self.onos_dhcp_relay_load(*intf_info[0])
Thangavelu K Sa61d1da2017-09-05 05:32:21 -0700398 return True
399
A.R Karthick369f89e2017-03-02 15:22:45 -0800400 def setup_vcpes(self, port_num = 0):
401 res = 0
402 for vcpe in self.vcpes:
403 port, s_tag, c_tag = vcpe['port'], vcpe['s_tag'], vcpe['c_tag']
404 if os.access('/sys/class/net/{}'.format(port), os.F_OK):
405 guest_port = 'vcpe{}'.format(port_num)
406 port_num += 1
A.R Karthick5968e0d2017-05-16 14:50:46 -0700407 print('Provisioning guest port %s for %s with host port: %s, s_tag: %d, c_tag: %d\n'
408 %(guest_port, self.name, port, s_tag, c_tag))
A.R Karthick369f89e2017-03-02 15:22:45 -0800409 cmd = 'pipework {} -i {} -l {} {} 0.0.0.0/24'.format(port, guest_port, guest_port, self.name)
410 res = os.system(cmd)
411 if res == 0:
412 vlan_outer_port = '{}.{}'.format(guest_port, s_tag)
413 vlan_inner_port = '{}.{}'.format(vlan_outer_port, c_tag)
414 #configure the s_tag/c_tag interfaces inside the guest container
415 cmds = ('ip link set {} up'.format(guest_port),
416 'ip link add link {} name {} type vlan id {}'.format(guest_port,
417 vlan_outer_port,
418 s_tag),
419 'ip link set {} up'.format(vlan_outer_port),
420 'ip link add link {} name {} type vlan id {}'.format(vlan_outer_port,
421 vlan_inner_port,
422 c_tag),
423 'ip link set {} up'.format(vlan_inner_port),
424 )
425 res += self.execute(cmds, shell = True)
426
427 @classmethod
428 def cleanup_vcpes(cls, vcpes):
429 port_num = 0
430 for vcpe in vcpes:
431 port = vcpe['port']
432 if os.access('/sys/class/net/{}'.format(port), os.F_OK):
433 local_port = 'vcpe{}'.format(port_num)
434 cmd = 'ip link del {}'.format(local_port)
435 os.system(cmd)
436 port_num += 1
437
A R Karthick1f03e912016-05-18 11:39:22 -0700438 def setup_intfs(self, port_num = 0):
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700439 tester_intf_subnet = '192.168.100'
440 res = 0
A.R Karthick88e80b92016-12-05 20:23:45 -0800441 switches = self.port_map['switches']
Chetan Gaonker5209fe82016-04-19 10:09:53 -0700442 start_vlan = self.port_map['start_vlan']
A.R Karthick4f583842017-06-09 17:15:47 -0700443 ponsim = self.port_map['ponsim']
A R Karthick07769362016-07-28 17:36:15 -0700444 start_vlan += port_num
445 uplink = self.port_map['uplink']
446 wan = self.port_map['wan']
A.R Karthick4f583842017-06-09 17:15:47 -0700447 if ponsim is True:
448 if not wan:
449 wan = 'ponmgmt'
A.R Karthick369f89e2017-03-02 15:22:45 -0800450 vcpe_port_num = port_num
A.R Karthick88e80b92016-12-05 20:23:45 -0800451 port_list = self.port_map['switch_port_list'] + self.port_map['switch_relay_port_list']
A R Karthickefcf1ab2017-09-08 18:24:16 -0700452 subscribers = self.port_map['num_ports'] * len(self.port_map['switch_port_list'])
453 subscriber_macs = get_radius_macs(subscribers,
454 start = RADIUS_USER_MAC_START,
455 end = RADIUS_USER_MAC_END)
A R Karthick52414732017-01-31 09:59:47 -0800456 print('Provisioning the ports for the test container\n')
A.R Karthick88e80b92016-12-05 20:23:45 -0800457 for host_intf, ports in port_list:
A R Karthicke8bd80e2017-08-01 12:36:24 -0700458 if self.switch_started is False and host_intf.startswith('br-int'):
459 continue
A.R Karthick4f583842017-06-09 17:15:47 -0700460 setup_ponsim = ponsim
461 host_index = 0
462 host_intf_base = 'pon1'
A R Karthickf97ee132017-03-23 09:48:37 -0700463 #if the host interface/switch does not exist, just create a dummy ovs switch
464 #needed if we are running with no-switch option
465 if not os.access('/sys/class/net/{}'.format(host_intf), os.F_OK):
466 os.system('ovs-vsctl add-br {}'.format(host_intf))
A.R Karthick88e80b92016-12-05 20:23:45 -0800467 uplink = self.port_map[host_intf]['uplink']
A.R Karthick4f583842017-06-09 17:15:47 -0700468 if setup_ponsim is True:
469 if host_intf.find('_') < 0:
470 print('Invalid host interface specified with ponsim. Disabling ponsim setup')
471 setup_ponsim = False
472 else:
473 try:
474 host_index = int(host_intf.split('_')[-1])
475 host_intf_base = host_intf.split('_')[0]
476 except:
477 print('Invalid host interface with ponsim. Disabling ponsim setup')
478 setup_ponsim = False
A.R Karthick88e80b92016-12-05 20:23:45 -0800479 for port in ports:
480 guest_if = port
A R Karthicke4660f52017-02-23 12:08:41 -0800481 local_if = 'l{}'.format(port_num+1) #port #'{0}_{1}'.format(guest_if, port_num+1)
A.R Karthick88e80b92016-12-05 20:23:45 -0800482 guest_ip = '{0}.{1}/24'.format(tester_intf_subnet, port_num+1)
A.R Karthick4f583842017-06-09 17:15:47 -0700483 if setup_ponsim is True:
484 if port != self.port_map[uplink]:
485 host_intf = '{}_{}'.format(host_intf_base, host_index)
486 host_index += 1
487
A R Karthickefcf1ab2017-09-08 18:24:16 -0700488 port_mac = ''
489 if len(subscriber_macs) > 0:
490 port_mac = subscriber_macs.pop(0)
491
A.R Karthick88e80b92016-12-05 20:23:45 -0800492 ##Use pipeworks to configure container interfaces on host/bridge interfaces
A R Karthickefcf1ab2017-09-08 18:24:16 -0700493 pipework_cmd = 'pipework {0} -i {1} -l {2} {3} {4} {5}'.format(host_intf, guest_if,
494 local_if, self.name, guest_ip,
495 port_mac)
A.R Karthick88e80b92016-12-05 20:23:45 -0800496 #if the wan interface is specified for uplink, then use it instead
497 if wan and port == self.port_map[uplink]:
A R Karthickefcf1ab2017-09-08 18:24:16 -0700498 pipework_cmd = 'pipework {0} -i {1} -l {2} {3} {4} {5}'.format(wan, guest_if,
499 local_if, self.name, guest_ip,
500 port_mac)
A.R Karthick88e80b92016-12-05 20:23:45 -0800501 else:
502 if start_vlan != 0:
503 pipework_cmd += ' @{}'.format(start_vlan)
504 start_vlan += 1
A.R Karthick4f583842017-06-09 17:15:47 -0700505 print('Running PIPEWORK cmd: %s' %pipework_cmd)
A.R Karthick88e80b92016-12-05 20:23:45 -0800506 res += os.system(pipework_cmd)
507 port_num += 1
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700508
A R Karthicke8bd80e2017-08-01 12:36:24 -0700509 if setup_ponsim is True:
510 ponsim = False
511 wan = None
512
A R Karthick1555c7c2017-09-07 14:59:41 -0700513 if self.switch_started and self.radius:
514 radius_macs = get_radius_macs(len(self.port_map['radius_ports']))
515 radius_intf_index = 0
A R Karthickefcf1ab2017-09-08 18:24:16 -0700516 radius_networks = get_radius_networks(len(self.port_map['switch_radius_port_list']))
517 index = 0
A R Karthick1555c7c2017-09-07 14:59:41 -0700518 for host_intf, ports in self.port_map['switch_radius_port_list']:
A R Karthickefcf1ab2017-09-08 18:24:16 -0700519 prefix, subnet, gw = radius_networks[index]
520 mask = subnet.split('/')[-1]
521 #configure the host interface as the gateway
522 cmds = ( #'ip addr add {}/{} dev {}'.format(gw, mask, host_intf),
523 'ip link set {} up'.format(host_intf),
524 )
525 print('Configuring host interface %s for radius server' %(host_intf))
526 for cmd in cmds:
527 print('Running command: %s' %(cmd))
528 res += os.system(cmd)
529 index += 1
A R Karthick1555c7c2017-09-07 14:59:41 -0700530 for port in ports:
531 guest_if = 'eth{}'.format(radius_intf_index+2)
532 port_index = self.port_map[port]
533 local_if = 'r{}'.format(port_index)
A R Karthickefcf1ab2017-09-08 18:24:16 -0700534 guest_ip = '{}.{}/{}'.format(prefix, port_index, mask)
A R Karthick1555c7c2017-09-07 14:59:41 -0700535 mac = radius_macs[radius_intf_index]
536 radius_intf_index += 1
537 port_num += 1
538 pipework_cmd = 'pipework {0} -i {1} -l {2} {3} {4} {5}'.format(host_intf, guest_if,
539 local_if, self.radius.name,
540 guest_ip, mac)
541 print('Configuring Radius port %s on OVS bridge %s' %(guest_if, host_intf))
542 print('Running pipework command: %s' %(pipework_cmd))
543 res += os.system(pipework_cmd)
A R Karthick4ad56272017-09-09 11:32:33 -0700544 rp_filter_disable = 'docker exec {} '\
545 'sysctl -w net.ipv4.conf.{}.rp_filter=2'.format(self.radius.name,
546 guest_if)
547 print('Disabling rp filter on radius interface %s' %(guest_if))
548 res += os.system(rp_filter_disable)
A R Karthickefcf1ab2017-09-08 18:24:16 -0700549 brd = '{}.255'.format(prefix)
550 brd_cmd = 'docker exec {} ifconfig {} broadcast {} up'.format(self.radius.name,
551 guest_if,
552 brd)
553 print('Setting broadcast address to %s for radius interface %s' %(brd, guest_if))
554 res += os.system(brd_cmd)
A R Karthick1555c7c2017-09-07 14:59:41 -0700555
A.R Karthick369f89e2017-03-02 15:22:45 -0800556 self.setup_vcpes(vcpe_port_num)
A R Karthick1f03e912016-05-18 11:39:22 -0700557 return res, port_num
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700558
559 @classmethod
A.R Karthick88e80b92016-12-05 20:23:45 -0800560 def get_intf_type(cls, intf):
561 intf_type = 0
562 if os.path.isdir('/sys/class/net/{}/bridge'.format(intf)):
563 intf_type = 1 ##linux bridge
564 else:
565 cmd = 'ovs-vsctl list-br | grep -q "^{0}$"'.format(intf)
566 res = os.system(cmd)
567 if res == 0: ##ovs bridge
568 intf_type = 2
569
570 return intf_type
571
572 @classmethod
A.R Karthick5968e0d2017-05-16 14:50:46 -0700573 def cleanup_intfs(cls, olt_conf_file):
574 if not os.access(olt_conf_file, os.F_OK):
575 olt_conf_file = os.path.join(cls.tester_base, os.path.basename(olt_conf_file))
A R Karthickb50f5592016-07-26 12:19:29 -0700576 olt_config = OltConfig(olt_conf_file)
A R Karthickb03cecd2016-07-27 10:27:55 -0700577 port_map, _ = olt_config.olt_port_map()
A.R Karthick369f89e2017-03-02 15:22:45 -0800578 vcpes = olt_config.get_vcpes()
A R Karthickb50f5592016-07-26 12:19:29 -0700579 port_num = 0
A R Karthickb50f5592016-07-26 12:19:29 -0700580 start_vlan = port_map['start_vlan']
A R Karthick07769362016-07-28 17:36:15 -0700581 wan = port_map['wan']
A R Karthickb50f5592016-07-26 12:19:29 -0700582 res = 0
A.R Karthick88e80b92016-12-05 20:23:45 -0800583 port_list = port_map['switch_port_list'] + port_map['switch_relay_port_list']
584 for intf_host, ports in port_list:
585 intf_type = cls.get_intf_type(intf_host)
586 for port in ports:
A R Karthick1555c7c2017-09-07 14:59:41 -0700587 cmds = ()
A R Karthicke4660f52017-02-23 12:08:41 -0800588 local_if = 'l{}'.format(port_num+1) #port #'{0}_{1}'.format(port, port_num+1)
A.R Karthick88e80b92016-12-05 20:23:45 -0800589 if intf_type == 0:
590 if start_vlan != 0:
591 cmds = ('ip link del {}.{}'.format(intf_host, start_vlan),)
592 start_vlan += 1
A R Karthickb50f5592016-07-26 12:19:29 -0700593 else:
A.R Karthick88e80b92016-12-05 20:23:45 -0800594 if intf_type == 1:
595 cmds = ('brctl delif {} {}'.format(intf_host, local_if),
596 'ip link del {}'.format(local_if))
597 else:
598 cmds = ('ovs-vsctl del-port {} {}'.format(intf_host, local_if),
599 'ip link del {}'.format(local_if))
A R Karthickb50f5592016-07-26 12:19:29 -0700600
A.R Karthick88e80b92016-12-05 20:23:45 -0800601 for cmd in cmds:
602 res += os.system(cmd)
603 port_num += 1
A R Karthickb50f5592016-07-26 12:19:29 -0700604
A R Karthick1555c7c2017-09-07 14:59:41 -0700605 for intf_host, ports in port_map['switch_radius_port_list']:
606 intf_type = cls.get_intf_type(intf_host)
607 for port in ports:
608 port_index = port_map[port]
609 local_if = 'r{}'.format(port_index)
610 cmds = ()
611 if intf_type == 2:
612 cmds = ('ovs-vsctl del-port {} {}'.format(intf_host, local_if),
613 'ip link del {}'.format(local_if))
614
615 for cmd in cmds:
616 res += os.system(cmd)
617
A.R Karthick369f89e2017-03-02 15:22:45 -0800618 cls.cleanup_vcpes(vcpes)
619
A R Karthickb50f5592016-07-26 12:19:29 -0700620 @classmethod
A R Karthicke4660f52017-02-23 12:08:41 -0800621 def get_name(cls, num_instances):
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700622 cnt_name = '/{0}'.format(cls.basename)
623 cnt_name_len = len(cnt_name)
624 names = list(flatten(n['Names'] for n in cls.dckr.containers(all=True)))
625 test_names = filter(lambda n: n.startswith(cnt_name), names)
626 last_cnt_number = 0
627 if test_names:
628 last_cnt_name = reduce(lambda n1, n2: n1 if int(n1[cnt_name_len:]) > \
629 int(n2[cnt_name_len:]) else n2,
630 test_names)
631 last_cnt_number = int(last_cnt_name[cnt_name_len:])
A R Karthicke4660f52017-02-23 12:08:41 -0800632 if num_instances == 1:
633 last_cnt_number -= 1
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700634 test_cnt_name = cls.basename + str(last_cnt_number+1)
635 return test_cnt_name
636
637 @classmethod
638 def build_image(cls, image):
639 print('Building test container docker image %s' %image)
Chetan Gaonkerb6064fa2016-05-02 16:29:57 -0700640 ovs_version = '2.5.0'
641 image_format = (ovs_version,)*4
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700642 dockerfile = '''
643FROM ubuntu:14.04
644MAINTAINER chetan@ciena.com
A R Karthickc762df42016-05-25 10:09:21 -0700645RUN apt-get update && \
646 apt-get install -y git git-core autoconf automake autotools-dev pkg-config \
647 make gcc g++ libtool libc6-dev cmake libpcap-dev libxerces-c2-dev \
648 unzip libpcre3-dev flex bison libboost-dev \
649 python python-pip python-setuptools python-scapy tcpdump doxygen doxypy wget \
650 openvswitch-common openvswitch-switch \
A R Karthick07608ef2016-08-23 16:51:19 -0700651 python-twisted python-sqlite sqlite3 python-pexpect telnet arping isc-dhcp-server \
Chetan Gaonker53f16382017-02-20 20:31:22 +0000652 python-paramiko python-maas-client python-keystoneclient python-neutronclient \
A R Karthickc28f6a92017-04-18 16:05:18 -0700653 python-glanceclient python-novaclient python-dev libffi-dev libssl-dev
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700654RUN easy_install nose
A R Karthickc28f6a92017-04-18 16:05:18 -0700655RUN python -m pip install --upgrade pip
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700656RUN mkdir -p /root/ovs
657WORKDIR /root
Chetan Gaonkerb6064fa2016-05-02 16:29:57 -0700658RUN wget http://openvswitch.org/releases/openvswitch-{}.tar.gz -O /root/ovs/openvswitch-{}.tar.gz && \
659(cd /root/ovs && tar zxpvf openvswitch-{}.tar.gz && \
660 cd openvswitch-{} && \
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700661 ./configure --prefix=/usr --sysconfdir=/etc --localstatedir=/var --disable-ssl && make && make install)
662RUN service openvswitch-switch restart || /bin/true
A.R Karthickec5b72a2016-11-03 09:53:07 -0700663RUN pip install scapy==2.3.2 scapy-ssl_tls==1.2.2 monotonic configObj docker-py pyyaml nsenter pyroute2 netaddr python-daemon
A R Karthickc28f6a92017-04-18 16:05:18 -0700664RUN pip install -U cryptography
665RUN pip install -U paramiko
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700666RUN mv /usr/sbin/tcpdump /sbin/
667RUN ln -sf /sbin/tcpdump /usr/sbin/tcpdump
A R Karthickf4999472016-07-01 16:42:13 -0700668RUN mv /usr/sbin/dhcpd /sbin/
669RUN ln -sf /sbin/dhcpd /usr/sbin/dhcpd
A.R Karthickcaf82ac2017-04-17 17:45:24 -0700670RUN mv /sbin/dhclient /usr/sbin/
671RUN ln -sf /usr/sbin/dhclient /sbin/dhclient
A R Karthickb7e80902016-05-17 09:38:31 -0700672WORKDIR /root
673RUN wget -nc http://de.archive.ubuntu.com/ubuntu/pool/main/b/bison/bison_2.5.dfsg-2.1_amd64.deb \
674 http://de.archive.ubuntu.com/ubuntu/pool/main/b/bison/libbison-dev_2.5.dfsg-2.1_amd64.deb
675RUN sudo dpkg -i bison_2.5.dfsg-2.1_amd64.deb libbison-dev_2.5.dfsg-2.1_amd64.deb
676RUN rm bison_2.5.dfsg-2.1_amd64.deb libbison-dev_2.5.dfsg-2.1_amd64.deb
677RUN wget -nc http://www.nbee.org/download/nbeesrc-jan-10-2013.zip && \
678 unzip nbeesrc-jan-10-2013.zip && \
679 cd nbeesrc-jan-10-2013/src && cmake . && make && \
680 cp ../bin/libn*.so /usr/local/lib && ldconfig && \
681 cp -R ../include/* /usr/include/
682WORKDIR /root
683RUN git clone https://github.com/CPqD/ofsoftswitch13.git && \
684 cd ofsoftswitch13 && \
A R Karthickb7e80902016-05-17 09:38:31 -0700685 ./boot.sh && \
686 ./configure --prefix=/usr --sysconfdir=/etc --localstatedir=/var --disable-ssl && \
687 make && make install
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700688CMD ["/bin/bash"]
Chetan Gaonkerb6064fa2016-05-02 16:29:57 -0700689'''.format(*image_format)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700690 super(CordTester, cls).build_image(dockerfile, image)
691 print('Done building docker image %s' %image)
692
A R Karthick1f03e912016-05-18 11:39:22 -0700693 def run_tests(self):
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700694 '''Run the list of tests'''
A R Karthick9a5edc42016-08-24 19:10:22 -0700695 res = 0
Thangavelu K Sef6f0a52016-12-14 19:57:05 +0000696 print('Modifying scapy tool files before running a test: %s' %self.tests)
697 self.modify_scapy_files_for_specific_tests()
A R Karthick1f03e912016-05-18 11:39:22 -0700698 print('Running tests: %s' %self.tests)
699 for t in self.tests:
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700700 test = t.split(':')[0]
A R Karthick24f1de62016-05-12 15:16:38 -0700701 test_file = '{}Test.py'.format(test)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700702 if t.find(':') >= 0:
A R Karthick24f1de62016-05-12 15:16:38 -0700703 test_case = '{0}:{1}'.format(test_file, t.split(':')[1])
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700704 else:
705 test_case = test_file
Chetan Gaonker7142a342016-04-07 14:53:12 -0700706 cmd = 'nosetests -v {0}/src/test/{1}/{2}'.format(self.sandbox, test, test_case)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700707 status = self.execute(cmd, shell = True)
A R Karthick9a5edc42016-08-24 19:10:22 -0700708 if status > 255:
709 status = 1
710 res |= status
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700711 print('Test %s %s' %(test_case, 'Success' if status == 0 else 'Failure'))
712 print('Done running tests')
713 if self.rm:
714 print('Removing test container %s' %self.name)
715 self.kill(remove=True)
716
A R Karthick9a5edc42016-08-24 19:10:22 -0700717 return res
718
Thangavelu K Sef6f0a52016-12-14 19:57:05 +0000719 def modify_scapy_files_for_specific_tests(self):
720 name = self.name
A R Karthickf7a613b2017-02-24 09:36:44 -0800721 container_cmd_exec = Container(name = name, image = CordTester.IMAGE)
Thangavelu K Sef6f0a52016-12-14 19:57:05 +0000722 tty = False
723 dckr = Client()
724 cmd = 'cp test/src/test/scapy/fields.py /usr/local/lib/python2.7/dist-packages/scapy/fields.py '
725 i = container_cmd_exec.execute(cmd = cmd, tty= tty, stream = True)
726
Chetan Gaonkerfb3cb5e2016-05-06 11:55:44 -0700727 @classmethod
728 def list_tests(cls, tests):
729 print('Listing test cases')
730 for test in tests:
A R Karthick24f1de62016-05-12 15:16:38 -0700731 test_file = '{}Test.py'.format(test)
Chetan Gaonkerfb3cb5e2016-05-06 11:55:44 -0700732 cmd = 'nosetests -v --collect-only {0}/../{1}/{2}'.format(cls.tester_base, test, test_file)
733 os.system(cmd)
734
A R Karthicka013a272016-08-16 16:40:19 -0700735
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700736##default onos/radius/test container images and names
737onos_image_default='onosproject/onos:latest'
A R Karthick07608ef2016-08-23 16:51:19 -0700738nose_image_default= '{}:candidate'.format(CordTester.IMAGE)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700739test_type_default='dhcp'
A R Karthickb608d402017-06-02 11:48:41 -0700740onos_app_version = '3.0-SNAPSHOT'
Chetan Gaonker4d842ad2016-04-26 10:04:24 -0700741cord_tester_base = os.path.dirname(os.path.realpath(__file__))
A.R Karthick5968e0d2017-05-16 14:50:46 -0700742olt_config_default = os.path.join(cord_tester_base, 'olt_config.json')
A R Karthick38d5df42017-07-10 13:33:26 -0700743onos_app_file = os.path.abspath('{0}/../apps/ciena-cordigmp-multitable-'.format(cord_tester_base) + onos_app_version + '.oar')
A R Karthick81acbff2016-06-17 14:45:16 -0700744cord_test_server_address = '{}:{}'.format(CORD_TEST_HOST, CORD_TEST_PORT)
A R Karthick07608ef2016-08-23 16:51:19 -0700745identity_file_default = '/etc/maas/ansible/id_rsa'
A R Karthicke14fc022016-12-08 14:50:29 -0800746onos_log_level = 'INFO'
A R Karthick07608ef2016-08-23 16:51:19 -0700747
748##sets up the ssh key file for the test container
749def set_ssh_key_file(identity_file):
750 ssh_key_file = None
751 if os.access(identity_file, os.F_OK):
752 ##copy it to setup directory
753 identity_dest = os.path.join(CordTester.tester_base, 'id_rsa')
754 if os.path.abspath(identity_file) != identity_dest:
755 try:
756 shutil.copy(identity_file, identity_dest)
757 ssh_key_file = os.path.join(CordTester.sandbox_setup, 'id_rsa')
758 except: pass
759
760 return ssh_key_file
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700761
A R Karthickcaac5542017-03-06 16:20:57 -0800762def openstack_setup(test_cnt_env):
763 admin_rc = os.path.join(os.getenv('HOME'), 'admin-openrc.sh')
A R Karthick8983cb02017-06-09 11:32:53 -0700764 if not os.access(admin_rc, os.F_OK):
765 admin_rc = os.path.join('/opt/cord_profile', 'admin-openrc.sh')
A R Karthickcaac5542017-03-06 16:20:57 -0800766 if os.access(admin_rc, os.F_OK):
767 dest = os.path.join(CordTester.tester_base, 'admin-openrc.sh')
768 shutil.copy(admin_rc, dest)
769 with open(dest, 'r') as f:
770 cfg = {}
771 for data in f.read().splitlines():
772 try:
773 k, v = data.split('=')
774 except:
775 continue
776
777 k = k.split()[-1]
778 cfg[k] = v
779
780 if 'REQUESTS_CA_BUNDLE' in cfg:
781 #copy the certificate to setup directory
782 cert_src = cfg['REQUESTS_CA_BUNDLE']
783 shutil.copy(cert_src, CordTester.tester_base)
784 test_cert_loc = os.path.join(CordTester.sandbox_setup,
785 os.path.basename(cert_src))
786 cfg['REQUESTS_CA_BUNDLE'] = test_cert_loc
787
788 for key, value in cfg.iteritems():
789 test_cnt_env[key] = value
790
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700791def runTest(args):
Chetan Gaonker823cdc52016-05-09 15:51:23 -0700792 #Start the cord test tcp server
A.R Karthickb17e2022017-01-27 11:29:26 -0800793 test_manifest = TestManifest(args = args)
794 test_server_params = test_manifest.server.split(':')
A R Karthick81acbff2016-06-17 14:45:16 -0700795 test_host = test_server_params[0]
796 test_port = CORD_TEST_PORT
797 if len(test_server_params) > 1:
798 test_port = int(test_server_params[1])
A R Karthick81acbff2016-06-17 14:45:16 -0700799
A R Karthick1f03e912016-05-18 11:39:22 -0700800 test_containers = []
801 #These tests end up restarting ONOS/quagga/radius
A R Karthick4e0c0912016-08-17 16:57:42 -0700802 tests_exempt = ('vrouter', 'cordSubscriber', 'proxyarp', 'dhcprelay')
Chetan Gaonker503032a2016-05-12 12:06:29 -0700803 if args.test_type.lower() == 'all':
804 tests = CordTester.ALL_TESTS
Chetan Gaonker503032a2016-05-12 12:06:29 -0700805 args.quagga = True
806 else:
A R Karthickacae3b42016-05-12 15:27:24 -0700807 tests = args.test_type.split('-')
Chetan Gaonker503032a2016-05-12 12:06:29 -0700808
A R Karthick1f03e912016-05-18 11:39:22 -0700809 tests_parallel = [ t for t in tests if t.split(':')[0] not in tests_exempt ]
810 tests_not_parallel = [ t for t in tests if t.split(':')[0] in tests_exempt ]
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700811 onos_cnt = {'tag':'latest'}
A R Karthick07608ef2016-08-23 16:51:19 -0700812 nose_cnt = {'image': CordTester.IMAGE, 'tag': 'candidate'}
Chetan Gaonker503032a2016-05-12 12:06:29 -0700813 update_map = { 'quagga' : False, 'test' : False, 'radius' : False }
814 update_map[args.update.lower()] = True
A.R Karthick95d044e2016-06-10 18:44:36 -0700815
Chetan Gaonker503032a2016-05-12 12:06:29 -0700816 if args.update.lower() == 'all':
817 for c in update_map.keys():
818 update_map[c] = True
A.R Karthick95d044e2016-06-10 18:44:36 -0700819
A R Karthick07608ef2016-08-23 16:51:19 -0700820 use_manifest = False
821 if args.manifest:
822 if os.access(args.manifest, os.F_OK):
823 ##copy it to setup directory
A.R Karthick8b2f1e32017-02-09 15:10:07 -0800824 dest = os.path.join(CordTester.tester_base,
825 os.path.basename(args.manifest))
A R Karthick07608ef2016-08-23 16:51:19 -0700826 if os.path.abspath(args.manifest) != dest:
827 try:
828 shutil.copy(args.manifest, dest)
829 except: pass
A R Karthick65d950d2016-12-19 19:41:55 -0800830 test_manifest = TestManifest(manifest = dest)
A R Karthick07608ef2016-08-23 16:51:19 -0700831 use_manifest = True
832 else:
833 print('Unable to access test manifest: %s' %args.manifest)
Chetan Gaonkerfb3cb5e2016-05-06 11:55:44 -0700834
A R Karthick65d950d2016-12-19 19:41:55 -0800835 onos_ip = test_manifest.onos_ip
836 radius_ip = test_manifest.radius_ip
A R Karthick1555c7c2017-09-07 14:59:41 -0700837 radius = None
A R Karthick65d950d2016-12-19 19:41:55 -0800838 head_node = test_manifest.head_node
A R Karthick5af23712017-01-20 09:49:24 -0800839 iterations = test_manifest.iterations
A.R Karthick263d3fc2017-01-27 12:52:53 -0800840 onos_cord_loc = test_manifest.onos_cord
A.R Karthickf184b342017-01-27 19:30:50 -0800841 service_profile = test_manifest.service_profile
842 synchronizer = test_manifest.synchronizer
A.R Karthick5968e0d2017-05-16 14:50:46 -0700843 olt_config_file = test_manifest.olt_config
A.R Karthick57fa9372017-05-24 12:47:03 -0700844 voltha_loc = test_manifest.voltha_loc
845 voltha_intf = test_manifest.voltha_intf
A.R Karthick5968e0d2017-05-16 14:50:46 -0700846 if not os.access(olt_config_file, os.F_OK):
847 olt_config_file = os.path.join(CordTester.tester_base, 'olt_config.json')
848 else:
849 dest = os.path.join(CordTester.tester_base,
850 os.path.basename(olt_config_file))
851 if os.path.abspath(olt_config_file) != dest:
852 try:
853 shutil.copy(olt_config_file, dest)
854 except: pass
855
A.R Karthickf184b342017-01-27 19:30:50 -0800856 onos_cord = None
A R Karthick973010f2017-02-06 16:41:51 -0800857 Onos.update_data_dir(test_manifest.karaf_version)
A R Karthick184945a2017-07-25 17:23:57 -0700858 Onos.set_expose_port(test_manifest.expose_port)
A R Karthick1555c7c2017-09-07 14:59:41 -0700859 Radius.create_network()
A.R Karthick263d3fc2017-01-27 12:52:53 -0800860 if onos_cord_loc:
861 if onos_cord_loc.find(os.path.sep) < 0:
862 onos_cord_loc = os.path.join(os.getenv('HOME'), onos_cord_loc)
A.R Karthickf184b342017-01-27 19:30:50 -0800863 if not os.access(onos_cord_loc, os.F_OK):
864 print('ONOS cord config location %s is not accessible' %onos_cord_loc)
865 sys.exit(1)
A.R Karthick263d3fc2017-01-27 12:52:53 -0800866 if not onos_ip:
867 ##Unexpected case. Specify the external controller ip when running on cord node
868 print('Specify ONOS ip using \"-e\" option when running the cord-tester on cord node')
869 sys.exit(1)
A.R Karthickf184b342017-01-27 19:30:50 -0800870 if not service_profile:
A R Karthick03bd2812017-03-03 17:49:17 -0800871 print('Specify service profile for the ONOS cord instance. Eg: rcord')
A.R Karthickf184b342017-01-27 19:30:50 -0800872 sys.exit(1)
873 if not synchronizer:
874 print('Specify synchronizer to use for the ONOS cord instance. Eg: vtn, fabric, cord')
875 sys.exit(1)
Kailash Khalasi1511ef02018-01-23 11:19:16 -0800876 onos_cord = OnosCord(onos_ip, onos_cord_loc, service_profile, synchronizer, skip = test_manifest.skip_onos_restart)
A.R Karthick263d3fc2017-01-27 12:52:53 -0800877
878 try:
879 test_server = cord_test_server_start(daemonize = False, cord_test_host = test_host, cord_test_port = test_port,
880 onos_cord = onos_cord)
881 except:
882 ##Most likely a server instance is already running (daemonized earlier)
883 test_server = None
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700884
A R Karthick65d950d2016-12-19 19:41:55 -0800885 Container.IMAGE_PREFIX = test_manifest.image_prefix
886 Onos.MAX_INSTANCES = test_manifest.onos_instances
A R Karthickc69d73e2017-01-20 11:44:34 -0800887 Onos.JVM_HEAP_SIZE = test_manifest.jvm_heap_size
A R Karthick65d950d2016-12-19 19:41:55 -0800888 cluster_mode = True if test_manifest.onos_instances > 1 else False
889 async_mode = cluster_mode and test_manifest.async_mode
890 existing_list = [ c['Names'][0][1:] for c in Container.dckr.containers() if c['Image'] == test_manifest.onos_image ]
891 setup_cluster = False if len(existing_list) == test_manifest.onos_instances else True
A.R Karthickc4e474d2016-12-12 15:24:57 -0800892 onos_ips = []
A R Karthick65d950d2016-12-19 19:41:55 -0800893 if cluster_mode is True and len(existing_list) > 1:
894 ##don't setup cluster config again
895 cluster_mode = False
A R Karthicka41c1c52017-08-02 15:30:48 -0700896 if voltha_loc:
897 voltha_key = os.path.join(voltha_loc, 'docker', 'onos_cfg', 'onos.jks')
898 Onos.update_ssl_key(voltha_key)
A R Karthick07608ef2016-08-23 16:51:19 -0700899 if onos_ip is None:
A R Karthick65d950d2016-12-19 19:41:55 -0800900 image_names = test_manifest.onos_image.rsplit(':', 1)
A R Karthick07608ef2016-08-23 16:51:19 -0700901 onos_cnt['image'] = image_names[0]
902 if len(image_names) > 1:
903 if image_names[1].find('/') < 0:
904 onos_cnt['tag'] = image_names[1]
905 else:
906 #tag cannot have slashes
A R Karthick65d950d2016-12-19 19:41:55 -0800907 onos_cnt['image'] = test_manifest.onos_image
A R Karthick07608ef2016-08-23 16:51:19 -0700908
909 Onos.IMAGE = onos_cnt['image']
A R Karthick65d950d2016-12-19 19:41:55 -0800910 Onos.PREFIX = test_manifest.image_prefix
A R Karthick07608ef2016-08-23 16:51:19 -0700911 Onos.TAG = onos_cnt['tag']
A R Karthick65d950d2016-12-19 19:41:55 -0800912 data_volume = '{}-data'.format(Onos.NAME) if test_manifest.shared_volume else None
A R Karthick07608ef2016-08-23 16:51:19 -0700913 onos = Onos(image = Onos.IMAGE,
A.R Karthickc4e474d2016-12-12 15:24:57 -0800914 tag = Onos.TAG, boot_delay = 60, cluster = cluster_mode,
A R Karthick85eb1862017-01-23 16:10:57 -0800915 data_volume = data_volume, async = async_mode, network = test_manifest.docker_network)
A.R Karthickc4e474d2016-12-12 15:24:57 -0800916 if onos.running:
A R Karthick65d950d2016-12-19 19:41:55 -0800917 onos_ips.append(onos.ipaddr)
A.R Karthickc4e474d2016-12-12 15:24:57 -0800918 else:
919 onos_ips.append(onos_ip)
920
A R Karthick65d950d2016-12-19 19:41:55 -0800921 num_onos_instances = test_manifest.onos_instances
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -0700922 if num_onos_instances > 1 and onos is not None:
923 onos_instances = []
924 onos_instances.append(onos)
925 for i in range(1, num_onos_instances):
926 name = '{}-{}'.format(Onos.NAME, i+1)
A R Karthick65d950d2016-12-19 19:41:55 -0800927 data_volume = '{}-data'.format(name) if test_manifest.shared_volume else None
A.R Karthickc4e474d2016-12-12 15:24:57 -0800928 quagga_config = Onos.get_quagga_config(i)
A R Karthickec2db322016-11-17 15:06:01 -0800929 onos = Onos(name = name, image = Onos.IMAGE, tag = Onos.TAG, boot_delay = 60, cluster = cluster_mode,
A R Karthick3b811152016-12-15 10:24:24 -0800930 data_volume = data_volume, async = async_mode,
A R Karthick184945a2017-07-25 17:23:57 -0700931 quagga_config = quagga_config, network = test_manifest.docker_network, instance = i)
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -0700932 onos_instances.append(onos)
A.R Karthickc4e474d2016-12-12 15:24:57 -0800933 if onos.running:
934 onos_ips.append(onos.ipaddr)
A R Karthick65d950d2016-12-19 19:41:55 -0800935 if async_mode is True and cluster_mode is True:
A.R Karthickc4e474d2016-12-12 15:24:57 -0800936 Onos.start_cluster_async(onos_instances)
937 if not onos_ips:
938 for onos in onos_instances:
939 onos_ips.append(onos.ipaddr)
A R Karthick65d950d2016-12-19 19:41:55 -0800940 if cluster_mode is True:
941 try:
942 for ip in onos_ips:
943 print('Installing cord tester ONOS app %s in ONOS instance %s' %(args.app,ip))
944 OnosCtrl.install_app(args.app, onos_ip = ip)
945 except: pass
A R Karthick1ef70552016-11-17 17:33:36 -0800946 if setup_cluster is True:
947 Onos.setup_cluster(onos_instances)
948 else:
949 print('ONOS instances already running. Skipping ONOS form cluster for %d instances' %num_onos_instances)
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -0700950 ctlr_addr = ','.join(onos_ips)
Chetan Gaonker5209fe82016-04-19 10:09:53 -0700951
A R Karthick65d950d2016-12-19 19:41:55 -0800952 print('Controller IP %s, Test type %s' %(onos_ips, args.test_type))
953 if onos_ip is not None:
A R Karthickbd9b8a32016-07-21 09:56:45 -0700954 print('Installing ONOS cord apps')
A R Karthick07608ef2016-08-23 16:51:19 -0700955 try:
956 Onos.install_cord_apps(onos_ip = onos_ip)
957 except: pass
A R Karthickeaf1c4e2016-07-19 12:22:35 -0700958
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -0700959 if not cluster_mode:
960 print('Installing cord tester ONOS app %s' %args.app)
961 try:
962 for ip in onos_ips:
A R Karthick65d950d2016-12-19 19:41:55 -0800963 OnosCtrl.install_app(args.app, onos_ip = ip)
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -0700964 except: pass
A R Karthickeaf1c4e2016-07-19 12:22:35 -0700965
A.R Karthick57fa9372017-05-24 12:47:03 -0700966 if voltha_loc:
967 #start voltha
A R Karthick168e2342017-08-15 16:13:10 -0700968 voltha = VolthaService(voltha_loc, onos_ips[0], interface = voltha_intf,
969 olt_config = olt_config_file, container_mode = test_manifest.voltha_container_mode)
A.R Karthick57fa9372017-05-24 12:47:03 -0700970 voltha.start()
971
A R Karthickeaf1c4e2016-07-19 12:22:35 -0700972 if radius_ip is None:
A R Karthick1555c7c2017-09-07 14:59:41 -0700973 ##Create Radius container
A R Karthick85eb1862017-01-23 16:10:57 -0800974 radius = Radius(prefix = Container.IMAGE_PREFIX, update = update_map['radius'],
A R Karthickefcf1ab2017-09-08 18:24:16 -0700975 network = test_manifest.docker_network,
976 olt_config = olt_config_file)
A R Karthick1555c7c2017-09-07 14:59:41 -0700977 radius_ip = radius.ip(network = Radius.NETWORK)
A.R Karthick95d044e2016-06-10 18:44:36 -0700978
A R Karthickeaf1c4e2016-07-19 12:22:35 -0700979 print('Radius server running with IP %s' %radius_ip)
A.R Karthick95d044e2016-06-10 18:44:36 -0700980
Chetan Gaonkerb84835f2016-04-19 15:12:10 -0700981 if args.quagga == True:
982 #Start quagga. Builds container if required
A R Karthick85eb1862017-01-23 16:10:57 -0800983 quagga = Quagga(prefix = Container.IMAGE_PREFIX, update = update_map['quagga'],
984 network = test_manifest.docker_network)
A R Karthick81acbff2016-06-17 14:45:16 -0700985
A R Karthick07608ef2016-08-23 16:51:19 -0700986 try:
987 maas_api_key = FabricMAAS.get_api_key()
988 except:
989 maas_api_key = 'UNKNOWN'
990
991 ssh_key_file = set_ssh_key_file(args.identity_file)
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -0700992 test_cnt_env = { 'ONOS_CONTROLLER_IP' : ctlr_addr,
Chetan Gaonkerc170f3f2016-04-19 17:24:45 -0700993 'ONOS_AAA_IP' : radius_ip if radius_ip is not None else '',
A R Karthick8d03cc52016-06-28 14:51:59 -0700994 'QUAGGA_IP': test_host,
A R Karthick81acbff2016-06-17 14:45:16 -0700995 'CORD_TEST_HOST' : test_host,
996 'CORD_TEST_PORT' : test_port,
A R Karthick65d950d2016-12-19 19:41:55 -0800997 'ONOS_RESTART' : 0 if test_manifest.olt and args.test_controller else 1,
998 'LOG_LEVEL': test_manifest.log_level,
A R Karthick07608ef2016-08-23 16:51:19 -0700999 'HEAD_NODE': head_node if head_node else CORD_TEST_HOST,
A R Karthick973010f2017-02-06 16:41:51 -08001000 'MAAS_API_KEY': maas_api_key,
A.R Karthick57fa9372017-05-24 12:47:03 -07001001 'KARAF_VERSION' : test_manifest.karaf_version,
A R Karthickfa2caab2017-07-24 18:04:46 -07001002 'VOLTHA_ENABLED' : int(test_manifest.voltha_enable)
Chetan Gaonker93e302d2016-04-05 10:51:07 -07001003 }
A R Karthick07608ef2016-08-23 16:51:19 -07001004
1005 if ssh_key_file:
1006 test_cnt_env['SSH_KEY_FILE'] = ssh_key_file
1007
A.R Karthick5968e0d2017-05-16 14:50:46 -07001008 olt_conf_test_loc = os.path.join(CordTester.sandbox_setup, os.path.basename(olt_config_file))
1009 test_cnt_env['OLT_CONFIG_FILE'] = olt_conf_test_loc
A R Karthick65d950d2016-12-19 19:41:55 -08001010 if test_manifest.olt:
Chetan Gaonker93e302d2016-04-05 10:51:07 -07001011 test_cnt_env['OLT_CONFIG'] = olt_conf_test_loc
1012
A.R Karthick8b2f1e32017-02-09 15:10:07 -08001013 if use_manifest:
1014 test_cnt_env['MANIFEST'] = os.path.join(CordTester.sandbox_setup,
1015 os.path.basename(args.manifest))
1016
A.R Karthick5968e0d2017-05-16 14:50:46 -07001017
A R Karthick5af23712017-01-20 09:49:24 -08001018 if iterations is not None:
1019 test_cnt_env['ITERATIONS'] = iterations
1020
A R Karthickcaac5542017-03-06 16:20:57 -08001021 openstack_setup(test_cnt_env)
1022
A R Karthicka013a272016-08-16 16:40:19 -07001023 if args.num_containers > 1 and args.container:
1024 print('Cannot specify number of containers with container option')
1025 sys.exit(1)
1026 if args.container:
1027 args.keep = True
A R Karthick1f03e912016-05-18 11:39:22 -07001028 port_num = 0
1029 num_tests = len(tests_parallel)
A R Karthick2b93d6a2016-09-06 15:19:09 -07001030 if num_tests > 0 and num_tests < args.num_containers:
1031 tests_parallel *= args.num_containers/num_tests
1032 num_tests = len(tests_parallel)
A R Karthick1f03e912016-05-18 11:39:22 -07001033 tests_per_container = max(1, num_tests/args.num_containers)
A R Karthicke4660f52017-02-23 12:08:41 -08001034 last_batch = num_tests % args.num_containers
A R Karthick1f03e912016-05-18 11:39:22 -07001035 test_slice_start = 0
1036 test_slice_end = test_slice_start + tests_per_container
1037 num_test_containers = min(num_tests, args.num_containers)
1038 if tests_parallel:
1039 print('Running %s tests across %d containers in parallel' %(tests_parallel, num_test_containers))
A R Karthicke4660f52017-02-23 12:08:41 -08001040 for container in xrange(num_test_containers):
1041 if container + 1 == num_test_containers:
1042 test_slice_end += last_batch
A R Karthick1f03e912016-05-18 11:39:22 -07001043 test_cnt = CordTester(tests_parallel[test_slice_start:test_slice_end],
1044 instance = container, num_instances = num_test_containers,
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -07001045 ctlr_ip = ctlr_addr,
A R Karthick07608ef2016-08-23 16:51:19 -07001046 name = args.container,
1047 image = nose_cnt['image'],
1048 prefix = Container.IMAGE_PREFIX,
1049 tag = nose_cnt['tag'],
A R Karthick1f03e912016-05-18 11:39:22 -07001050 env = test_cnt_env,
1051 rm = False if args.keep else True,
A R Karthick85eb1862017-01-23 16:10:57 -08001052 update = update_map['test'],
A R Karthick1555c7c2017-09-07 14:59:41 -07001053 network = test_manifest.docker_network,
1054 radius = radius)
A R Karthick1f03e912016-05-18 11:39:22 -07001055 test_slice_start = test_slice_end
1056 test_slice_end = test_slice_start + tests_per_container
1057 update_map['test'] = False
1058 test_containers.append(test_cnt)
A R Karthicka013a272016-08-16 16:40:19 -07001059 if not test_cnt.create:
1060 continue
A R Karthick65d950d2016-12-19 19:41:55 -08001061 if test_cnt.create and (test_manifest.start_switch or not test_manifest.olt):
A R Karthick07608ef2016-08-23 16:51:19 -07001062 if not args.no_switch:
A R Karthicke8bd80e2017-08-01 12:36:24 -07001063 test_cnt.start_switch(test_manifest)
A R Karthicka013a272016-08-16 16:40:19 -07001064 if test_cnt.create and test_cnt.olt:
A R Karthick1f03e912016-05-18 11:39:22 -07001065 _, port_num = test_cnt.setup_intfs(port_num = port_num)
1066
A R Karthick9a5edc42016-08-24 19:10:22 -07001067 status = 0
1068 if len(test_containers) > 1:
A R Karthicke4660f52017-02-23 12:08:41 -08001069 thread_pool = ThreadPool(len(test_containers), queue_size = 1, wait_timeout=1)
1070 for test_cnt in test_containers:
1071 thread_pool.addTask(test_cnt.run_tests)
1072 thread_pool.cleanUpThreads()
A R Karthick9a5edc42016-08-24 19:10:22 -07001073 else:
A R Karthickcee37412016-08-29 10:10:56 -07001074 if test_containers:
1075 status = test_containers[0].run_tests()
A R Karthick1f03e912016-05-18 11:39:22 -07001076
1077 ##Run the linear tests
1078 if tests_not_parallel:
1079 test_cnt = CordTester(tests_not_parallel,
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -07001080 ctlr_ip = ctlr_addr,
A R Karthick07608ef2016-08-23 16:51:19 -07001081 name = args.container,
1082 image = nose_cnt['image'],
1083 prefix = Container.IMAGE_PREFIX,
1084 tag = nose_cnt['tag'],
A R Karthick1f03e912016-05-18 11:39:22 -07001085 env = test_cnt_env,
1086 rm = False if args.keep else True,
A R Karthick85eb1862017-01-23 16:10:57 -08001087 update = update_map['test'],
1088 network = test_manifest.docker_network)
A R Karthick65d950d2016-12-19 19:41:55 -08001089 if test_cnt.create and (test_manifest.start_switch or not test_manifest.olt):
A R Karthick36cfcef2016-08-18 15:20:07 -07001090 #For non parallel tests, we just restart the switch also for OLT's
1091 CordTester.switch_on_olt = False
A R Karthick07608ef2016-08-23 16:51:19 -07001092 if not args.no_switch:
A R Karthicke8bd80e2017-08-01 12:36:24 -07001093 test_cnt.start_switch(test_manifest)
A R Karthicka013a272016-08-16 16:40:19 -07001094 if test_cnt.create and test_cnt.olt:
A R Karthick1f03e912016-05-18 11:39:22 -07001095 test_cnt.setup_intfs(port_num = port_num)
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -07001096 test_cnt.run_tests()
A R Karthick1f03e912016-05-18 11:39:22 -07001097
A R Karthick81acbff2016-06-17 14:45:16 -07001098 if test_server:
A.R Karthickf184b342017-01-27 19:30:50 -08001099 if onos_cord:
1100 onos_cord.restore()
A R Karthick81acbff2016-06-17 14:45:16 -07001101 cord_test_server_stop(test_server)
1102
A R Karthick9a5edc42016-08-24 19:10:22 -07001103 return status
1104
A R Karthick81acbff2016-06-17 14:45:16 -07001105##Starts onos/radius/quagga containers as appropriate
1106def setupCordTester(args):
1107 onos_cnt = {'tag':'latest'}
A R Karthick07608ef2016-08-23 16:51:19 -07001108 nose_cnt = {'image': CordTester.IMAGE, 'tag': 'candidate'}
A R Karthick92a0e5a2016-06-22 17:11:05 -07001109 update_map = { 'quagga' : False, 'radius' : False, 'test': False }
A R Karthick81acbff2016-06-17 14:45:16 -07001110 update_map[args.update.lower()] = True
A R Karthick65d950d2016-12-19 19:41:55 -08001111 test_manifest = TestManifest(args = args)
1112
A R Karthick81acbff2016-06-17 14:45:16 -07001113 if args.update.lower() == 'all':
1114 for c in update_map.keys():
1115 update_map[c] = True
1116
A R Karthick07608ef2016-08-23 16:51:19 -07001117 use_manifest = False
1118 if args.manifest:
1119 if os.access(args.manifest, os.F_OK):
1120 ##copy it to setup directory
A.R Karthick8b2f1e32017-02-09 15:10:07 -08001121 dest = os.path.join(CordTester.tester_base,
1122 os.path.basename(args.manifest))
A R Karthick07608ef2016-08-23 16:51:19 -07001123 if os.path.abspath(args.manifest) != dest:
1124 try:
1125 shutil.copy(args.manifest, dest)
1126 except: pass
A R Karthick65d950d2016-12-19 19:41:55 -08001127 test_manifest = TestManifest(manifest = dest)
A R Karthick07608ef2016-08-23 16:51:19 -07001128 use_manifest = True
1129
A.R Karthickf184b342017-01-27 19:30:50 -08001130 onos_ip = test_manifest.onos_ip
1131 radius_ip = test_manifest.radius_ip
A R Karthick1555c7c2017-09-07 14:59:41 -07001132 radius = None
A.R Karthickf184b342017-01-27 19:30:50 -08001133 head_node = test_manifest.head_node
1134 iterations = test_manifest.iterations
1135 service_profile = test_manifest.service_profile
1136 synchronizer = test_manifest.synchronizer
A.R Karthick57fa9372017-05-24 12:47:03 -07001137 voltha_loc = test_manifest.voltha_loc
1138 voltha_intf = test_manifest.voltha_intf
A.R Karthickf184b342017-01-27 19:30:50 -08001139 onos_cord = None
A.R Karthick263d3fc2017-01-27 12:52:53 -08001140 onos_cord_loc = test_manifest.onos_cord
A R Karthick973010f2017-02-06 16:41:51 -08001141 Onos.update_data_dir(test_manifest.karaf_version)
A R Karthick184945a2017-07-25 17:23:57 -07001142 Onos.set_expose_port(test_manifest.expose_port)
A.R Karthick5968e0d2017-05-16 14:50:46 -07001143 olt_config_file = test_manifest.olt_config
1144 if not os.access(olt_config_file, os.F_OK):
1145 olt_config_file = os.path.join(CordTester.tester_base, 'olt_config.json')
1146 else:
1147 dest = os.path.join(CordTester.tester_base,
1148 os.path.basename(olt_config_file))
1149 if os.path.abspath(olt_config_file) != dest:
1150 try:
1151 shutil.copy(olt_config_file, dest)
1152 except: pass
A R Karthick973010f2017-02-06 16:41:51 -08001153
A R Karthick1555c7c2017-09-07 14:59:41 -07001154 Radius.create_network()
A.R Karthick263d3fc2017-01-27 12:52:53 -08001155 if onos_cord_loc:
1156 if onos_cord_loc.find(os.path.sep) < 0:
1157 onos_cord_loc = os.path.join(os.getenv('HOME'), onos_cord_loc)
1158 if not os.access(onos_cord_loc, os.F_OK):
1159 print('ONOS cord config location %s is not accessible' %onos_cord_loc)
1160 sys.exit(1)
A.R Karthick263d3fc2017-01-27 12:52:53 -08001161 if not onos_ip:
A R Karthickd44cea12016-07-20 12:16:41 -07001162 ##Unexpected case. Specify the external controller ip when running on cord node
1163 print('Specify ONOS ip using \"-e\" option when running the cord-tester on cord node')
1164 sys.exit(1)
A.R Karthickf184b342017-01-27 19:30:50 -08001165 if not service_profile:
A R Karthick03bd2812017-03-03 17:49:17 -08001166 print('Specify service profile for the ONOS cord instance. Eg: rcord')
A.R Karthickf184b342017-01-27 19:30:50 -08001167 sys.exit(1)
1168 if not synchronizer:
1169 print('Specify synchronizer to use for the ONOS cord instance. Eg: vtn, fabric, cord')
1170 sys.exit(1)
Kailash Khalasi1511ef02018-01-23 11:19:16 -08001171 onos_cord = OnosCord(onos_ip, onos_cord_loc, service_profile, synchronizer, skip = test_manifest.skip_onos_restart)
A R Karthickd44cea12016-07-20 12:16:41 -07001172
A R Karthick65d950d2016-12-19 19:41:55 -08001173 Container.IMAGE_PREFIX = test_manifest.image_prefix
A R Karthick81acbff2016-06-17 14:45:16 -07001174 #don't spawn onos if the user had started it externally
A R Karthick65d950d2016-12-19 19:41:55 -08001175 image_names = test_manifest.onos_image.rsplit(':', 1)
A R Karthick07608ef2016-08-23 16:51:19 -07001176 onos_cnt['image'] = image_names[0]
1177 if len(image_names) > 1:
1178 if image_names[1].find('/') < 0:
1179 onos_cnt['tag'] = image_names[1]
1180 else:
1181 #tag cannot have slashes
A R Karthick65d950d2016-12-19 19:41:55 -08001182 onos_cnt['image'] = test_manifest.onos_image
A R Karthick81acbff2016-06-17 14:45:16 -07001183
A R Karthick07608ef2016-08-23 16:51:19 -07001184 Onos.IMAGE = onos_cnt['image']
A R Karthick65d950d2016-12-19 19:41:55 -08001185 Onos.PREFIX = test_manifest.image_prefix
A R Karthick07608ef2016-08-23 16:51:19 -07001186 Onos.TAG = onos_cnt['tag']
A R Karthick65d950d2016-12-19 19:41:55 -08001187 Onos.MAX_INSTANCES = test_manifest.onos_instances
A R Karthickc69d73e2017-01-20 11:44:34 -08001188 Onos.JVM_HEAP_SIZE = test_manifest.jvm_heap_size
A R Karthick65d950d2016-12-19 19:41:55 -08001189 cluster_mode = True if test_manifest.onos_instances > 1 else False
1190 async_mode = cluster_mode and test_manifest.async_mode
1191 existing_list = [ c['Names'][0][1:] for c in Container.dckr.containers() if c['Image'] == test_manifest.onos_image ]
1192 setup_cluster = False if len(existing_list) == test_manifest.onos_instances else True
A R Karthickc41c2422016-12-09 10:59:19 -08001193 #cleanup existing volumes before forming a new cluster
1194 if setup_cluster is True:
1195 print('Cleaning up existing cluster volumes')
1196 data_dir = os.path.join(Onos.setup_dir, 'cord-onos*-data')
1197 try:
1198 os.system('rm -rf {}'.format(data_dir))
1199 except: pass
1200
A R Karthick2b93d6a2016-09-06 15:19:09 -07001201 onos = None
A.R Karthickc4e474d2016-12-12 15:24:57 -08001202 onos_ips = []
A R Karthicka41c1c52017-08-02 15:30:48 -07001203 if voltha_loc:
1204 voltha_key = os.path.join(voltha_loc, 'docker', 'onos_cfg', 'onos.jks')
1205 Onos.update_ssl_key(voltha_key)
A R Karthick81acbff2016-06-17 14:45:16 -07001206 if onos_ip is None:
A R Karthick65d950d2016-12-19 19:41:55 -08001207 data_volume = '{}-data'.format(Onos.NAME) if test_manifest.shared_volume else None
A R Karthickec2db322016-11-17 15:06:01 -08001208 onos = Onos(image = Onos.IMAGE, tag = Onos.TAG, boot_delay = 60, cluster = cluster_mode,
A R Karthick85eb1862017-01-23 16:10:57 -08001209 data_volume = data_volume, async = async_mode, network = test_manifest.docker_network)
A.R Karthickc4e474d2016-12-12 15:24:57 -08001210 if onos.running:
A R Karthick65d950d2016-12-19 19:41:55 -08001211 onos_ips.append(onos.ipaddr)
A.R Karthickc4e474d2016-12-12 15:24:57 -08001212 else:
1213 onos_ips.append(onos_ip)
A R Karthick81acbff2016-06-17 14:45:16 -07001214
A R Karthick65d950d2016-12-19 19:41:55 -08001215 num_onos_instances = test_manifest.onos_instances
A R Karthick2b93d6a2016-09-06 15:19:09 -07001216 if num_onos_instances > 1 and onos is not None:
1217 onos_instances = []
1218 onos_instances.append(onos)
1219 for i in range(1, num_onos_instances):
1220 name = '{}-{}'.format(Onos.NAME, i+1)
A R Karthick65d950d2016-12-19 19:41:55 -08001221 data_volume = '{}-data'.format(name) if test_manifest.shared_volume else None
A.R Karthickc4e474d2016-12-12 15:24:57 -08001222 quagga_config = Onos.get_quagga_config(i)
A R Karthickec2db322016-11-17 15:06:01 -08001223 onos = Onos(name = name, image = Onos.IMAGE, tag = Onos.TAG, boot_delay = 60, cluster = cluster_mode,
A R Karthick3b811152016-12-15 10:24:24 -08001224 data_volume = data_volume, async = async_mode,
A R Karthick184945a2017-07-25 17:23:57 -07001225 quagga_config = quagga_config, network = test_manifest.docker_network, instance = i)
A R Karthick2b93d6a2016-09-06 15:19:09 -07001226 onos_instances.append(onos)
A.R Karthickc4e474d2016-12-12 15:24:57 -08001227 if onos.running:
1228 onos_ips.append(onos.ipaddr)
1229 if async_mode is True:
1230 Onos.start_cluster_async(onos_instances)
1231 if not onos_ips:
1232 for onos in onos_instances:
1233 onos_ips.append(onos.ipaddr)
A R Karthick51e6fd82016-11-22 14:39:19 -08001234 if setup_cluster is True:
1235 Onos.setup_cluster(onos_instances)
A R Karthick2b93d6a2016-09-06 15:19:09 -07001236
1237 ctlr_addr = ','.join(onos_ips)
1238 print('Onos IP %s' %ctlr_addr)
Kailash Khalasi1511ef02018-01-23 11:19:16 -08001239 if not test_manifest.skip_onos_restart:
1240 if onos_ip is not None:
1241 print('Installing ONOS cord apps')
1242 try:
1243 Onos.install_cord_apps(onos_ip = onos_ip)
1244 except: pass
A R Karthickbd9b8a32016-07-21 09:56:45 -07001245
Kailash Khalasi1511ef02018-01-23 11:19:16 -08001246 print('Installing cord tester ONOS app %s' %args.app)
1247 try:
1248 for ip in onos_ips:
1249 OnosCtrl.install_app(args.app, onos_ip = ip)
1250 except: pass
A R Karthickeaf1c4e2016-07-19 12:22:35 -07001251
A.R Karthick57fa9372017-05-24 12:47:03 -07001252 if voltha_loc:
1253 #start voltha
A R Karthick168e2342017-08-15 16:13:10 -07001254 voltha = VolthaService(voltha_loc, onos_ips[0], interface = voltha_intf,
1255 olt_config = olt_config_file, container_mode = test_manifest.voltha_container_mode)
A.R Karthick57fa9372017-05-24 12:47:03 -07001256 voltha.start()
1257
A R Karthick81acbff2016-06-17 14:45:16 -07001258 ##Start Radius container if not started
1259 if radius_ip is None:
A R Karthick85eb1862017-01-23 16:10:57 -08001260 radius = Radius(prefix = Container.IMAGE_PREFIX, update = update_map['radius'],
A R Karthickefcf1ab2017-09-08 18:24:16 -07001261 network = test_manifest.docker_network,
1262 olt_config = olt_config_file)
A R Karthick1555c7c2017-09-07 14:59:41 -07001263 radius_ip = radius.ip(network = Radius.NETWORK)
A R Karthick81acbff2016-06-17 14:45:16 -07001264
1265 print('Radius server running with IP %s' %radius_ip)
A R Karthick81acbff2016-06-17 14:45:16 -07001266
1267 if args.quagga == True:
1268 #Start quagga. Builds container if required
A R Karthick85eb1862017-01-23 16:10:57 -08001269 quagga = Quagga(prefix = Container.IMAGE_PREFIX, update = update_map['quagga'],
1270 network = test_manifest.docker_network)
A R Karthick8d03cc52016-06-28 14:51:59 -07001271 print('Quagga started')
A R Karthick81acbff2016-06-17 14:45:16 -07001272
A.R Karthickb17e2022017-01-27 11:29:26 -08001273 params = test_manifest.server.split(':')
A R Karthick81acbff2016-06-17 14:45:16 -07001274 ip = params[0]
1275 port = CORD_TEST_PORT
1276 if len(params) > 1:
1277 port = int(params[1])
A R Karthick92a0e5a2016-06-22 17:11:05 -07001278
A R Karthick07608ef2016-08-23 16:51:19 -07001279 try:
1280 maas_api_key = FabricMAAS.get_api_key()
1281 except:
1282 maas_api_key = 'UNKNOWN'
1283
1284 ssh_key_file = set_ssh_key_file(args.identity_file)
1285
A R Karthick92a0e5a2016-06-22 17:11:05 -07001286 #provision the test container
1287 if not args.dont_provision:
A R Karthick2b93d6a2016-09-06 15:19:09 -07001288 test_cnt_env = { 'ONOS_CONTROLLER_IP' : ctlr_addr,
A R Karthick1555c7c2017-09-07 14:59:41 -07001289 'ONOS_AAA_IP' : radius_ip if radius_ip is not None else '',
A R Karthick8d03cc52016-06-28 14:51:59 -07001290 'QUAGGA_IP': ip,
A R Karthick92a0e5a2016-06-22 17:11:05 -07001291 'CORD_TEST_HOST' : ip,
1292 'CORD_TEST_PORT' : port,
A R Karthick65d950d2016-12-19 19:41:55 -08001293 'ONOS_RESTART' : 0 if test_manifest.olt and args.test_controller else 1,
1294 'LOG_LEVEL': test_manifest.log_level,
A R Karthick07608ef2016-08-23 16:51:19 -07001295 'HEAD_NODE': head_node if head_node else CORD_TEST_HOST,
A R Karthick973010f2017-02-06 16:41:51 -08001296 'MAAS_API_KEY': maas_api_key,
A.R Karthick57fa9372017-05-24 12:47:03 -07001297 'KARAF_VERSION' : test_manifest.karaf_version,
A R Karthickfa2caab2017-07-24 18:04:46 -07001298 'VOLTHA_ENABLED' : int(test_manifest.voltha_enable)
A R Karthick92a0e5a2016-06-22 17:11:05 -07001299 }
A R Karthick07608ef2016-08-23 16:51:19 -07001300
1301 if ssh_key_file:
1302 test_cnt_env['SSH_KEY_FILE'] = ssh_key_file
A.R Karthick5968e0d2017-05-16 14:50:46 -07001303 olt_conf_test_loc = os.path.join(CordTester.sandbox_setup, os.path.basename(olt_config_file))
1304 test_cnt_env['OLT_CONFIG_FILE'] = olt_conf_test_loc
A R Karthick65d950d2016-12-19 19:41:55 -08001305 if test_manifest.olt:
A R Karthick92a0e5a2016-06-22 17:11:05 -07001306 test_cnt_env['OLT_CONFIG'] = olt_conf_test_loc
A R Karthick5af23712017-01-20 09:49:24 -08001307 if test_manifest.iterations is not None:
1308 test_cnt_env['ITERATIONS'] = iterations
A.R Karthick8b2f1e32017-02-09 15:10:07 -08001309 if use_manifest:
1310 test_cnt_env['MANIFEST'] = os.path.join(CordTester.sandbox_setup,
1311 os.path.basename(args.manifest))
A R Karthickcaac5542017-03-06 16:20:57 -08001312
1313 openstack_setup(test_cnt_env)
1314
A R Karthick92a0e5a2016-06-22 17:11:05 -07001315 test_cnt = CordTester((),
A R Karthick2b93d6a2016-09-06 15:19:09 -07001316 ctlr_ip = ctlr_addr,
A R Karthick92a0e5a2016-06-22 17:11:05 -07001317 image = nose_cnt['image'],
A R Karthick07608ef2016-08-23 16:51:19 -07001318 prefix = Container.IMAGE_PREFIX,
A R Karthick92a0e5a2016-06-22 17:11:05 -07001319 tag = nose_cnt['tag'],
1320 env = test_cnt_env,
1321 rm = False,
A R Karthick85eb1862017-01-23 16:10:57 -08001322 update = update_map['test'],
A R Karthick1555c7c2017-09-07 14:59:41 -07001323 network = test_manifest.docker_network,
1324 radius = radius)
A R Karthick92a0e5a2016-06-22 17:11:05 -07001325
A R Karthick65d950d2016-12-19 19:41:55 -08001326 if test_manifest.start_switch or not test_manifest.olt:
A R Karthicke8bd80e2017-08-01 12:36:24 -07001327 test_cnt.start_switch(test_manifest)
A R Karthick92a0e5a2016-06-22 17:11:05 -07001328 if test_cnt.olt:
1329 test_cnt.setup_intfs(port_num = 0)
Thangavelu K Sa61d1da2017-09-05 05:32:21 -07001330 if test_manifest.setup_dhcpd and test_manifest.start_switch:
1331 test_cnt.setup_dhcpd(test_manifest)
A R Karthick92a0e5a2016-06-22 17:11:05 -07001332 print('Test container %s started and provisioned to run tests using nosetests' %(test_cnt.name))
1333
1334 #Finally start the test server and daemonize
A R Karthick14118c62016-07-27 14:54:04 -07001335 try:
A R Karthickbd82f362016-11-10 15:08:52 -08001336 cord_test_server_start(daemonize = not args.foreground, cord_test_host = ip, cord_test_port = port,
1337 onos_cord = onos_cord, foreground = args.foreground)
A R Karthick14118c62016-07-27 14:54:04 -07001338 except socket.error, e:
1339 #the test agent address could be remote or already running. Exit gracefully
1340 sys.exit(0)
Chetan Gaonker93e302d2016-04-05 10:51:07 -07001341
A R Karthick9a5edc42016-08-24 19:10:22 -07001342 return 0
1343
Chetan Gaonker503032a2016-05-12 12:06:29 -07001344def cleanupTests(args):
A R Karthick757eb4d2017-01-09 14:51:16 -08001345 if args.manifest and os.access(args.manifest, os.F_OK):
1346 manifest = TestManifest(manifest = args.manifest)
1347 args.prefix = manifest.image_prefix
1348 args.olt = manifest.olt
A.R Karthick5968e0d2017-05-16 14:50:46 -07001349 args.olt_config = manifest.olt_config
A R Karthick757eb4d2017-01-09 14:51:16 -08001350 args.onos = manifest.onos_image
1351 args.server = manifest.server
A.R Karthickb17e2022017-01-27 11:29:26 -08001352 args.onos_ip = manifest.onos_ip
1353 args.radius_ip = manifest.radius_ip
1354 args.onos_cord = manifest.onos_cord
A.R Karthickf184b342017-01-27 19:30:50 -08001355 args.service_profile = manifest.service_profile
1356 args.synchronizer = manifest.synchronizer
A.R Karthick57fa9372017-05-24 12:47:03 -07001357 args.voltha_loc = manifest.voltha_loc
A.R Karthickb17e2022017-01-27 11:29:26 -08001358 else:
1359 args.onos_ip = None
1360 args.radius_ip = None
1361 if args.test_controller:
1362 ips = args.test_controller.split('/')
1363 args.onos_ip = ips[0]
1364 if len(ips) > 1:
1365 args.radius_ip = ips[1]
A R Karthick757eb4d2017-01-09 14:51:16 -08001366
A R Karthick2b93d6a2016-09-06 15:19:09 -07001367 image_name = args.onos
A R Karthick07608ef2016-08-23 16:51:19 -07001368 prefix = args.prefix
1369 if prefix:
1370 prefix += '/'
1371 test_container = '{}{}:candidate'.format(prefix, CordTester.IMAGE)
Chetan Gaonker503032a2016-05-12 12:06:29 -07001372 print('Cleaning up Test containers ...')
1373 Container.cleanup(test_container)
A R Karthickb50f5592016-07-26 12:19:29 -07001374 if args.olt:
1375 print('Cleaning up test container OLT configuration')
A.R Karthick5968e0d2017-05-16 14:50:46 -07001376 CordTester.cleanup_intfs(args.olt_config)
A R Karthick2b93d6a2016-09-06 15:19:09 -07001377
1378 onos_list = [ c['Names'][0][1:] for c in Container.dckr.containers() if c['Image'] == image_name ]
1379 if len(onos_list) > 1:
1380 for onos in onos_list:
1381 Container.dckr.kill(onos)
1382 Container.dckr.remove_container(onos, force=True)
A R Karthickec2db322016-11-17 15:06:01 -08001383 for index in range(len(onos_list)):
1384 volume = '{}-data'.format(Onos.NAME) if index == 0 else '{}-{}-data'.format(Onos.NAME, index+1)
1385 Onos.remove_data_map(volume, Onos.guest_data_dir)
A R Karthick9d48c652016-09-15 09:16:36 -07001386 Onos.cleanup_runtime()
A R Karthickec2db322016-11-17 15:06:01 -08001387
A R Karthick394976f2017-01-31 14:25:16 -08001388 radius_container = '{}{}:candidate'.format(prefix, Radius.IMAGE)
1389 quagga_container = '{}{}:candidate'.format(prefix, Quagga.IMAGE)
1390 Container.cleanup(radius_container)
A R Karthickefcf1ab2017-09-08 18:24:16 -07001391 radius_restore_users()
A R Karthick394976f2017-01-31 14:25:16 -08001392 Container.cleanup(quagga_container)
A.R Karthick57fa9372017-05-24 12:47:03 -07001393 if args.voltha_loc:
1394 voltha = VolthaService(args.voltha_loc, args.onos_ip)
1395 voltha.stop()
1396
A.R Karthickb17e2022017-01-27 11:29:26 -08001397 if args.onos_cord:
A.R Karthickf184b342017-01-27 19:30:50 -08001398 #try restoring the onos cord instance
1399 try:
Kailash Khalasi1511ef02018-01-23 11:19:16 -08001400 onos_cord = OnosCord(args.onos_ip, args.onos_cord, args.service_profile, args.synchronizer, start = False, skip = test_manifest.skip_onos_restart)
A.R Karthickf184b342017-01-27 19:30:50 -08001401 onos_cord.restore(force = True)
1402 except Exception as e:
1403 print(e)
A.R Karthickb17e2022017-01-27 11:29:26 -08001404
A.R Karthick842f0122016-09-28 14:48:47 -07001405 if args.xos:
1406 ##cleanup XOS images
1407 xos_images = ( '{}:{}'.format(XosServer.IMAGE,XosServer.TAG),
1408 '{}:{}'.format(XosSynchronizerOpenstack.IMAGE,
1409 XosSynchronizerOpenstack.TAG),
1410 '{}:{}'.format(XosSynchronizerOnboarding.IMAGE,
1411 XosSynchronizerOnboarding.TAG),
1412 '{}:{}'.format(XosSynchronizerOpenvpn.IMAGE,
1413 XosSynchronizerOpenvpn.TAG),
1414 '{}:{}'.format(XosPostgresql.IMAGE,
1415 XosPostgresql.TAG),
1416 '{}:{}'.format(XosSyndicateMs.IMAGE,
1417 XosSyndicateMs.TAG),
1418 )
1419 for img in xos_images:
1420 print('Cleaning up XOS image: %s' %img)
1421 Container.cleanup(img)
1422
A R Karthicke99ab5c2016-09-30 13:59:57 -07001423 server_params = args.server.split(':')
1424 server_host = server_params[0]
1425 server_port = CORD_TEST_PORT
1426 if len(server_params) > 1:
1427 server_port = int(server_params[1])
1428 cord_test_server_shutdown(server_host, server_port)
A R Karthick9a5edc42016-08-24 19:10:22 -07001429 return 0
Chetan Gaonker503032a2016-05-12 12:06:29 -07001430
1431def listTests(args):
1432 if args.test == 'all':
1433 tests = CordTester.ALL_TESTS
1434 else:
A R Karthickacae3b42016-05-12 15:27:24 -07001435 tests = args.test.split('-')
Chetan Gaonker503032a2016-05-12 12:06:29 -07001436 CordTester.list_tests(tests)
A R Karthickcee37412016-08-29 10:10:56 -07001437 return 0
ChetanGaonkereadad482016-08-26 01:21:47 -07001438
1439def getMetrics(args):
A R Karthickcee37412016-08-29 10:10:56 -07001440 try:
1441 detail = c.inspect_container(args.container)
1442 except:
1443 print('Unknown container %s' %args.container)
1444 return 0
1445 user_hz = os.sysconf(os.sysconf_names['SC_CLK_TCK'])
ChetanGaonkereadad482016-08-26 01:21:47 -07001446 state = detail["State"]
1447 if bool(state["Paused"]):
1448 print("Container is in Paused State")
1449 elif bool(state["Running"]):
1450 print("Container is in Running State")
1451 elif int(state["ExitCode"]) == 0:
1452 print("Container is in Stopped State")
1453 else:
1454 print("Container is in Crashed State")
1455
A R Karthickcee37412016-08-29 10:10:56 -07001456 print("Ip Address of the container: " +detail['NetworkSettings']['IPAddress'])
ChetanGaonkereadad482016-08-26 01:21:47 -07001457
1458 if bool(detail["State"]["Running"]):
1459 container_id = detail['Id']
1460 cpu_usage = {}
A R Karthickcee37412016-08-29 10:10:56 -07001461 cur_usage = 0
1462 last_usage = 0
1463 for i in range(2):
1464 with open('/sys/fs/cgroup/cpuacct/docker/' + container_id + '/cpuacct.stat', 'r') as f:
1465 for line in f:
1466 m = re.search(r"(system|user)\s+(\d+)", line)
1467 if m:
1468 cpu_usage[m.group(1)] = int(m.group(2))
1469 cpu = cpu_usage["system"] + cpu_usage["user"]
1470 last_usage = cur_usage
1471 cur_usage = cpu
1472 time.sleep(1)
1473 cpu_percent = (cur_usage - last_usage)*100.0/user_hz
1474 print("CPU Usage: %.2f %%" %(cpu_percent))
ChetanGaonkereadad482016-08-26 01:21:47 -07001475 else:
1476 print(0)
1477
1478 if bool(detail["State"]["Running"]):
1479 container_id = detail['Id']
A R Karthickcee37412016-08-29 10:10:56 -07001480 print("Docker Port Info:")
ChetanGaonkereadad482016-08-26 01:21:47 -07001481 cmd = "sudo docker port {}".format(container_id)
1482 os.system(cmd)
1483
1484 if bool(detail["State"]["Running"]):
1485 container_id = detail['Id']
1486 with open('/sys/fs/cgroup/memory/docker/' + container_id + '/memory.stat', 'r') as f:
1487 for line in f:
1488 m = re.search(r"total_rss\s+(\d+)", line)
1489 if m:
A R Karthickcee37412016-08-29 10:10:56 -07001490 mem = int(m.group(1))
1491 print("Memory: %s KB "%(mem/1024.0))
ChetanGaonkereadad482016-08-26 01:21:47 -07001492 o = re.search(r"usage\s+(\d+)", line)
1493 if o:
A R Karthickcee37412016-08-29 10:10:56 -07001494 print("Usage: %s "%(o.group(1)))
ChetanGaonkereadad482016-08-26 01:21:47 -07001495 p = re.search(r"max_usage\s+(\d+)", line)
1496 if p:
A R Karthickcee37412016-08-29 10:10:56 -07001497 print("Max Usage: %s "%(p.group(1)))
ChetanGaonkereadad482016-08-26 01:21:47 -07001498
1499 if bool(detail["State"]["Running"]):
1500 container_id = detail['Id']
1501 with open('/sys/fs/cgroup/cpuacct/docker/' + container_id + '/cpuacct.stat', 'r') as f:
1502 for line in f:
1503 m = re.search(r"user\s+(\d+)", line)
1504 if m:
A R Karthickcee37412016-08-29 10:10:56 -07001505 user_ticks = int(m.group(1))
1506 print("Time spent by running processes: %.2f ms"%(user_ticks*1000.0/user_hz))
1507 print("List Networks:")
ChetanGaonkereadad482016-08-26 01:21:47 -07001508 cmd = "docker network ls"
1509 os.system(cmd)
A R Karthick9a5edc42016-08-24 19:10:22 -07001510 return 0
Chetan Gaonker503032a2016-05-12 12:06:29 -07001511
1512def buildImages(args):
A R Karthick07608ef2016-08-23 16:51:19 -07001513 tag = 'candidate'
1514 prefix = args.prefix
1515 if prefix:
1516 prefix += '/'
Chetan Gaonker503032a2016-05-12 12:06:29 -07001517 if args.image == 'all' or args.image == 'quagga':
A R Karthick07608ef2016-08-23 16:51:19 -07001518 image_name = '{}{}:{}'.format(prefix, Quagga.IMAGE, tag)
1519 Quagga.build_image(image_name)
A.R Karthick95d044e2016-06-10 18:44:36 -07001520
Chetan Gaonker503032a2016-05-12 12:06:29 -07001521 if args.image == 'all' or args.image == 'radius':
A R Karthick07608ef2016-08-23 16:51:19 -07001522 image_name = '{}{}:{}'.format(prefix, Radius.IMAGE, tag)
1523 Radius.build_image(image_name)
Chetan Gaonker503032a2016-05-12 12:06:29 -07001524
1525 if args.image == 'all' or args.image == 'test':
A R Karthick07608ef2016-08-23 16:51:19 -07001526 image_name = '{}{}:{}'.format(prefix, CordTester.IMAGE, tag)
1527 CordTester.build_image(image_name)
Chetan Gaonker503032a2016-05-12 12:06:29 -07001528
A R Karthick9a5edc42016-08-24 19:10:22 -07001529 return 0
1530
A R Karthickbec27762016-07-28 10:59:34 -07001531def startImages(args):
A R Karthickbec27762016-07-28 10:59:34 -07001532 ##starts the latest ONOS image
A R Karthick07608ef2016-08-23 16:51:19 -07001533 onos_cnt = {'tag': 'latest'}
1534 image_names = args.onos.rsplit(':', 1)
1535 onos_cnt['image'] = image_names[0]
1536 if len(image_names) > 1:
1537 if image_names[1].find('/') < 0:
1538 onos_cnt['tag'] = image_names[1]
1539 else:
1540 #tag cannot have slashes
1541 onos_cnt['image'] = args.onos
1542
A R Karthickbec27762016-07-28 10:59:34 -07001543 if args.image == 'all' or args.image == 'onos':
A R Karthick07608ef2016-08-23 16:51:19 -07001544 onos = Onos(image = onos_cnt['image'], tag = onos_cnt['tag'])
A R Karthickbec27762016-07-28 10:59:34 -07001545 print('ONOS started with ip %s' %(onos.ip()))
1546
1547 if args.image == 'all' or args.image == 'quagga':
A R Karthick07608ef2016-08-23 16:51:19 -07001548 quagga = Quagga(prefix = args.prefix)
A R Karthickbec27762016-07-28 10:59:34 -07001549 print('Quagga started with ip %s' %(quagga.ip()))
1550
1551 if args.image == 'all' or args.image == 'radius':
A R Karthick07608ef2016-08-23 16:51:19 -07001552 radius = Radius(prefix = args.prefix)
A R Karthickbec27762016-07-28 10:59:34 -07001553 print('Radius started with ip %s' %(radius.ip()))
1554
A R Karthick9a5edc42016-08-24 19:10:22 -07001555 return 0
1556
A R Karthickea8bfce2016-10-13 16:32:07 -07001557def xosCommand(args):
1558 update = False
1559 profile = args.profile
1560 if args.command == 'update':
1561 update = True
1562 xos = XosServiceProfile(profile = profile, update = update)
1563 if args.command == 'build':
1564 xos.build_images(force = True)
1565 if args.command == 'start':
1566 xos.start_services()
1567 if args.command == 'stop':
1568 xos.stop_services(rm = True)
ChetanGaonker2c0e9bb2016-09-21 13:38:37 -07001569 return 0
1570
Chetan Gaonker93e302d2016-04-05 10:51:07 -07001571if __name__ == '__main__':
Chetan Gaonker678743f2016-04-26 09:54:31 -07001572 parser = ArgumentParser(description='Cord Tester')
Chetan Gaonker503032a2016-05-12 12:06:29 -07001573 subparser = parser.add_subparsers()
1574 parser_run = subparser.add_parser('run', help='Run cord tester')
1575 parser_run.add_argument('-t', '--test-type', default=test_type_default, help='Specify test type or test case to run')
1576 parser_run.add_argument('-o', '--onos', default=onos_image_default, type=str, help='ONOS container image')
Chetan Gaonker503032a2016-05-12 12:06:29 -07001577 parser_run.add_argument('-q', '--quagga',action='store_true',help='Provision quagga container for vrouter')
1578 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 -07001579 parser_run.add_argument('-l', '--olt', action='store_true', help='Use OLT config')
A.R Karthick5968e0d2017-05-16 14:50:46 -07001580 parser_run.add_argument('-olt-config', '--olt-config', default=olt_config_default, type=str, help='Provide OLT configuration')
Chetan Gaonker503032a2016-05-12 12:06:29 -07001581 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 -07001582 '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 -07001583 parser_run.add_argument('-r', '--server', default=cord_test_server_address, type=str,
1584 help='ip:port address to connect for cord test server for container requests')
Chetan Gaonker503032a2016-05-12 12:06:29 -07001585 parser_run.add_argument('-k', '--keep', action='store_true', help='Keep test container after tests')
1586 parser_run.add_argument('-s', '--start-switch', action='store_true', help='Start OVS when running under OLT config')
A R Karthick6708f562017-09-05 12:15:31 -07001587 parser_run.add_argument('-dh', '--setup-dhcpd', action='store_true', help='Start dhcpd Server in cord-tester test container')
Chetan Gaonker503032a2016-05-12 12:06:29 -07001588 parser_run.add_argument('-u', '--update', default='none', choices=['test','quagga','radius', 'all'], type=str, help='Update cord tester container images. '
1589 'Eg: --update=quagga to rebuild quagga image.'
1590 ' --update=radius to rebuild radius server image.'
1591 ' --update=test to rebuild cord test image.(Default)'
1592 ' --update=all to rebuild all cord tester images.')
A R Karthick1f03e912016-05-18 11:39:22 -07001593 parser_run.add_argument('-n', '--num-containers', default=1, type=int,
1594 help='Specify number of test containers to spawn for tests')
A R Karthicka013a272016-08-16 16:40:19 -07001595 parser_run.add_argument('-c', '--container', default='', type=str, help='Test container name for running tests')
A R Karthick07608ef2016-08-23 16:51:19 -07001596 parser_run.add_argument('-m', '--manifest', default='', type=str, help='Provide test configuration manifest')
1597 parser_run.add_argument('-p', '--prefix', default='', type=str, help='Provide container image prefix')
1598 parser_run.add_argument('-d', '--no-switch', action='store_true', help='Dont start test switch.')
1599 parser_run.add_argument('-i', '--identity-file', default=identity_file_default,
1600 type=str, help='ssh identity file to access compute nodes from test container')
ChetanGaonkerdbd4e4b2016-10-28 17:40:11 -07001601 parser_run.add_argument('-j', '--onos-instances', default=1, type=int,
1602 help='Specify number to test onos instances to form cluster')
A R Karthick09dbc6d2016-11-22 10:37:42 -08001603 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 -08001604 parser_run.add_argument('-async', '--async-mode', action='store_true',
1605 help='Start ONOS cluster instances in async mode')
A R Karthicke14fc022016-12-08 14:50:29 -08001606 parser_run.add_argument('-log', '--log-level', default=onos_log_level,
1607 choices=['DEBUG','TRACE','ERROR','WARN','INFO'],
1608 type=str,
1609 help='Specify the log level for the test cases')
A R Karthickc69d73e2017-01-20 11:44:34 -08001610 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 -08001611 parser_run.add_argument('-network', '--network', default='', type=str, help='Docker network to attach')
A.R Karthickb17e2022017-01-27 11:29:26 -08001612 parser_run.add_argument('-onos-cord', '--onos-cord', default='', type=str,
1613 help='Specify config location for ONOS cord when running on podd')
A.R Karthickf184b342017-01-27 19:30:50 -08001614 parser_run.add_argument('-service-profile', '--service-profile', default='', type=str,
1615 help='Specify config location for ONOS cord service profile when running on podd.'
1616 'Eg: $HOME/service-profile/cord-pod')
1617 parser_run.add_argument('-synchronizer', '--synchronizer', default='', type=str,
1618 help='Specify the synchronizer to use for ONOS cord instance when running on podd.'
1619 'Eg: vtn,fabric,cord')
A.R Karthickdda22062017-02-09 14:39:20 -08001620 parser_run.add_argument('-karaf', '--karaf', default='3.0.8', type=str, help='Karaf version for ONOS')
A.R Karthick57fa9372017-05-24 12:47:03 -07001621 parser_run.add_argument('-voltha-loc', '--voltha-loc', default='', type=str,
1622 help='Specify the voltha location in order to start voltha')
1623 parser_run.add_argument('-voltha-intf', '--voltha-intf', default='eth0', type=str,
1624 help='Specify the voltha interface for voltha to listen')
A R Karthickfa2caab2017-07-24 18:04:46 -07001625 parser_run.add_argument('-voltha-enable', '--voltha-enable', action='store_true',
1626 help='Run the tests with voltha environment enabled')
A R Karthick168e2342017-08-15 16:13:10 -07001627 parser_run.add_argument('-voltha-container-mode', '--voltha-container-mode', action='store_true',
1628 help='Run the tests with voltha container environment enabled')
A R Karthick184945a2017-07-25 17:23:57 -07001629 parser_run.add_argument('-expose-port', '--expose-port', action='store_true',
1630 help='Start ONOS by exposing the controller ports to the host.'
1631 'Add +1 for every other onos/cluster instance when running more than 1 ONOS instances')
Kailash Khalasi1511ef02018-01-23 11:19:16 -08001632 parser_run.add_argument('-skip-onos-restart', '--skip-onos-restart', action='store_true',
1633 help = 'Skips restarting/configuring of onoscord')
Chetan Gaonker503032a2016-05-12 12:06:29 -07001634 parser_run.set_defaults(func=runTest)
1635
A R Karthick81acbff2016-06-17 14:45:16 -07001636 parser_setup = subparser.add_parser('setup', help='Setup cord tester environment')
1637 parser_setup.add_argument('-o', '--onos', default=onos_image_default, type=str, help='ONOS container image')
1638 parser_setup.add_argument('-r', '--server', default=cord_test_server_address, type=str,
1639 help='ip:port address for cord test server to listen for container restart requests')
1640 parser_setup.add_argument('-q', '--quagga',action='store_true',help='Provision quagga container for vrouter')
1641 parser_setup.add_argument('-a', '--app', default=onos_app_file, type=str, help='Cord ONOS app filename')
1642 parser_setup.add_argument('-e', '--test-controller', default='', type=str, help='External test controller ip for Onos and/or radius server. '
1643 'Eg: 10.0.0.2/10.0.0.3 to specify ONOS and Radius ip to connect')
1644 parser_setup.add_argument('-u', '--update', default='none', choices=['quagga','radius', 'all'], type=str, help='Update cord tester container images. '
1645 'Eg: --update=quagga to rebuild quagga image.'
1646 ' --update=radius to rebuild radius server image.'
1647 ' --update=all to rebuild all cord tester images.')
A R Karthick92a0e5a2016-06-22 17:11:05 -07001648 parser_setup.add_argument('-d', '--dont-provision', action='store_true', help='Dont start test container.')
A R Karthick07608ef2016-08-23 16:51:19 -07001649 parser_setup.add_argument('-l', '--olt', action='store_true', help='Use OLT config')
A.R Karthick5968e0d2017-05-16 14:50:46 -07001650 parser_setup.add_argument('-olt-config', '--olt-config', default=olt_config_default, type=str, help='Provide OLT configuration')
A R Karthicke14fc022016-12-08 14:50:29 -08001651 parser_setup.add_argument('-log', '--log-level', default=onos_log_level, type=str,
1652 choices=['DEBUG','TRACE','ERROR','WARN','INFO'],
1653 help='Specify the log level for the test cases')
A R Karthick92a0e5a2016-06-22 17:11:05 -07001654 parser_setup.add_argument('-s', '--start-switch', action='store_true', help='Start OVS when running under OLT config')
A R Karthick6708f562017-09-05 12:15:31 -07001655 parser_setup.add_argument('-dh', '--setup-dhcpd', action='store_true', help='Start dhcpd Server in cord-tester container')
A.R Karthickb17e2022017-01-27 11:29:26 -08001656 parser_setup.add_argument('-onos-cord', '--onos-cord', default='', type=str,
1657 help='Specify config location for ONOS cord when running on podd')
A.R Karthickf184b342017-01-27 19:30:50 -08001658 parser_setup.add_argument('-service-profile', '--service-profile', default='', type=str,
1659 help='Specify config location for ONOS cord service profile when running on podd.'
1660 'Eg: $HOME/service-profile/cord-pod')
1661 parser_setup.add_argument('-synchronizer', '--synchronizer', default='', type=str,
1662 help='Specify the synchronizer to use for ONOS cord instance when running on podd.'
1663 'Eg: vtn,fabric,cord')
A R Karthick07608ef2016-08-23 16:51:19 -07001664 parser_setup.add_argument('-m', '--manifest', default='', type=str, help='Provide test configuration manifest')
1665 parser_setup.add_argument('-p', '--prefix', default='', type=str, help='Provide container image prefix')
1666 parser_setup.add_argument('-i', '--identity-file', default=identity_file_default,
1667 type=str, help='ssh identity file to access compute nodes from test container')
A R Karthick2b93d6a2016-09-06 15:19:09 -07001668 parser_setup.add_argument('-n', '--onos-instances', default=1, type=int,
A R Karthickbd82f362016-11-10 15:08:52 -08001669 help='Specify number of test onos instances to spawn')
A R Karthick09dbc6d2016-11-22 10:37:42 -08001670 parser_setup.add_argument('-v', '--shared-volume', action='store_true',
1671 help='Start ONOS cluster instances with shared volume')
A.R Karthickc4e474d2016-12-12 15:24:57 -08001672 parser_setup.add_argument('-async', '--async-mode', action='store_true',
1673 help='Start ONOS cluster instances in async mode')
A R Karthickbd82f362016-11-10 15:08:52 -08001674 parser_setup.add_argument('-f', '--foreground', action='store_true', help='Run in foreground')
A R Karthickc69d73e2017-01-20 11:44:34 -08001675 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 -08001676 parser_setup.add_argument('-network', '--network', default='', type=str, help='Docker network to attach')
A.R Karthickdda22062017-02-09 14:39:20 -08001677 parser_setup.add_argument('-karaf', '--karaf', default='3.0.8', type=str, help='Karaf version for ONOS')
A.R Karthick57fa9372017-05-24 12:47:03 -07001678 parser_setup.add_argument('-voltha-loc', '--voltha-loc', default='', type=str,
1679 help='Specify the voltha location in order to start voltha')
1680 parser_setup.add_argument('-voltha-intf', '--voltha-intf', default='eth0', type=str,
1681 help='Specify the voltha interface for voltha to listen')
A R Karthickfa2caab2017-07-24 18:04:46 -07001682 parser_setup.add_argument('-voltha-enable', '--voltha-enable', action='store_true',
1683 help='Run the tests with voltha environment enabled')
A R Karthick168e2342017-08-15 16:13:10 -07001684 parser_setup.add_argument('-voltha-container-mode', '--voltha-container-mode', action='store_true',
1685 help='Run the tests with voltha container environment enabled')
A R Karthick184945a2017-07-25 17:23:57 -07001686 parser_setup.add_argument('-expose-port', '--expose-port', action='store_true',
1687 help='Start ONOS by exposing the controller ports to the host.'
1688 'Add +1 for every other onos/cluster instance when running more than 1 ONOS instances')
Kailash Khalasi1511ef02018-01-23 11:19:16 -08001689 parser_setup.add_argument('-skip-onos-restart', '--skip-onos-restart', action='store_true',
1690 help = 'Skips restarting/configuring of onoscord')
A R Karthick81acbff2016-06-17 14:45:16 -07001691 parser_setup.set_defaults(func=setupCordTester)
1692
ChetanGaonker2c0e9bb2016-09-21 13:38:37 -07001693 parser_xos = subparser.add_parser('xos', help='Building xos into cord tester environment')
A R Karthickea8bfce2016-10-13 16:32:07 -07001694 parser_xos.add_argument('command', choices=['build', 'update', 'start', 'stop'])
1695 parser_xos.add_argument('-p', '--profile', default='cord-pod', type=str, help='Provide service profile')
1696 parser_xos.set_defaults(func=xosCommand)
ChetanGaonker2c0e9bb2016-09-21 13:38:37 -07001697
Chetan Gaonker503032a2016-05-12 12:06:29 -07001698 parser_list = subparser.add_parser('list', help='List test cases')
1699 parser_list.add_argument('-t', '--test', default='all', help='Specify test type to list test cases. '
1700 'Eg: -t tls to list tls test cases.'
1701 ' -t tls-dhcp-vrouter to list tls,dhcp and vrouter test cases.'
1702 ' -t all to list all test cases.')
1703 parser_list.set_defaults(func=listTests)
1704
1705 parser_build = subparser.add_parser('build', help='Build cord test container images')
ChetanGaonker2c0e9bb2016-09-21 13:38:37 -07001706 parser_build.add_argument('image', choices=['quagga', 'radius', 'test','all'])
A.R Karthick842f0122016-09-28 14:48:47 -07001707 parser_build.add_argument('-p', '--prefix', default='', type=str, help='Provide container image prefix')
Chetan Gaonker503032a2016-05-12 12:06:29 -07001708 parser_build.set_defaults(func=buildImages)
1709
ChetanGaonkereadad482016-08-26 01:21:47 -07001710 parser_metrics = subparser.add_parser('metrics', help='Info of container')
1711 parser_metrics.add_argument("container", help="Container name")
1712 parser_metrics.set_defaults(func=getMetrics)
1713
A R Karthickbec27762016-07-28 10:59:34 -07001714 parser_start = subparser.add_parser('start', help='Start cord tester containers')
A R Karthick07608ef2016-08-23 16:51:19 -07001715 parser_start.add_argument('-p', '--prefix', default='', type=str, help='Provide container image prefix')
1716 parser_start.add_argument('-o', '--onos', default=onos_image_default, type=str, help='ONOS container image')
A R Karthickbec27762016-07-28 10:59:34 -07001717 parser_start.add_argument('image', choices=['onos', 'quagga', 'radius', 'all'])
1718 parser_start.set_defaults(func=startImages)
1719
Chetan Gaonker503032a2016-05-12 12:06:29 -07001720 parser_cleanup = subparser.add_parser('cleanup', help='Cleanup test containers')
A R Karthick07608ef2016-08-23 16:51:19 -07001721 parser_cleanup.add_argument('-p', '--prefix', default='', type=str, help='Provide container image prefix')
1722 parser_cleanup.add_argument('-l', '--olt', action = 'store_true', help = 'Cleanup OLT config')
A.R Karthick5968e0d2017-05-16 14:50:46 -07001723 parser_cleanup.add_argument('-olt-config', '--olt-config', default=olt_config_default, type=str, help='Provide OLT configuration')
A R Karthick2b93d6a2016-09-06 15:19:09 -07001724 parser_cleanup.add_argument('-o', '--onos', default=onos_image_default, type=str,
1725 help='ONOS container image to cleanup')
A.R Karthick842f0122016-09-28 14:48:47 -07001726 parser_cleanup.add_argument('-x', '--xos', action='store_true',
1727 help='Cleanup XOS containers')
A R Karthicke99ab5c2016-09-30 13:59:57 -07001728 parser_cleanup.add_argument('-r', '--server', default=cord_test_server_address, type=str,
1729 help='ip:port address for cord test server to cleanup')
A.R Karthickb17e2022017-01-27 11:29:26 -08001730 parser_cleanup.add_argument('-e', '--test-controller', default='', type=str,
1731 help='External test controller ip for Onos and/or radius server. '
1732 'Eg: 10.0.0.2/10.0.0.3 to specify ONOS and Radius ip')
1733 parser_cleanup.add_argument('-onos-cord', '--onos-cord', default='', type=str,
1734 help='Specify config location for ONOS cord instance when running on podd to restore')
A.R Karthickf184b342017-01-27 19:30:50 -08001735 parser_cleanup.add_argument('-service-profile', '--service-profile', default='', type=str,
1736 help='Specify config location for ONOS cord service profile when running on podd.'
1737 'Eg: $HOME/service-profile/cord-pod')
1738 parser_cleanup.add_argument('-synchronizer', '--synchronizer', default='', type=str,
1739 help='Specify the synchronizer to use for ONOS cord instance when running on podd.'
1740 'Eg: vtn,fabric,cord')
A R Karthick757eb4d2017-01-09 14:51:16 -08001741 parser_cleanup.add_argument('-m', '--manifest', default='', type=str, help='Provide test manifest')
A.R Karthick57fa9372017-05-24 12:47:03 -07001742 parser_cleanup.add_argument('-voltha-loc', '--voltha-loc', default='', type=str,
1743 help='Specify the voltha location')
Kailash Khalasi1511ef02018-01-23 11:19:16 -08001744 parser_cleanup.add_argument('-skip-onos-restart', '--skip-onos-restart', action='store_true',
1745 help = 'Skips restarting/configuring of onoscord')
Chetan Gaonker503032a2016-05-12 12:06:29 -07001746 parser_cleanup.set_defaults(func=cleanupTests)
1747
ChetanGaonkereadad482016-08-26 01:21:47 -07001748 c = Client(**(kwargs_from_env()))
1749
Chetan Gaonker93e302d2016-04-05 10:51:07 -07001750 args = parser.parse_args()
A R Karthick9a5edc42016-08-24 19:10:22 -07001751 res = args.func(args)
1752 sys.exit(res)