blob: e05a960892c035023dfa11a586b011b378927606 [file] [log] [blame]
Chetan Gaonkercb122cc2016-05-10 10:58:34 -07001#!/usr/bin/env python
Chetan Gaonkercfcce782016-05-10 10:10:42 -07002#
3# Copyright 2016-present Ciena Corporation
4#
5# Licensed under the Apache License, Version 2.0 (the "License");
6# you may not use this file except in compliance with the License.
7# You may obtain a copy of the License at
8#
9# http://www.apache.org/licenses/LICENSE-2.0
10#
11# Unless required by applicable law or agreed to in writing, software
12# distributed under the License is distributed on an "AS IS" BASIS,
13# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14# See the License for the specific language governing permissions and
15# limitations under the License.
16#
Chetan Gaonker93e302d2016-04-05 10:51:07 -070017from argparse import ArgumentParser
18import os,sys,time
Chetan Gaonker4d842ad2016-04-26 10:04:24 -070019utils_dir = os.path.join( os.path.dirname(os.path.realpath(__file__)), '../utils')
Chetan Gaonker7142a342016-04-07 14:53:12 -070020sys.path.append(utils_dir)
Chetan Gaonker93e302d2016-04-05 10:51:07 -070021from OnosCtrl import OnosCtrl
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -070022from OltConfig import OltConfig
A R Karthick1f03e912016-05-18 11:39:22 -070023from threadPool import ThreadPool
Chetan Gaonker3533faa2016-04-25 17:50:14 -070024from CordContainer import *
25from CordTestServer import cord_test_server_start, cord_test_server_stop
Chetan Gaonker93e302d2016-04-05 10:51:07 -070026
Chetan Gaonker93e302d2016-04-05 10:51:07 -070027class CordTester(Container):
Chetan Gaonker93e302d2016-04-05 10:51:07 -070028 sandbox = '/root/test'
Chetan Gaonker7142a342016-04-07 14:53:12 -070029 sandbox_setup = '/root/test/src/test/setup'
Chetan Gaonker4d842ad2016-04-26 10:04:24 -070030 tester_base = os.path.dirname(os.path.realpath(__file__))
31 tester_paths = os.path.realpath(__file__).split(os.path.sep)
A R Karthickb7e80902016-05-17 09:38:31 -070032 tester_path_index = tester_paths.index('src') - 1
Chetan Gaonker7142a342016-04-07 14:53:12 -070033 sandbox_host = os.path.sep.join(tester_paths[:tester_path_index+1])
Chetan Gaonker93e302d2016-04-05 10:51:07 -070034
35 host_guest_map = ( (sandbox_host, sandbox),
Chetan Gaonker85b7bd52016-04-20 10:29:12 -070036 ('/lib/modules', '/lib/modules'),
37 ('/var/run/docker.sock', '/var/run/docker.sock')
Chetan Gaonker93e302d2016-04-05 10:51:07 -070038 )
39 basename = 'cord-tester'
Chetan Gaonker503032a2016-05-12 12:06:29 -070040 IMAGE = 'cord-test/nose'
41 ALL_TESTS = ('tls', 'dhcp', 'igmp', 'subscriber', 'vrouter', 'flows')
Chetan Gaonker93e302d2016-04-05 10:51:07 -070042
A R Karthick1f03e912016-05-18 11:39:22 -070043 def __init__(self, tests, instance = 0, num_instances = 1, ctlr_ip = None, image = IMAGE, tag = 'latest',
Chetan Gaonker85b7bd52016-04-20 10:29:12 -070044 env = None, rm = False, update = False):
A R Karthick1f03e912016-05-18 11:39:22 -070045 self.tests = tests
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -070046 self.ctlr_ip = ctlr_ip
Chetan Gaonker93e302d2016-04-05 10:51:07 -070047 self.rm = rm
48 self.name = self.get_name()
49 super(CordTester, self).__init__(self.name, image = image, tag = tag)
50 host_config = self.create_host_config(host_guest_map = self.host_guest_map, privileged = True)
51 volumes = []
Chetan Gaonkerb84835f2016-04-19 15:12:10 -070052 for _, g in self.host_guest_map:
Chetan Gaonker93e302d2016-04-05 10:51:07 -070053 volumes.append(g)
Chetan Gaonker85b7bd52016-04-20 10:29:12 -070054 if update is True or not self.img_exists():
Chetan Gaonkerb84835f2016-04-19 15:12:10 -070055 self.build_image(image)
Chetan Gaonker7142a342016-04-07 14:53:12 -070056 ##Remove test container if any
57 self.remove_container(self.name, force=True)
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -070058 if env is not None and env.has_key('OLT_CONFIG'):
59 self.olt = True
Chetan Gaonker5209fe82016-04-19 10:09:53 -070060 olt_conf_file = os.path.join(self.tester_base, 'olt_config.json')
61 olt_config = OltConfig(olt_conf_file)
62 self.port_map = olt_config.olt_port_map()
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -070063 else:
64 self.olt = False
Chetan Gaonker5209fe82016-04-19 10:09:53 -070065 self.port_map = None
A R Karthick1f03e912016-05-18 11:39:22 -070066 if env is not None:
67 env['TEST_HOST'] = self.name
68 env['TEST_INSTANCE'] = instance
69 env['TEST_INSTANCES'] = num_instances
Chetan Gaonker93e302d2016-04-05 10:51:07 -070070 print('Starting test container %s, image %s, tag %s' %(self.name, self.image, self.tag))
71 self.start(rm = False, volumes = volumes, environment = env,
72 host_config = host_config, tty = True)
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -070073
74 def execute_switch(self, cmd, shell = False):
75 if self.olt:
76 return os.system(cmd)
77 return self.execute(cmd, shell = shell)
78
79 def start_switch(self, bridge = 'ovsbr0', boot_delay = 2):
80 """Start OVS"""
81 ##Determine if OVS has to be started locally or not
82 s_file,s_sandbox = ('of-bridge-local.sh',self.tester_base) if self.olt else ('of-bridge.sh',self.sandbox_setup)
83 ovs_cmd = os.path.join(s_sandbox, '{0}'.format(s_file)) + ' {0}'.format(bridge)
84 if self.olt:
85 ovs_cmd += ' {0}'.format(self.ctlr_ip)
86 print('Starting OVS on the host')
87 else:
88 print('Starting OVS on test container %s' %self.name)
89 self.execute_switch(ovs_cmd)
Chetan Gaonker93e302d2016-04-05 10:51:07 -070090 status = 1
91 ## Wait for the LLDP flows to be added to the switch
92 tries = 0
Chetan Gaonkera52016e2016-05-05 15:19:59 -070093 while status != 0 and tries < 200:
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -070094 cmd = 'sudo ovs-ofctl dump-flows {0} | grep \"type=0x8942\"'.format(bridge)
95 status = self.execute_switch(cmd, shell = True)
Chetan Gaonker93e302d2016-04-05 10:51:07 -070096 tries += 1
97 if tries % 10 == 0:
98 print('Waiting for test switch to be connected to ONOS controller ...')
99
100 if status != 0:
101 print('Test Switch not connected to ONOS container.'
102 'Please remove ONOS container and restart the test')
103 if self.rm:
104 self.kill()
105 sys.exit(1)
106
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700107 if boot_delay:
108 time.sleep(boot_delay)
109
A R Karthick1f03e912016-05-18 11:39:22 -0700110 def setup_intfs(self, port_num = 0):
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700111 tester_intf_subnet = '192.168.100'
112 res = 0
Chetan Gaonker5209fe82016-04-19 10:09:53 -0700113 host_intf = self.port_map['host']
114 start_vlan = self.port_map['start_vlan']
115 for port in self.port_map['ports']:
116 guest_if = port
A R Karthick1f03e912016-05-18 11:39:22 -0700117 local_if = '{0}_{1}'.format(guest_if, port_num+1)
118 guest_ip = '{0}.{1}/24'.format(tester_intf_subnet, port_num+1)
Chetan Gaonker5209fe82016-04-19 10:09:53 -0700119 ##Use pipeworks to configure container interfaces on host/bridge interfaces
120 pipework_cmd = 'pipework {0} -i {1} -l {2} {3} {4}'.format(host_intf, guest_if, local_if, self.name, guest_ip)
121 if start_vlan != 0:
A R Karthick1f03e912016-05-18 11:39:22 -0700122 pipework_cmd += ' @{}'.format(start_vlan + port_num)
Chetan Gaonker5209fe82016-04-19 10:09:53 -0700123
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700124 res += os.system(pipework_cmd)
Chetan Gaonker5209fe82016-04-19 10:09:53 -0700125 port_num += 1
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700126
A R Karthick1f03e912016-05-18 11:39:22 -0700127 return res, port_num
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700128
129 @classmethod
130 def get_name(cls):
131 cnt_name = '/{0}'.format(cls.basename)
132 cnt_name_len = len(cnt_name)
133 names = list(flatten(n['Names'] for n in cls.dckr.containers(all=True)))
134 test_names = filter(lambda n: n.startswith(cnt_name), names)
135 last_cnt_number = 0
136 if test_names:
137 last_cnt_name = reduce(lambda n1, n2: n1 if int(n1[cnt_name_len:]) > \
138 int(n2[cnt_name_len:]) else n2,
139 test_names)
140 last_cnt_number = int(last_cnt_name[cnt_name_len:])
141 test_cnt_name = cls.basename + str(last_cnt_number+1)
142 return test_cnt_name
143
144 @classmethod
145 def build_image(cls, image):
146 print('Building test container docker image %s' %image)
Chetan Gaonkerb6064fa2016-05-02 16:29:57 -0700147 ovs_version = '2.5.0'
148 image_format = (ovs_version,)*4
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700149 dockerfile = '''
150FROM ubuntu:14.04
151MAINTAINER chetan@ciena.com
A R Karthickc762df42016-05-25 10:09:21 -0700152RUN apt-get update && \
153 apt-get install -y git git-core autoconf automake autotools-dev pkg-config \
154 make gcc g++ libtool libc6-dev cmake libpcap-dev libxerces-c2-dev \
155 unzip libpcre3-dev flex bison libboost-dev \
156 python python-pip python-setuptools python-scapy tcpdump doxygen doxypy wget \
157 openvswitch-common openvswitch-switch \
158 python-twisted python-sqlite sqlite3 python-pexpect telnet arping
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700159RUN easy_install nose
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700160RUN mkdir -p /root/ovs
161WORKDIR /root
Chetan Gaonkerb6064fa2016-05-02 16:29:57 -0700162RUN wget http://openvswitch.org/releases/openvswitch-{}.tar.gz -O /root/ovs/openvswitch-{}.tar.gz && \
163(cd /root/ovs && tar zxpvf openvswitch-{}.tar.gz && \
164 cd openvswitch-{} && \
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700165 ./configure --prefix=/usr --sysconfdir=/etc --localstatedir=/var --disable-ssl && make && make install)
166RUN service openvswitch-switch restart || /bin/true
A R Karthickc762df42016-05-25 10:09:21 -0700167RUN pip install -U scapy scapy-ssl_tls monotonic configObj docker-py pyyaml nsenter pyroute2 netaddr
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700168RUN mv /usr/sbin/tcpdump /sbin/
169RUN ln -sf /sbin/tcpdump /usr/sbin/tcpdump
A R Karthickb7e80902016-05-17 09:38:31 -0700170WORKDIR /root
171RUN wget -nc http://de.archive.ubuntu.com/ubuntu/pool/main/b/bison/bison_2.5.dfsg-2.1_amd64.deb \
172 http://de.archive.ubuntu.com/ubuntu/pool/main/b/bison/libbison-dev_2.5.dfsg-2.1_amd64.deb
173RUN sudo dpkg -i bison_2.5.dfsg-2.1_amd64.deb libbison-dev_2.5.dfsg-2.1_amd64.deb
174RUN rm bison_2.5.dfsg-2.1_amd64.deb libbison-dev_2.5.dfsg-2.1_amd64.deb
175RUN wget -nc http://www.nbee.org/download/nbeesrc-jan-10-2013.zip && \
176 unzip nbeesrc-jan-10-2013.zip && \
177 cd nbeesrc-jan-10-2013/src && cmake . && make && \
178 cp ../bin/libn*.so /usr/local/lib && ldconfig && \
179 cp -R ../include/* /usr/include/
180WORKDIR /root
181RUN git clone https://github.com/CPqD/ofsoftswitch13.git && \
182 cd ofsoftswitch13 && \
A R Karthickb7e80902016-05-17 09:38:31 -0700183 ./boot.sh && \
184 ./configure --prefix=/usr --sysconfdir=/etc --localstatedir=/var --disable-ssl && \
185 make && make install
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700186CMD ["/bin/bash"]
Chetan Gaonkerb6064fa2016-05-02 16:29:57 -0700187'''.format(*image_format)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700188 super(CordTester, cls).build_image(dockerfile, image)
189 print('Done building docker image %s' %image)
190
A R Karthick1f03e912016-05-18 11:39:22 -0700191 def run_tests(self):
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700192 '''Run the list of tests'''
A R Karthick1f03e912016-05-18 11:39:22 -0700193 print('Running tests: %s' %self.tests)
194 for t in self.tests:
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700195 test = t.split(':')[0]
A R Karthick24f1de62016-05-12 15:16:38 -0700196 test_file = '{}Test.py'.format(test)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700197 if t.find(':') >= 0:
A R Karthick24f1de62016-05-12 15:16:38 -0700198 test_case = '{0}:{1}'.format(test_file, t.split(':')[1])
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700199 else:
200 test_case = test_file
Chetan Gaonker7142a342016-04-07 14:53:12 -0700201 cmd = 'nosetests -v {0}/src/test/{1}/{2}'.format(self.sandbox, test, test_case)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700202 status = self.execute(cmd, shell = True)
203 print('Test %s %s' %(test_case, 'Success' if status == 0 else 'Failure'))
204 print('Done running tests')
205 if self.rm:
206 print('Removing test container %s' %self.name)
207 self.kill(remove=True)
208
Chetan Gaonkerfb3cb5e2016-05-06 11:55:44 -0700209 @classmethod
210 def list_tests(cls, tests):
211 print('Listing test cases')
212 for test in tests:
A R Karthick24f1de62016-05-12 15:16:38 -0700213 test_file = '{}Test.py'.format(test)
Chetan Gaonkerfb3cb5e2016-05-06 11:55:44 -0700214 cmd = 'nosetests -v --collect-only {0}/../{1}/{2}'.format(cls.tester_base, test, test_file)
215 os.system(cmd)
216
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700217##default onos/radius/test container images and names
218onos_image_default='onosproject/onos:latest'
Chetan Gaonker503032a2016-05-12 12:06:29 -0700219nose_image_default= '{}:latest'.format(CordTester.IMAGE)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700220test_type_default='dhcp'
221onos_app_version = '1.0-SNAPSHOT'
Chetan Gaonker4d842ad2016-04-26 10:04:24 -0700222cord_tester_base = os.path.dirname(os.path.realpath(__file__))
Chetan Gaonker4ca5cca2016-04-11 13:59:35 -0700223onos_app_file = os.path.abspath('{0}/../apps/ciena-cordigmp-'.format(cord_tester_base) + onos_app_version + '.oar')
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700224
225def runTest(args):
Chetan Gaonker823cdc52016-05-09 15:51:23 -0700226 #Start the cord test tcp server
227 test_server = cord_test_server_start()
A R Karthick1f03e912016-05-18 11:39:22 -0700228 test_containers = []
229 #These tests end up restarting ONOS/quagga/radius
230 tests_exempt = ('vrouter',)
Chetan Gaonker503032a2016-05-12 12:06:29 -0700231 if args.test_type.lower() == 'all':
232 tests = CordTester.ALL_TESTS
Chetan Gaonker503032a2016-05-12 12:06:29 -0700233 args.quagga = True
234 else:
A R Karthickacae3b42016-05-12 15:27:24 -0700235 tests = args.test_type.split('-')
Chetan Gaonker503032a2016-05-12 12:06:29 -0700236
A R Karthick1f03e912016-05-18 11:39:22 -0700237 tests_parallel = [ t for t in tests if t.split(':')[0] not in tests_exempt ]
238 tests_not_parallel = [ t for t in tests if t.split(':')[0] in tests_exempt ]
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700239 onos_cnt = {'tag':'latest'}
Chetan Gaonker503032a2016-05-12 12:06:29 -0700240 nose_cnt = {'image': CordTester.IMAGE, 'tag': 'latest'}
241 update_map = { 'quagga' : False, 'test' : False, 'radius' : False }
242 update_map[args.update.lower()] = True
243
244 if args.update.lower() == 'all':
245 for c in update_map.keys():
246 update_map[c] = True
247
Chetan Gaonker5209fe82016-04-19 10:09:53 -0700248 radius_ip = None
Chetan Gaonkerc170f3f2016-04-19 17:24:45 -0700249 quagga_ip = None
Chetan Gaonkerfb3cb5e2016-05-06 11:55:44 -0700250
Chetan Gaonker5209fe82016-04-19 10:09:53 -0700251 #don't spawn onos if the user has specified external test controller with test interface config
252 if args.test_controller:
253 ips = args.test_controller.split('/')
254 onos_ip = ips[0]
255 if len(ips) > 1:
256 radius_ip = ips[1]
257 else:
258 radius_ip = None
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700259 else:
Chetan Gaonker5209fe82016-04-19 10:09:53 -0700260 onos_cnt['image'] = args.onos.split(':')[0]
261 if args.onos.find(':') >= 0:
262 onos_cnt['tag'] = args.onos.split(':')[1]
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700263
Chetan Gaonker5209fe82016-04-19 10:09:53 -0700264 onos = Onos(image = onos_cnt['image'], tag = onos_cnt['tag'], boot_delay = 60)
265 onos_ip = onos.ip()
266
A R Karthicka661b552016-05-25 10:18:50 -0700267 ##Start Radius container
268 radius = Radius( update = update_map['radius'])
269 radius_ip = radius.ip()
270 print('Radius server running with IP %s' %radius_ip)
Chetan Gaonker5209fe82016-04-19 10:09:53 -0700271
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700272 print('Onos IP %s, Test type %s' %(onos_ip, args.test_type))
273 print('Installing ONOS app %s' %onos_app_file)
Chetan Gaonker5209fe82016-04-19 10:09:53 -0700274 OnosCtrl.install_app(args.app, onos_ip = onos_ip)
Chetan Gaonkerb84835f2016-04-19 15:12:10 -0700275
276 if args.quagga == True:
277 #Start quagga. Builds container if required
Chetan Gaonker503032a2016-05-12 12:06:29 -0700278 quagga = Quagga(update = update_map['quagga'])
Chetan Gaonkerc170f3f2016-04-19 17:24:45 -0700279 quagga_ip = quagga.ip()
Chetan Gaonkerb84835f2016-04-19 15:12:10 -0700280
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700281 test_cnt_env = { 'ONOS_CONTROLLER_IP' : onos_ip,
Chetan Gaonkerc170f3f2016-04-19 17:24:45 -0700282 'ONOS_AAA_IP' : radius_ip if radius_ip is not None else '',
283 'QUAGGA_IP': quagga_ip if quagga_ip is not None else '',
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700284 }
285 if args.olt:
Chetan Gaonker7142a342016-04-07 14:53:12 -0700286 olt_conf_test_loc = os.path.join(CordTester.sandbox_setup, 'olt_config.json')
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700287 test_cnt_env['OLT_CONFIG'] = olt_conf_test_loc
288
A R Karthick1f03e912016-05-18 11:39:22 -0700289 port_num = 0
290 num_tests = len(tests_parallel)
291 tests_per_container = max(1, num_tests/args.num_containers)
292 test_slice_start = 0
293 test_slice_end = test_slice_start + tests_per_container
294 num_test_containers = min(num_tests, args.num_containers)
295 if tests_parallel:
296 print('Running %s tests across %d containers in parallel' %(tests_parallel, num_test_containers))
297 for container in range(num_test_containers):
298 test_cnt = CordTester(tests_parallel[test_slice_start:test_slice_end],
299 instance = container, num_instances = num_test_containers,
300 ctlr_ip = onos_ip, image = nose_cnt['image'], tag = nose_cnt['tag'],
301 env = test_cnt_env,
302 rm = False if args.keep else True,
303 update = update_map['test'])
304 test_slice_start = test_slice_end
305 test_slice_end = test_slice_start + tests_per_container
306 update_map['test'] = False
307 test_containers.append(test_cnt)
308 if args.start_switch or not args.olt:
309 test_cnt.start_switch()
310 if test_cnt.olt:
311 _, port_num = test_cnt.setup_intfs(port_num = port_num)
312
313 thread_pool = ThreadPool(len(test_containers), queue_size = 1, wait_timeout=1)
314 for test_cnt in test_containers:
315 thread_pool.addTask(test_cnt.run_tests)
316 thread_pool.cleanUpThreads()
317
318 ##Run the linear tests
319 if tests_not_parallel:
320 test_cnt = CordTester(tests_not_parallel,
321 ctlr_ip = onos_ip, image = nose_cnt['image'], tag = nose_cnt['tag'],
322 env = test_cnt_env,
323 rm = False if args.keep else True,
324 update = update_map['test'])
325 if args.start_switch or not args.olt:
326 test_cnt.start_switch()
327 if test_cnt.olt:
328 test_cnt.setup_intfs(port_num = port_num)
329 test_cnt.run_tests()
330
Chetan Gaonker3533faa2016-04-25 17:50:14 -0700331 cord_test_server_stop(test_server)
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700332
Chetan Gaonker503032a2016-05-12 12:06:29 -0700333def cleanupTests(args):
334 test_container = '{}:latest'.format(CordTester.IMAGE)
335 print('Cleaning up Test containers ...')
336 Container.cleanup(test_container)
337
338def listTests(args):
339 if args.test == 'all':
340 tests = CordTester.ALL_TESTS
341 else:
A R Karthickacae3b42016-05-12 15:27:24 -0700342 tests = args.test.split('-')
Chetan Gaonker503032a2016-05-12 12:06:29 -0700343 CordTester.list_tests(tests)
344
345def buildImages(args):
346 if args.image == 'all' or args.image == 'quagga':
347 Quagga.build_image(Quagga.IMAGE)
348
349 if args.image == 'all' or args.image == 'radius':
350 Radius.build_image(Radius.IMAGE)
351
352 if args.image == 'all' or args.image == 'test':
353 CordTester.build_image(CordTester.IMAGE)
354
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700355if __name__ == '__main__':
Chetan Gaonker678743f2016-04-26 09:54:31 -0700356 parser = ArgumentParser(description='Cord Tester')
Chetan Gaonker503032a2016-05-12 12:06:29 -0700357 subparser = parser.add_subparsers()
358 parser_run = subparser.add_parser('run', help='Run cord tester')
359 parser_run.add_argument('-t', '--test-type', default=test_type_default, help='Specify test type or test case to run')
360 parser_run.add_argument('-o', '--onos', default=onos_image_default, type=str, help='ONOS container image')
Chetan Gaonker503032a2016-05-12 12:06:29 -0700361 parser_run.add_argument('-q', '--quagga',action='store_true',help='Provision quagga container for vrouter')
362 parser_run.add_argument('-a', '--app', default=onos_app_file, type=str, help='Cord ONOS app filename')
363 parser_run.add_argument('-p', '--olt', action='store_true', help='Use OLT config')
364 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 -0700365 'Eg: 10.0.0.2/10.0.0.3 to specify ONOS and Radius ip to connect')
Chetan Gaonker503032a2016-05-12 12:06:29 -0700366 parser_run.add_argument('-k', '--keep', action='store_true', help='Keep test container after tests')
367 parser_run.add_argument('-s', '--start-switch', action='store_true', help='Start OVS when running under OLT config')
368 parser_run.add_argument('-u', '--update', default='none', choices=['test','quagga','radius', 'all'], type=str, help='Update cord tester container images. '
369 'Eg: --update=quagga to rebuild quagga image.'
370 ' --update=radius to rebuild radius server image.'
371 ' --update=test to rebuild cord test image.(Default)'
372 ' --update=all to rebuild all cord tester images.')
A R Karthick1f03e912016-05-18 11:39:22 -0700373 parser_run.add_argument('-n', '--num-containers', default=1, type=int,
374 help='Specify number of test containers to spawn for tests')
Chetan Gaonker503032a2016-05-12 12:06:29 -0700375 parser_run.set_defaults(func=runTest)
376
377 parser_list = subparser.add_parser('list', help='List test cases')
378 parser_list.add_argument('-t', '--test', default='all', help='Specify test type to list test cases. '
379 'Eg: -t tls to list tls test cases.'
380 ' -t tls-dhcp-vrouter to list tls,dhcp and vrouter test cases.'
381 ' -t all to list all test cases.')
382 parser_list.set_defaults(func=listTests)
383
384 parser_build = subparser.add_parser('build', help='Build cord test container images')
385 parser_build.add_argument('image', choices=['quagga', 'radius', 'test', 'all'])
386 parser_build.set_defaults(func=buildImages)
387
388 parser_cleanup = subparser.add_parser('cleanup', help='Cleanup test containers')
389 parser_cleanup.set_defaults(func=cleanupTests)
390
Chetan Gaonker93e302d2016-04-05 10:51:07 -0700391 args = parser.parse_args()
392 args.func(args)