blob: fe41a68ae204fda99eff10d3b1b460c44e7bbeaf [file] [log] [blame]
andrewonlab95ce8322014-10-13 14:12:04 -04001#!/usr/bin/env python
2
kelvin8ec71442015-01-15 16:57:00 -08003"""
Jeremy Ronquillob27ce4c2017-07-17 12:41:28 -07004OCT 13 2014
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00005Copyright 2014 Open Networking Foundation (ONF)
Jeremy Ronquillob27ce4c2017-07-17 12:41:28 -07006
7Please refer questions to either the onos test mailing list at <onos-test@onosproject.org>,
8the System Testing Plans and Results wiki page at <https://wiki.onosproject.org/x/voMg>,
9or the System Testing Guide page at <https://wiki.onosproject.org/x/WYQg>
10
11 TestON is free software: you can redistribute it and/or modify
12 it under the terms of the GNU General Public License as published by
13 the Free Software Foundation, either version 2 of the License, or
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +000014 (at your option) any later version.
Jeremy Ronquillob27ce4c2017-07-17 12:41:28 -070015
16 TestON is distributed in the hope that it will be useful,
17 but WITHOUT ANY WARRANTY; without even the implied warranty of
18 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 GNU General Public License for more details.
20
21 You should have received a copy of the GNU General Public License
22 along with TestON. If not, see <http://www.gnu.org/licenses/>.
23"""
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +000024
Jeremy Ronquillob27ce4c2017-07-17 12:41:28 -070025"""
andrewonlab95ce8322014-10-13 14:12:04 -040026This driver enters the onos> prompt to issue commands.
27
kelvin8ec71442015-01-15 16:57:00 -080028Please follow the coding style demonstrated by existing
andrewonlab95ce8322014-10-13 14:12:04 -040029functions and document properly.
30
31If you are a contributor to the driver, please
32list your email here for future contact:
33
34jhall@onlab.us
35andrew@onlab.us
Jon Halle8217482014-10-17 13:49:14 -040036shreya@onlab.us
Jeremy Ronquillo818bc7c2017-08-09 17:14:53 +000037jeremyr@opennetworking.org
kelvin8ec71442015-01-15 16:57:00 -080038"""
andrewonlab95ce8322014-10-13 14:12:04 -040039import pexpect
40import re
Jon Hall30b82fa2015-03-04 17:15:43 -080041import json
42import types
Jon Hallbd16b922015-03-26 17:53:15 -070043import time
kelvin-onlaba4074292015-07-09 15:19:49 -070044import os
andrewonlab95ce8322014-10-13 14:12:04 -040045from drivers.common.clidriver import CLI
You Wangdb8cd0a2016-05-26 15:19:45 -070046from core.graph import Graph
Shreya Chowdhary6fbb96c2017-05-02 16:20:19 -070047from cStringIO import StringIO
48from itertools import izip
andrewonlab95ce8322014-10-13 14:12:04 -040049
kelvin8ec71442015-01-15 16:57:00 -080050class OnosCliDriver( CLI ):
andrewonlab95ce8322014-10-13 14:12:04 -040051
kelvin8ec71442015-01-15 16:57:00 -080052 def __init__( self ):
53 """
54 Initialize client
55 """
Jon Hallefbd9792015-03-05 16:11:36 -080056 self.name = None
57 self.home = None
58 self.handle = None
Devin Limdc78e202017-06-09 18:30:07 -070059 self.karafUser = None
60 self.karafPass = None
You Wangdb8cd0a2016-05-26 15:19:45 -070061 self.graph = Graph()
Devin Limdc78e202017-06-09 18:30:07 -070062 super( OnosCliDriver, self ).__init__()
kelvin8ec71442015-01-15 16:57:00 -080063
Jeremy Ronquillo82705492017-10-18 14:19:55 -070064 def checkOptions( self, var, defaultVar ):
Devin Limdc78e202017-06-09 18:30:07 -070065 if var is None or var == "":
66 return defaultVar
67 return var
Jeremy Ronquillo82705492017-10-18 14:19:55 -070068
kelvin8ec71442015-01-15 16:57:00 -080069 def connect( self, **connectargs ):
70 """
andrewonlab95ce8322014-10-13 14:12:04 -040071 Creates ssh handle for ONOS cli.
kelvin8ec71442015-01-15 16:57:00 -080072 """
andrewonlab95ce8322014-10-13 14:12:04 -040073 try:
74 for key in connectargs:
kelvin8ec71442015-01-15 16:57:00 -080075 vars( self )[ key ] = connectargs[ key ]
andrew@onlab.us658ec012015-03-11 15:13:09 -070076 self.home = "~/onos"
andrewonlab95ce8322014-10-13 14:12:04 -040077 for key in self.options:
78 if key == "home":
Devin Limdc78e202017-06-09 18:30:07 -070079 self.home = self.options[ key ]
80 elif key == "karaf_username":
81 self.karafUser = self.options[ key ]
82 elif key == "karaf_password":
83 self.karafPass = self.options[ key ]
84
Jeremy Ronquillo82705492017-10-18 14:19:55 -070085 self.home = self.checkOptions( self.home, "~/onos" )
86 self.karafUser = self.checkOptions( self.karafUser, self.user_name )
87 self.karafPass = self.checkOptions( self.karafPass, self.pwd )
andrewonlab95ce8322014-10-13 14:12:04 -040088
kelvin-onlaba4074292015-07-09 15:19:49 -070089 for key in self.options:
90 if key == 'onosIp':
91 self.onosIp = self.options[ 'onosIp' ]
92 break
93
kelvin8ec71442015-01-15 16:57:00 -080094 self.name = self.options[ 'name' ]
kelvin-onlaba4074292015-07-09 15:19:49 -070095
96 try:
Jon Hallc6793552016-01-19 14:18:37 -080097 if os.getenv( str( self.ip_address ) ) is not None:
kelvin-onlaba4074292015-07-09 15:19:49 -070098 self.ip_address = os.getenv( str( self.ip_address ) )
99 else:
100 main.log.info( self.name +
101 ": Trying to connect to " +
102 self.ip_address )
103
104 except KeyError:
105 main.log.info( "Invalid host name," +
106 " connecting to local host instead" )
107 self.ip_address = 'localhost'
108 except Exception as inst:
109 main.log.error( "Uncaught exception: " + str( inst ) )
110
kelvin8ec71442015-01-15 16:57:00 -0800111 self.handle = super( OnosCliDriver, self ).connect(
kelvin-onlab08679eb2015-01-21 16:11:48 -0800112 user_name=self.user_name,
113 ip_address=self.ip_address,
kelvin-onlab898a6c62015-01-16 14:13:53 -0800114 port=self.port,
115 pwd=self.pwd,
116 home=self.home )
andrewonlab95ce8322014-10-13 14:12:04 -0400117
kelvin8ec71442015-01-15 16:57:00 -0800118 self.handle.sendline( "cd " + self.home )
Devin Limdc78e202017-06-09 18:30:07 -0700119 self.handle.expect( self.prompt )
andrewonlab95ce8322014-10-13 14:12:04 -0400120 if self.handle:
121 return self.handle
kelvin8ec71442015-01-15 16:57:00 -0800122 else:
123 main.log.info( "NO ONOS HANDLE" )
andrewonlab95ce8322014-10-13 14:12:04 -0400124 return main.FALSE
Jon Halld4d4b372015-01-28 16:02:41 -0800125 except TypeError:
126 main.log.exception( self.name + ": Object not as expected" )
127 return None
andrewonlab95ce8322014-10-13 14:12:04 -0400128 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -0800129 main.log.error( self.name + ": EOF exception found" )
130 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700131 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -0800132 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -0800133 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700134 main.cleanAndExit()
andrewonlab95ce8322014-10-13 14:12:04 -0400135
kelvin8ec71442015-01-15 16:57:00 -0800136 def disconnect( self ):
137 """
andrewonlab95ce8322014-10-13 14:12:04 -0400138 Called when Test is complete to disconnect the ONOS handle.
kelvin8ec71442015-01-15 16:57:00 -0800139 """
Jon Halld61331b2015-02-17 16:35:47 -0800140 response = main.TRUE
andrewonlab95ce8322014-10-13 14:12:04 -0400141 try:
Jon Hall61282e32015-03-19 11:34:11 -0700142 if self.handle:
143 i = self.logout()
144 if i == main.TRUE:
145 self.handle.sendline( "" )
Devin Limdc78e202017-06-09 18:30:07 -0700146 self.handle.expect( self.prompt )
Jon Hall61282e32015-03-19 11:34:11 -0700147 self.handle.sendline( "exit" )
148 self.handle.expect( "closed" )
Jon Halld4d4b372015-01-28 16:02:41 -0800149 except TypeError:
150 main.log.exception( self.name + ": Object not as expected" )
Jon Halld61331b2015-02-17 16:35:47 -0800151 response = main.FALSE
andrewonlab95ce8322014-10-13 14:12:04 -0400152 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -0800153 main.log.error( self.name + ": EOF exception found" )
154 main.log.error( self.name + ": " + self.handle.before )
Jon Hall61282e32015-03-19 11:34:11 -0700155 except ValueError:
Jon Hall1a77a1e2015-04-06 10:41:13 -0700156 main.log.exception( "Exception in disconnect of " + self.name )
Jon Hall61282e32015-03-19 11:34:11 -0700157 response = main.TRUE
Jon Hallfebb1c72015-03-05 13:30:09 -0800158 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -0800159 main.log.exception( self.name + ": Connection failed to the host" )
andrewonlab95ce8322014-10-13 14:12:04 -0400160 response = main.FALSE
161 return response
162
kelvin8ec71442015-01-15 16:57:00 -0800163 def logout( self ):
164 """
andrewonlab38d2b4a2014-11-13 16:28:47 -0500165 Sends 'logout' command to ONOS cli
Jon Hall61282e32015-03-19 11:34:11 -0700166 Returns main.TRUE if exited CLI and
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +0000167 main.FALSE on timeout (not guranteed you are disconnected)
Jon Hall61282e32015-03-19 11:34:11 -0700168 None on TypeError
169 Exits test on unknown error or pexpect exits unexpectedly
kelvin8ec71442015-01-15 16:57:00 -0800170 """
andrewonlab38d2b4a2014-11-13 16:28:47 -0500171 try:
Jon Hall61282e32015-03-19 11:34:11 -0700172 if self.handle:
173 self.handle.sendline( "" )
Devin Limdc78e202017-06-09 18:30:07 -0700174 i = self.handle.expect( [ "onos>", self.prompt, pexpect.TIMEOUT ],
Jon Hall61282e32015-03-19 11:34:11 -0700175 timeout=10 )
176 if i == 0: # In ONOS CLI
177 self.handle.sendline( "logout" )
Devin Limdc78e202017-06-09 18:30:07 -0700178 j = self.handle.expect( [ self.prompt,
Jon Hallbfe00002016-04-05 10:23:54 -0700179 "Command not found:",
180 pexpect.TIMEOUT ] )
181 if j == 0: # Successfully logged out
182 return main.TRUE
183 elif j == 1 or j == 2:
184 # ONOS didn't fully load, and logout command isn't working
185 # or the command timed out
186 self.handle.send( "\x04" ) # send ctrl-d
Jon Hall64ab3bd2016-05-13 11:29:44 -0700187 try:
Devin Limdc78e202017-06-09 18:30:07 -0700188 self.handle.expect( self.prompt )
Jon Hall64ab3bd2016-05-13 11:29:44 -0700189 except pexpect.TIMEOUT:
190 main.log.error( "ONOS did not respond to 'logout' or CTRL-d" )
Jon Hallbfe00002016-04-05 10:23:54 -0700191 return main.TRUE
Jon Halle0f0b342017-04-18 11:43:47 -0700192 else: # some other output
Jon Hallbfe00002016-04-05 10:23:54 -0700193 main.log.warn( "Unknown repsonse to logout command: '{}'",
194 repr( self.handle.before ) )
195 return main.FALSE
Jon Hall61282e32015-03-19 11:34:11 -0700196 elif i == 1: # not in CLI
197 return main.TRUE
198 elif i == 3: # Timeout
199 return main.FALSE
200 else:
andrewonlab9627f432014-11-14 12:45:10 -0500201 return main.TRUE
Jon Halld4d4b372015-01-28 16:02:41 -0800202 except TypeError:
203 main.log.exception( self.name + ": Object not as expected" )
204 return None
andrewonlab38d2b4a2014-11-13 16:28:47 -0500205 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -0800206 main.log.error( self.name + ": eof exception found" )
Jon Hall61282e32015-03-19 11:34:11 -0700207 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700208 main.cleanAndExit()
Jon Hall61282e32015-03-19 11:34:11 -0700209 except ValueError:
Jon Hall5aa168b2015-03-23 14:23:09 -0700210 main.log.error( self.name +
211 "ValueError exception in logout method" )
Jon Hallfebb1c72015-03-05 13:30:09 -0800212 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -0800213 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700214 main.cleanAndExit()
andrewonlab38d2b4a2014-11-13 16:28:47 -0500215
kelvin-onlabd3b64892015-01-20 13:26:24 -0800216 def setCell( self, cellname ):
kelvin8ec71442015-01-15 16:57:00 -0800217 """
andrewonlab95ce8322014-10-13 14:12:04 -0400218 Calls 'cell <name>' to set the environment variables on ONOSbench
kelvin8ec71442015-01-15 16:57:00 -0800219
andrewonlab95ce8322014-10-13 14:12:04 -0400220 Before issuing any cli commands, set the environment variable first.
kelvin8ec71442015-01-15 16:57:00 -0800221 """
andrewonlab95ce8322014-10-13 14:12:04 -0400222 try:
223 if not cellname:
kelvin8ec71442015-01-15 16:57:00 -0800224 main.log.error( "Must define cellname" )
Devin Lim44075962017-08-11 10:56:37 -0700225 main.cleanAndExit()
andrewonlab95ce8322014-10-13 14:12:04 -0400226 else:
kelvin8ec71442015-01-15 16:57:00 -0800227 self.handle.sendline( "cell " + str( cellname ) )
kelvin-onlabd3b64892015-01-20 13:26:24 -0800228 # Expect the cellname in the ONOSCELL variable.
kelvin8ec71442015-01-15 16:57:00 -0800229 # Note that this variable name is subject to change
andrewonlab95ce8322014-10-13 14:12:04 -0400230 # and that this driver will have to change accordingly
Jeremy Ronquillo82705492017-10-18 14:19:55 -0700231 self.handle.expect( str( cellname ) )
andrew@onlab.usc400b112015-01-21 15:33:19 -0800232 handleBefore = self.handle.before
233 handleAfter = self.handle.after
kelvin8ec71442015-01-15 16:57:00 -0800234 # Get the rest of the handle
Jeremy Ronquillo82705492017-10-18 14:19:55 -0700235 self.handle.sendline( "" )
236 self.handle.expect( self.prompt )
andrew@onlab.usc400b112015-01-21 15:33:19 -0800237 handleMore = self.handle.before
andrewonlab95ce8322014-10-13 14:12:04 -0400238
kelvin-onlabd3b64892015-01-20 13:26:24 -0800239 main.log.info( "Cell call returned: " + handleBefore +
240 handleAfter + handleMore )
andrewonlab95ce8322014-10-13 14:12:04 -0400241
242 return main.TRUE
243
Jon Halld4d4b372015-01-28 16:02:41 -0800244 except TypeError:
245 main.log.exception( self.name + ": Object not as expected" )
246 return None
andrewonlab95ce8322014-10-13 14:12:04 -0400247 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -0800248 main.log.error( self.name + ": eof exception found" )
249 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700250 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -0800251 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -0800252 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700253 main.cleanAndExit()
kelvin8ec71442015-01-15 16:57:00 -0800254
pingping-lin57a56ce2015-05-20 16:43:48 -0700255 def startOnosCli( self, ONOSIp, karafTimeout="",
Chiyu Chengef109502016-11-21 15:51:38 -0800256 commandlineTimeout=10, onosStartTimeout=60, waitForStart=False ):
kelvin8ec71442015-01-15 16:57:00 -0800257 """
Jon Hallefbd9792015-03-05 16:11:36 -0800258 karafTimeout is an optional argument. karafTimeout value passed
kelvin-onlabd3b64892015-01-20 13:26:24 -0800259 by user would be used to set the current karaf shell idle timeout.
260 Note that when ever this property is modified the shell will exit and
Hari Krishnad7b9c202015-01-05 10:38:14 -0800261 the subsequent login would reflect new idle timeout.
kelvin-onlabd3b64892015-01-20 13:26:24 -0800262 Below is an example to start a session with 60 seconds idle timeout
263 ( input value is in milliseconds ):
kelvin8ec71442015-01-15 16:57:00 -0800264
Hari Krishna25d42f72015-01-05 15:08:28 -0800265 tValue = "60000"
kelvin-onlabd3b64892015-01-20 13:26:24 -0800266 main.ONOScli1.startOnosCli( ONOSIp, karafTimeout=tValue )
kelvin8ec71442015-01-15 16:57:00 -0800267
kelvin-onlabd3b64892015-01-20 13:26:24 -0800268 Note: karafTimeout is left as str so that this could be read
269 and passed to startOnosCli from PARAMS file as str.
kelvin8ec71442015-01-15 16:57:00 -0800270 """
You Wangf69ab392016-01-26 16:34:38 -0800271 self.onosIp = ONOSIp
andrewonlab95ce8322014-10-13 14:12:04 -0400272 try:
Jon Hall67253832016-12-05 09:47:13 -0800273 # Check if we are already in the cli
kelvin8ec71442015-01-15 16:57:00 -0800274 self.handle.sendline( "" )
275 x = self.handle.expect( [
Jeremy Ronquillo82705492017-10-18 14:19:55 -0700276 self.prompt, "onos>" ], commandlineTimeout )
andrewonlab48829f62014-11-17 13:49:01 -0500277 if x == 1:
kelvin8ec71442015-01-15 16:57:00 -0800278 main.log.info( "ONOS cli is already running" )
andrewonlab48829f62014-11-17 13:49:01 -0500279 return main.TRUE
andrewonlab95ce8322014-10-13 14:12:04 -0400280
Jon Hall67253832016-12-05 09:47:13 -0800281 # Not in CLI so login
Chiyu Chengef109502016-11-21 15:51:38 -0800282 if waitForStart:
Jeremy Ronquilloec916a42018-02-02 13:05:57 -0800283 # Wait for onos start ( onos-wait-for-start ) and enter onos cli
284 startCliCommand = "onos-wait-for-start "
Chiyu Chengef109502016-11-21 15:51:38 -0800285 else:
286 startCliCommand = "onos "
287 self.handle.sendline( startCliCommand + str( ONOSIp ) )
kelvin8ec71442015-01-15 16:57:00 -0800288 i = self.handle.expect( [
289 "onos>",
pingping-lin57a56ce2015-05-20 16:43:48 -0700290 pexpect.TIMEOUT ], onosStartTimeout )
andrewonlab2a7ea9b2014-10-24 12:21:05 -0400291
292 if i == 0:
kelvin-onlabd3b64892015-01-20 13:26:24 -0800293 main.log.info( str( ONOSIp ) + " CLI Started successfully" )
Hari Krishnae36ef212015-01-04 14:09:13 -0800294 if karafTimeout:
kelvin8ec71442015-01-15 16:57:00 -0800295 self.handle.sendline(
Hari Krishnaac4e1782015-01-26 12:09:12 -0800296 "config:property-set -p org.apache.karaf.shell\
297 sshIdleTimeout " +
kelvin8ec71442015-01-15 16:57:00 -0800298 karafTimeout )
Devin Limdc78e202017-06-09 18:30:07 -0700299 self.handle.expect( self.prompt )
Chiyu Chengef109502016-11-21 15:51:38 -0800300 self.handle.sendline( startCliCommand + str( ONOSIp ) )
kelvin8ec71442015-01-15 16:57:00 -0800301 self.handle.expect( "onos>" )
andrewonlab2a7ea9b2014-10-24 12:21:05 -0400302 return main.TRUE
303 else:
kelvin8ec71442015-01-15 16:57:00 -0800304 # If failed, send ctrl+c to process and try again
305 main.log.info( "Starting CLI failed. Retrying..." )
306 self.handle.send( "\x03" )
Chiyu Chengef109502016-11-21 15:51:38 -0800307 self.handle.sendline( startCliCommand + str( ONOSIp ) )
kelvin8ec71442015-01-15 16:57:00 -0800308 i = self.handle.expect( [ "onos>", pexpect.TIMEOUT ],
309 timeout=30 )
andrewonlab3a7c3c72014-10-24 17:21:03 -0400310 if i == 0:
kelvin-onlabd3b64892015-01-20 13:26:24 -0800311 main.log.info( str( ONOSIp ) + " CLI Started " +
kelvin8ec71442015-01-15 16:57:00 -0800312 "successfully after retry attempt" )
Hari Krishnae36ef212015-01-04 14:09:13 -0800313 if karafTimeout:
kelvin8ec71442015-01-15 16:57:00 -0800314 self.handle.sendline(
kelvin-onlabd3b64892015-01-20 13:26:24 -0800315 "config:property-set -p org.apache.karaf.shell\
316 sshIdleTimeout " +
kelvin8ec71442015-01-15 16:57:00 -0800317 karafTimeout )
Devin Limdc78e202017-06-09 18:30:07 -0700318 self.handle.expect( self.prompt )
Chiyu Chengef109502016-11-21 15:51:38 -0800319 self.handle.sendline( startCliCommand + str( ONOSIp ) )
kelvin8ec71442015-01-15 16:57:00 -0800320 self.handle.expect( "onos>" )
andrewonlab3a7c3c72014-10-24 17:21:03 -0400321 return main.TRUE
322 else:
kelvin8ec71442015-01-15 16:57:00 -0800323 main.log.error( "Connection to CLI " +
kelvin-onlabd3b64892015-01-20 13:26:24 -0800324 str( ONOSIp ) + " timeout" )
andrewonlab3a7c3c72014-10-24 17:21:03 -0400325 return main.FALSE
andrewonlab95ce8322014-10-13 14:12:04 -0400326
Jon Halld4d4b372015-01-28 16:02:41 -0800327 except TypeError:
328 main.log.exception( self.name + ": Object not as expected" )
329 return None
andrewonlab95ce8322014-10-13 14:12:04 -0400330 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -0800331 main.log.error( self.name + ": EOF exception found" )
332 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700333 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -0800334 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -0800335 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700336 main.cleanAndExit()
andrewonlab95ce8322014-10-13 14:12:04 -0400337
suibin zhang116647a2016-05-06 16:30:09 -0700338 def startCellCli( self, karafTimeout="",
339 commandlineTimeout=10, onosStartTimeout=60 ):
340 """
341 Start CLI on onos ecll handle.
342
343 karafTimeout is an optional argument. karafTimeout value passed
344 by user would be used to set the current karaf shell idle timeout.
345 Note that when ever this property is modified the shell will exit and
346 the subsequent login would reflect new idle timeout.
347 Below is an example to start a session with 60 seconds idle timeout
348 ( input value is in milliseconds ):
349
350 tValue = "60000"
351
352 Note: karafTimeout is left as str so that this could be read
353 and passed to startOnosCli from PARAMS file as str.
354 """
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +0000355
suibin zhang116647a2016-05-06 16:30:09 -0700356 try:
357 self.handle.sendline( "" )
358 x = self.handle.expect( [
Jeremy Ronquillo82705492017-10-18 14:19:55 -0700359 self.prompt, "onos>" ], commandlineTimeout )
suibin zhang116647a2016-05-06 16:30:09 -0700360
361 if x == 1:
362 main.log.info( "ONOS cli is already running" )
363 return main.TRUE
364
Jeremy Ronquilloec916a42018-02-02 13:05:57 -0800365 # Wait for onos start ( onos-wait-for-start ) and enter onos cli
suibin zhang116647a2016-05-06 16:30:09 -0700366 self.handle.sendline( "/opt/onos/bin/onos" )
367 i = self.handle.expect( [
368 "onos>",
369 pexpect.TIMEOUT ], onosStartTimeout )
370
371 if i == 0:
372 main.log.info( self.name + " CLI Started successfully" )
373 if karafTimeout:
374 self.handle.sendline(
375 "config:property-set -p org.apache.karaf.shell\
376 sshIdleTimeout " +
377 karafTimeout )
Devin Limdc78e202017-06-09 18:30:07 -0700378 self.handle.expect( self.prompt )
suibin zhang116647a2016-05-06 16:30:09 -0700379 self.handle.sendline( "/opt/onos/bin/onos" )
380 self.handle.expect( "onos>" )
381 return main.TRUE
382 else:
383 # If failed, send ctrl+c to process and try again
384 main.log.info( "Starting CLI failed. Retrying..." )
385 self.handle.send( "\x03" )
386 self.handle.sendline( "/opt/onos/bin/onos" )
387 i = self.handle.expect( [ "onos>", pexpect.TIMEOUT ],
388 timeout=30 )
389 if i == 0:
390 main.log.info( self.name + " CLI Started " +
391 "successfully after retry attempt" )
392 if karafTimeout:
393 self.handle.sendline(
394 "config:property-set -p org.apache.karaf.shell\
395 sshIdleTimeout " +
396 karafTimeout )
Devin Limdc78e202017-06-09 18:30:07 -0700397 self.handle.expect( self.prompt )
suibin zhang116647a2016-05-06 16:30:09 -0700398 self.handle.sendline( "/opt/onos/bin/onos" )
399 self.handle.expect( "onos>" )
400 return main.TRUE
401 else:
402 main.log.error( "Connection to CLI " +
403 self.name + " timeout" )
404 return main.FALSE
405
406 except TypeError:
407 main.log.exception( self.name + ": Object not as expected" )
408 return None
409 except pexpect.EOF:
410 main.log.error( self.name + ": EOF exception found" )
411 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700412 main.cleanAndExit()
suibin zhang116647a2016-05-06 16:30:09 -0700413 except Exception:
414 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700415 main.cleanAndExit()
suibin zhang116647a2016-05-06 16:30:09 -0700416
Pratik Parab3b2ab5a2017-02-14 13:15:14 -0800417 def log( self, cmdStr, level="", noExit=False ):
kelvin-onlab9f541032015-02-04 16:19:53 -0800418 """
419 log the commands in the onos CLI.
kelvin-onlab338f5512015-02-06 10:53:16 -0800420 returns main.TRUE on success
Jon Hallefbd9792015-03-05 16:11:36 -0800421 returns main.FALSE if Error occurred
YPZhangebf9eb52016-05-12 15:20:24 -0700422 if noExit is True, TestON will not exit, but clean up
kelvin-onlab338f5512015-02-06 10:53:16 -0800423 Available level: DEBUG, TRACE, INFO, WARN, ERROR
424 Level defaults to INFO
Pratik Parab3b2ab5a2017-02-14 13:15:14 -0800425 if cmdStr has spaces then put quotes in the passed string
kelvin-onlab9f541032015-02-04 16:19:53 -0800426 """
427 try:
kelvin-onlab338f5512015-02-06 10:53:16 -0800428 lvlStr = ""
429 if level:
430 lvlStr = "--level=" + level
431
kelvin-onlab338f5512015-02-06 10:53:16 -0800432 self.handle.sendline( "log:log " + lvlStr + " " + cmdStr )
Jon Hall390696c2015-05-05 17:13:41 -0700433 self.handle.expect( "log:log" )
kelvin-onlab9f541032015-02-04 16:19:53 -0800434 self.handle.expect( "onos>" )
kelvin-onlabfb521662015-02-27 09:52:40 -0800435
kelvin-onlab9f541032015-02-04 16:19:53 -0800436 response = self.handle.before
437 if re.search( "Error", response ):
438 return main.FALSE
439 return main.TRUE
Jon Hall80daded2015-05-27 16:07:00 -0700440 except pexpect.TIMEOUT:
441 main.log.exception( self.name + ": TIMEOUT exception found" )
YPZhangebf9eb52016-05-12 15:20:24 -0700442 if noExit:
443 main.cleanup()
444 return None
445 else:
Devin Lim44075962017-08-11 10:56:37 -0700446 main.cleanAndExit()
kelvin-onlab9f541032015-02-04 16:19:53 -0800447 except pexpect.EOF:
448 main.log.error( self.name + ": EOF exception found" )
449 main.log.error( self.name + ": " + self.handle.before )
YPZhangebf9eb52016-05-12 15:20:24 -0700450 if noExit:
451 main.cleanup()
452 return None
453 else:
Devin Lim44075962017-08-11 10:56:37 -0700454 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -0800455 except Exception:
kelvin-onlabfb521662015-02-27 09:52:40 -0800456 main.log.exception( self.name + ": Uncaught exception!" )
YPZhangebf9eb52016-05-12 15:20:24 -0700457 if noExit:
458 main.cleanup()
459 return None
460 else:
Devin Lim44075962017-08-11 10:56:37 -0700461 main.cleanAndExit()
andrewonlab95ce8322014-10-13 14:12:04 -0400462
Shreya Chowdhary6fbb96c2017-05-02 16:20:19 -0700463 def sendline( self, cmdStr, showResponse=False, debug=False, timeout=10, noExit=False, dollarSign=False ):
kelvin8ec71442015-01-15 16:57:00 -0800464 """
Jon Halle3f39ff2015-01-13 11:50:53 -0800465 Send a completely user specified string to
466 the onos> prompt. Use this function if you have
andrewonlaba18f6bf2014-10-13 19:31:54 -0400467 a very specific command to send.
Jon Halle3f39ff2015-01-13 11:50:53 -0800468
YPZhang14a4aa92016-07-15 13:37:15 -0700469 if noExit is True, TestON will not exit, and return None
Shreya Chowdhary6fbb96c2017-05-02 16:20:19 -0700470 if dollarSign is True, TestON will not expect for '$' as a new CLI or onos> prompt
471 since '$' can be in the output.
YPZhangebf9eb52016-05-12 15:20:24 -0700472
andrewonlaba18f6bf2014-10-13 19:31:54 -0400473 Warning: There are no sanity checking to commands
474 sent using this method.
GlennRCed771242016-01-13 17:02:47 -0800475
kelvin8ec71442015-01-15 16:57:00 -0800476 """
andrewonlaba18f6bf2014-10-13 19:31:54 -0400477 try:
Jon Halla495f562016-05-16 18:03:26 -0700478 # Try to reconnect if disconnected from cli
479 self.handle.sendline( "" )
Devin Limdc78e202017-06-09 18:30:07 -0700480 i = self.handle.expect( [ "onos>", self.prompt, pexpect.TIMEOUT ] )
Jon Halla495f562016-05-16 18:03:26 -0700481 if i == 1:
Jeremy Ronquillo82705492017-10-18 14:19:55 -0700482 main.log.error( self.name + ": onos cli session closed. " )
Jon Halla495f562016-05-16 18:03:26 -0700483 if self.onosIp:
484 main.log.warn( "Trying to reconnect " + self.onosIp )
485 reconnectResult = self.startOnosCli( self.onosIp )
486 if reconnectResult:
487 main.log.info( self.name + ": onos cli session reconnected." )
488 else:
489 main.log.error( self.name + ": reconnection failed." )
YPZhang14a4aa92016-07-15 13:37:15 -0700490 if noExit:
491 return None
492 else:
Devin Lim44075962017-08-11 10:56:37 -0700493 main.cleanAndExit()
Jon Halla495f562016-05-16 18:03:26 -0700494 else:
Devin Lim44075962017-08-11 10:56:37 -0700495 main.cleanAndExit()
Jon Halla495f562016-05-16 18:03:26 -0700496 if i == 2:
Jon Hall7a6ebfd2017-03-13 10:58:58 -0700497 main.log.warn( "Timeout when testing cli responsiveness" )
498 main.log.debug( self.handle.before )
499 self.handle.send( "\x03" ) # Send ctrl-c to clear previous output
Jon Halla495f562016-05-16 18:03:26 -0700500 self.handle.expect( "onos>" )
501
Jon Hall14a03b52016-05-11 12:07:30 -0700502 if debug:
503 # NOTE: This adds and average of .4 seconds per call
504 logStr = "\"Sending CLI command: '" + cmdStr + "'\""
Jon Halle0f0b342017-04-18 11:43:47 -0700505 self.log( logStr, noExit=noExit )
kelvin-onlabd3b64892015-01-20 13:26:24 -0800506 self.handle.sendline( cmdStr )
Shreya Chowdhary6fbb96c2017-05-02 16:20:19 -0700507 if dollarSign:
Jeremy Ronquillo82705492017-10-18 14:19:55 -0700508 i = self.handle.expect( [ "onos>" ], timeout )
Shreya Chowdhary6fbb96c2017-05-02 16:20:19 -0700509 else:
Jeremy Ronquillo82705492017-10-18 14:19:55 -0700510 i = self.handle.expect( [ "onos>", self.prompt ], timeout )
Jon Hall63604932015-02-26 17:09:50 -0800511 response = self.handle.before
Jon Hall63604932015-02-26 17:09:50 -0800512 # TODO: do something with i
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +0000513 main.log.info( "Command '" + str( cmdStr ) + "' sent to "
Jon Hallc6793552016-01-19 14:18:37 -0800514 + self.name + "." )
Jon Hallc6358dd2015-04-10 12:44:28 -0700515 if debug:
Jon Hall390696c2015-05-05 17:13:41 -0700516 main.log.debug( self.name + ": Raw output" )
517 main.log.debug( self.name + ": " + repr( response ) )
Jon Hallc6358dd2015-04-10 12:44:28 -0700518
519 # Remove ANSI color control strings from output
kelvin-onlabd3b64892015-01-20 13:26:24 -0800520 ansiEscape = re.compile( r'\x1b[^m]*m' )
Jon Hall63604932015-02-26 17:09:50 -0800521 response = ansiEscape.sub( '', response )
Jon Hallc6358dd2015-04-10 12:44:28 -0700522 if debug:
Jon Hall390696c2015-05-05 17:13:41 -0700523 main.log.debug( self.name + ": ansiEscape output" )
524 main.log.debug( self.name + ": " + repr( response ) )
Jon Hallc6358dd2015-04-10 12:44:28 -0700525
kelvin-onlabfb521662015-02-27 09:52:40 -0800526 # Remove extra return chars that get added
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +0000527 response = re.sub( r"\s\r", "", response )
Jon Hallc6358dd2015-04-10 12:44:28 -0700528 if debug:
Jon Hall390696c2015-05-05 17:13:41 -0700529 main.log.debug( self.name + ": Removed extra returns " +
530 "from output" )
531 main.log.debug( self.name + ": " + repr( response ) )
Jon Hallc6358dd2015-04-10 12:44:28 -0700532
533 # Strip excess whitespace
Jon Hall63604932015-02-26 17:09:50 -0800534 response = response.strip()
Jon Hallc6358dd2015-04-10 12:44:28 -0700535 if debug:
Jon Hall390696c2015-05-05 17:13:41 -0700536 main.log.debug( self.name + ": parsed and stripped output" )
537 main.log.debug( self.name + ": " + repr( response ) )
Jon Hallc6358dd2015-04-10 12:44:28 -0700538
Jon Hall63604932015-02-26 17:09:50 -0800539 # parse for just the output, remove the cmd from response
Jon Hallc6358dd2015-04-10 12:44:28 -0700540 output = response.split( cmdStr.strip(), 1 )
541 if debug:
Jon Hall390696c2015-05-05 17:13:41 -0700542 main.log.debug( self.name + ": split output" )
Jon Hallc6358dd2015-04-10 12:44:28 -0700543 for r in output:
Jon Hall390696c2015-05-05 17:13:41 -0700544 main.log.debug( self.name + ": " + repr( r ) )
Jeremy Ronquillo82705492017-10-18 14:19:55 -0700545 output = output[ 1 ].strip()
GlennRC85870432015-11-23 11:45:51 -0800546 if showResponse:
GlennRCed771242016-01-13 17:02:47 -0800547 main.log.info( "Response from ONOS: {}".format( output ) )
GlennRC85870432015-11-23 11:45:51 -0800548 return output
GlennRCed771242016-01-13 17:02:47 -0800549 except pexpect.TIMEOUT:
550 main.log.error( self.name + ":ONOS timeout" )
551 if debug:
552 main.log.debug( self.handle.before )
553 return None
Jon Hallc6358dd2015-04-10 12:44:28 -0700554 except IndexError:
555 main.log.exception( self.name + ": Object not as expected" )
Jon Halla495f562016-05-16 18:03:26 -0700556 main.log.debug( "response: {}".format( repr( response ) ) )
Jon Hallc6358dd2015-04-10 12:44:28 -0700557 return None
Jon Halld4d4b372015-01-28 16:02:41 -0800558 except TypeError:
559 main.log.exception( self.name + ": Object not as expected" )
560 return None
andrewonlaba18f6bf2014-10-13 19:31:54 -0400561 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -0800562 main.log.error( self.name + ": EOF exception found" )
563 main.log.error( self.name + ": " + self.handle.before )
YPZhangebf9eb52016-05-12 15:20:24 -0700564 if noExit:
YPZhangebf9eb52016-05-12 15:20:24 -0700565 return None
566 else:
Devin Lim44075962017-08-11 10:56:37 -0700567 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -0800568 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -0800569 main.log.exception( self.name + ": Uncaught exception!" )
YPZhangebf9eb52016-05-12 15:20:24 -0700570 if noExit:
YPZhangebf9eb52016-05-12 15:20:24 -0700571 return None
572 else:
Devin Lim44075962017-08-11 10:56:37 -0700573 main.cleanAndExit()
andrewonlaba18f6bf2014-10-13 19:31:54 -0400574
kelvin8ec71442015-01-15 16:57:00 -0800575 # IMPORTANT NOTE:
576 # For all cli commands, naming convention should match
kelvin-onlabd3b64892015-01-20 13:26:24 -0800577 # the cli command changing 'a:b' with 'aB'.
578 # Ex ) onos:topology > onosTopology
579 # onos:links > onosLinks
580 # feature:list > featureList
Jon Halle3f39ff2015-01-13 11:50:53 -0800581
kelvin-onlabd3b64892015-01-20 13:26:24 -0800582 def addNode( self, nodeId, ONOSIp, tcpPort="" ):
kelvin8ec71442015-01-15 16:57:00 -0800583 """
andrewonlabc2d05aa2014-10-13 16:51:10 -0400584 Adds a new cluster node by ID and address information.
585 Required:
kelvin-onlabd3b64892015-01-20 13:26:24 -0800586 * nodeId
587 * ONOSIp
andrewonlabc2d05aa2014-10-13 16:51:10 -0400588 Optional:
kelvin-onlabd3b64892015-01-20 13:26:24 -0800589 * tcpPort
kelvin8ec71442015-01-15 16:57:00 -0800590 """
andrewonlabc2d05aa2014-10-13 16:51:10 -0400591 try:
kelvin-onlabd3b64892015-01-20 13:26:24 -0800592 cmdStr = "add-node " + str( nodeId ) + " " +\
593 str( ONOSIp ) + " " + str( tcpPort )
594 handle = self.sendline( cmdStr )
Jon Halla495f562016-05-16 18:03:26 -0700595 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -0800596 assert "Command not found:" not in handle, handle
kelvin-onlab898a6c62015-01-16 14:13:53 -0800597 if re.search( "Error", handle ):
kelvin8ec71442015-01-15 16:57:00 -0800598 main.log.error( "Error in adding node" )
599 main.log.error( handle )
Jon Halle3f39ff2015-01-13 11:50:53 -0800600 return main.FALSE
andrewonlabc2d05aa2014-10-13 16:51:10 -0400601 else:
kelvin-onlabd3b64892015-01-20 13:26:24 -0800602 main.log.info( "Node " + str( ONOSIp ) + " added" )
andrewonlabc2d05aa2014-10-13 16:51:10 -0400603 return main.TRUE
Jon Hallc6793552016-01-19 14:18:37 -0800604 except AssertionError:
605 main.log.exception( "" )
606 return None
Jon Halld4d4b372015-01-28 16:02:41 -0800607 except TypeError:
608 main.log.exception( self.name + ": Object not as expected" )
609 return None
andrewonlabc2d05aa2014-10-13 16:51:10 -0400610 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -0800611 main.log.error( self.name + ": EOF exception found" )
612 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700613 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -0800614 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -0800615 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700616 main.cleanAndExit()
andrewonlabc2d05aa2014-10-13 16:51:10 -0400617
kelvin-onlabd3b64892015-01-20 13:26:24 -0800618 def removeNode( self, nodeId ):
kelvin8ec71442015-01-15 16:57:00 -0800619 """
andrewonlab86dc3082014-10-13 18:18:38 -0400620 Removes a cluster by ID
621 Issues command: 'remove-node [<node-id>]'
622 Required:
kelvin-onlabd3b64892015-01-20 13:26:24 -0800623 * nodeId
kelvin8ec71442015-01-15 16:57:00 -0800624 """
andrewonlab86dc3082014-10-13 18:18:38 -0400625 try:
andrewonlab86dc3082014-10-13 18:18:38 -0400626
kelvin-onlabd3b64892015-01-20 13:26:24 -0800627 cmdStr = "remove-node " + str( nodeId )
Jon Hall08f61bc2015-04-13 16:00:30 -0700628 handle = self.sendline( cmdStr )
Jon Halla495f562016-05-16 18:03:26 -0700629 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -0800630 assert "Command not found:" not in handle, handle
Jon Hallc6358dd2015-04-10 12:44:28 -0700631 if re.search( "Error", handle ):
632 main.log.error( "Error in removing node" )
633 main.log.error( handle )
634 return main.FALSE
635 else:
636 return main.TRUE
Jon Hallc6793552016-01-19 14:18:37 -0800637 except AssertionError:
638 main.log.exception( "" )
639 return None
Jon Halld4d4b372015-01-28 16:02:41 -0800640 except TypeError:
641 main.log.exception( self.name + ": Object not as expected" )
642 return None
andrewonlab86dc3082014-10-13 18:18:38 -0400643 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -0800644 main.log.error( self.name + ": EOF exception found" )
645 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700646 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -0800647 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -0800648 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700649 main.cleanAndExit()
andrewonlabc2d05aa2014-10-13 16:51:10 -0400650
Jeremy Ronquillo82705492017-10-18 14:19:55 -0700651 def nodes( self, jsonFormat=True ):
kelvin8ec71442015-01-15 16:57:00 -0800652 """
andrewonlab7c211572014-10-15 16:45:20 -0400653 List the nodes currently visible
654 Issues command: 'nodes'
Jon Hall61282e32015-03-19 11:34:11 -0700655 Optional argument:
656 * jsonFormat - boolean indicating if you want output in json
kelvin8ec71442015-01-15 16:57:00 -0800657 """
andrewonlab7c211572014-10-15 16:45:20 -0400658 try:
Jon Hallc6358dd2015-04-10 12:44:28 -0700659 cmdStr = "nodes"
Jon Hall61282e32015-03-19 11:34:11 -0700660 if jsonFormat:
Jon Hallc6358dd2015-04-10 12:44:28 -0700661 cmdStr += " -j"
662 output = self.sendline( cmdStr )
Jon Halla495f562016-05-16 18:03:26 -0700663 assert output is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -0800664 assert "Command not found:" not in output, output
Jon Hallc6358dd2015-04-10 12:44:28 -0700665 return output
Jon Hallc6793552016-01-19 14:18:37 -0800666 except AssertionError:
667 main.log.exception( "" )
668 return None
Jon Halld4d4b372015-01-28 16:02:41 -0800669 except TypeError:
670 main.log.exception( self.name + ": Object not as expected" )
671 return None
andrewonlab7c211572014-10-15 16:45:20 -0400672 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -0800673 main.log.error( self.name + ": EOF exception found" )
674 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700675 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -0800676 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -0800677 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700678 main.cleanAndExit()
andrewonlab7c211572014-10-15 16:45:20 -0400679
kelvin8ec71442015-01-15 16:57:00 -0800680 def topology( self ):
681 """
Hari Krishnaef1bd4e2015-03-12 16:55:30 -0700682 Definition:
Jon Hall390696c2015-05-05 17:13:41 -0700683 Returns the output of topology command.
Hari Krishnaef1bd4e2015-03-12 16:55:30 -0700684 Return:
685 topology = current ONOS topology
kelvin8ec71442015-01-15 16:57:00 -0800686 """
andrewonlab95ce8322014-10-13 14:12:04 -0400687 try:
Hari Krishnaef1bd4e2015-03-12 16:55:30 -0700688 cmdStr = "topology -j"
kelvin-onlabd3b64892015-01-20 13:26:24 -0800689 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -0800690 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -0800691 assert "Command not found:" not in handle, handle
Jon Hallc6358dd2015-04-10 12:44:28 -0700692 main.log.info( cmdStr + " returned: " + str( handle ) )
andrewonlab95ce8322014-10-13 14:12:04 -0400693 return handle
Jon Hallc6793552016-01-19 14:18:37 -0800694 except AssertionError:
695 main.log.exception( "" )
Jon Halld4d4b372015-01-28 16:02:41 -0800696 return None
Jon Halld4d4b372015-01-28 16:02:41 -0800697 except TypeError:
698 main.log.exception( self.name + ": Object not as expected" )
699 return None
andrewonlabc2d05aa2014-10-13 16:51:10 -0400700 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -0800701 main.log.error( self.name + ": EOF exception found" )
702 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700703 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -0800704 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -0800705 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700706 main.cleanAndExit()
Jon Hallffb386d2014-11-21 13:43:38 -0800707
jenkins7ead5a82015-03-13 10:28:21 -0700708 def deviceRemove( self, deviceId ):
709 """
710 Removes particular device from storage
711
712 TODO: refactor this function
713 """
714 try:
Jon Hallc6358dd2015-04-10 12:44:28 -0700715 cmdStr = "device-remove " + str( deviceId )
716 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -0800717 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -0800718 assert "Command not found:" not in handle, handle
Jon Hallc6358dd2015-04-10 12:44:28 -0700719 if re.search( "Error", handle ):
720 main.log.error( "Error in removing device" )
721 main.log.error( handle )
722 return main.FALSE
723 else:
724 return main.TRUE
Jon Hallc6793552016-01-19 14:18:37 -0800725 except AssertionError:
726 main.log.exception( "" )
727 return None
jenkins7ead5a82015-03-13 10:28:21 -0700728 except TypeError:
729 main.log.exception( self.name + ": Object not as expected" )
730 return None
731 except pexpect.EOF:
732 main.log.error( self.name + ": EOF exception found" )
733 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700734 main.cleanAndExit()
jenkins7ead5a82015-03-13 10:28:21 -0700735 except Exception:
736 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700737 main.cleanAndExit()
jenkins7ead5a82015-03-13 10:28:21 -0700738
kelvin-onlabd3b64892015-01-20 13:26:24 -0800739 def devices( self, jsonFormat=True ):
kelvin8ec71442015-01-15 16:57:00 -0800740 """
Jon Hall7b02d952014-10-17 20:14:54 -0400741 Lists all infrastructure devices or switches
andrewonlab86dc3082014-10-13 18:18:38 -0400742 Optional argument:
kelvin-onlabd3b64892015-01-20 13:26:24 -0800743 * jsonFormat - boolean indicating if you want output in json
kelvin8ec71442015-01-15 16:57:00 -0800744 """
andrewonlab86dc3082014-10-13 18:18:38 -0400745 try:
Jon Hallc6358dd2015-04-10 12:44:28 -0700746 cmdStr = "devices"
kelvin-onlabd3b64892015-01-20 13:26:24 -0800747 if jsonFormat:
Jon Hallc6358dd2015-04-10 12:44:28 -0700748 cmdStr += " -j"
749 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -0800750 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -0800751 assert "Command not found:" not in handle, handle
Jon Hallc6358dd2015-04-10 12:44:28 -0700752 return handle
Jon Hallc6793552016-01-19 14:18:37 -0800753 except AssertionError:
754 main.log.exception( "" )
755 return None
Jon Halld4d4b372015-01-28 16:02:41 -0800756 except TypeError:
757 main.log.exception( self.name + ": Object not as expected" )
758 return None
andrewonlab7c211572014-10-15 16:45:20 -0400759 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -0800760 main.log.error( self.name + ": EOF exception found" )
761 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700762 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -0800763 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -0800764 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700765 main.cleanAndExit()
andrewonlab7c211572014-10-15 16:45:20 -0400766
kelvin-onlabd3b64892015-01-20 13:26:24 -0800767 def balanceMasters( self ):
kelvin8ec71442015-01-15 16:57:00 -0800768 """
Hari Krishnaa43d4e92014-12-19 13:22:40 -0800769 This balances the devices across all controllers
770 by issuing command: 'onos> onos:balance-masters'
771 If required this could be extended to return devices balanced output.
kelvin8ec71442015-01-15 16:57:00 -0800772 """
Hari Krishnaa43d4e92014-12-19 13:22:40 -0800773 try:
kelvin-onlabd3b64892015-01-20 13:26:24 -0800774 cmdStr = "onos:balance-masters"
Jon Hallc6358dd2015-04-10 12:44:28 -0700775 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -0800776 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -0800777 assert "Command not found:" not in handle, handle
Jon Hallc6358dd2015-04-10 12:44:28 -0700778 if re.search( "Error", handle ):
779 main.log.error( "Error in balancing masters" )
780 main.log.error( handle )
781 return main.FALSE
782 else:
783 return main.TRUE
Jon Hallc6793552016-01-19 14:18:37 -0800784 except AssertionError:
785 main.log.exception( "" )
786 return None
Jon Halld4d4b372015-01-28 16:02:41 -0800787 except TypeError:
788 main.log.exception( self.name + ": Object not as expected" )
789 return None
Hari Krishnaa43d4e92014-12-19 13:22:40 -0800790 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -0800791 main.log.error( self.name + ": EOF exception found" )
792 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700793 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -0800794 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -0800795 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700796 main.cleanAndExit()
Hari Krishnaa43d4e92014-12-19 13:22:40 -0800797
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +0000798 def checkMasters( self, jsonFormat=True ):
acsmars24950022015-07-30 18:00:43 -0700799 """
800 Returns the output of the masters command.
801 Optional argument:
802 * jsonFormat - boolean indicating if you want output in json
803 """
804 try:
805 cmdStr = "onos:masters"
806 if jsonFormat:
807 cmdStr += " -j"
808 output = self.sendline( cmdStr )
Jon Halla495f562016-05-16 18:03:26 -0700809 assert output is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -0800810 assert "Command not found:" not in output, output
acsmars24950022015-07-30 18:00:43 -0700811 return output
Jon Hallc6793552016-01-19 14:18:37 -0800812 except AssertionError:
813 main.log.exception( "" )
814 return None
acsmars24950022015-07-30 18:00:43 -0700815 except TypeError:
816 main.log.exception( self.name + ": Object not as expected" )
817 return None
818 except pexpect.EOF:
819 main.log.error( self.name + ": EOF exception found" )
820 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700821 main.cleanAndExit()
acsmars24950022015-07-30 18:00:43 -0700822 except Exception:
823 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700824 main.cleanAndExit()
acsmars24950022015-07-30 18:00:43 -0700825
Jon Hallc6793552016-01-19 14:18:37 -0800826 def checkBalanceMasters( self, jsonFormat=True ):
acsmars24950022015-07-30 18:00:43 -0700827 """
828 Uses the master command to check that the devices' leadership
829 is evenly divided
830
831 Dependencies: checkMasters() and summary()
832
Jon Hall6509dbf2016-06-21 17:01:17 -0700833 Returns main.TRUE if the devices are balanced
834 Returns main.FALSE if the devices are unbalanced
acsmars24950022015-07-30 18:00:43 -0700835 Exits on Exception
836 Returns None on TypeError
837 """
838 try:
Jon Hallc6793552016-01-19 14:18:37 -0800839 summaryOutput = self.summary()
840 totalDevices = json.loads( summaryOutput )[ "devices" ]
841 except ( TypeError, ValueError ):
842 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, summaryOutput ) )
843 return None
844 try:
acsmars24950022015-07-30 18:00:43 -0700845 totalOwnedDevices = 0
Jon Hallc6793552016-01-19 14:18:37 -0800846 mastersOutput = self.checkMasters()
847 masters = json.loads( mastersOutput )
acsmars24950022015-07-30 18:00:43 -0700848 first = masters[ 0 ][ "size" ]
849 for master in masters:
850 totalOwnedDevices += master[ "size" ]
851 if master[ "size" ] > first + 1 or master[ "size" ] < first - 1:
852 main.log.error( "Mastership not balanced" )
853 main.log.info( "\n" + self.checkMasters( False ) )
854 return main.FALSE
Jon Halle0f0b342017-04-18 11:43:47 -0700855 main.log.info( "Mastership balanced between " +
Jeremy Ronquillo82705492017-10-18 14:19:55 -0700856 str( len( masters ) ) + " masters" )
acsmars24950022015-07-30 18:00:43 -0700857 return main.TRUE
Jon Hallc6793552016-01-19 14:18:37 -0800858 except ( TypeError, ValueError ):
859 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, mastersOutput ) )
acsmars24950022015-07-30 18:00:43 -0700860 return None
861 except pexpect.EOF:
862 main.log.error( self.name + ": EOF exception found" )
863 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700864 main.cleanAndExit()
acsmars24950022015-07-30 18:00:43 -0700865 except Exception:
866 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700867 main.cleanAndExit()
acsmars24950022015-07-30 18:00:43 -0700868
YPZhangfebf7302016-05-24 16:45:56 -0700869 def links( self, jsonFormat=True, timeout=30 ):
kelvin8ec71442015-01-15 16:57:00 -0800870 """
Jon Halle8217482014-10-17 13:49:14 -0400871 Lists all core links
872 Optional argument:
kelvin-onlabd3b64892015-01-20 13:26:24 -0800873 * jsonFormat - boolean indicating if you want output in json
kelvin8ec71442015-01-15 16:57:00 -0800874 """
Jon Halle8217482014-10-17 13:49:14 -0400875 try:
Jon Hallc6358dd2015-04-10 12:44:28 -0700876 cmdStr = "links"
kelvin-onlabd3b64892015-01-20 13:26:24 -0800877 if jsonFormat:
Jon Hallc6358dd2015-04-10 12:44:28 -0700878 cmdStr += " -j"
YPZhangfebf7302016-05-24 16:45:56 -0700879 handle = self.sendline( cmdStr, timeout=timeout )
You Wangb5a55f72017-03-03 12:51:05 -0800880 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -0800881 assert "Command not found:" not in handle, handle
Jon Hallc6358dd2015-04-10 12:44:28 -0700882 return handle
Jon Hallc6793552016-01-19 14:18:37 -0800883 except AssertionError:
884 main.log.exception( "" )
885 return None
Jon Halld4d4b372015-01-28 16:02:41 -0800886 except TypeError:
887 main.log.exception( self.name + ": Object not as expected" )
888 return None
Jon Halle8217482014-10-17 13:49:14 -0400889 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -0800890 main.log.error( self.name + ": EOF exception found" )
891 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700892 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -0800893 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -0800894 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700895 main.cleanAndExit()
Jon Halle8217482014-10-17 13:49:14 -0400896
kelvin-onlabd3b64892015-01-20 13:26:24 -0800897 def ports( self, jsonFormat=True ):
kelvin8ec71442015-01-15 16:57:00 -0800898 """
Jon Halle8217482014-10-17 13:49:14 -0400899 Lists all ports
900 Optional argument:
kelvin-onlabd3b64892015-01-20 13:26:24 -0800901 * jsonFormat - boolean indicating if you want output in json
kelvin8ec71442015-01-15 16:57:00 -0800902 """
Jon Halle8217482014-10-17 13:49:14 -0400903 try:
Jon Hallc6358dd2015-04-10 12:44:28 -0700904 cmdStr = "ports"
kelvin-onlabd3b64892015-01-20 13:26:24 -0800905 if jsonFormat:
Jon Hallc6358dd2015-04-10 12:44:28 -0700906 cmdStr += " -j"
907 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -0800908 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -0800909 assert "Command not found:" not in handle, handle
Jon Hallc6358dd2015-04-10 12:44:28 -0700910 return handle
Jon Hallc6793552016-01-19 14:18:37 -0800911 except AssertionError:
912 main.log.exception( "" )
913 return None
Jon Halld4d4b372015-01-28 16:02:41 -0800914 except TypeError:
915 main.log.exception( self.name + ": Object not as expected" )
916 return None
Jon Halle8217482014-10-17 13:49:14 -0400917 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -0800918 main.log.error( self.name + ": EOF exception found" )
919 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700920 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -0800921 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -0800922 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700923 main.cleanAndExit()
Jon Halle8217482014-10-17 13:49:14 -0400924
kelvin-onlabd3b64892015-01-20 13:26:24 -0800925 def roles( self, jsonFormat=True ):
kelvin8ec71442015-01-15 16:57:00 -0800926 """
Jon Hall983a1702014-10-28 18:44:22 -0400927 Lists all devices and the controllers with roles assigned to them
928 Optional argument:
kelvin-onlabd3b64892015-01-20 13:26:24 -0800929 * jsonFormat - boolean indicating if you want output in json
kelvin8ec71442015-01-15 16:57:00 -0800930 """
andrewonlab7c211572014-10-15 16:45:20 -0400931 try:
Jon Hallc6358dd2015-04-10 12:44:28 -0700932 cmdStr = "roles"
kelvin-onlabd3b64892015-01-20 13:26:24 -0800933 if jsonFormat:
Jon Hallc6358dd2015-04-10 12:44:28 -0700934 cmdStr += " -j"
935 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -0800936 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -0800937 assert "Command not found:" not in handle, handle
Jon Hallc6358dd2015-04-10 12:44:28 -0700938 return handle
Jon Hallc6793552016-01-19 14:18:37 -0800939 except AssertionError:
940 main.log.exception( "" )
941 return None
Jon Halld4d4b372015-01-28 16:02:41 -0800942 except TypeError:
943 main.log.exception( self.name + ": Object not as expected" )
944 return None
Jon Hall983a1702014-10-28 18:44:22 -0400945 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -0800946 main.log.error( self.name + ": EOF exception found" )
947 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700948 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -0800949 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -0800950 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700951 main.cleanAndExit()
Jon Hall983a1702014-10-28 18:44:22 -0400952
kelvin-onlabd3b64892015-01-20 13:26:24 -0800953 def getRole( self, deviceId ):
kelvin-onlab898a6c62015-01-16 14:13:53 -0800954 """
Jon Halle3f39ff2015-01-13 11:50:53 -0800955 Given the a string containing the json representation of the "roles"
956 cli command and a partial or whole device id, returns a json object
957 containing the roles output for the first device whose id contains
958 "device_id"
Jon Hall983a1702014-10-28 18:44:22 -0400959
960 Returns:
Jon Halle3f39ff2015-01-13 11:50:53 -0800961 A dict of the role assignments for the given device or
962 None if no match
kelvin8ec71442015-01-15 16:57:00 -0800963 """
Jon Hall983a1702014-10-28 18:44:22 -0400964 try:
kelvin-onlabd3b64892015-01-20 13:26:24 -0800965 if deviceId is None:
Jon Hall983a1702014-10-28 18:44:22 -0400966 return None
967 else:
kelvin-onlabd3b64892015-01-20 13:26:24 -0800968 rawRoles = self.roles()
969 rolesJson = json.loads( rawRoles )
kelvin8ec71442015-01-15 16:57:00 -0800970 # search json for the device with id then return the device
kelvin-onlabd3b64892015-01-20 13:26:24 -0800971 for device in rolesJson:
kelvin8ec71442015-01-15 16:57:00 -0800972 # print device
kelvin-onlabd3b64892015-01-20 13:26:24 -0800973 if str( deviceId ) in device[ 'id' ]:
Jon Hall983a1702014-10-28 18:44:22 -0400974 return device
975 return None
Jon Hallc6793552016-01-19 14:18:37 -0800976 except ( TypeError, ValueError ):
977 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, rawRoles ) )
Jon Halld4d4b372015-01-28 16:02:41 -0800978 return None
andrewonlab86dc3082014-10-13 18:18:38 -0400979 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -0800980 main.log.error( self.name + ": EOF exception found" )
981 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -0700982 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -0800983 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -0800984 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -0700985 main.cleanAndExit()
Jon Hall94fd0472014-12-08 11:52:42 -0800986
kelvin-onlabd3b64892015-01-20 13:26:24 -0800987 def rolesNotNull( self ):
kelvin8ec71442015-01-15 16:57:00 -0800988 """
Jon Hall94fd0472014-12-08 11:52:42 -0800989 Iterates through each device and checks if there is a master assigned
990 Returns: main.TRUE if each device has a master
991 main.FALSE any device has no master
kelvin8ec71442015-01-15 16:57:00 -0800992 """
Jon Hall94fd0472014-12-08 11:52:42 -0800993 try:
kelvin-onlabd3b64892015-01-20 13:26:24 -0800994 rawRoles = self.roles()
995 rolesJson = json.loads( rawRoles )
kelvin8ec71442015-01-15 16:57:00 -0800996 # search json for the device with id then return the device
kelvin-onlabd3b64892015-01-20 13:26:24 -0800997 for device in rolesJson:
kelvin8ec71442015-01-15 16:57:00 -0800998 # print device
999 if device[ 'master' ] == "none":
1000 main.log.warn( "Device has no master: " + str( device ) )
Jon Hall94fd0472014-12-08 11:52:42 -08001001 return main.FALSE
1002 return main.TRUE
Jon Hallc6793552016-01-19 14:18:37 -08001003 except ( TypeError, ValueError ):
1004 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, rawRoles ) )
Jon Halld4d4b372015-01-28 16:02:41 -08001005 return None
Jon Hall94fd0472014-12-08 11:52:42 -08001006 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08001007 main.log.error( self.name + ": EOF exception found" )
1008 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07001009 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08001010 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08001011 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07001012 main.cleanAndExit()
Jon Hall94fd0472014-12-08 11:52:42 -08001013
kelvin-onlabd3b64892015-01-20 13:26:24 -08001014 def paths( self, srcId, dstId ):
kelvin8ec71442015-01-15 16:57:00 -08001015 """
andrewonlab3e15ead2014-10-15 14:21:34 -04001016 Returns string of paths, and the cost.
1017 Issues command: onos:paths <src> <dst>
kelvin8ec71442015-01-15 16:57:00 -08001018 """
andrewonlab3e15ead2014-10-15 14:21:34 -04001019 try:
kelvin-onlabd3b64892015-01-20 13:26:24 -08001020 cmdStr = "onos:paths " + str( srcId ) + " " + str( dstId )
1021 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -08001022 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08001023 assert "Command not found:" not in handle, handle
Jon Halle3f39ff2015-01-13 11:50:53 -08001024 if re.search( "Error", handle ):
kelvin8ec71442015-01-15 16:57:00 -08001025 main.log.error( "Error in getting paths" )
1026 return ( handle, "Error" )
andrewonlab3e15ead2014-10-15 14:21:34 -04001027 else:
kelvin8ec71442015-01-15 16:57:00 -08001028 path = handle.split( ";" )[ 0 ]
1029 cost = handle.split( ";" )[ 1 ]
1030 return ( path, cost )
Jon Hallc6793552016-01-19 14:18:37 -08001031 except AssertionError:
1032 main.log.exception( "" )
1033 return ( handle, "Error" )
Jon Halld4d4b372015-01-28 16:02:41 -08001034 except TypeError:
1035 main.log.exception( self.name + ": Object not as expected" )
1036 return ( handle, "Error" )
andrewonlab3e15ead2014-10-15 14:21:34 -04001037 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08001038 main.log.error( self.name + ": EOF exception found" )
1039 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07001040 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08001041 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08001042 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07001043 main.cleanAndExit()
Jon Hallffb386d2014-11-21 13:43:38 -08001044
kelvin-onlabd3b64892015-01-20 13:26:24 -08001045 def hosts( self, jsonFormat=True ):
kelvin8ec71442015-01-15 16:57:00 -08001046 """
Jon Hallffb386d2014-11-21 13:43:38 -08001047 Lists all discovered hosts
Jon Hall42db6dc2014-10-24 19:03:48 -04001048 Optional argument:
kelvin-onlabd3b64892015-01-20 13:26:24 -08001049 * jsonFormat - boolean indicating if you want output in json
kelvin8ec71442015-01-15 16:57:00 -08001050 """
Jon Hall42db6dc2014-10-24 19:03:48 -04001051 try:
Jon Hallc6358dd2015-04-10 12:44:28 -07001052 cmdStr = "hosts"
kelvin-onlabd3b64892015-01-20 13:26:24 -08001053 if jsonFormat:
Jon Hallc6358dd2015-04-10 12:44:28 -07001054 cmdStr += " -j"
1055 handle = self.sendline( cmdStr )
Jeremyd9e4eb12016-04-13 12:09:06 -07001056 if handle:
1057 assert "Command not found:" not in handle, handle
Jon Hallbaf53162015-12-17 17:04:34 -08001058 # TODO: Maybe make this less hardcoded
1059 # ConsistentMap Exceptions
1060 assert "org.onosproject.store.service" not in handle
1061 # Node not leader
1062 assert "java.lang.IllegalStateException" not in handle
Jon Hallc6358dd2015-04-10 12:44:28 -07001063 return handle
Jon Hallc6793552016-01-19 14:18:37 -08001064 except AssertionError:
Jeremyd9e4eb12016-04-13 12:09:06 -07001065 main.log.exception( "Error in processing '" + cmdStr + "' " +
Jeremy Songster6949cea2016-04-19 18:13:18 -07001066 "command: " + str( handle ) )
Jon Hallc6793552016-01-19 14:18:37 -08001067 return None
Jon Halld4d4b372015-01-28 16:02:41 -08001068 except TypeError:
1069 main.log.exception( self.name + ": Object not as expected" )
1070 return None
Jon Hall42db6dc2014-10-24 19:03:48 -04001071 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08001072 main.log.error( self.name + ": EOF exception found" )
1073 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07001074 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08001075 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08001076 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07001077 main.cleanAndExit()
Jon Hall42db6dc2014-10-24 19:03:48 -04001078
kelvin-onlabd3b64892015-01-20 13:26:24 -08001079 def getHost( self, mac ):
kelvin8ec71442015-01-15 16:57:00 -08001080 """
Jon Hall42db6dc2014-10-24 19:03:48 -04001081 Return the first host from the hosts api whose 'id' contains 'mac'
Jon Halle3f39ff2015-01-13 11:50:53 -08001082
Jon Hallefbd9792015-03-05 16:11:36 -08001083 Note: mac must be a colon separated mac address, but could be a
Jon Halle3f39ff2015-01-13 11:50:53 -08001084 partial mac address
1085
Jon Hall42db6dc2014-10-24 19:03:48 -04001086 Return None if there is no match
kelvin8ec71442015-01-15 16:57:00 -08001087 """
Jon Hall42db6dc2014-10-24 19:03:48 -04001088 try:
kelvin8ec71442015-01-15 16:57:00 -08001089 if mac is None:
Jon Hall42db6dc2014-10-24 19:03:48 -04001090 return None
1091 else:
1092 mac = mac
kelvin-onlabd3b64892015-01-20 13:26:24 -08001093 rawHosts = self.hosts()
1094 hostsJson = json.loads( rawHosts )
kelvin8ec71442015-01-15 16:57:00 -08001095 # search json for the host with mac then return the device
kelvin-onlabd3b64892015-01-20 13:26:24 -08001096 for host in hostsJson:
kelvin8ec71442015-01-15 16:57:00 -08001097 # print "%s in %s?" % ( mac, host[ 'id' ] )
Jon Halld4d4b372015-01-28 16:02:41 -08001098 if not host:
1099 pass
1100 elif mac in host[ 'id' ]:
Jon Hall42db6dc2014-10-24 19:03:48 -04001101 return host
1102 return None
Jon Hallc6793552016-01-19 14:18:37 -08001103 except ( TypeError, ValueError ):
1104 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, rawHosts ) )
Jon Halld4d4b372015-01-28 16:02:41 -08001105 return None
Jon Hall42db6dc2014-10-24 19:03:48 -04001106 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08001107 main.log.error( self.name + ": EOF exception found" )
1108 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07001109 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08001110 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08001111 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07001112 main.cleanAndExit()
Jon Hall42db6dc2014-10-24 19:03:48 -04001113
kelvin-onlabd3b64892015-01-20 13:26:24 -08001114 def getHostsId( self, hostList ):
kelvin8ec71442015-01-15 16:57:00 -08001115 """
1116 Obtain list of hosts
andrewonlab3f0a4af2014-10-17 12:25:14 -04001117 Issues command: 'onos> hosts'
kelvin8ec71442015-01-15 16:57:00 -08001118
andrewonlab3f0a4af2014-10-17 12:25:14 -04001119 Required:
kelvin-onlabd3b64892015-01-20 13:26:24 -08001120 * hostList: List of hosts obtained by Mininet
andrewonlab3f0a4af2014-10-17 12:25:14 -04001121 IMPORTANT:
1122 This function assumes that you started your
kelvin8ec71442015-01-15 16:57:00 -08001123 topology with the option '--mac'.
andrewonlab3f0a4af2014-10-17 12:25:14 -04001124 Furthermore, it assumes that value of VLAN is '-1'
1125 Description:
kelvin8ec71442015-01-15 16:57:00 -08001126 Converts mininet hosts ( h1, h2, h3... ) into
1127 ONOS format ( 00:00:00:00:00:01/-1 , ... )
1128 """
andrewonlab3f0a4af2014-10-17 12:25:14 -04001129 try:
kelvin-onlabd3b64892015-01-20 13:26:24 -08001130 onosHostList = []
andrewonlab3f0a4af2014-10-17 12:25:14 -04001131
kelvin-onlabd3b64892015-01-20 13:26:24 -08001132 for host in hostList:
kelvin8ec71442015-01-15 16:57:00 -08001133 host = host.replace( "h", "" )
kelvin-onlabd3b64892015-01-20 13:26:24 -08001134 hostHex = hex( int( host ) ).zfill( 12 )
1135 hostHex = str( hostHex ).replace( 'x', '0' )
1136 i = iter( str( hostHex ) )
1137 hostHex = ":".join( a + b for a, b in zip( i, i ) )
1138 hostHex = hostHex + "/-1"
1139 onosHostList.append( hostHex )
andrewonlab3f0a4af2014-10-17 12:25:14 -04001140
kelvin-onlabd3b64892015-01-20 13:26:24 -08001141 return onosHostList
andrewonlab3f0a4af2014-10-17 12:25:14 -04001142
Jon Halld4d4b372015-01-28 16:02:41 -08001143 except TypeError:
1144 main.log.exception( self.name + ": Object not as expected" )
1145 return None
andrewonlab3f0a4af2014-10-17 12:25:14 -04001146 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08001147 main.log.error( self.name + ": EOF exception found" )
1148 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07001149 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08001150 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08001151 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07001152 main.cleanAndExit()
andrewonlab3e15ead2014-10-15 14:21:34 -04001153
You Wang53dba1e2018-02-02 17:45:44 -08001154 def verifyHostIp( self, hostList=[], prefix="" ):
1155 """
1156 Description:
1157 Verify that all hosts have IP address assigned to them
1158 Optional:
1159 hostList: If specified, verifications only happen to the hosts
1160 in hostList
1161 prefix: at least one of the ip address assigned to the host
1162 needs to have the specified prefix
1163 Returns:
1164 main.TRUE if all hosts have specific IP address assigned;
1165 main.FALSE otherwise
1166 """
1167 import json
1168 try:
1169 hosts = self.hosts()
1170 hosts = json.loads( hosts )
1171 if not hostList:
1172 hostList = [ host[ "id" ] for host in hosts ]
1173 for host in hosts:
1174 hostId = host[ "id" ]
1175 if hostId not in hostList:
1176 continue
1177 ipList = host[ "ipAddresses" ]
1178 main.log.debug( self.name + ": IP list on host " + str( hostId ) + ": " + str( ipList ) )
1179 if not ipList:
1180 main.log.warn( self.name + ": Failed to discover any IP addresses on host " + str( hostId ) )
1181 else:
1182 if not any( ip.startswith( str( prefix ) ) for ip in ipList ):
1183 main.log.warn( self.name + ": None of the IPs on host " + str( hostId ) + " has prefix " + str( prefix ) )
1184 else:
1185 main.log.debug( self.name + ": Found matching IP on host " + str( hostId ) )
1186 hostList.remove( hostId )
1187 if hostList:
1188 main.log.warn( self.name + ": failed to verify IP on following hosts: " + str( hostList) )
1189 return main.FALSE
1190 else:
1191 return main.TRUE
1192 except KeyError:
1193 main.log.exception( self.name + ": host data not as expected: " + hosts )
1194 return None
1195 except pexpect.EOF:
1196 main.log.error( self.name + ": EOF exception found" )
1197 main.log.error( self.name + ": " + self.handle.before )
1198 main.cleanAndExit()
1199 except Exception:
1200 main.log.exception( self.name + ": Uncaught exception" )
1201 return None
1202
Shreya Chowdhary6fbb96c2017-05-02 16:20:19 -07001203 def addHostIntent( self, hostIdOne, hostIdTwo, vlanId="", setVlan="", encap="", bandwidth="" ):
kelvin8ec71442015-01-15 16:57:00 -08001204 """
andrewonlabe6745342014-10-17 14:29:13 -04001205 Required:
kelvin-onlabd3b64892015-01-20 13:26:24 -08001206 * hostIdOne: ONOS host id for host1
1207 * hostIdTwo: ONOS host id for host2
Jeremy Songster832f9e92016-05-05 14:30:49 -07001208 Optional:
1209 * vlanId: specify a VLAN id for the intent
Jeremy Songsterff553672016-05-12 17:06:23 -07001210 * setVlan: specify a VLAN id treatment
Jeremy Songsterc032f162016-08-04 17:14:49 -07001211 * encap: specify an encapsulation type
andrewonlabe6745342014-10-17 14:29:13 -04001212 Description:
Jon Hallefbd9792015-03-05 16:11:36 -08001213 Adds a host-to-host intent ( bidirectional ) by
Jon Hallb1290e82014-11-18 16:17:48 -05001214 specifying the two hosts.
kelvin-onlabfb521662015-02-27 09:52:40 -08001215 Returns:
1216 A string of the intent id or None on Error
kelvin8ec71442015-01-15 16:57:00 -08001217 """
andrewonlabe6745342014-10-17 14:29:13 -04001218 try:
Jeremy Songster832f9e92016-05-05 14:30:49 -07001219 cmdStr = "add-host-intent "
1220 if vlanId:
1221 cmdStr += "-v " + str( vlanId ) + " "
Jeremy Songsterff553672016-05-12 17:06:23 -07001222 if setVlan:
1223 cmdStr += "--setVlan " + str( vlanId ) + " "
Jeremy Songsterc032f162016-08-04 17:14:49 -07001224 if encap:
1225 cmdStr += "--encapsulation " + str( encap ) + " "
Shreya Chowdhary6fbb96c2017-05-02 16:20:19 -07001226 if bandwidth:
1227 cmdStr += "-b " + str( bandwidth ) + " "
Jeremy Songster832f9e92016-05-05 14:30:49 -07001228 cmdStr += str( hostIdOne ) + " " + str( hostIdTwo )
kelvin-onlabd3b64892015-01-20 13:26:24 -08001229 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -08001230 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08001231 assert "Command not found:" not in handle, handle
Hari Krishnaac4e1782015-01-26 12:09:12 -08001232 if re.search( "Error", handle ):
1233 main.log.error( "Error in adding Host intent" )
Jon Hall61282e32015-03-19 11:34:11 -07001234 main.log.debug( "Response from ONOS was: " + repr( handle ) )
kelvin-onlabfb521662015-02-27 09:52:40 -08001235 return None
Hari Krishnaac4e1782015-01-26 12:09:12 -08001236 else:
1237 main.log.info( "Host intent installed between " +
kelvin-onlabfb521662015-02-27 09:52:40 -08001238 str( hostIdOne ) + " and " + str( hostIdTwo ) )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07001239 match = re.search( 'id=0x([\da-f]+),', handle )
kelvin-onlabfb521662015-02-27 09:52:40 -08001240 if match:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07001241 return match.group()[ 3:-1 ]
kelvin-onlabfb521662015-02-27 09:52:40 -08001242 else:
1243 main.log.error( "Error, intent ID not found" )
Jon Hall61282e32015-03-19 11:34:11 -07001244 main.log.debug( "Response from ONOS was: " +
1245 repr( handle ) )
kelvin-onlabfb521662015-02-27 09:52:40 -08001246 return None
Jon Hallc6793552016-01-19 14:18:37 -08001247 except AssertionError:
1248 main.log.exception( "" )
1249 return None
Jon Halld4d4b372015-01-28 16:02:41 -08001250 except TypeError:
1251 main.log.exception( self.name + ": Object not as expected" )
1252 return None
andrewonlabe6745342014-10-17 14:29:13 -04001253 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08001254 main.log.error( self.name + ": EOF exception found" )
1255 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07001256 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08001257 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08001258 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07001259 main.cleanAndExit()
andrewonlabe6745342014-10-17 14:29:13 -04001260
kelvin-onlabd3b64892015-01-20 13:26:24 -08001261 def addOpticalIntent( self, ingressDevice, egressDevice ):
kelvin8ec71442015-01-15 16:57:00 -08001262 """
andrewonlab7b31d232014-10-24 13:31:47 -04001263 Required:
kelvin-onlabd3b64892015-01-20 13:26:24 -08001264 * ingressDevice: device id of ingress device
1265 * egressDevice: device id of egress device
andrewonlab7b31d232014-10-24 13:31:47 -04001266 Optional:
1267 TODO: Still needs to be implemented via dev side
kelvin-onlabfb521662015-02-27 09:52:40 -08001268 Description:
1269 Adds an optical intent by specifying an ingress and egress device
1270 Returns:
1271 A string of the intent id or None on error
kelvin-onlab898a6c62015-01-16 14:13:53 -08001272 """
andrewonlab7b31d232014-10-24 13:31:47 -04001273 try:
kelvin-onlabd3b64892015-01-20 13:26:24 -08001274 cmdStr = "add-optical-intent " + str( ingressDevice ) +\
1275 " " + str( egressDevice )
1276 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -08001277 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08001278 assert "Command not found:" not in handle, handle
kelvin-onlab898a6c62015-01-16 14:13:53 -08001279 # If error, return error message
Jon Halle3f39ff2015-01-13 11:50:53 -08001280 if re.search( "Error", handle ):
kelvin-onlabfb521662015-02-27 09:52:40 -08001281 main.log.error( "Error in adding Optical intent" )
1282 return None
andrewonlab7b31d232014-10-24 13:31:47 -04001283 else:
kelvin-onlabfb521662015-02-27 09:52:40 -08001284 main.log.info( "Optical intent installed between " +
1285 str( ingressDevice ) + " and " +
1286 str( egressDevice ) )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07001287 match = re.search( 'id=0x([\da-f]+),', handle )
kelvin-onlabfb521662015-02-27 09:52:40 -08001288 if match:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07001289 return match.group()[ 3:-1 ]
kelvin-onlabfb521662015-02-27 09:52:40 -08001290 else:
1291 main.log.error( "Error, intent ID not found" )
1292 return None
Jon Hallc6793552016-01-19 14:18:37 -08001293 except AssertionError:
1294 main.log.exception( "" )
1295 return None
Jon Halld4d4b372015-01-28 16:02:41 -08001296 except TypeError:
1297 main.log.exception( self.name + ": Object not as expected" )
1298 return None
andrewonlab7b31d232014-10-24 13:31:47 -04001299 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08001300 main.log.error( self.name + ": EOF exception found" )
1301 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07001302 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08001303 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08001304 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07001305 main.cleanAndExit()
andrewonlab7b31d232014-10-24 13:31:47 -04001306
kelvin-onlabd3b64892015-01-20 13:26:24 -08001307 def addPointIntent(
kelvin-onlab898a6c62015-01-16 14:13:53 -08001308 self,
kelvin-onlabd3b64892015-01-20 13:26:24 -08001309 ingressDevice,
1310 egressDevice,
1311 portIngress="",
1312 portEgress="",
kelvin-onlab898a6c62015-01-16 14:13:53 -08001313 ethType="",
1314 ethSrc="",
1315 ethDst="",
1316 bandwidth="",
kelvin-onlabd3b64892015-01-20 13:26:24 -08001317 lambdaAlloc=False,
alisonda157272016-12-22 01:13:21 -08001318 protected=False,
kelvin-onlab898a6c62015-01-16 14:13:53 -08001319 ipProto="",
1320 ipSrc="",
1321 ipDst="",
1322 tcpSrc="",
Jeremy Songster832f9e92016-05-05 14:30:49 -07001323 tcpDst="",
Jeremy Songsterff553672016-05-12 17:06:23 -07001324 vlanId="",
Jeremy Songsterc032f162016-08-04 17:14:49 -07001325 setVlan="",
1326 encap="" ):
kelvin8ec71442015-01-15 16:57:00 -08001327 """
andrewonlab4dbb4d82014-10-17 18:22:31 -04001328 Required:
kelvin-onlabd3b64892015-01-20 13:26:24 -08001329 * ingressDevice: device id of ingress device
1330 * egressDevice: device id of egress device
andrewonlab289e4b72014-10-21 21:24:18 -04001331 Optional:
1332 * ethType: specify ethType
kelvin8ec71442015-01-15 16:57:00 -08001333 * ethSrc: specify ethSrc ( i.e. src mac addr )
1334 * ethDst: specify ethDst ( i.e. dst mac addr )
andrewonlab0dbb6ec2014-11-06 13:46:55 -05001335 * bandwidth: specify bandwidth capacity of link
kelvin-onlabd3b64892015-01-20 13:26:24 -08001336 * lambdaAlloc: if True, intent will allocate lambda
andrewonlab40ccd8b2014-11-06 16:23:34 -05001337 for the specified intent
Jon Halle3f39ff2015-01-13 11:50:53 -08001338 * ipProto: specify ip protocol
andrewonlabf77e0cb2014-11-11 17:17:59 -05001339 * ipSrc: specify ip source address
1340 * ipDst: specify ip destination address
1341 * tcpSrc: specify tcp source port
1342 * tcpDst: specify tcp destination port
Jeremy Songster832f9e92016-05-05 14:30:49 -07001343 * vlanId: specify vlan ID
Jeremy Songsterff553672016-05-12 17:06:23 -07001344 * setVlan: specify a VLAN id treatment
Jeremy Songsterc032f162016-08-04 17:14:49 -07001345 * encap: specify an Encapsulation type to use
andrewonlab4dbb4d82014-10-17 18:22:31 -04001346 Description:
kelvin8ec71442015-01-15 16:57:00 -08001347 Adds a point-to-point intent ( uni-directional ) by
andrewonlab289e4b72014-10-21 21:24:18 -04001348 specifying device id's and optional fields
kelvin-onlabfb521662015-02-27 09:52:40 -08001349 Returns:
1350 A string of the intent id or None on error
andrewonlab289e4b72014-10-21 21:24:18 -04001351
Jon Halle3f39ff2015-01-13 11:50:53 -08001352 NOTE: This function may change depending on the
andrewonlab4dbb4d82014-10-17 18:22:31 -04001353 options developers provide for point-to-point
1354 intent via cli
kelvin8ec71442015-01-15 16:57:00 -08001355 """
andrewonlab4dbb4d82014-10-17 18:22:31 -04001356 try:
Jeremy Songsterff553672016-05-12 17:06:23 -07001357 cmd = "add-point-intent"
andrewonlab36af3822014-11-18 17:48:18 -05001358
Jeremy Songsterff553672016-05-12 17:06:23 -07001359 if ethType:
1360 cmd += " --ethType " + str( ethType )
1361 if ethSrc:
1362 cmd += " --ethSrc " + str( ethSrc )
1363 if ethDst:
1364 cmd += " --ethDst " + str( ethDst )
1365 if bandwidth:
1366 cmd += " --bandwidth " + str( bandwidth )
1367 if lambdaAlloc:
1368 cmd += " --lambda "
1369 if ipProto:
1370 cmd += " --ipProto " + str( ipProto )
1371 if ipSrc:
1372 cmd += " --ipSrc " + str( ipSrc )
1373 if ipDst:
1374 cmd += " --ipDst " + str( ipDst )
1375 if tcpSrc:
1376 cmd += " --tcpSrc " + str( tcpSrc )
1377 if tcpDst:
1378 cmd += " --tcpDst " + str( tcpDst )
1379 if vlanId:
1380 cmd += " -v " + str( vlanId )
1381 if setVlan:
1382 cmd += " --setVlan " + str( setVlan )
Jeremy Songsterc032f162016-08-04 17:14:49 -07001383 if encap:
1384 cmd += " --encapsulation " + str( encap )
alisonda157272016-12-22 01:13:21 -08001385 if protected:
1386 cmd += " --protect "
andrewonlab289e4b72014-10-21 21:24:18 -04001387
kelvin8ec71442015-01-15 16:57:00 -08001388 # Check whether the user appended the port
1389 # or provided it as an input
kelvin-onlabd3b64892015-01-20 13:26:24 -08001390 if "/" in ingressDevice:
1391 cmd += " " + str( ingressDevice )
andrewonlab36af3822014-11-18 17:48:18 -05001392 else:
kelvin-onlabd3b64892015-01-20 13:26:24 -08001393 if not portIngress:
kelvin-onlabfb521662015-02-27 09:52:40 -08001394 main.log.error( "You must specify the ingress port" )
kelvin8ec71442015-01-15 16:57:00 -08001395 # TODO: perhaps more meaningful return
kelvin-onlabfb521662015-02-27 09:52:40 -08001396 # Would it make sense to throw an exception and exit
1397 # the test?
1398 return None
andrewonlab36af3822014-11-18 17:48:18 -05001399
kelvin8ec71442015-01-15 16:57:00 -08001400 cmd += " " + \
kelvin-onlabd3b64892015-01-20 13:26:24 -08001401 str( ingressDevice ) + "/" +\
1402 str( portIngress ) + " "
andrewonlab36af3822014-11-18 17:48:18 -05001403
kelvin-onlabd3b64892015-01-20 13:26:24 -08001404 if "/" in egressDevice:
1405 cmd += " " + str( egressDevice )
andrewonlab36af3822014-11-18 17:48:18 -05001406 else:
kelvin-onlabd3b64892015-01-20 13:26:24 -08001407 if not portEgress:
kelvin-onlabfb521662015-02-27 09:52:40 -08001408 main.log.error( "You must specify the egress port" )
1409 return None
Jon Halle3f39ff2015-01-13 11:50:53 -08001410
kelvin8ec71442015-01-15 16:57:00 -08001411 cmd += " " +\
kelvin-onlabd3b64892015-01-20 13:26:24 -08001412 str( egressDevice ) + "/" +\
1413 str( portEgress )
kelvin8ec71442015-01-15 16:57:00 -08001414
kelvin-onlab898a6c62015-01-16 14:13:53 -08001415 handle = self.sendline( cmd )
You Wangb5a55f72017-03-03 12:51:05 -08001416 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08001417 assert "Command not found:" not in handle, handle
kelvin-onlabfb521662015-02-27 09:52:40 -08001418 # If error, return error message
kelvin-onlab898a6c62015-01-16 14:13:53 -08001419 if re.search( "Error", handle ):
kelvin8ec71442015-01-15 16:57:00 -08001420 main.log.error( "Error in adding point-to-point intent" )
kelvin-onlabfb521662015-02-27 09:52:40 -08001421 return None
andrewonlab4dbb4d82014-10-17 18:22:31 -04001422 else:
kelvin-onlabfb521662015-02-27 09:52:40 -08001423 # TODO: print out all the options in this message?
1424 main.log.info( "Point-to-point intent installed between " +
1425 str( ingressDevice ) + " and " +
1426 str( egressDevice ) )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07001427 match = re.search( 'id=0x([\da-f]+),', handle )
kelvin-onlabfb521662015-02-27 09:52:40 -08001428 if match:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07001429 return match.group()[ 3:-1 ]
kelvin-onlabfb521662015-02-27 09:52:40 -08001430 else:
1431 main.log.error( "Error, intent ID not found" )
1432 return None
Jon Hallc6793552016-01-19 14:18:37 -08001433 except AssertionError:
1434 main.log.exception( "" )
1435 return None
Jon Halld4d4b372015-01-28 16:02:41 -08001436 except TypeError:
1437 main.log.exception( self.name + ": Object not as expected" )
1438 return None
andrewonlab4dbb4d82014-10-17 18:22:31 -04001439 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08001440 main.log.error( self.name + ": EOF exception found" )
1441 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07001442 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08001443 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08001444 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07001445 main.cleanAndExit()
andrewonlab4dbb4d82014-10-17 18:22:31 -04001446
kelvin-onlabd3b64892015-01-20 13:26:24 -08001447 def addMultipointToSinglepointIntent(
kelvin-onlab898a6c62015-01-16 14:13:53 -08001448 self,
shahshreyac2f97072015-03-19 17:04:29 -07001449 ingressDeviceList,
kelvin-onlabd3b64892015-01-20 13:26:24 -08001450 egressDevice,
shahshreyac2f97072015-03-19 17:04:29 -07001451 portIngressList=None,
kelvin-onlabd3b64892015-01-20 13:26:24 -08001452 portEgress="",
kelvin-onlab898a6c62015-01-16 14:13:53 -08001453 ethType="",
1454 ethSrc="",
1455 ethDst="",
1456 bandwidth="",
kelvin-onlabd3b64892015-01-20 13:26:24 -08001457 lambdaAlloc=False,
kelvin-onlab898a6c62015-01-16 14:13:53 -08001458 ipProto="",
1459 ipSrc="",
1460 ipDst="",
1461 tcpSrc="",
1462 tcpDst="",
1463 setEthSrc="",
Jeremy Songster832f9e92016-05-05 14:30:49 -07001464 setEthDst="",
Jeremy Songsterff553672016-05-12 17:06:23 -07001465 vlanId="",
Jeremy Songster9385d412016-06-02 17:57:36 -07001466 setVlan="",
Jeremy Songsterc032f162016-08-04 17:14:49 -07001467 partial=False,
1468 encap="" ):
kelvin8ec71442015-01-15 16:57:00 -08001469 """
shahshreyad0c80432014-12-04 16:56:05 -08001470 Note:
shahshreya70622b12015-03-19 17:19:00 -07001471 This function assumes the format of all ingress devices
Jon Hallbe379602015-03-24 13:39:32 -07001472 is same. That is, all ingress devices include port numbers
1473 with a "/" or all ingress devices could specify device
1474 ids and port numbers seperately.
shahshreyad0c80432014-12-04 16:56:05 -08001475 Required:
Jon Hallbe379602015-03-24 13:39:32 -07001476 * ingressDeviceList: List of device ids of ingress device
shahshreyac2f97072015-03-19 17:04:29 -07001477 ( Atleast 2 ingress devices required in the list )
kelvin-onlabd3b64892015-01-20 13:26:24 -08001478 * egressDevice: device id of egress device
shahshreyad0c80432014-12-04 16:56:05 -08001479 Optional:
1480 * ethType: specify ethType
kelvin8ec71442015-01-15 16:57:00 -08001481 * ethSrc: specify ethSrc ( i.e. src mac addr )
1482 * ethDst: specify ethDst ( i.e. dst mac addr )
shahshreyad0c80432014-12-04 16:56:05 -08001483 * bandwidth: specify bandwidth capacity of link
kelvin-onlabd3b64892015-01-20 13:26:24 -08001484 * lambdaAlloc: if True, intent will allocate lambda
shahshreyad0c80432014-12-04 16:56:05 -08001485 for the specified intent
Jon Halle3f39ff2015-01-13 11:50:53 -08001486 * ipProto: specify ip protocol
shahshreyad0c80432014-12-04 16:56:05 -08001487 * ipSrc: specify ip source address
1488 * ipDst: specify ip destination address
1489 * tcpSrc: specify tcp source port
1490 * tcpDst: specify tcp destination port
1491 * setEthSrc: action to Rewrite Source MAC Address
1492 * setEthDst: action to Rewrite Destination MAC Address
Jeremy Songster832f9e92016-05-05 14:30:49 -07001493 * vlanId: specify vlan Id
Jeremy Songsterff553672016-05-12 17:06:23 -07001494 * setVlan: specify VLAN Id treatment
Jeremy Songsterc032f162016-08-04 17:14:49 -07001495 * encap: specify a type of encapsulation
shahshreyad0c80432014-12-04 16:56:05 -08001496 Description:
kelvin8ec71442015-01-15 16:57:00 -08001497 Adds a multipoint-to-singlepoint intent ( uni-directional ) by
shahshreyad0c80432014-12-04 16:56:05 -08001498 specifying device id's and optional fields
kelvin-onlabfb521662015-02-27 09:52:40 -08001499 Returns:
1500 A string of the intent id or None on error
shahshreyad0c80432014-12-04 16:56:05 -08001501
Jon Halle3f39ff2015-01-13 11:50:53 -08001502 NOTE: This function may change depending on the
Jon Hallefbd9792015-03-05 16:11:36 -08001503 options developers provide for multipoint-to-singlepoint
shahshreyad0c80432014-12-04 16:56:05 -08001504 intent via cli
kelvin8ec71442015-01-15 16:57:00 -08001505 """
shahshreyad0c80432014-12-04 16:56:05 -08001506 try:
Jeremy Songsterff553672016-05-12 17:06:23 -07001507 cmd = "add-multi-to-single-intent"
shahshreyad0c80432014-12-04 16:56:05 -08001508
Jeremy Songsterff553672016-05-12 17:06:23 -07001509 if ethType:
1510 cmd += " --ethType " + str( ethType )
1511 if ethSrc:
1512 cmd += " --ethSrc " + str( ethSrc )
1513 if ethDst:
1514 cmd += " --ethDst " + str( ethDst )
1515 if bandwidth:
1516 cmd += " --bandwidth " + str( bandwidth )
1517 if lambdaAlloc:
1518 cmd += " --lambda "
1519 if ipProto:
1520 cmd += " --ipProto " + str( ipProto )
1521 if ipSrc:
1522 cmd += " --ipSrc " + str( ipSrc )
1523 if ipDst:
1524 cmd += " --ipDst " + str( ipDst )
1525 if tcpSrc:
1526 cmd += " --tcpSrc " + str( tcpSrc )
1527 if tcpDst:
1528 cmd += " --tcpDst " + str( tcpDst )
1529 if setEthSrc:
1530 cmd += " --setEthSrc " + str( setEthSrc )
1531 if setEthDst:
1532 cmd += " --setEthDst " + str( setEthDst )
1533 if vlanId:
1534 cmd += " -v " + str( vlanId )
1535 if setVlan:
1536 cmd += " --setVlan " + str( setVlan )
Jeremy Songster9385d412016-06-02 17:57:36 -07001537 if partial:
1538 cmd += " --partial"
Jeremy Songsterc032f162016-08-04 17:14:49 -07001539 if encap:
1540 cmd += " --encapsulation " + str( encap )
shahshreyad0c80432014-12-04 16:56:05 -08001541
kelvin8ec71442015-01-15 16:57:00 -08001542 # Check whether the user appended the port
1543 # or provided it as an input
shahshreyac2f97072015-03-19 17:04:29 -07001544
1545 if portIngressList is None:
1546 for ingressDevice in ingressDeviceList:
1547 if "/" in ingressDevice:
1548 cmd += " " + str( ingressDevice )
1549 else:
1550 main.log.error( "You must specify " +
Jon Hallbe379602015-03-24 13:39:32 -07001551 "the ingress port" )
shahshreyac2f97072015-03-19 17:04:29 -07001552 # TODO: perhaps more meaningful return
1553 return main.FALSE
shahshreyad0c80432014-12-04 16:56:05 -08001554 else:
Jon Hall71ce4e72015-03-23 14:05:58 -07001555 if len( ingressDeviceList ) == len( portIngressList ):
Jon Hall08f61bc2015-04-13 16:00:30 -07001556 for ingressDevice, portIngress in zip( ingressDeviceList,
1557 portIngressList ):
shahshreya70622b12015-03-19 17:19:00 -07001558 cmd += " " + \
1559 str( ingressDevice ) + "/" +\
1560 str( portIngress ) + " "
kelvin-onlab38143812015-04-01 15:03:01 -07001561 else:
Jon Hall08f61bc2015-04-13 16:00:30 -07001562 main.log.error( "Device list and port list does not " +
1563 "have the same length" )
kelvin-onlab38143812015-04-01 15:03:01 -07001564 return main.FALSE
kelvin-onlabd3b64892015-01-20 13:26:24 -08001565 if "/" in egressDevice:
1566 cmd += " " + str( egressDevice )
shahshreyad0c80432014-12-04 16:56:05 -08001567 else:
kelvin-onlabd3b64892015-01-20 13:26:24 -08001568 if not portEgress:
kelvin8ec71442015-01-15 16:57:00 -08001569 main.log.error( "You must specify " +
1570 "the egress port" )
shahshreyad0c80432014-12-04 16:56:05 -08001571 return main.FALSE
Jon Halle3f39ff2015-01-13 11:50:53 -08001572
kelvin8ec71442015-01-15 16:57:00 -08001573 cmd += " " +\
kelvin-onlabd3b64892015-01-20 13:26:24 -08001574 str( egressDevice ) + "/" +\
1575 str( portEgress )
kelvin-onlab898a6c62015-01-16 14:13:53 -08001576 handle = self.sendline( cmd )
You Wangb5a55f72017-03-03 12:51:05 -08001577 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08001578 assert "Command not found:" not in handle, handle
kelvin-onlabfb521662015-02-27 09:52:40 -08001579 # If error, return error message
kelvin-onlab898a6c62015-01-16 14:13:53 -08001580 if re.search( "Error", handle ):
kelvin-onlabfb521662015-02-27 09:52:40 -08001581 main.log.error( "Error in adding multipoint-to-singlepoint " +
1582 "intent" )
1583 return None
shahshreyad0c80432014-12-04 16:56:05 -08001584 else:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07001585 match = re.search( 'id=0x([\da-f]+),', handle )
kelvin-onlabb9408212015-04-01 13:34:04 -07001586 if match:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07001587 return match.group()[ 3:-1 ]
kelvin-onlabb9408212015-04-01 13:34:04 -07001588 else:
1589 main.log.error( "Error, intent ID not found" )
1590 return None
Jon Hallc6793552016-01-19 14:18:37 -08001591 except AssertionError:
1592 main.log.exception( "" )
1593 return None
kelvin-onlabb9408212015-04-01 13:34:04 -07001594 except TypeError:
1595 main.log.exception( self.name + ": Object not as expected" )
1596 return None
1597 except pexpect.EOF:
1598 main.log.error( self.name + ": EOF exception found" )
1599 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07001600 main.cleanAndExit()
kelvin-onlabb9408212015-04-01 13:34:04 -07001601 except Exception:
1602 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07001603 main.cleanAndExit()
kelvin-onlabb9408212015-04-01 13:34:04 -07001604
1605 def addSinglepointToMultipointIntent(
1606 self,
1607 ingressDevice,
1608 egressDeviceList,
1609 portIngress="",
1610 portEgressList=None,
1611 ethType="",
1612 ethSrc="",
1613 ethDst="",
1614 bandwidth="",
1615 lambdaAlloc=False,
1616 ipProto="",
1617 ipSrc="",
1618 ipDst="",
1619 tcpSrc="",
1620 tcpDst="",
1621 setEthSrc="",
Jeremy Songster832f9e92016-05-05 14:30:49 -07001622 setEthDst="",
Jeremy Songsterff553672016-05-12 17:06:23 -07001623 vlanId="",
Jeremy Songster9385d412016-06-02 17:57:36 -07001624 setVlan="",
Jeremy Songsterc032f162016-08-04 17:14:49 -07001625 partial=False,
1626 encap="" ):
kelvin-onlabb9408212015-04-01 13:34:04 -07001627 """
1628 Note:
1629 This function assumes the format of all egress devices
1630 is same. That is, all egress devices include port numbers
1631 with a "/" or all egress devices could specify device
1632 ids and port numbers seperately.
1633 Required:
1634 * EgressDeviceList: List of device ids of egress device
1635 ( Atleast 2 eress devices required in the list )
1636 * ingressDevice: device id of ingress device
1637 Optional:
1638 * ethType: specify ethType
1639 * ethSrc: specify ethSrc ( i.e. src mac addr )
1640 * ethDst: specify ethDst ( i.e. dst mac addr )
1641 * bandwidth: specify bandwidth capacity of link
1642 * lambdaAlloc: if True, intent will allocate lambda
1643 for the specified intent
1644 * ipProto: specify ip protocol
1645 * ipSrc: specify ip source address
1646 * ipDst: specify ip destination address
1647 * tcpSrc: specify tcp source port
1648 * tcpDst: specify tcp destination port
1649 * setEthSrc: action to Rewrite Source MAC Address
1650 * setEthDst: action to Rewrite Destination MAC Address
Jeremy Songster832f9e92016-05-05 14:30:49 -07001651 * vlanId: specify vlan Id
Jeremy Songsterff553672016-05-12 17:06:23 -07001652 * setVlan: specify VLAN ID treatment
Jeremy Songsterc032f162016-08-04 17:14:49 -07001653 * encap: specify an encapsulation type
kelvin-onlabb9408212015-04-01 13:34:04 -07001654 Description:
1655 Adds a singlepoint-to-multipoint intent ( uni-directional ) by
1656 specifying device id's and optional fields
1657 Returns:
1658 A string of the intent id or None on error
1659
1660 NOTE: This function may change depending on the
1661 options developers provide for singlepoint-to-multipoint
1662 intent via cli
1663 """
1664 try:
Jeremy Songsterff553672016-05-12 17:06:23 -07001665 cmd = "add-single-to-multi-intent"
kelvin-onlabb9408212015-04-01 13:34:04 -07001666
Jeremy Songsterff553672016-05-12 17:06:23 -07001667 if ethType:
1668 cmd += " --ethType " + str( ethType )
1669 if ethSrc:
1670 cmd += " --ethSrc " + str( ethSrc )
1671 if ethDst:
1672 cmd += " --ethDst " + str( ethDst )
1673 if bandwidth:
1674 cmd += " --bandwidth " + str( bandwidth )
1675 if lambdaAlloc:
1676 cmd += " --lambda "
1677 if ipProto:
1678 cmd += " --ipProto " + str( ipProto )
1679 if ipSrc:
1680 cmd += " --ipSrc " + str( ipSrc )
1681 if ipDst:
1682 cmd += " --ipDst " + str( ipDst )
1683 if tcpSrc:
1684 cmd += " --tcpSrc " + str( tcpSrc )
1685 if tcpDst:
1686 cmd += " --tcpDst " + str( tcpDst )
1687 if setEthSrc:
1688 cmd += " --setEthSrc " + str( setEthSrc )
1689 if setEthDst:
1690 cmd += " --setEthDst " + str( setEthDst )
1691 if vlanId:
1692 cmd += " -v " + str( vlanId )
1693 if setVlan:
1694 cmd += " --setVlan " + str( setVlan )
Jeremy Songster9385d412016-06-02 17:57:36 -07001695 if partial:
1696 cmd += " --partial"
Jeremy Songsterc032f162016-08-04 17:14:49 -07001697 if encap:
1698 cmd += " --encapsulation " + str( encap )
kelvin-onlabb9408212015-04-01 13:34:04 -07001699
1700 # Check whether the user appended the port
1701 # or provided it as an input
Jon Hall08f61bc2015-04-13 16:00:30 -07001702
kelvin-onlabb9408212015-04-01 13:34:04 -07001703 if "/" in ingressDevice:
1704 cmd += " " + str( ingressDevice )
1705 else:
1706 if not portIngress:
1707 main.log.error( "You must specify " +
1708 "the Ingress port" )
1709 return main.FALSE
1710
1711 cmd += " " +\
1712 str( ingressDevice ) + "/" +\
1713 str( portIngress )
1714
1715 if portEgressList is None:
1716 for egressDevice in egressDeviceList:
1717 if "/" in egressDevice:
1718 cmd += " " + str( egressDevice )
1719 else:
1720 main.log.error( "You must specify " +
1721 "the egress port" )
1722 # TODO: perhaps more meaningful return
1723 return main.FALSE
1724 else:
1725 if len( egressDeviceList ) == len( portEgressList ):
Jon Hall08f61bc2015-04-13 16:00:30 -07001726 for egressDevice, portEgress in zip( egressDeviceList,
1727 portEgressList ):
kelvin-onlabb9408212015-04-01 13:34:04 -07001728 cmd += " " + \
1729 str( egressDevice ) + "/" +\
1730 str( portEgress )
kelvin-onlab38143812015-04-01 15:03:01 -07001731 else:
Jon Hall08f61bc2015-04-13 16:00:30 -07001732 main.log.error( "Device list and port list does not " +
1733 "have the same length" )
kelvin-onlab38143812015-04-01 15:03:01 -07001734 return main.FALSE
kelvin-onlabb9408212015-04-01 13:34:04 -07001735 handle = self.sendline( cmd )
You Wangb5a55f72017-03-03 12:51:05 -08001736 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08001737 assert "Command not found:" not in handle, handle
kelvin-onlabb9408212015-04-01 13:34:04 -07001738 # If error, return error message
1739 if re.search( "Error", handle ):
1740 main.log.error( "Error in adding singlepoint-to-multipoint " +
1741 "intent" )
shahshreyac2f97072015-03-19 17:04:29 -07001742 return None
kelvin-onlabb9408212015-04-01 13:34:04 -07001743 else:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07001744 match = re.search( 'id=0x([\da-f]+),', handle )
kelvin-onlabb9408212015-04-01 13:34:04 -07001745 if match:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07001746 return match.group()[ 3:-1 ]
kelvin-onlabb9408212015-04-01 13:34:04 -07001747 else:
1748 main.log.error( "Error, intent ID not found" )
1749 return None
Jon Hallc6793552016-01-19 14:18:37 -08001750 except AssertionError:
1751 main.log.exception( "" )
1752 return None
Jon Halld4d4b372015-01-28 16:02:41 -08001753 except TypeError:
1754 main.log.exception( self.name + ": Object not as expected" )
1755 return None
shahshreyad0c80432014-12-04 16:56:05 -08001756 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08001757 main.log.error( self.name + ": EOF exception found" )
1758 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07001759 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08001760 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08001761 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07001762 main.cleanAndExit()
shahshreyad0c80432014-12-04 16:56:05 -08001763
Hari Krishna9e232602015-04-13 17:29:08 -07001764 def addMplsIntent(
1765 self,
1766 ingressDevice,
1767 egressDevice,
Hari Krishna87a17f12015-04-13 17:42:23 -07001768 ingressPort="",
1769 egressPort="",
Hari Krishna9e232602015-04-13 17:29:08 -07001770 ethType="",
1771 ethSrc="",
1772 ethDst="",
1773 bandwidth="",
1774 lambdaAlloc=False,
1775 ipProto="",
1776 ipSrc="",
1777 ipDst="",
1778 tcpSrc="",
1779 tcpDst="",
Hari Krishna87a17f12015-04-13 17:42:23 -07001780 ingressLabel="",
Hari Krishnadfff6672015-04-13 17:53:27 -07001781 egressLabel="",
Jeremy Ronquillo82705492017-10-18 14:19:55 -07001782 priority="" ):
Hari Krishna9e232602015-04-13 17:29:08 -07001783 """
1784 Required:
1785 * ingressDevice: device id of ingress device
1786 * egressDevice: device id of egress device
1787 Optional:
1788 * ethType: specify ethType
1789 * ethSrc: specify ethSrc ( i.e. src mac addr )
1790 * ethDst: specify ethDst ( i.e. dst mac addr )
1791 * bandwidth: specify bandwidth capacity of link
1792 * lambdaAlloc: if True, intent will allocate lambda
1793 for the specified intent
1794 * ipProto: specify ip protocol
1795 * ipSrc: specify ip source address
1796 * ipDst: specify ip destination address
1797 * tcpSrc: specify tcp source port
1798 * tcpDst: specify tcp destination port
1799 * ingressLabel: Ingress MPLS label
1800 * egressLabel: Egress MPLS label
1801 Description:
1802 Adds MPLS intent by
1803 specifying device id's and optional fields
1804 Returns:
1805 A string of the intent id or None on error
1806
1807 NOTE: This function may change depending on the
1808 options developers provide for MPLS
1809 intent via cli
1810 """
1811 try:
Jeremy Songsterff553672016-05-12 17:06:23 -07001812 cmd = "add-mpls-intent"
Hari Krishna9e232602015-04-13 17:29:08 -07001813
Jeremy Songsterff553672016-05-12 17:06:23 -07001814 if ethType:
1815 cmd += " --ethType " + str( ethType )
1816 if ethSrc:
1817 cmd += " --ethSrc " + str( ethSrc )
1818 if ethDst:
1819 cmd += " --ethDst " + str( ethDst )
1820 if bandwidth:
1821 cmd += " --bandwidth " + str( bandwidth )
1822 if lambdaAlloc:
1823 cmd += " --lambda "
1824 if ipProto:
1825 cmd += " --ipProto " + str( ipProto )
1826 if ipSrc:
1827 cmd += " --ipSrc " + str( ipSrc )
1828 if ipDst:
1829 cmd += " --ipDst " + str( ipDst )
1830 if tcpSrc:
1831 cmd += " --tcpSrc " + str( tcpSrc )
1832 if tcpDst:
1833 cmd += " --tcpDst " + str( tcpDst )
1834 if ingressLabel:
1835 cmd += " --ingressLabel " + str( ingressLabel )
1836 if egressLabel:
1837 cmd += " --egressLabel " + str( egressLabel )
1838 if priority:
1839 cmd += " --priority " + str( priority )
Hari Krishna9e232602015-04-13 17:29:08 -07001840
1841 # Check whether the user appended the port
1842 # or provided it as an input
1843 if "/" in ingressDevice:
1844 cmd += " " + str( ingressDevice )
1845 else:
Hari Krishna87a17f12015-04-13 17:42:23 -07001846 if not ingressPort:
Hari Krishna9e232602015-04-13 17:29:08 -07001847 main.log.error( "You must specify the ingress port" )
1848 return None
1849
1850 cmd += " " + \
1851 str( ingressDevice ) + "/" +\
Hari Krishna87a17f12015-04-13 17:42:23 -07001852 str( ingressPort ) + " "
Hari Krishna9e232602015-04-13 17:29:08 -07001853
1854 if "/" in egressDevice:
1855 cmd += " " + str( egressDevice )
1856 else:
Hari Krishna87a17f12015-04-13 17:42:23 -07001857 if not egressPort:
Hari Krishna9e232602015-04-13 17:29:08 -07001858 main.log.error( "You must specify the egress port" )
1859 return None
1860
1861 cmd += " " +\
1862 str( egressDevice ) + "/" +\
Hari Krishna87a17f12015-04-13 17:42:23 -07001863 str( egressPort )
Hari Krishna9e232602015-04-13 17:29:08 -07001864
1865 handle = self.sendline( cmd )
You Wangb5a55f72017-03-03 12:51:05 -08001866 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08001867 assert "Command not found:" not in handle, handle
Hari Krishna9e232602015-04-13 17:29:08 -07001868 # If error, return error message
1869 if re.search( "Error", handle ):
1870 main.log.error( "Error in adding mpls intent" )
1871 return None
1872 else:
1873 # TODO: print out all the options in this message?
1874 main.log.info( "MPLS intent installed between " +
1875 str( ingressDevice ) + " and " +
1876 str( egressDevice ) )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07001877 match = re.search( 'id=0x([\da-f]+),', handle )
Hari Krishna9e232602015-04-13 17:29:08 -07001878 if match:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07001879 return match.group()[ 3:-1 ]
Hari Krishna9e232602015-04-13 17:29:08 -07001880 else:
1881 main.log.error( "Error, intent ID not found" )
1882 return None
Jon Hallc6793552016-01-19 14:18:37 -08001883 except AssertionError:
1884 main.log.exception( "" )
1885 return None
Hari Krishna9e232602015-04-13 17:29:08 -07001886 except TypeError:
1887 main.log.exception( self.name + ": Object not as expected" )
1888 return None
1889 except pexpect.EOF:
1890 main.log.error( self.name + ": EOF exception found" )
1891 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07001892 main.cleanAndExit()
Hari Krishna9e232602015-04-13 17:29:08 -07001893 except Exception:
1894 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07001895 main.cleanAndExit()
Hari Krishna9e232602015-04-13 17:29:08 -07001896
Jon Hallefbd9792015-03-05 16:11:36 -08001897 def removeIntent( self, intentId, app='org.onosproject.cli',
1898 purge=False, sync=False ):
kelvin-onlab898a6c62015-01-16 14:13:53 -08001899 """
shahshreya1c818fc2015-02-26 13:44:08 -08001900 Remove intent for specified application id and intent id
Jon Hall61282e32015-03-19 11:34:11 -07001901 Optional args:-
shahshreya1c818fc2015-02-26 13:44:08 -08001902 -s or --sync: Waits for the removal before returning
Jon Hall61282e32015-03-19 11:34:11 -07001903 -p or --purge: Purge the intent from the store after removal
1904
Jon Halle3f39ff2015-01-13 11:50:53 -08001905 Returns:
Jon Hall6509dbf2016-06-21 17:01:17 -07001906 main.FALSE on error and
Jon Halle3f39ff2015-01-13 11:50:53 -08001907 cli output otherwise
kelvin-onlab898a6c62015-01-16 14:13:53 -08001908 """
andrewonlab9a50dfe2014-10-17 17:22:31 -04001909 try:
Jon Hallc6358dd2015-04-10 12:44:28 -07001910 cmdStr = "remove-intent"
shahshreya1c818fc2015-02-26 13:44:08 -08001911 if purge:
1912 cmdStr += " -p"
1913 if sync:
1914 cmdStr += " -s"
1915
1916 cmdStr += " " + app + " " + str( intentId )
kelvin-onlabd3b64892015-01-20 13:26:24 -08001917 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -08001918 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08001919 assert "Command not found:" not in handle, handle
Jon Halle3f39ff2015-01-13 11:50:53 -08001920 if re.search( "Error", handle ):
kelvin-onlab898a6c62015-01-16 14:13:53 -08001921 main.log.error( "Error in removing intent" )
Jon Halle3f39ff2015-01-13 11:50:53 -08001922 return main.FALSE
andrewonlab9a50dfe2014-10-17 17:22:31 -04001923 else:
Jon Halle3f39ff2015-01-13 11:50:53 -08001924 # TODO: Should this be main.TRUE
1925 return handle
Jon Hallc6793552016-01-19 14:18:37 -08001926 except AssertionError:
1927 main.log.exception( "" )
1928 return None
Jon Halld4d4b372015-01-28 16:02:41 -08001929 except TypeError:
1930 main.log.exception( self.name + ": Object not as expected" )
1931 return None
andrewonlab9a50dfe2014-10-17 17:22:31 -04001932 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08001933 main.log.error( self.name + ": EOF exception found" )
1934 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07001935 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08001936 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08001937 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07001938 main.cleanAndExit()
andrewonlab9a50dfe2014-10-17 17:22:31 -04001939
YPZhangfebf7302016-05-24 16:45:56 -07001940 def removeAllIntents( self, purge=False, sync=False, app='org.onosproject.cli', timeout=30 ):
Jeremy42df2e72016-02-23 16:37:46 -08001941 """
1942 Description:
1943 Remove all the intents
1944 Optional args:-
1945 -s or --sync: Waits for the removal before returning
1946 -p or --purge: Purge the intent from the store after removal
1947 Returns:
1948 Returns main.TRUE if all intents are removed, otherwise returns
1949 main.FALSE; Returns None for exception
1950 """
1951 try:
1952 cmdStr = "remove-intent"
1953 if purge:
1954 cmdStr += " -p"
1955 if sync:
1956 cmdStr += " -s"
1957
1958 cmdStr += " " + app
YPZhangfebf7302016-05-24 16:45:56 -07001959 handle = self.sendline( cmdStr, timeout=timeout )
You Wangb5a55f72017-03-03 12:51:05 -08001960 assert handle is not None, "Error in sendline"
Jeremy42df2e72016-02-23 16:37:46 -08001961 assert "Command not found:" not in handle, handle
1962 if re.search( "Error", handle ):
1963 main.log.error( "Error in removing intent" )
1964 return main.FALSE
1965 else:
1966 return main.TRUE
1967 except AssertionError:
1968 main.log.exception( "" )
1969 return None
1970 except TypeError:
1971 main.log.exception( self.name + ": Object not as expected" )
1972 return None
1973 except pexpect.EOF:
1974 main.log.error( self.name + ": EOF exception found" )
1975 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07001976 main.cleanAndExit()
Jeremy42df2e72016-02-23 16:37:46 -08001977 except Exception:
1978 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07001979 main.cleanAndExit()
Jeremy42df2e72016-02-23 16:37:46 -08001980
Hari Krishnaacabd5a2015-07-01 17:10:19 -07001981 def purgeWithdrawnIntents( self ):
Hari Krishna0ce0e152015-06-23 09:55:29 -07001982 """
1983 Purges all WITHDRAWN Intents
1984 """
1985 try:
1986 cmdStr = "purge-intents"
1987 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -08001988 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08001989 assert "Command not found:" not in handle, handle
Hari Krishna0ce0e152015-06-23 09:55:29 -07001990 if re.search( "Error", handle ):
1991 main.log.error( "Error in purging intents" )
1992 return main.FALSE
1993 else:
1994 return main.TRUE
Jon Hallc6793552016-01-19 14:18:37 -08001995 except AssertionError:
1996 main.log.exception( "" )
1997 return None
Hari Krishna0ce0e152015-06-23 09:55:29 -07001998 except TypeError:
1999 main.log.exception( self.name + ": Object not as expected" )
2000 return None
2001 except pexpect.EOF:
2002 main.log.error( self.name + ": EOF exception found" )
2003 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002004 main.cleanAndExit()
Hari Krishna0ce0e152015-06-23 09:55:29 -07002005 except Exception:
2006 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002007 main.cleanAndExit()
Hari Krishna0ce0e152015-06-23 09:55:29 -07002008
Devin Lime6fe3c42017-10-18 16:28:40 -07002009 def wipeout( self ):
2010 """
2011 Wipe out the flows,intents,links,devices,hosts, and groups from the ONOS.
2012 """
2013 try:
2014 cmdStr = "wipe-out please"
2015 handle = self.sendline( cmdStr, timeout=60 )
2016 assert handle is not None, "Error in sendline"
2017 assert "Command not found:" not in handle, handle
2018 return main.TRUE
2019 except AssertionError:
2020 main.log.exception( "" )
2021 return None
2022 except TypeError:
2023 main.log.exception( self.name + ": Object not as expected" )
2024 return None
2025 except pexpect.EOF:
2026 main.log.error( self.name + ": EOF exception found" )
2027 main.log.error( self.name + ": " + self.handle.before )
2028 main.cleanAndExit()
2029 except Exception:
2030 main.log.exception( self.name + ": Uncaught exception!" )
2031 main.cleanAndExit()
2032
kelvin-onlabd3b64892015-01-20 13:26:24 -08002033 def routes( self, jsonFormat=False ):
kelvin8ec71442015-01-15 16:57:00 -08002034 """
kelvin-onlab898a6c62015-01-16 14:13:53 -08002035 NOTE: This method should be used after installing application:
2036 onos-app-sdnip
pingping-lin8b306ac2014-11-17 18:13:51 -08002037 Optional:
kelvin-onlabd3b64892015-01-20 13:26:24 -08002038 * jsonFormat: enable output formatting in json
pingping-lin8b306ac2014-11-17 18:13:51 -08002039 Description:
2040 Obtain all routes in the system
kelvin8ec71442015-01-15 16:57:00 -08002041 """
pingping-lin8b306ac2014-11-17 18:13:51 -08002042 try:
Jon Hallc6358dd2015-04-10 12:44:28 -07002043 cmdStr = "routes"
kelvin-onlabd3b64892015-01-20 13:26:24 -08002044 if jsonFormat:
Jon Hallc6358dd2015-04-10 12:44:28 -07002045 cmdStr += " -j"
2046 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -08002047 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08002048 assert "Command not found:" not in handle, handle
pingping-lin8b306ac2014-11-17 18:13:51 -08002049 return handle
Jon Hallc6793552016-01-19 14:18:37 -08002050 except AssertionError:
2051 main.log.exception( "" )
2052 return None
Jon Halld4d4b372015-01-28 16:02:41 -08002053 except TypeError:
2054 main.log.exception( self.name + ": Object not as expected" )
2055 return None
pingping-lin8b306ac2014-11-17 18:13:51 -08002056 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08002057 main.log.error( self.name + ": EOF exception found" )
2058 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002059 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08002060 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08002061 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002062 main.cleanAndExit()
pingping-lin8b306ac2014-11-17 18:13:51 -08002063
pingping-lin54b03372015-08-13 14:43:10 -07002064 def ipv4RouteNumber( self ):
2065 """
2066 NOTE: This method should be used after installing application:
2067 onos-app-sdnip
2068 Description:
2069 Obtain the total IPv4 routes number in the system
2070 """
2071 try:
Pratik Parab57963572017-05-09 11:37:54 -07002072 cmdStr = "routes -j"
pingping-lin54b03372015-08-13 14:43:10 -07002073 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -08002074 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08002075 assert "Command not found:" not in handle, handle
pingping-lin54b03372015-08-13 14:43:10 -07002076 jsonResult = json.loads( handle )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002077 return len( jsonResult[ 'routes4' ] )
Jon Hallc6793552016-01-19 14:18:37 -08002078 except AssertionError:
2079 main.log.exception( "" )
2080 return None
2081 except ( TypeError, ValueError ):
2082 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, handle ) )
pingping-lin54b03372015-08-13 14:43:10 -07002083 return None
2084 except pexpect.EOF:
2085 main.log.error( self.name + ": EOF exception found" )
2086 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002087 main.cleanAndExit()
pingping-lin54b03372015-08-13 14:43:10 -07002088 except Exception:
2089 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002090 main.cleanAndExit()
pingping-lin54b03372015-08-13 14:43:10 -07002091
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002092 # =============Function to check Bandwidth allocation========
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00002093 def allocations( self, jsonFormat = True, dollarSign = True ):
Shreya Chowdhary6fbb96c2017-05-02 16:20:19 -07002094 """
2095 Description:
2096 Obtain Bandwidth Allocation Information from ONOS cli.
2097 """
2098 try:
2099 cmdStr = "allocations"
2100 if jsonFormat:
2101 cmdStr += " -j"
2102 handle = self.sendline( cmdStr, timeout=300, dollarSign=True )
2103 assert handle is not None, "Error in sendline"
2104 assert "Command not found:" not in handle, handle
2105 return handle
2106 except AssertionError:
2107 main.log.exception( "" )
2108 return None
2109 except ( TypeError, ValueError ):
2110 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, handle ) )
2111 return None
2112 except pexpect.EOF:
2113 main.log.error( self.name + ": EOF exception found" )
2114 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002115 main.cleanAndExit()
Shreya Chowdhary6fbb96c2017-05-02 16:20:19 -07002116 except Exception:
2117 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002118 main.cleanAndExit()
Shreya Chowdhary6fbb96c2017-05-02 16:20:19 -07002119
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002120 def intents( self, jsonFormat = True, summary = False, **intentargs ):
kelvin8ec71442015-01-15 16:57:00 -08002121 """
andrewonlabe6745342014-10-17 14:29:13 -04002122 Description:
Jon Hallff566d52016-01-15 14:45:36 -08002123 Obtain intents from the ONOS cli.
2124 Optional:
2125 * jsonFormat: Enable output formatting in json, default to True
2126 * summary: Whether only output the intent summary, defaults to False
2127 * type: Only output a certain type of intent. This options is valid
2128 only when jsonFormat is True and summary is True.
kelvin-onlab898a6c62015-01-16 14:13:53 -08002129 """
andrewonlabe6745342014-10-17 14:29:13 -04002130 try:
Jon Hallc6358dd2015-04-10 12:44:28 -07002131 cmdStr = "intents"
pingping-lin8244a3b2015-09-16 13:36:56 -07002132 if summary:
2133 cmdStr += " -s"
kelvin-onlabd3b64892015-01-20 13:26:24 -08002134 if jsonFormat:
Jon Hallc6358dd2015-04-10 12:44:28 -07002135 cmdStr += " -j"
Shreya Chowdhary6fbb96c2017-05-02 16:20:19 -07002136 handle = self.sendline( cmdStr, timeout=300 )
You Wangb5a55f72017-03-03 12:51:05 -08002137 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08002138 assert "Command not found:" not in handle, handle
pingping-lin8244a3b2015-09-16 13:36:56 -07002139 args = utilities.parse_args( [ "TYPE" ], **intentargs )
acsmars5b5fbaf2015-09-18 10:38:20 -07002140 if "TYPE" in args.keys():
Jon Hallff566d52016-01-15 14:45:36 -08002141 intentType = args[ "TYPE" ]
acsmars5b5fbaf2015-09-18 10:38:20 -07002142 else:
Jon Hallff566d52016-01-15 14:45:36 -08002143 intentType = ""
2144 # IF we want the summary of a specific intent type
2145 if jsonFormat and summary and ( intentType != "" ):
pingping-lin8244a3b2015-09-16 13:36:56 -07002146 jsonResult = json.loads( handle )
Jon Hallff566d52016-01-15 14:45:36 -08002147 if intentType in jsonResult.keys():
2148 return jsonResult[ intentType ]
pingping-lin8244a3b2015-09-16 13:36:56 -07002149 else:
Jon Hallff566d52016-01-15 14:45:36 -08002150 main.log.error( "unknown TYPE, returning all types of intents" )
pingping-lin8244a3b2015-09-16 13:36:56 -07002151 return handle
2152 else:
2153 return handle
Jon Hallc6793552016-01-19 14:18:37 -08002154 except AssertionError:
2155 main.log.exception( "" )
2156 return None
2157 except ( TypeError, ValueError ):
2158 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, handle ) )
pingping-lin54b03372015-08-13 14:43:10 -07002159 return None
2160 except pexpect.EOF:
2161 main.log.error( self.name + ": EOF exception found" )
2162 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002163 main.cleanAndExit()
pingping-lin54b03372015-08-13 14:43:10 -07002164 except Exception:
2165 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002166 main.cleanAndExit()
pingping-lin54b03372015-08-13 14:43:10 -07002167
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002168 def getIntentState( self, intentsId, intentsJson=None ):
kelvin-onlab54400a92015-02-26 18:05:51 -08002169 """
You Wangfdcbfc42016-05-16 12:16:53 -07002170 Description:
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00002171 Gets intent state. Accepts a single intent ID (string type) or a
You Wangfdcbfc42016-05-16 12:16:53 -07002172 list of intent IDs.
2173 Parameters:
2174 intentsId: intent ID, both string type and list type are acceptable
kelvin-onlab54400a92015-02-26 18:05:51 -08002175 intentsJson: parsed json object from the onos:intents api
You Wangfdcbfc42016-05-16 12:16:53 -07002176 Returns:
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00002177 Returns the state (string type) of the ID if a single intent ID is
You Wangfdcbfc42016-05-16 12:16:53 -07002178 accepted.
2179 Returns a list of dictionaries if a list of intent IDs is accepted,
2180 and each dictionary maps 'id' to the Intent ID and 'state' to
2181 corresponding intent state.
kelvin-onlab54400a92015-02-26 18:05:51 -08002182 """
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00002183
kelvin-onlab54400a92015-02-26 18:05:51 -08002184 try:
2185 state = "State is Undefined"
2186 if not intentsJson:
Jon Hallc6793552016-01-19 14:18:37 -08002187 rawJson = self.intents()
kelvin-onlab54400a92015-02-26 18:05:51 -08002188 else:
Jon Hallc6793552016-01-19 14:18:37 -08002189 rawJson = intentsJson
2190 parsedIntentsJson = json.loads( rawJson )
Jon Hallefbd9792015-03-05 16:11:36 -08002191 if isinstance( intentsId, types.StringType ):
Jon Hallc6793552016-01-19 14:18:37 -08002192 for intent in parsedIntentsJson:
kelvin-onlabc2dcd3f2015-04-09 16:40:02 -07002193 if intentsId == intent[ 'id' ]:
2194 state = intent[ 'state' ]
kelvin-onlab54400a92015-02-26 18:05:51 -08002195 return state
Jon Hallefbd9792015-03-05 16:11:36 -08002196 main.log.info( "Cannot find intent ID" + str( intentsId ) +
Jon Hall53158082017-05-18 11:17:00 -07002197 " in the list" )
kelvin-onlab54400a92015-02-26 18:05:51 -08002198 return state
Jon Hallefbd9792015-03-05 16:11:36 -08002199 elif isinstance( intentsId, types.ListType ):
kelvin-onlab07dbd012015-03-04 16:29:39 -08002200 dictList = []
kelvin-onlabc2dcd3f2015-04-09 16:40:02 -07002201 for i in xrange( len( intentsId ) ):
kelvin-onlab07dbd012015-03-04 16:29:39 -08002202 stateDict = {}
Jon Hall53158082017-05-18 11:17:00 -07002203 for intent in parsedIntentsJson:
2204 if intentsId[ i ] == intent[ 'id' ]:
2205 stateDict[ 'state' ] = intent[ 'state' ]
kelvin-onlabc2dcd3f2015-04-09 16:40:02 -07002206 stateDict[ 'id' ] = intentsId[ i ]
Jon Hallefbd9792015-03-05 16:11:36 -08002207 dictList.append( stateDict )
kelvin-onlab54400a92015-02-26 18:05:51 -08002208 break
Jon Hallefbd9792015-03-05 16:11:36 -08002209 if len( intentsId ) != len( dictList ):
Jon Hall53158082017-05-18 11:17:00 -07002210 main.log.warn( "Could not find all intents in ONOS output" )
2211 main.log.debug( "expected ids: {} \n ONOS intents: {}".format( intentsId, parsedIntentsJson ) )
kelvin-onlab07dbd012015-03-04 16:29:39 -08002212 return dictList
kelvin-onlab54400a92015-02-26 18:05:51 -08002213 else:
Jon Hall53158082017-05-18 11:17:00 -07002214 main.log.info( "Invalid type for intentsId argument" )
kelvin-onlab54400a92015-02-26 18:05:51 -08002215 return None
Jon Hallc6793552016-01-19 14:18:37 -08002216 except ( TypeError, ValueError ):
2217 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, rawJson ) )
kelvin-onlab54400a92015-02-26 18:05:51 -08002218 return None
2219 except pexpect.EOF:
2220 main.log.error( self.name + ": EOF exception found" )
2221 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002222 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08002223 except Exception:
kelvin-onlab54400a92015-02-26 18:05:51 -08002224 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002225 main.cleanAndExit()
Jon Hall390696c2015-05-05 17:13:41 -07002226
Jon Hallf539eb92017-05-22 17:18:42 -07002227 def checkIntentState( self, intentsId, expectedState='INSTALLED' ):
kelvin-onlabc2dcd3f2015-04-09 16:40:02 -07002228 """
2229 Description:
2230 Check intents state
2231 Required:
2232 intentsId - List of intents ID to be checked
2233 Optional:
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00002234 expectedState - Check the expected state(s) of each intents
kelvin-onlabf512e942015-06-08 19:42:59 -07002235 state in the list.
2236 *NOTE: You can pass in a list of expected state,
2237 Eg: expectedState = [ 'INSTALLED' , 'INSTALLING' ]
kelvin-onlabc2dcd3f2015-04-09 16:40:02 -07002238 Return:
Jon Hall53158082017-05-18 11:17:00 -07002239 Returns main.TRUE only if all intent are the same as expected states,
2240 otherwise returns main.FALSE.
kelvin-onlabc2dcd3f2015-04-09 16:40:02 -07002241 """
2242 try:
kelvin-onlabf512e942015-06-08 19:42:59 -07002243 returnValue = main.TRUE
Jon Hallf539eb92017-05-22 17:18:42 -07002244 # Generating a dictionary: intent id as a key and state as value
Devin Lim752dd7b2017-06-27 14:40:03 -07002245
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002246 # intentsDict = self.getIntentState( intentsId )
Devin Lim752dd7b2017-06-27 14:40:03 -07002247 intentsDict = []
2248 for intent in json.loads( self.intents() ):
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002249 if isinstance( intentsId, types.StringType ) \
2250 and intent.get( 'id' ) == intentsId:
2251 intentsDict.append( intent )
2252 elif isinstance( intentsId, types.ListType ) \
Devin Lim752dd7b2017-06-27 14:40:03 -07002253 and any( intent.get( 'id' ) == ids for ids in intentsId ):
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002254 intentsDict.append( intent )
Devin Lim752dd7b2017-06-27 14:40:03 -07002255
2256 if not intentsDict:
Jon Hallae04e622016-01-27 10:38:05 -08002257 main.log.info( self.name + ": There is something wrong " +
kelvin-onlabc2dcd3f2015-04-09 16:40:02 -07002258 "getting intents state" )
2259 return main.FALSE
kelvin-onlabf512e942015-06-08 19:42:59 -07002260
2261 if isinstance( expectedState, types.StringType ):
2262 for intents in intentsDict:
2263 if intents.get( 'state' ) != expectedState:
kelvin-onlaba297c4d2015-06-01 13:53:55 -07002264 main.log.debug( self.name + " : Intent ID - " +
2265 intents.get( 'id' ) +
kelvin-onlabf512e942015-06-08 19:42:59 -07002266 " actual state = " +
2267 intents.get( 'state' )
2268 + " does not equal expected state = "
2269 + expectedState )
kelvin-onlaba297c4d2015-06-01 13:53:55 -07002270 returnValue = main.FALSE
kelvin-onlabf512e942015-06-08 19:42:59 -07002271 elif isinstance( expectedState, types.ListType ):
2272 for intents in intentsDict:
2273 if not any( state == intents.get( 'state' ) for state in
2274 expectedState ):
2275 main.log.debug( self.name + " : Intent ID - " +
2276 intents.get( 'id' ) +
2277 " actual state = " +
2278 intents.get( 'state' ) +
2279 " does not equal expected states = "
2280 + str( expectedState ) )
2281 returnValue = main.FALSE
2282
kelvin-onlabc2dcd3f2015-04-09 16:40:02 -07002283 if returnValue == main.TRUE:
2284 main.log.info( self.name + ": All " +
2285 str( len( intentsDict ) ) +
kelvin-onlabf512e942015-06-08 19:42:59 -07002286 " intents are in " + str( expectedState ) +
2287 " state" )
kelvin-onlabc2dcd3f2015-04-09 16:40:02 -07002288 return returnValue
2289 except TypeError:
2290 main.log.exception( self.name + ": Object not as expected" )
2291 return None
2292 except pexpect.EOF:
2293 main.log.error( self.name + ": EOF exception found" )
2294 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002295 main.cleanAndExit()
kelvin-onlabc2dcd3f2015-04-09 16:40:02 -07002296 except Exception:
2297 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002298 main.cleanAndExit()
andrewonlabe6745342014-10-17 14:29:13 -04002299
Jon Hallf539eb92017-05-22 17:18:42 -07002300 def compareBandwidthAllocations( self, expectedAllocations ):
2301 """
2302 Description:
2303 Compare the allocated bandwidth with the given allocations
2304 Required:
2305 expectedAllocations - The expected ONOS output of the allocations command
2306 Return:
2307 Returns main.TRUE only if all intent are the same as expected states,
2308 otherwise returns main.FALSE.
2309 """
2310 # FIXME: Convert these string comparisons to object comparisons
2311 try:
2312 returnValue = main.TRUE
2313 bandwidthFailed = False
2314 rawAlloc = self.allocations()
2315 expectedFormat = StringIO( expectedAllocations )
2316 ONOSOutput = StringIO( rawAlloc )
2317 main.log.debug( "ONOSOutput: {}\nexpected output: {}".format( str( ONOSOutput ),
2318 str( expectedFormat ) ) )
2319
2320 for actual, expected in izip( ONOSOutput, expectedFormat ):
2321 actual = actual.rstrip()
2322 expected = expected.rstrip()
2323 main.log.debug( "Expect: {}\nactual: {}".format( expected, actual ) )
2324 if actual != expected and 'allocated' in actual and 'allocated' in expected:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002325 marker1 = actual.find( 'allocated' )
2326 m1 = actual[ :marker1 ]
2327 marker2 = expected.find( 'allocated' )
2328 m2 = expected[ :marker2 ]
Jon Hallf539eb92017-05-22 17:18:42 -07002329 if m1 != m2:
2330 bandwidthFailed = True
2331 elif actual != expected and 'allocated' not in actual and 'allocated' not in expected:
2332 bandwidthFailed = True
2333 expectedFormat.close()
2334 ONOSOutput.close()
2335
2336 if bandwidthFailed:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002337 main.log.error( "Bandwidth not allocated correctly using Intents!!" )
Jon Hallf539eb92017-05-22 17:18:42 -07002338 returnValue = main.FALSE
2339 return returnValue
2340 except TypeError:
2341 main.log.exception( self.name + ": Object not as expected" )
2342 return None
2343 except pexpect.EOF:
2344 main.log.error( self.name + ": EOF exception found" )
2345 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002346 main.cleanAndExit()
Jon Hallf539eb92017-05-22 17:18:42 -07002347 except Exception:
2348 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002349 main.cleanAndExit()
Jon Hallf539eb92017-05-22 17:18:42 -07002350
You Wang66518af2016-05-16 15:32:59 -07002351 def compareIntent( self, intentDict ):
2352 """
2353 Description:
2354 Compare the intent ids and states provided in the argument with all intents in ONOS
2355 Return:
2356 Returns main.TRUE if the two sets of intents match exactly, otherwise main.FALSE
2357 Arguments:
2358 intentDict: a dictionary which maps intent ids to intent states
2359 """
2360 try:
2361 intentsRaw = self.intents()
2362 intentsJson = json.loads( intentsRaw )
2363 intentDictONOS = {}
2364 for intent in intentsJson:
2365 intentDictONOS[ intent[ 'id' ] ] = intent[ 'state' ]
You Wang58d04452016-09-21 15:13:05 -07002366 returnValue = main.TRUE
You Wang66518af2016-05-16 15:32:59 -07002367 if len( intentDict ) != len( intentDictONOS ):
You Wang58d04452016-09-21 15:13:05 -07002368 main.log.warn( self.name + ": expected intent count does not match that in ONOS, " +
You Wang66518af2016-05-16 15:32:59 -07002369 str( len( intentDict ) ) + " expected and " +
2370 str( len( intentDictONOS ) ) + " actual" )
You Wang58d04452016-09-21 15:13:05 -07002371 returnValue = main.FALSE
You Wang66518af2016-05-16 15:32:59 -07002372 for intentID in intentDict.keys():
Jon Halle0f0b342017-04-18 11:43:47 -07002373 if intentID not in intentDictONOS.keys():
You Wang66518af2016-05-16 15:32:59 -07002374 main.log.debug( self.name + ": intent ID - " + intentID + " is not in ONOS" )
2375 returnValue = main.FALSE
You Wang58d04452016-09-21 15:13:05 -07002376 else:
2377 if intentDict[ intentID ] != intentDictONOS[ intentID ]:
2378 main.log.debug( self.name + ": intent ID - " + intentID +
2379 " expected state is " + intentDict[ intentID ] +
2380 " but actual state is " + intentDictONOS[ intentID ] )
2381 returnValue = main.FALSE
2382 intentDictONOS.pop( intentID )
2383 if len( intentDictONOS ) > 0:
2384 returnValue = main.FALSE
2385 for intentID in intentDictONOS.keys():
2386 main.log.debug( self.name + ": find extra intent in ONOS: intent ID " + intentID )
You Wang66518af2016-05-16 15:32:59 -07002387 if returnValue == main.TRUE:
2388 main.log.info( self.name + ": all intent IDs and states match that in ONOS" )
2389 return returnValue
You Wang1be9a512016-05-26 16:54:17 -07002390 except KeyError:
2391 main.log.exception( self.name + ": KeyError exception found" )
2392 return main.ERROR
You Wang66518af2016-05-16 15:32:59 -07002393 except ( TypeError, ValueError ):
2394 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, intentsRaw ) )
You Wang85560372016-05-18 10:44:33 -07002395 return main.ERROR
You Wang66518af2016-05-16 15:32:59 -07002396 except pexpect.EOF:
2397 main.log.error( self.name + ": EOF exception found" )
2398 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002399 main.cleanAndExit()
You Wang66518af2016-05-16 15:32:59 -07002400 except Exception:
2401 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002402 main.cleanAndExit()
You Wang66518af2016-05-16 15:32:59 -07002403
YPZhang14a4aa92016-07-15 13:37:15 -07002404 def checkIntentSummary( self, timeout=60, noExit=True ):
GlennRCed771242016-01-13 17:02:47 -08002405 """
2406 Description:
2407 Check the number of installed intents.
2408 Optional:
2409 timeout - the timeout for pexcept
YPZhang14a4aa92016-07-15 13:37:15 -07002410 noExit - If noExit, TestON will not exit if any except.
GlennRCed771242016-01-13 17:02:47 -08002411 Return:
2412 Returns main.TRUE only if the number of all installed intents are the same as total intents number
2413 , otherwise, returns main.FALSE.
2414 """
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00002415
GlennRCed771242016-01-13 17:02:47 -08002416 try:
2417 cmd = "intents -s -j"
2418
2419 # Check response if something wrong
YPZhang14a4aa92016-07-15 13:37:15 -07002420 response = self.sendline( cmd, timeout=timeout, noExit=noExit )
Jon Halle0f0b342017-04-18 11:43:47 -07002421 if response is None:
YPZhang0584d432016-06-21 15:20:13 -07002422 return main.FALSE
GlennRCed771242016-01-13 17:02:47 -08002423 response = json.loads( response )
2424
2425 # get total and installed number, see if they are match
2426 allState = response.get( 'all' )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002427 if allState.get( 'total' ) == allState.get( 'installed' ):
Jon Halla478b852017-12-04 15:00:15 -08002428 main.log.info( 'Total Intents: {} Installed Intents: {}'.format(
2429 allState.get( 'total' ), allState.get( 'installed' ) ) )
GlennRCed771242016-01-13 17:02:47 -08002430 return main.TRUE
Jon Halla478b852017-12-04 15:00:15 -08002431 main.log.info( 'Verified Intents failed Expected intents: {} installed intents: {}'.format(
2432 allState.get( 'total' ), allState.get( 'installed' ) ) )
GlennRCed771242016-01-13 17:02:47 -08002433 return main.FALSE
2434
Jon Hallc6793552016-01-19 14:18:37 -08002435 except ( TypeError, ValueError ):
2436 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, response ) )
GlennRCed771242016-01-13 17:02:47 -08002437 return None
2438 except pexpect.EOF:
2439 main.log.error( self.name + ": EOF exception found" )
2440 main.log.error( self.name + ": " + self.handle.before )
YPZhang14a4aa92016-07-15 13:37:15 -07002441 if noExit:
2442 return main.FALSE
2443 else:
Devin Lim44075962017-08-11 10:56:37 -07002444 main.cleanAndExit()
Jon Halle0f0b342017-04-18 11:43:47 -07002445 except pexpect.TIMEOUT:
2446 main.log.error( self.name + ": ONOS timeout" )
2447 return None
GlennRCed771242016-01-13 17:02:47 -08002448 except Exception:
2449 main.log.exception( self.name + ": Uncaught exception!" )
YPZhang14a4aa92016-07-15 13:37:15 -07002450 if noExit:
2451 return main.FALSE
2452 else:
Devin Lim44075962017-08-11 10:56:37 -07002453 main.cleanAndExit()
GlennRCed771242016-01-13 17:02:47 -08002454
Jeremy Songster306ed7a2016-07-19 10:59:07 -07002455 def flows( self, state="", jsonFormat=True, timeout=60, noExit=False, noCore=False ):
kelvin8ec71442015-01-15 16:57:00 -08002456 """
Shreya Shah0f01c812014-10-26 20:15:28 -04002457 Optional:
kelvin-onlabd3b64892015-01-20 13:26:24 -08002458 * jsonFormat: enable output formatting in json
Jeremy Songster306ed7a2016-07-19 10:59:07 -07002459 * noCore: suppress core flows
Shreya Shah0f01c812014-10-26 20:15:28 -04002460 Description:
Jon Halle3f39ff2015-01-13 11:50:53 -08002461 Obtain flows currently installed
kelvin-onlab898a6c62015-01-16 14:13:53 -08002462 """
Shreya Shah0f01c812014-10-26 20:15:28 -04002463 try:
Jon Hallc6358dd2015-04-10 12:44:28 -07002464 cmdStr = "flows"
kelvin-onlabd3b64892015-01-20 13:26:24 -08002465 if jsonFormat:
GlennRCed771242016-01-13 17:02:47 -08002466 cmdStr += " -j "
Jeremy Songster306ed7a2016-07-19 10:59:07 -07002467 if noCore:
2468 cmdStr += " -n "
GlennRCed771242016-01-13 17:02:47 -08002469 cmdStr += state
YPZhangebf9eb52016-05-12 15:20:24 -07002470 handle = self.sendline( cmdStr, timeout=timeout, noExit=noExit )
You Wangb5a55f72017-03-03 12:51:05 -08002471 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08002472 assert "Command not found:" not in handle, handle
2473 if re.search( "Error:", handle ):
2474 main.log.error( self.name + ": flows() response: " +
2475 str( handle ) )
2476 return handle
2477 except AssertionError:
2478 main.log.exception( "" )
GlennRCed771242016-01-13 17:02:47 -08002479 return None
Jon Halld4d4b372015-01-28 16:02:41 -08002480 except TypeError:
2481 main.log.exception( self.name + ": Object not as expected" )
2482 return None
Jon Hallc6793552016-01-19 14:18:37 -08002483 except pexpect.TIMEOUT:
2484 main.log.error( self.name + ": ONOS timeout" )
2485 return None
Shreya Shah0f01c812014-10-26 20:15:28 -04002486 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08002487 main.log.error( self.name + ": EOF exception found" )
2488 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002489 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08002490 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08002491 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002492 main.cleanAndExit()
Shreya Shah0f01c812014-10-26 20:15:28 -04002493
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002494 def checkFlowCount( self, min=0, timeout=60 ):
Flavio Castroa1286fe2016-07-25 14:48:51 -07002495 count = self.getTotalFlowsNum( timeout=timeout )
Jon Halle0f0b342017-04-18 11:43:47 -07002496 count = int( count ) if count else 0
2497 return count if ( count > min ) else False
GlennRCed771242016-01-13 17:02:47 -08002498
Jon Halle0f0b342017-04-18 11:43:47 -07002499 def checkFlowsState( self, isPENDING=True, timeout=60, noExit=False ):
kelvin-onlab4df89f22015-04-13 18:10:23 -07002500 """
2501 Description:
GlennRCed771242016-01-13 17:02:47 -08002502 Check the if all the current flows are in ADDED state
Jon Hallc6793552016-01-19 14:18:37 -08002503 We check PENDING_ADD, PENDING_REMOVE, REMOVED, and FAILED flows,
2504 if the count of those states is 0, which means all current flows
2505 are in ADDED state, and return main.TRUE otherwise return main.FALSE
pingping-linbab7f8a2015-09-21 17:33:36 -07002506 Optional:
GlennRCed771242016-01-13 17:02:47 -08002507 * isPENDING: whether the PENDING_ADD is also a correct status
kelvin-onlab4df89f22015-04-13 18:10:23 -07002508 Return:
2509 returnValue - Returns main.TRUE only if all flows are in
Jon Hallc6793552016-01-19 14:18:37 -08002510 ADDED state or PENDING_ADD if the isPENDING
pingping-linbab7f8a2015-09-21 17:33:36 -07002511 parameter is set true, return main.FALSE otherwise.
kelvin-onlab4df89f22015-04-13 18:10:23 -07002512 """
2513 try:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002514 states = [ "PENDING_ADD", "PENDING_REMOVE", "REMOVED", "FAILED" ]
GlennRCed771242016-01-13 17:02:47 -08002515 checkedStates = []
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002516 statesCount = [ 0, 0, 0, 0 ]
GlennRCed771242016-01-13 17:02:47 -08002517 for s in states:
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00002518 rawFlows = self.flows( state=s, timeout = timeout )
YPZhang240842b2016-05-17 12:00:50 -07002519 if rawFlows:
2520 # if we didn't get flows or flows function return None, we should return
2521 # main.Flase
2522 checkedStates.append( json.loads( rawFlows ) )
2523 else:
2524 return main.FALSE
Jon Hallc6793552016-01-19 14:18:37 -08002525 for i in range( len( states ) ):
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002526 for c in checkedStates[ i ]:
Jon Hallc6793552016-01-19 14:18:37 -08002527 try:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002528 statesCount[ i ] += int( c.get( "flowCount" ) )
Jon Hallc6793552016-01-19 14:18:37 -08002529 except TypeError:
2530 main.log.exception( "Json object not as expected" )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002531 main.log.info( states[ i ] + " flows: " + str( statesCount[ i ] ) )
kelvin-onlabf2ec6e02015-05-27 14:15:28 -07002532
GlennRCed771242016-01-13 17:02:47 -08002533 # We want to count PENDING_ADD if isPENDING is true
2534 if isPENDING:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002535 if statesCount[ 1 ] + statesCount[ 2 ] + statesCount[ 3 ] > 0:
GlennRCed771242016-01-13 17:02:47 -08002536 return main.FALSE
pingping-linbab7f8a2015-09-21 17:33:36 -07002537 else:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002538 if statesCount[ 0 ] + statesCount[ 1 ] + statesCount[ 2 ] + statesCount[ 3 ] > 0:
GlennRCed771242016-01-13 17:02:47 -08002539 return main.FALSE
GlennRCed771242016-01-13 17:02:47 -08002540 return main.TRUE
Jon Hallc6793552016-01-19 14:18:37 -08002541 except ( TypeError, ValueError ):
2542 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, rawFlows ) )
kelvin-onlab4df89f22015-04-13 18:10:23 -07002543 return None
Jeremy Songster9385d412016-06-02 17:57:36 -07002544
YPZhang240842b2016-05-17 12:00:50 -07002545 except AssertionError:
2546 main.log.exception( "" )
2547 return None
Jon Halle0f0b342017-04-18 11:43:47 -07002548 except pexpect.TIMEOUT:
2549 main.log.error( self.name + ": ONOS timeout" )
2550 return None
kelvin-onlab4df89f22015-04-13 18:10:23 -07002551 except pexpect.EOF:
2552 main.log.error( self.name + ": EOF exception found" )
2553 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002554 main.cleanAndExit()
kelvin-onlab4df89f22015-04-13 18:10:23 -07002555 except Exception:
2556 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002557 main.cleanAndExit()
kelvin-onlab4df89f22015-04-13 18:10:23 -07002558
GlennRCed771242016-01-13 17:02:47 -08002559 def pushTestIntents( self, ingress, egress, batchSize, offset="",
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00002560 options="", timeout=10, background = False, noExit=False, getResponse=False ):
kelvin8ec71442015-01-15 16:57:00 -08002561 """
andrewonlab87852b02014-11-19 18:44:19 -05002562 Description:
Jon Halle3f39ff2015-01-13 11:50:53 -08002563 Push a number of intents in a batch format to
andrewonlab87852b02014-11-19 18:44:19 -05002564 a specific point-to-point intent definition
2565 Required:
GlennRCed771242016-01-13 17:02:47 -08002566 * ingress: specify source dpid
2567 * egress: specify destination dpid
2568 * batchSize: specify number of intents to push
andrewonlab87852b02014-11-19 18:44:19 -05002569 Optional:
GlennRCed771242016-01-13 17:02:47 -08002570 * offset: the keyOffset is where the next batch of intents
2571 will be installed
YPZhangb34b7e12016-06-14 14:28:19 -07002572 * noExit: If set to True, TestON will not exit if any error when issus command
2573 * getResponse: If set to True, function will return ONOS response.
2574
GlennRCed771242016-01-13 17:02:47 -08002575 Returns: If failed to push test intents, it will returen None,
2576 if successful, return true.
2577 Timeout expection will return None,
2578 TypeError will return false
2579 other expections will exit()
kelvin8ec71442015-01-15 16:57:00 -08002580 """
andrewonlab87852b02014-11-19 18:44:19 -05002581 try:
GlennRCed771242016-01-13 17:02:47 -08002582 if background:
2583 back = "&"
andrewonlab87852b02014-11-19 18:44:19 -05002584 else:
GlennRCed771242016-01-13 17:02:47 -08002585 back = ""
2586 cmd = "push-test-intents {} {} {} {} {} {}".format( options,
Jon Hallc6793552016-01-19 14:18:37 -08002587 ingress,
2588 egress,
2589 batchSize,
2590 offset,
2591 back )
YPZhangebf9eb52016-05-12 15:20:24 -07002592 response = self.sendline( cmd, timeout=timeout, noExit=noExit )
You Wangb5a55f72017-03-03 12:51:05 -08002593 assert response is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08002594 assert "Command not found:" not in response, response
GlennRCed771242016-01-13 17:02:47 -08002595 main.log.info( response )
YPZhangb34b7e12016-06-14 14:28:19 -07002596 if getResponse:
2597 return response
2598
GlennRCed771242016-01-13 17:02:47 -08002599 # TODO: We should handle if there is failure in installation
2600 return main.TRUE
2601
Jon Hallc6793552016-01-19 14:18:37 -08002602 except AssertionError:
2603 main.log.exception( "" )
2604 return None
GlennRCed771242016-01-13 17:02:47 -08002605 except pexpect.TIMEOUT:
2606 main.log.error( self.name + ": ONOS timeout" )
Jon Halld4d4b372015-01-28 16:02:41 -08002607 return None
andrewonlab87852b02014-11-19 18:44:19 -05002608 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08002609 main.log.error( self.name + ": EOF exception found" )
2610 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002611 main.cleanAndExit()
GlennRCed771242016-01-13 17:02:47 -08002612 except TypeError:
2613 main.log.exception( self.name + ": Object not as expected" )
Jon Hallc6793552016-01-19 14:18:37 -08002614 return None
Jon Hallfebb1c72015-03-05 13:30:09 -08002615 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08002616 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002617 main.cleanAndExit()
andrewonlab87852b02014-11-19 18:44:19 -05002618
YPZhangebf9eb52016-05-12 15:20:24 -07002619 def getTotalFlowsNum( self, timeout=60, noExit=False ):
YPZhangb5d3f832016-01-23 22:54:26 -08002620 """
2621 Description:
YPZhangf6f14a02016-01-28 15:17:31 -08002622 Get the number of ADDED flows.
YPZhangb5d3f832016-01-23 22:54:26 -08002623 Return:
YPZhangf6f14a02016-01-28 15:17:31 -08002624 The number of ADDED flows
YPZhang14a4aa92016-07-15 13:37:15 -07002625 Or return None if any exceptions
YPZhangb5d3f832016-01-23 22:54:26 -08002626 """
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00002627
YPZhangb5d3f832016-01-23 22:54:26 -08002628 try:
YPZhange3109a72016-02-02 11:25:37 -08002629 # get total added flows number
YPZhang14a4aa92016-07-15 13:37:15 -07002630 cmd = "flows -c added"
2631 rawFlows = self.sendline( cmd, timeout=timeout, noExit=noExit )
2632 if rawFlows:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002633 rawFlows = rawFlows.split( "\n" )
YPZhange3109a72016-02-02 11:25:37 -08002634 totalFlows = 0
YPZhang14a4aa92016-07-15 13:37:15 -07002635 for l in rawFlows:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002636 totalFlows += int( l.split( "Count=" )[ 1 ] )
YPZhang14a4aa92016-07-15 13:37:15 -07002637 else:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002638 main.log.error( "Response not as expected!" )
YPZhang14a4aa92016-07-15 13:37:15 -07002639 return None
2640 return totalFlows
YPZhange3109a72016-02-02 11:25:37 -08002641
You Wangd3cb2ce2016-05-16 14:01:24 -07002642 except ( TypeError, ValueError ):
YPZhang14a4aa92016-07-15 13:37:15 -07002643 main.log.exception( "{}: Object not as expected!".format( self.name ) )
YPZhangb5d3f832016-01-23 22:54:26 -08002644 return None
2645 except pexpect.EOF:
2646 main.log.error( self.name + ": EOF exception found" )
2647 main.log.error( self.name + ": " + self.handle.before )
YPZhang14a4aa92016-07-15 13:37:15 -07002648 if not noExit:
Devin Lim44075962017-08-11 10:56:37 -07002649 main.cleanAndExit()
YPZhang14a4aa92016-07-15 13:37:15 -07002650 return None
Jon Halle0f0b342017-04-18 11:43:47 -07002651 except pexpect.TIMEOUT:
2652 main.log.error( self.name + ": ONOS timeout" )
2653 return None
YPZhangb5d3f832016-01-23 22:54:26 -08002654 except Exception:
2655 main.log.exception( self.name + ": Uncaught exception!" )
YPZhang14a4aa92016-07-15 13:37:15 -07002656 if not noExit:
Devin Lim44075962017-08-11 10:56:37 -07002657 main.cleanAndExit()
YPZhang14a4aa92016-07-15 13:37:15 -07002658 return None
YPZhangb5d3f832016-01-23 22:54:26 -08002659
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00002660 def getTotalIntentsNum( self, timeout=60, noExit = False ):
YPZhangb5d3f832016-01-23 22:54:26 -08002661 """
2662 Description:
2663 Get the total number of intents, include every states.
YPZhang14a4aa92016-07-15 13:37:15 -07002664 Optional:
2665 noExit - If noExit, TestON will not exit if any except.
YPZhangb5d3f832016-01-23 22:54:26 -08002666 Return:
2667 The number of intents
2668 """
2669 try:
2670 cmd = "summary -j"
YPZhang14a4aa92016-07-15 13:37:15 -07002671 response = self.sendline( cmd, timeout=timeout, noExit=noExit )
Jon Halle0f0b342017-04-18 11:43:47 -07002672 if response is None:
2673 return -1
YPZhangb5d3f832016-01-23 22:54:26 -08002674 response = json.loads( response )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002675 return int( response.get( "intents" ) )
You Wangd3cb2ce2016-05-16 14:01:24 -07002676 except ( TypeError, ValueError ):
2677 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, response ) )
YPZhangb5d3f832016-01-23 22:54:26 -08002678 return None
2679 except pexpect.EOF:
2680 main.log.error( self.name + ": EOF exception found" )
2681 main.log.error( self.name + ": " + self.handle.before )
YPZhang14a4aa92016-07-15 13:37:15 -07002682 if noExit:
2683 return -1
2684 else:
Devin Lim44075962017-08-11 10:56:37 -07002685 main.cleanAndExit()
YPZhangb5d3f832016-01-23 22:54:26 -08002686 except Exception:
2687 main.log.exception( self.name + ": Uncaught exception!" )
YPZhang14a4aa92016-07-15 13:37:15 -07002688 if noExit:
2689 return -1
2690 else:
Devin Lim44075962017-08-11 10:56:37 -07002691 main.cleanAndExit()
YPZhangb5d3f832016-01-23 22:54:26 -08002692
kelvin-onlabd3b64892015-01-20 13:26:24 -08002693 def intentsEventsMetrics( self, jsonFormat=True ):
kelvin8ec71442015-01-15 16:57:00 -08002694 """
Jon Halle3f39ff2015-01-13 11:50:53 -08002695 Description:Returns topology metrics
andrewonlab0dbb6ec2014-11-06 13:46:55 -05002696 Optional:
kelvin-onlabd3b64892015-01-20 13:26:24 -08002697 * jsonFormat: enable json formatting of output
kelvin8ec71442015-01-15 16:57:00 -08002698 """
andrewonlab0dbb6ec2014-11-06 13:46:55 -05002699 try:
Jon Hallc6358dd2015-04-10 12:44:28 -07002700 cmdStr = "intents-events-metrics"
kelvin-onlabd3b64892015-01-20 13:26:24 -08002701 if jsonFormat:
Jon Hallc6358dd2015-04-10 12:44:28 -07002702 cmdStr += " -j"
2703 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -08002704 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08002705 assert "Command not found:" not in handle, handle
andrewonlab0dbb6ec2014-11-06 13:46:55 -05002706 return handle
Jon Hallc6793552016-01-19 14:18:37 -08002707 except AssertionError:
2708 main.log.exception( "" )
2709 return None
Jon Halld4d4b372015-01-28 16:02:41 -08002710 except TypeError:
2711 main.log.exception( self.name + ": Object not as expected" )
2712 return None
andrewonlab0dbb6ec2014-11-06 13:46:55 -05002713 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08002714 main.log.error( self.name + ": EOF exception found" )
2715 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002716 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08002717 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08002718 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002719 main.cleanAndExit()
Shreya Shah0f01c812014-10-26 20:15:28 -04002720
kelvin-onlabd3b64892015-01-20 13:26:24 -08002721 def topologyEventsMetrics( self, jsonFormat=True ):
kelvin8ec71442015-01-15 16:57:00 -08002722 """
2723 Description:Returns topology metrics
andrewonlab867212a2014-10-22 20:13:38 -04002724 Optional:
kelvin-onlabd3b64892015-01-20 13:26:24 -08002725 * jsonFormat: enable json formatting of output
kelvin8ec71442015-01-15 16:57:00 -08002726 """
andrewonlab867212a2014-10-22 20:13:38 -04002727 try:
Jon Hallc6358dd2015-04-10 12:44:28 -07002728 cmdStr = "topology-events-metrics"
kelvin-onlabd3b64892015-01-20 13:26:24 -08002729 if jsonFormat:
Jon Hallc6358dd2015-04-10 12:44:28 -07002730 cmdStr += " -j"
2731 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -08002732 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08002733 assert "Command not found:" not in handle, handle
jenkins7ead5a82015-03-13 10:28:21 -07002734 if handle:
2735 return handle
Jon Hallc6358dd2015-04-10 12:44:28 -07002736 elif jsonFormat:
Jon Hallbe379602015-03-24 13:39:32 -07002737 # Return empty json
jenkins7ead5a82015-03-13 10:28:21 -07002738 return '{}'
Jon Hallc6358dd2015-04-10 12:44:28 -07002739 else:
2740 return handle
Jon Hallc6793552016-01-19 14:18:37 -08002741 except AssertionError:
2742 main.log.exception( "" )
2743 return None
Jon Halld4d4b372015-01-28 16:02:41 -08002744 except TypeError:
2745 main.log.exception( self.name + ": Object not as expected" )
2746 return None
andrewonlab867212a2014-10-22 20:13:38 -04002747 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08002748 main.log.error( self.name + ": EOF exception found" )
2749 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002750 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08002751 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08002752 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002753 main.cleanAndExit()
andrewonlab867212a2014-10-22 20:13:38 -04002754
kelvin8ec71442015-01-15 16:57:00 -08002755 # Wrapper functions ****************
2756 # Wrapper functions use existing driver
2757 # functions and extends their use case.
2758 # For example, we may use the output of
2759 # a normal driver function, and parse it
2760 # using a wrapper function
andrewonlabc2d05aa2014-10-13 16:51:10 -04002761
kelvin-onlabd3b64892015-01-20 13:26:24 -08002762 def getAllIntentsId( self ):
kelvin8ec71442015-01-15 16:57:00 -08002763 """
andrewonlab9a50dfe2014-10-17 17:22:31 -04002764 Description:
2765 Obtain all intent id's in a list
kelvin8ec71442015-01-15 16:57:00 -08002766 """
andrewonlab9a50dfe2014-10-17 17:22:31 -04002767 try:
kelvin8ec71442015-01-15 16:57:00 -08002768 # Obtain output of intents function
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002769 intentsStr = self.intents( jsonFormat=True )
Jon Hall7a6ebfd2017-03-13 10:58:58 -07002770 if intentsStr is None:
2771 raise TypeError
Jon Hall6021e062017-01-30 11:10:06 -08002772 # Convert to a dictionary
2773 intents = json.loads( intentsStr )
kelvin-onlabd3b64892015-01-20 13:26:24 -08002774 intentIdList = []
Jon Hall6021e062017-01-30 11:10:06 -08002775 for intent in intents:
2776 intentIdList.append( intent[ 'id' ] )
kelvin-onlabd3b64892015-01-20 13:26:24 -08002777 return intentIdList
Jon Halld4d4b372015-01-28 16:02:41 -08002778 except TypeError:
2779 main.log.exception( self.name + ": Object not as expected" )
2780 return None
andrewonlab9a50dfe2014-10-17 17:22:31 -04002781 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08002782 main.log.error( self.name + ": EOF exception found" )
2783 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002784 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08002785 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08002786 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002787 main.cleanAndExit()
andrewonlab9a50dfe2014-10-17 17:22:31 -04002788
You Wang3c276252016-09-21 15:21:36 -07002789 def flowAddedCount( self, deviceId, core=False ):
Jon Hall30b82fa2015-03-04 17:15:43 -08002790 """
2791 Determine the number of flow rules for the given device id that are
2792 in the added state
You Wang3c276252016-09-21 15:21:36 -07002793 Params:
2794 core: if True, only return the number of core flows added
Jon Hall30b82fa2015-03-04 17:15:43 -08002795 """
2796 try:
You Wang3c276252016-09-21 15:21:36 -07002797 if core:
2798 cmdStr = "flows any " + str( deviceId ) + " | " +\
2799 "grep 'state=ADDED' | grep org.onosproject.core | wc -l"
2800 else:
2801 cmdStr = "flows any " + str( deviceId ) + " | " +\
2802 "grep 'state=ADDED' | wc -l"
Jon Hall30b82fa2015-03-04 17:15:43 -08002803 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -08002804 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08002805 assert "Command not found:" not in handle, handle
Jon Hall30b82fa2015-03-04 17:15:43 -08002806 return handle
Jon Hallc6793552016-01-19 14:18:37 -08002807 except AssertionError:
2808 main.log.exception( "" )
2809 return None
Jon Hall30b82fa2015-03-04 17:15:43 -08002810 except pexpect.EOF:
2811 main.log.error( self.name + ": EOF exception found" )
2812 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002813 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08002814 except Exception:
Jon Hall30b82fa2015-03-04 17:15:43 -08002815 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002816 main.cleanAndExit()
andrewonlab95ce8322014-10-13 14:12:04 -04002817
Andreas Pantelopoulos9173d442018-03-01 17:07:37 -08002818 def groupAddedCount( self, deviceId, core=False ):
2819 """
2820 Determine the number of group rules for the given device id that are
2821 in the added state
2822 Params:
2823 core: if True, only return the number of core groups added
2824 """
2825 try:
2826 if core:
2827 cmdStr = "groups any " + str( deviceId ) + " | " +\
2828 "grep 'state=ADDED' | grep org.onosproject.core | wc -l"
2829 else:
2830 cmdStr = "groups any " + str( deviceId ) + " | " +\
2831 "grep 'state=ADDED' | wc -l"
2832 handle = self.sendline( cmdStr )
2833 assert handle is not None, "Error in sendline"
2834 assert "Command not found:" not in handle, handle
2835 return handle
2836 except AssertionError:
2837 main.log.exception( "" )
2838 return None
2839 except pexpect.EOF:
2840 main.log.error( self.name + ": EOF exception found" )
2841 main.log.error( self.name + ": " + self.handle.before )
2842 main.cleanAndExit()
2843 except Exception:
2844 main.log.exception( self.name + ": Uncaught exception!" )
2845 main.cleanAndExit()
2846
2847 def checkGroupAddedCount( self, deviceId, expectedGroupCount=0, core=False, comparison=0):
2848 """
2849 Description:
2850 Check whether the number of groups for the given device id that
2851 are in ADDED state is bigger than minGroupCount.
2852 Required:
2853 * deviceId: device id to check the number of added group rules
2854 Optional:
2855 * minGroupCount: the number of groups to compare
2856 * core: if True, only check the number of core groups added
2857 * comparison: if 0, compare with greater than minFlowCount
2858 * if 1, compare with equal to minFlowCount
2859 Return:
2860 Returns the number of groups if it is bigger than minGroupCount,
2861 returns main.FALSE otherwise.
2862 """
2863 count = self.groupAddedCount( deviceId, core )
2864 count = int( count ) if count else 0
2865 return count if ((count > expectedGroupCount) if (comparison == 0) else (count == expectedGroupCount)) else main.FALSE
2866
2867 def checkFlowAddedCount( self, deviceId, expectedFlowCount=0, core=False, comparison=0):
Jonghwan Hyuncf2345c2018-02-26 11:07:54 -08002868 """
2869 Description:
2870 Check whether the number of flow rules for the given device id that
2871 are in ADDED state is bigger than minFlowCount.
2872 Required:
2873 * deviceId: device id to check the number of added flow rules
2874 Optional:
2875 * minFlowCount: the number of flow rules to compare
2876 * core: if True, only check the number of core flows added
Andreas Pantelopoulos9173d442018-03-01 17:07:37 -08002877 * comparison: if 0, compare with greater than minFlowCount
2878 * if 1, compare with equal to minFlowCount
Jonghwan Hyuncf2345c2018-02-26 11:07:54 -08002879 Return:
2880 Returns the number of flow rules if it is bigger than minFlowCount,
2881 returns main.FALSE otherwise.
2882 """
2883 count = self.flowAddedCount( deviceId, core )
2884 count = int( count ) if count else 0
Andreas Pantelopoulos9173d442018-03-01 17:07:37 -08002885 return count if ((count > expectedFlowCount) if (comparison == 0) else (count == expectedFlowCount)) else main.FALSE
Jonghwan Hyuncf2345c2018-02-26 11:07:54 -08002886
kelvin-onlabd3b64892015-01-20 13:26:24 -08002887 def getAllDevicesId( self ):
kelvin8ec71442015-01-15 16:57:00 -08002888 """
andrewonlab7e4d2d32014-10-15 13:23:21 -04002889 Use 'devices' function to obtain list of all devices
2890 and parse the result to obtain a list of all device
2891 id's. Returns this list. Returns empty list if no
2892 devices exist
kelvin8ec71442015-01-15 16:57:00 -08002893 List is ordered sequentially
2894
andrewonlab3e15ead2014-10-15 14:21:34 -04002895 This function may be useful if you are not sure of the
kelvin8ec71442015-01-15 16:57:00 -08002896 device id, and wish to execute other commands using
andrewonlab3e15ead2014-10-15 14:21:34 -04002897 the ids. By obtaining the list of device ids on the fly,
2898 you can iterate through the list to get mastership, etc.
kelvin8ec71442015-01-15 16:57:00 -08002899 """
andrewonlab7e4d2d32014-10-15 13:23:21 -04002900 try:
kelvin8ec71442015-01-15 16:57:00 -08002901 # Call devices and store result string
kelvin-onlabd3b64892015-01-20 13:26:24 -08002902 devicesStr = self.devices( jsonFormat=False )
2903 idList = []
kelvin8ec71442015-01-15 16:57:00 -08002904
kelvin-onlabd3b64892015-01-20 13:26:24 -08002905 if not devicesStr:
kelvin8ec71442015-01-15 16:57:00 -08002906 main.log.info( "There are no devices to get id from" )
kelvin-onlabd3b64892015-01-20 13:26:24 -08002907 return idList
kelvin8ec71442015-01-15 16:57:00 -08002908
2909 # Split the string into list by comma
kelvin-onlabd3b64892015-01-20 13:26:24 -08002910 deviceList = devicesStr.split( "," )
kelvin8ec71442015-01-15 16:57:00 -08002911 # Get temporary list of all arguments with string 'id='
kelvin-onlabd3b64892015-01-20 13:26:24 -08002912 tempList = [ dev for dev in deviceList if "id=" in dev ]
kelvin8ec71442015-01-15 16:57:00 -08002913 # Split list further into arguments before and after string
2914 # 'id='. Get the latter portion ( the actual device id ) and
kelvin-onlabd3b64892015-01-20 13:26:24 -08002915 # append to idList
2916 for arg in tempList:
2917 idList.append( arg.split( "id=" )[ 1 ] )
2918 return idList
andrewonlab7e4d2d32014-10-15 13:23:21 -04002919
Jon Halld4d4b372015-01-28 16:02:41 -08002920 except TypeError:
2921 main.log.exception( self.name + ": Object not as expected" )
2922 return None
andrewonlab7e4d2d32014-10-15 13:23:21 -04002923 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08002924 main.log.error( self.name + ": EOF exception found" )
2925 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002926 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08002927 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08002928 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002929 main.cleanAndExit()
andrewonlab7e4d2d32014-10-15 13:23:21 -04002930
kelvin-onlabd3b64892015-01-20 13:26:24 -08002931 def getAllNodesId( self ):
kelvin8ec71442015-01-15 16:57:00 -08002932 """
andrewonlab7c211572014-10-15 16:45:20 -04002933 Uses 'nodes' function to obtain list of all nodes
2934 and parse the result of nodes to obtain just the
kelvin8ec71442015-01-15 16:57:00 -08002935 node id's.
andrewonlab7c211572014-10-15 16:45:20 -04002936 Returns:
2937 list of node id's
kelvin8ec71442015-01-15 16:57:00 -08002938 """
andrewonlab7c211572014-10-15 16:45:20 -04002939 try:
Jon Hall5aa168b2015-03-23 14:23:09 -07002940 nodesStr = self.nodes( jsonFormat=True )
kelvin-onlabd3b64892015-01-20 13:26:24 -08002941 idList = []
Jon Hall5aa168b2015-03-23 14:23:09 -07002942 # Sample nodesStr output
Jon Hallbd182782016-03-28 16:42:22 -07002943 # id=local, address=127.0.0.1:9876, state=READY *
kelvin-onlabd3b64892015-01-20 13:26:24 -08002944 if not nodesStr:
kelvin8ec71442015-01-15 16:57:00 -08002945 main.log.info( "There are no nodes to get id from" )
kelvin-onlabd3b64892015-01-20 13:26:24 -08002946 return idList
Jon Hall5aa168b2015-03-23 14:23:09 -07002947 nodesJson = json.loads( nodesStr )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07002948 idList = [ node.get( 'id' ) for node in nodesJson ]
kelvin-onlabd3b64892015-01-20 13:26:24 -08002949 return idList
Jon Hallc6793552016-01-19 14:18:37 -08002950 except ( TypeError, ValueError ):
2951 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, nodesStr ) )
Jon Halld4d4b372015-01-28 16:02:41 -08002952 return None
andrewonlab7c211572014-10-15 16:45:20 -04002953 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08002954 main.log.error( self.name + ": EOF exception found" )
2955 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002956 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08002957 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08002958 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002959 main.cleanAndExit()
andrewonlab7e4d2d32014-10-15 13:23:21 -04002960
kelvin-onlabd3b64892015-01-20 13:26:24 -08002961 def getDevice( self, dpid=None ):
kelvin8ec71442015-01-15 16:57:00 -08002962 """
Jon Halla91c4dc2014-10-22 12:57:04 -04002963 Return the first device from the devices api whose 'id' contains 'dpid'
2964 Return None if there is no match
kelvin8ec71442015-01-15 16:57:00 -08002965 """
Jon Halla91c4dc2014-10-22 12:57:04 -04002966 try:
kelvin8ec71442015-01-15 16:57:00 -08002967 if dpid is None:
Jon Halla91c4dc2014-10-22 12:57:04 -04002968 return None
2969 else:
kelvin8ec71442015-01-15 16:57:00 -08002970 dpid = dpid.replace( ':', '' )
kelvin-onlabd3b64892015-01-20 13:26:24 -08002971 rawDevices = self.devices()
2972 devicesJson = json.loads( rawDevices )
kelvin8ec71442015-01-15 16:57:00 -08002973 # search json for the device with dpid then return the device
kelvin-onlabd3b64892015-01-20 13:26:24 -08002974 for device in devicesJson:
kelvin8ec71442015-01-15 16:57:00 -08002975 # print "%s in %s?" % ( dpid, device[ 'id' ] )
2976 if dpid in device[ 'id' ]:
Jon Halla91c4dc2014-10-22 12:57:04 -04002977 return device
2978 return None
Jon Hallc6793552016-01-19 14:18:37 -08002979 except ( TypeError, ValueError ):
2980 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, rawDevices ) )
Jon Halld4d4b372015-01-28 16:02:41 -08002981 return None
Jon Halla91c4dc2014-10-22 12:57:04 -04002982 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08002983 main.log.error( self.name + ": EOF exception found" )
2984 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07002985 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08002986 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08002987 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07002988 main.cleanAndExit()
Jon Halla91c4dc2014-10-22 12:57:04 -04002989
You Wang24139872016-05-03 11:48:47 -07002990 def getTopology( self, topologyOutput ):
2991 """
2992 Definition:
2993 Loads a json topology output
2994 Return:
2995 topology = current ONOS topology
2996 """
2997 import json
2998 try:
2999 # either onos:topology or 'topology' will work in CLI
Jeremy Ronquillo82705492017-10-18 14:19:55 -07003000 topology = json.loads( topologyOutput )
Jeremy Songsterbc2d8ac2016-05-04 11:25:42 -07003001 main.log.debug( topology )
You Wang24139872016-05-03 11:48:47 -07003002 return topology
You Wangd3cb2ce2016-05-16 14:01:24 -07003003 except ( TypeError, ValueError ):
3004 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, topologyOutput ) )
3005 return None
You Wang24139872016-05-03 11:48:47 -07003006 except pexpect.EOF:
3007 main.log.error( self.name + ": EOF exception found" )
3008 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003009 main.cleanAndExit()
You Wang24139872016-05-03 11:48:47 -07003010 except Exception:
3011 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003012 main.cleanAndExit()
You Wang24139872016-05-03 11:48:47 -07003013
Jeremy Ronquillo82705492017-10-18 14:19:55 -07003014 def checkStatus( self, numoswitch, numolink, numoctrl = -1, logLevel="info" ):
kelvin8ec71442015-01-15 16:57:00 -08003015 """
Jon Hallefbd9792015-03-05 16:11:36 -08003016 Checks the number of switches & links that ONOS sees against the
kelvin8ec71442015-01-15 16:57:00 -08003017 supplied values. By default this will report to main.log, but the
You Wang24139872016-05-03 11:48:47 -07003018 log level can be specific.
kelvin8ec71442015-01-15 16:57:00 -08003019
Flavio Castro82ee2f62016-06-07 15:04:12 -07003020 Params: numoswitch = expected number of switches
Jon Hallefbd9792015-03-05 16:11:36 -08003021 numolink = expected number of links
Flavio Castro82ee2f62016-06-07 15:04:12 -07003022 numoctrl = expected number of controllers
You Wang24139872016-05-03 11:48:47 -07003023 logLevel = level to log to.
3024 Currently accepts 'info', 'warn' and 'report'
Jon Hall42db6dc2014-10-24 19:03:48 -04003025
Jon Hallefbd9792015-03-05 16:11:36 -08003026 Returns: main.TRUE if the number of switches and links are correct,
3027 main.FALSE if the number of switches and links is incorrect,
Jon Hall42db6dc2014-10-24 19:03:48 -04003028 and main.ERROR otherwise
kelvin8ec71442015-01-15 16:57:00 -08003029 """
Flavio Castro82ee2f62016-06-07 15:04:12 -07003030 import json
Jon Hall42db6dc2014-10-24 19:03:48 -04003031 try:
You Wang13310252016-07-31 10:56:14 -07003032 summary = self.summary()
3033 summary = json.loads( summary )
Flavio Castrof5b3f872016-06-23 17:52:31 -07003034 except ( TypeError, ValueError ):
3035 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, summary ) )
3036 return main.ERROR
3037 try:
3038 topology = self.getTopology( self.topology() )
Jon Halle0f0b342017-04-18 11:43:47 -07003039 if topology == {} or topology is None or summary == {} or summary is None:
Jon Hall42db6dc2014-10-24 19:03:48 -04003040 return main.ERROR
3041 output = ""
kelvin8ec71442015-01-15 16:57:00 -08003042 # Is the number of switches is what we expected
3043 devices = topology.get( 'devices', False )
3044 links = topology.get( 'links', False )
Flavio Castro82ee2f62016-06-07 15:04:12 -07003045 nodes = summary.get( 'nodes', False )
3046 if devices is False or links is False or nodes is False:
Jon Hall42db6dc2014-10-24 19:03:48 -04003047 return main.ERROR
kelvin-onlabd3b64892015-01-20 13:26:24 -08003048 switchCheck = ( int( devices ) == int( numoswitch ) )
kelvin8ec71442015-01-15 16:57:00 -08003049 # Is the number of links is what we expected
kelvin-onlabd3b64892015-01-20 13:26:24 -08003050 linkCheck = ( int( links ) == int( numolink ) )
Flavio Castro82ee2f62016-06-07 15:04:12 -07003051 nodeCheck = ( int( nodes ) == int( numoctrl ) ) or int( numoctrl ) == -1
3052 if switchCheck and linkCheck and nodeCheck:
kelvin8ec71442015-01-15 16:57:00 -08003053 # We expected the correct numbers
You Wang24139872016-05-03 11:48:47 -07003054 output = output + "The number of links and switches match "\
3055 + "what was expected"
Jon Hall42db6dc2014-10-24 19:03:48 -04003056 result = main.TRUE
3057 else:
You Wang24139872016-05-03 11:48:47 -07003058 output = output + \
3059 "The number of links and switches does not match " + \
3060 "what was expected"
Jon Hall42db6dc2014-10-24 19:03:48 -04003061 result = main.FALSE
You Wang24139872016-05-03 11:48:47 -07003062 output = output + "\n ONOS sees %i devices" % int( devices )
3063 output = output + " (%i expected) " % int( numoswitch )
3064 output = output + "and %i links " % int( links )
3065 output = output + "(%i expected)" % int( numolink )
YPZhangd7e4b6e2016-06-17 16:07:55 -07003066 if int( numoctrl ) > 0:
Flavio Castro82ee2f62016-06-07 15:04:12 -07003067 output = output + "and %i controllers " % int( nodes )
3068 output = output + "(%i expected)" % int( numoctrl )
kelvin-onlabd3b64892015-01-20 13:26:24 -08003069 if logLevel == "report":
kelvin8ec71442015-01-15 16:57:00 -08003070 main.log.report( output )
kelvin-onlabd3b64892015-01-20 13:26:24 -08003071 elif logLevel == "warn":
kelvin8ec71442015-01-15 16:57:00 -08003072 main.log.warn( output )
Jon Hall42db6dc2014-10-24 19:03:48 -04003073 else:
You Wang24139872016-05-03 11:48:47 -07003074 main.log.info( output )
kelvin8ec71442015-01-15 16:57:00 -08003075 return result
Jon Hall42db6dc2014-10-24 19:03:48 -04003076 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08003077 main.log.error( self.name + ": EOF exception found" )
3078 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003079 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08003080 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08003081 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003082 main.cleanAndExit()
Jon Hall1c9e8732014-10-27 19:29:27 -04003083
kelvin-onlabd3b64892015-01-20 13:26:24 -08003084 def deviceRole( self, deviceId, onosNode, role="master" ):
kelvin8ec71442015-01-15 16:57:00 -08003085 """
Jon Hall1c9e8732014-10-27 19:29:27 -04003086 Calls the device-role cli command.
kelvin-onlabd3b64892015-01-20 13:26:24 -08003087 deviceId must be the id of a device as seen in the onos devices command
3088 onosNode is the ip of one of the onos nodes in the cluster
Jon Hall1c9e8732014-10-27 19:29:27 -04003089 role must be either master, standby, or none
3090
Jon Halle3f39ff2015-01-13 11:50:53 -08003091 Returns:
3092 main.TRUE or main.FALSE based on argument verification and
3093 main.ERROR if command returns and error
kelvin-onlab898a6c62015-01-16 14:13:53 -08003094 """
Jon Hall1c9e8732014-10-27 19:29:27 -04003095 try:
Jon Halle3f39ff2015-01-13 11:50:53 -08003096 if role.lower() == "master" or role.lower() == "standby" or\
Jon Hall1c9e8732014-10-27 19:29:27 -04003097 role.lower() == "none":
kelvin-onlabd3b64892015-01-20 13:26:24 -08003098 cmdStr = "device-role " +\
3099 str( deviceId ) + " " +\
3100 str( onosNode ) + " " +\
kelvin-onlab898a6c62015-01-16 14:13:53 -08003101 str( role )
kelvin-onlabd3b64892015-01-20 13:26:24 -08003102 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -08003103 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08003104 assert "Command not found:" not in handle, handle
kelvin-onlab898a6c62015-01-16 14:13:53 -08003105 if re.search( "Error", handle ):
3106 # end color output to escape any colours
3107 # from the cli
kelvin8ec71442015-01-15 16:57:00 -08003108 main.log.error( self.name + ": " +
kelvin-onlab898a6c62015-01-16 14:13:53 -08003109 handle + '\033[0m' )
kelvin8ec71442015-01-15 16:57:00 -08003110 return main.ERROR
kelvin8ec71442015-01-15 16:57:00 -08003111 return main.TRUE
Jon Hall1c9e8732014-10-27 19:29:27 -04003112 else:
kelvin-onlab898a6c62015-01-16 14:13:53 -08003113 main.log.error( "Invalid 'role' given to device_role(). " +
Jeremy Ronquillo82705492017-10-18 14:19:55 -07003114 "Value was '" + str( role ) + "'." )
Jon Hall1c9e8732014-10-27 19:29:27 -04003115 return main.FALSE
Jon Hallc6793552016-01-19 14:18:37 -08003116 except AssertionError:
3117 main.log.exception( "" )
3118 return None
Jon Halld4d4b372015-01-28 16:02:41 -08003119 except TypeError:
3120 main.log.exception( self.name + ": Object not as expected" )
3121 return None
Jon Hall1c9e8732014-10-27 19:29:27 -04003122 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08003123 main.log.error( self.name + ": EOF exception found" )
3124 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003125 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08003126 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08003127 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003128 main.cleanAndExit()
Jon Hall1c9e8732014-10-27 19:29:27 -04003129
kelvin-onlabd3b64892015-01-20 13:26:24 -08003130 def clusters( self, jsonFormat=True ):
kelvin8ec71442015-01-15 16:57:00 -08003131 """
Jon Hall73cf9cc2014-11-20 22:28:38 -08003132 Lists all clusters
Jon Hallffb386d2014-11-21 13:43:38 -08003133 Optional argument:
kelvin-onlabd3b64892015-01-20 13:26:24 -08003134 * jsonFormat - boolean indicating if you want output in json
kelvin8ec71442015-01-15 16:57:00 -08003135 """
Jon Hall73cf9cc2014-11-20 22:28:38 -08003136 try:
Jon Hallc6358dd2015-04-10 12:44:28 -07003137 cmdStr = "clusters"
kelvin-onlabd3b64892015-01-20 13:26:24 -08003138 if jsonFormat:
Jon Hallc6358dd2015-04-10 12:44:28 -07003139 cmdStr += " -j"
3140 handle = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -08003141 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08003142 assert "Command not found:" not in handle, handle
Jon Hallc6358dd2015-04-10 12:44:28 -07003143 return handle
Jon Hallc6793552016-01-19 14:18:37 -08003144 except AssertionError:
3145 main.log.exception( "" )
3146 return None
Jon Halld4d4b372015-01-28 16:02:41 -08003147 except TypeError:
3148 main.log.exception( self.name + ": Object not as expected" )
3149 return None
Jon Hall73cf9cc2014-11-20 22:28:38 -08003150 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08003151 main.log.error( self.name + ": EOF exception found" )
3152 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003153 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08003154 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08003155 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003156 main.cleanAndExit()
Jon Hall73cf9cc2014-11-20 22:28:38 -08003157
kelvin-onlabd3b64892015-01-20 13:26:24 -08003158 def electionTestLeader( self ):
kelvin-onlab898a6c62015-01-16 14:13:53 -08003159 """
Jon Halle3f39ff2015-01-13 11:50:53 -08003160 CLI command to get the current leader for the Election test application
3161 NOTE: Requires installation of the onos-app-election feature
3162 Returns: Node IP of the leader if one exists
3163 None if none exists
3164 Main.FALSE on error
kelvin-onlab898a6c62015-01-16 14:13:53 -08003165 """
Jon Hall94fd0472014-12-08 11:52:42 -08003166 try:
kelvin-onlabd3b64892015-01-20 13:26:24 -08003167 cmdStr = "election-test-leader"
3168 response = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -08003169 assert response is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08003170 assert "Command not found:" not in response, response
Jon Halle3f39ff2015-01-13 11:50:53 -08003171 # Leader
3172 leaderPattern = "The\scurrent\sleader\sfor\sthe\sElection\s" +\
kelvin-onlab898a6c62015-01-16 14:13:53 -08003173 "app\sis\s(?P<node>.+)\."
kelvin-onlabd3b64892015-01-20 13:26:24 -08003174 nodeSearch = re.search( leaderPattern, response )
3175 if nodeSearch:
3176 node = nodeSearch.group( 'node' )
Jon Halle3f39ff2015-01-13 11:50:53 -08003177 main.log.info( "Election-test-leader on " + str( self.name ) +
kelvin-onlab898a6c62015-01-16 14:13:53 -08003178 " found " + node + " as the leader" )
Jon Hall94fd0472014-12-08 11:52:42 -08003179 return node
Jon Halle3f39ff2015-01-13 11:50:53 -08003180 # no leader
3181 nullPattern = "There\sis\scurrently\sno\sleader\selected\sfor\s" +\
kelvin-onlab898a6c62015-01-16 14:13:53 -08003182 "the\sElection\sapp"
kelvin-onlabd3b64892015-01-20 13:26:24 -08003183 nullSearch = re.search( nullPattern, response )
3184 if nullSearch:
Jon Halle3f39ff2015-01-13 11:50:53 -08003185 main.log.info( "Election-test-leader found no leader on " +
kelvin-onlab898a6c62015-01-16 14:13:53 -08003186 self.name )
Jon Hall94fd0472014-12-08 11:52:42 -08003187 return None
kelvin-onlab898a6c62015-01-16 14:13:53 -08003188 # error
Jon Hall97cf84a2016-06-20 13:35:58 -07003189 main.log.error( "Error in electionTestLeader on " + self.name +
3190 ": " + "unexpected response" )
3191 main.log.error( repr( response ) )
3192 return main.FALSE
Jon Hallc6793552016-01-19 14:18:37 -08003193 except AssertionError:
3194 main.log.exception( "" )
3195 return None
Jon Halld4d4b372015-01-28 16:02:41 -08003196 except TypeError:
3197 main.log.exception( self.name + ": Object not as expected" )
3198 return main.FALSE
Jon Hall94fd0472014-12-08 11:52:42 -08003199 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08003200 main.log.error( self.name + ": EOF exception found" )
3201 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003202 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08003203 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08003204 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003205 main.cleanAndExit()
Jon Hall94fd0472014-12-08 11:52:42 -08003206
kelvin-onlabd3b64892015-01-20 13:26:24 -08003207 def electionTestRun( self ):
kelvin-onlab898a6c62015-01-16 14:13:53 -08003208 """
Jon Halle3f39ff2015-01-13 11:50:53 -08003209 CLI command to run for leadership of the Election test application.
3210 NOTE: Requires installation of the onos-app-election feature
3211 Returns: Main.TRUE on success
3212 Main.FALSE on error
kelvin-onlab898a6c62015-01-16 14:13:53 -08003213 """
Jon Hall94fd0472014-12-08 11:52:42 -08003214 try:
kelvin-onlabd3b64892015-01-20 13:26:24 -08003215 cmdStr = "election-test-run"
3216 response = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -08003217 assert response is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08003218 assert "Command not found:" not in response, response
kelvin-onlab898a6c62015-01-16 14:13:53 -08003219 # success
Jon Halle3f39ff2015-01-13 11:50:53 -08003220 successPattern = "Entering\sleadership\selections\sfor\sthe\s" +\
kelvin-onlab898a6c62015-01-16 14:13:53 -08003221 "Election\sapp."
Jon Halle3f39ff2015-01-13 11:50:53 -08003222 search = re.search( successPattern, response )
Jon Hall94fd0472014-12-08 11:52:42 -08003223 if search:
Jon Halle3f39ff2015-01-13 11:50:53 -08003224 main.log.info( self.name + " entering leadership elections " +
kelvin-onlab898a6c62015-01-16 14:13:53 -08003225 "for the Election app." )
Jon Hall94fd0472014-12-08 11:52:42 -08003226 return main.TRUE
kelvin-onlab898a6c62015-01-16 14:13:53 -08003227 # error
Jon Hall97cf84a2016-06-20 13:35:58 -07003228 main.log.error( "Error in electionTestRun on " + self.name +
3229 ": " + "unexpected response" )
3230 main.log.error( repr( response ) )
3231 return main.FALSE
Jon Hallc6793552016-01-19 14:18:37 -08003232 except AssertionError:
3233 main.log.exception( "" )
3234 return None
Jon Halld4d4b372015-01-28 16:02:41 -08003235 except TypeError:
3236 main.log.exception( self.name + ": Object not as expected" )
3237 return main.FALSE
Jon Hall94fd0472014-12-08 11:52:42 -08003238 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08003239 main.log.error( self.name + ": EOF exception found" )
3240 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003241 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08003242 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08003243 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003244 main.cleanAndExit()
Jon Hall94fd0472014-12-08 11:52:42 -08003245
kelvin-onlabd3b64892015-01-20 13:26:24 -08003246 def electionTestWithdraw( self ):
kelvin8ec71442015-01-15 16:57:00 -08003247 """
Jon Hall94fd0472014-12-08 11:52:42 -08003248 * CLI command to withdraw the local node from leadership election for
3249 * the Election test application.
3250 #NOTE: Requires installation of the onos-app-election feature
3251 Returns: Main.TRUE on success
3252 Main.FALSE on error
kelvin8ec71442015-01-15 16:57:00 -08003253 """
Jon Hall94fd0472014-12-08 11:52:42 -08003254 try:
kelvin-onlabd3b64892015-01-20 13:26:24 -08003255 cmdStr = "election-test-withdraw"
3256 response = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -08003257 assert response is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08003258 assert "Command not found:" not in response, response
kelvin-onlab898a6c62015-01-16 14:13:53 -08003259 # success
Jon Halle3f39ff2015-01-13 11:50:53 -08003260 successPattern = "Withdrawing\sfrom\sleadership\selections\sfor" +\
kelvin-onlab898a6c62015-01-16 14:13:53 -08003261 "\sthe\sElection\sapp."
Jon Halle3f39ff2015-01-13 11:50:53 -08003262 if re.search( successPattern, response ):
3263 main.log.info( self.name + " withdrawing from leadership " +
kelvin-onlab898a6c62015-01-16 14:13:53 -08003264 "elections for the Election app." )
Jon Hall94fd0472014-12-08 11:52:42 -08003265 return main.TRUE
kelvin-onlab898a6c62015-01-16 14:13:53 -08003266 # error
Jon Hall97cf84a2016-06-20 13:35:58 -07003267 main.log.error( "Error in electionTestWithdraw on " +
3268 self.name + ": " + "unexpected response" )
3269 main.log.error( repr( response ) )
3270 return main.FALSE
Jon Hallc6793552016-01-19 14:18:37 -08003271 except AssertionError:
3272 main.log.exception( "" )
3273 return None
Jon Halld4d4b372015-01-28 16:02:41 -08003274 except TypeError:
3275 main.log.exception( self.name + ": Object not as expected" )
3276 return main.FALSE
Jon Hall94fd0472014-12-08 11:52:42 -08003277 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08003278 main.log.error( self.name + ": EOF exception found" )
3279 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003280 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08003281 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08003282 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003283 main.cleanAndExit()
Jon Hall1c9e8732014-10-27 19:29:27 -04003284
kelvin8ec71442015-01-15 16:57:00 -08003285 def getDevicePortsEnabledCount( self, dpid ):
3286 """
Hari Krishnaa43d4e92014-12-19 13:22:40 -08003287 Get the count of all enabled ports on a particular device/switch
kelvin8ec71442015-01-15 16:57:00 -08003288 """
Hari Krishnaa43d4e92014-12-19 13:22:40 -08003289 try:
Jon Halle3f39ff2015-01-13 11:50:53 -08003290 dpid = str( dpid )
kelvin-onlabd3b64892015-01-20 13:26:24 -08003291 cmdStr = "onos:ports -e " + dpid + " | wc -l"
3292 output = self.sendline( cmdStr )
Jon Halla495f562016-05-16 18:03:26 -07003293 assert output is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08003294 assert "Command not found:" not in output, output
Jon Halle3f39ff2015-01-13 11:50:53 -08003295 if re.search( "No such device", output ):
3296 main.log.error( "Error in getting ports" )
3297 return ( output, "Error" )
Jon Halla495f562016-05-16 18:03:26 -07003298 return output
Jon Hallc6793552016-01-19 14:18:37 -08003299 except AssertionError:
3300 main.log.exception( "" )
3301 return None
Jon Halld4d4b372015-01-28 16:02:41 -08003302 except TypeError:
3303 main.log.exception( self.name + ": Object not as expected" )
3304 return ( output, "Error" )
Hari Krishnaa43d4e92014-12-19 13:22:40 -08003305 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08003306 main.log.error( self.name + ": EOF exception found" )
3307 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003308 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08003309 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08003310 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003311 main.cleanAndExit()
Hari Krishnaa43d4e92014-12-19 13:22:40 -08003312
kelvin8ec71442015-01-15 16:57:00 -08003313 def getDeviceLinksActiveCount( self, dpid ):
3314 """
Hari Krishnaa43d4e92014-12-19 13:22:40 -08003315 Get the count of all enabled ports on a particular device/switch
kelvin8ec71442015-01-15 16:57:00 -08003316 """
Hari Krishnaa43d4e92014-12-19 13:22:40 -08003317 try:
kelvin-onlab898a6c62015-01-16 14:13:53 -08003318 dpid = str( dpid )
kelvin-onlabd3b64892015-01-20 13:26:24 -08003319 cmdStr = "onos:links " + dpid + " | grep ACTIVE | wc -l"
3320 output = self.sendline( cmdStr )
Jon Halla495f562016-05-16 18:03:26 -07003321 assert output is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08003322 assert "Command not found:" not in output, output
Jon Halle3f39ff2015-01-13 11:50:53 -08003323 if re.search( "No such device", output ):
kelvin-onlab898a6c62015-01-16 14:13:53 -08003324 main.log.error( "Error in getting ports " )
3325 return ( output, "Error " )
Jon Halla495f562016-05-16 18:03:26 -07003326 return output
Jon Hallc6793552016-01-19 14:18:37 -08003327 except AssertionError:
3328 main.log.exception( "" )
3329 return None
Jon Halld4d4b372015-01-28 16:02:41 -08003330 except TypeError:
3331 main.log.exception( self.name + ": Object not as expected" )
3332 return ( output, "Error " )
Hari Krishnaa43d4e92014-12-19 13:22:40 -08003333 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08003334 main.log.error( self.name + ": EOF exception found" )
3335 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003336 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08003337 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08003338 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003339 main.cleanAndExit()
Hari Krishnaa43d4e92014-12-19 13:22:40 -08003340
kelvin8ec71442015-01-15 16:57:00 -08003341 def getAllIntentIds( self ):
3342 """
Hari Krishnaa43d4e92014-12-19 13:22:40 -08003343 Return a list of all Intent IDs
kelvin8ec71442015-01-15 16:57:00 -08003344 """
Hari Krishnaa43d4e92014-12-19 13:22:40 -08003345 try:
kelvin-onlabd3b64892015-01-20 13:26:24 -08003346 cmdStr = "onos:intents | grep id="
3347 output = self.sendline( cmdStr )
Jon Halla495f562016-05-16 18:03:26 -07003348 assert output is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08003349 assert "Command not found:" not in output, output
Jon Halle3f39ff2015-01-13 11:50:53 -08003350 if re.search( "Error", output ):
3351 main.log.error( "Error in getting ports" )
3352 return ( output, "Error" )
Jon Halla495f562016-05-16 18:03:26 -07003353 return output
Jon Hallc6793552016-01-19 14:18:37 -08003354 except AssertionError:
3355 main.log.exception( "" )
3356 return None
Jon Halld4d4b372015-01-28 16:02:41 -08003357 except TypeError:
3358 main.log.exception( self.name + ": Object not as expected" )
3359 return ( output, "Error" )
Hari Krishnaa43d4e92014-12-19 13:22:40 -08003360 except pexpect.EOF:
kelvin8ec71442015-01-15 16:57:00 -08003361 main.log.error( self.name + ": EOF exception found" )
3362 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003363 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08003364 except Exception:
Jon Halld4d4b372015-01-28 16:02:41 -08003365 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003366 main.cleanAndExit()
Jon Halld4d4b372015-01-28 16:02:41 -08003367
Jon Hall73509952015-02-24 16:42:56 -08003368 def intentSummary( self ):
3369 """
Jon Hallefbd9792015-03-05 16:11:36 -08003370 Returns a dictionary containing the current intent states and the count
Jon Hall73509952015-02-24 16:42:56 -08003371 """
3372 try:
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00003373 intents = self.intents( )
Jon Hall08f61bc2015-04-13 16:00:30 -07003374 states = []
Jon Hall5aa168b2015-03-23 14:23:09 -07003375 for intent in json.loads( intents ):
Jon Hall08f61bc2015-04-13 16:00:30 -07003376 states.append( intent.get( 'state', None ) )
3377 out = [ ( i, states.count( i ) ) for i in set( states ) ]
Jon Hall63604932015-02-26 17:09:50 -08003378 main.log.info( dict( out ) )
Jon Hall73509952015-02-24 16:42:56 -08003379 return dict( out )
Jon Hallc6793552016-01-19 14:18:37 -08003380 except ( TypeError, ValueError ):
3381 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, intents ) )
Jon Hall73509952015-02-24 16:42:56 -08003382 return None
3383 except pexpect.EOF:
3384 main.log.error( self.name + ": EOF exception found" )
3385 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003386 main.cleanAndExit()
Jon Hallfebb1c72015-03-05 13:30:09 -08003387 except Exception:
Jon Hall73509952015-02-24 16:42:56 -08003388 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003389 main.cleanAndExit()
Jon Hall63604932015-02-26 17:09:50 -08003390
Jon Hall61282e32015-03-19 11:34:11 -07003391 def leaders( self, jsonFormat=True ):
Jon Hall63604932015-02-26 17:09:50 -08003392 """
3393 Returns the output of the leaders command.
Jon Hall61282e32015-03-19 11:34:11 -07003394 Optional argument:
3395 * jsonFormat - boolean indicating if you want output in json
Jon Hall63604932015-02-26 17:09:50 -08003396 """
Jon Hall63604932015-02-26 17:09:50 -08003397 try:
Jon Hallc6358dd2015-04-10 12:44:28 -07003398 cmdStr = "onos:leaders"
Jon Hall61282e32015-03-19 11:34:11 -07003399 if jsonFormat:
Jon Hallc6358dd2015-04-10 12:44:28 -07003400 cmdStr += " -j"
3401 output = self.sendline( cmdStr )
Jon Halla495f562016-05-16 18:03:26 -07003402 assert output is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08003403 assert "Command not found:" not in output, output
Jon Hallc6358dd2015-04-10 12:44:28 -07003404 return output
Jon Hallc6793552016-01-19 14:18:37 -08003405 except AssertionError:
3406 main.log.exception( "" )
3407 return None
Jon Hall63604932015-02-26 17:09:50 -08003408 except TypeError:
3409 main.log.exception( self.name + ": Object not as expected" )
3410 return None
Hari Krishnaa43d4e92014-12-19 13:22:40 -08003411 except pexpect.EOF:
3412 main.log.error( self.name + ": EOF exception found" )
3413 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003414 main.cleanAndExit()
Jon Hall77ba41c2015-04-06 10:25:40 -07003415 except Exception:
Jon Hall63604932015-02-26 17:09:50 -08003416 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003417 main.cleanAndExit()
Jon Hall63604932015-02-26 17:09:50 -08003418
acsmarsa4a4d1e2015-07-10 16:01:24 -07003419 def leaderCandidates( self, jsonFormat=True ):
3420 """
3421 Returns the output of the leaders -c command.
3422 Optional argument:
3423 * jsonFormat - boolean indicating if you want output in json
3424 """
3425 try:
3426 cmdStr = "onos:leaders -c"
3427 if jsonFormat:
3428 cmdStr += " -j"
3429 output = self.sendline( cmdStr )
Jon Halla495f562016-05-16 18:03:26 -07003430 assert output is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08003431 assert "Command not found:" not in output, output
acsmarsa4a4d1e2015-07-10 16:01:24 -07003432 return output
Jon Hallc6793552016-01-19 14:18:37 -08003433 except AssertionError:
3434 main.log.exception( "" )
3435 return None
acsmarsa4a4d1e2015-07-10 16:01:24 -07003436 except TypeError:
3437 main.log.exception( self.name + ": Object not as expected" )
3438 return None
3439 except pexpect.EOF:
3440 main.log.error( self.name + ": EOF exception found" )
3441 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003442 main.cleanAndExit()
acsmarsa4a4d1e2015-07-10 16:01:24 -07003443 except Exception:
3444 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003445 main.cleanAndExit()
acsmarsa4a4d1e2015-07-10 16:01:24 -07003446
Jon Hallc6793552016-01-19 14:18:37 -08003447 def specificLeaderCandidate( self, topic ):
acsmarsa4a4d1e2015-07-10 16:01:24 -07003448 """
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00003449 Returns a list in format [leader,candidate1,candidate2,...] for a given
acsmarsa4a4d1e2015-07-10 16:01:24 -07003450 topic parameter and an empty list if the topic doesn't exist
3451 If no leader is elected leader in the returned list will be "none"
3452 Returns None if there is a type error processing the json object
3453 """
3454 try:
Jon Hall6e709752016-02-01 13:38:46 -08003455 cmdStr = "onos:leaders -j"
Jon Hallc6793552016-01-19 14:18:37 -08003456 rawOutput = self.sendline( cmdStr )
Jon Halla495f562016-05-16 18:03:26 -07003457 assert rawOutput is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08003458 assert "Command not found:" not in rawOutput, rawOutput
3459 output = json.loads( rawOutput )
acsmarsa4a4d1e2015-07-10 16:01:24 -07003460 results = []
3461 for dict in output:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07003462 if dict[ "topic" ] == topic:
3463 leader = dict[ "leader" ]
3464 candidates = re.split( ", ", dict[ "candidates" ][ 1:-1 ] )
Jon Hallc6793552016-01-19 14:18:37 -08003465 results.append( leader )
3466 results.extend( candidates )
acsmarsa4a4d1e2015-07-10 16:01:24 -07003467 return results
Jon Hallc6793552016-01-19 14:18:37 -08003468 except AssertionError:
3469 main.log.exception( "" )
3470 return None
3471 except ( TypeError, ValueError ):
3472 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, rawOutput ) )
acsmarsa4a4d1e2015-07-10 16:01:24 -07003473 return None
3474 except pexpect.EOF:
3475 main.log.error( self.name + ": EOF exception found" )
3476 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003477 main.cleanAndExit()
acsmarsa4a4d1e2015-07-10 16:01:24 -07003478 except Exception:
3479 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003480 main.cleanAndExit()
acsmarsa4a4d1e2015-07-10 16:01:24 -07003481
Jon Hall61282e32015-03-19 11:34:11 -07003482 def pendingMap( self, jsonFormat=True ):
Jon Hall63604932015-02-26 17:09:50 -08003483 """
3484 Returns the output of the intent Pending map.
3485 """
Jon Hall63604932015-02-26 17:09:50 -08003486 try:
Jon Hallc6358dd2015-04-10 12:44:28 -07003487 cmdStr = "onos:intents -p"
Jon Hall61282e32015-03-19 11:34:11 -07003488 if jsonFormat:
Jon Hallc6358dd2015-04-10 12:44:28 -07003489 cmdStr += " -j"
3490 output = self.sendline( cmdStr )
Jon Halla495f562016-05-16 18:03:26 -07003491 assert output is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08003492 assert "Command not found:" not in output, output
Jon Hallc6358dd2015-04-10 12:44:28 -07003493 return output
Jon Hallc6793552016-01-19 14:18:37 -08003494 except AssertionError:
3495 main.log.exception( "" )
3496 return None
Jon Hall63604932015-02-26 17:09:50 -08003497 except TypeError:
3498 main.log.exception( self.name + ": Object not as expected" )
3499 return None
3500 except pexpect.EOF:
3501 main.log.error( self.name + ": EOF exception found" )
3502 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003503 main.cleanAndExit()
Jon Hall77ba41c2015-04-06 10:25:40 -07003504 except Exception:
Jon Hall63604932015-02-26 17:09:50 -08003505 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003506 main.cleanAndExit()
Jon Hall63604932015-02-26 17:09:50 -08003507
Jon Hall2c8959e2016-12-16 12:17:34 -08003508 def partitions( self, candidates=False, jsonFormat=True ):
Jon Hall63604932015-02-26 17:09:50 -08003509 """
3510 Returns the output of the raft partitions command for ONOS.
3511 """
Jon Hall61282e32015-03-19 11:34:11 -07003512 # Sample JSON
3513 # {
3514 # "leader": "tcp://10.128.30.11:7238",
3515 # "members": [
3516 # "tcp://10.128.30.11:7238",
3517 # "tcp://10.128.30.17:7238",
3518 # "tcp://10.128.30.13:7238",
3519 # ],
3520 # "name": "p1",
3521 # "term": 3
3522 # },
Jon Hall63604932015-02-26 17:09:50 -08003523 try:
Jon Hallc6358dd2015-04-10 12:44:28 -07003524 cmdStr = "onos:partitions"
Jon Hall2c8959e2016-12-16 12:17:34 -08003525 if candidates:
3526 cmdStr += " -c"
Jon Hall61282e32015-03-19 11:34:11 -07003527 if jsonFormat:
Jon Hallc6358dd2015-04-10 12:44:28 -07003528 cmdStr += " -j"
3529 output = self.sendline( cmdStr )
Jon Halla495f562016-05-16 18:03:26 -07003530 assert output is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08003531 assert "Command not found:" not in output, output
Jon Hallc6358dd2015-04-10 12:44:28 -07003532 return output
Jon Hallc6793552016-01-19 14:18:37 -08003533 except AssertionError:
3534 main.log.exception( "" )
3535 return None
Jon Hall63604932015-02-26 17:09:50 -08003536 except TypeError:
3537 main.log.exception( self.name + ": Object not as expected" )
3538 return None
3539 except pexpect.EOF:
3540 main.log.error( self.name + ": EOF exception found" )
3541 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003542 main.cleanAndExit()
Jon Hall77ba41c2015-04-06 10:25:40 -07003543 except Exception:
Jon Hall63604932015-02-26 17:09:50 -08003544 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003545 main.cleanAndExit()
Jon Hall63604932015-02-26 17:09:50 -08003546
Jon Halle9f909e2016-09-23 10:43:12 -07003547 def apps( self, summary=False, active=False, jsonFormat=True ):
Jon Hallbe379602015-03-24 13:39:32 -07003548 """
3549 Returns the output of the apps command for ONOS. This command lists
3550 information about installed ONOS applications
3551 """
3552 # Sample JSON object
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00003553 # [{"name":"org.onosproject.openflow","id":0,"version":"1.2.0",
Jon Hallbe379602015-03-24 13:39:32 -07003554 # "description":"ONOS OpenFlow protocol southbound providers",
3555 # "origin":"ON.Lab","permissions":"[]","featuresRepo":"",
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00003556 # "features":"[onos-openflow]","state":"ACTIVE"}]
Jon Hallbe379602015-03-24 13:39:32 -07003557 try:
Jon Hallc6358dd2015-04-10 12:44:28 -07003558 cmdStr = "onos:apps"
Jon Halle9f909e2016-09-23 10:43:12 -07003559 if summary:
3560 cmdStr += " -s"
3561 if active:
3562 cmdStr += " -a"
Jon Hallbe379602015-03-24 13:39:32 -07003563 if jsonFormat:
Jon Hallc6358dd2015-04-10 12:44:28 -07003564 cmdStr += " -j"
3565 output = self.sendline( cmdStr )
Jon Halla495f562016-05-16 18:03:26 -07003566 assert output is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08003567 assert "Command not found:" not in output, output
3568 assert "Error executing command" not in output, output
Jon Hallc6358dd2015-04-10 12:44:28 -07003569 return output
Jon Hallbe379602015-03-24 13:39:32 -07003570 # FIXME: look at specific exceptions/Errors
3571 except AssertionError:
Jon Hallc6793552016-01-19 14:18:37 -08003572 main.log.exception( "Error in processing onos:app command." )
Jon Hallbe379602015-03-24 13:39:32 -07003573 return None
3574 except TypeError:
3575 main.log.exception( self.name + ": Object not as expected" )
3576 return None
3577 except pexpect.EOF:
3578 main.log.error( self.name + ": EOF exception found" )
3579 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003580 main.cleanAndExit()
Jon Hall77ba41c2015-04-06 10:25:40 -07003581 except Exception:
Jon Hallbe379602015-03-24 13:39:32 -07003582 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003583 main.cleanAndExit()
Jon Hallbe379602015-03-24 13:39:32 -07003584
Jon Hall146f1522015-03-24 15:33:24 -07003585 def appStatus( self, appName ):
3586 """
3587 Uses the onos:apps cli command to return the status of an application.
3588 Returns:
3589 "ACTIVE" - If app is installed and activated
3590 "INSTALLED" - If app is installed and deactivated
3591 "UNINSTALLED" - If app is not installed
3592 None - on error
3593 """
Jon Hall146f1522015-03-24 15:33:24 -07003594 try:
3595 if not isinstance( appName, types.StringType ):
3596 main.log.error( self.name + ".appStatus(): appName must be" +
3597 " a string" )
3598 return None
3599 output = self.apps( jsonFormat=True )
3600 appsJson = json.loads( output )
3601 state = None
3602 for app in appsJson:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07003603 if appName == app.get( 'name' ):
3604 state = app.get( 'state' )
Jon Hall146f1522015-03-24 15:33:24 -07003605 break
3606 if state == "ACTIVE" or state == "INSTALLED":
3607 return state
3608 elif state is None:
Jon Hall8bafdc02017-09-05 11:36:26 -07003609 main.log.warn( "{} app not found", appName )
Jon Hall146f1522015-03-24 15:33:24 -07003610 return "UNINSTALLED"
3611 elif state:
3612 main.log.error( "Unexpected state from 'onos:apps': " +
3613 str( state ) )
3614 return state
Jon Hallc6793552016-01-19 14:18:37 -08003615 except ( TypeError, ValueError ):
3616 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, output ) )
Jon Hall146f1522015-03-24 15:33:24 -07003617 return None
3618 except pexpect.EOF:
3619 main.log.error( self.name + ": EOF exception found" )
3620 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003621 main.cleanAndExit()
Jon Hall77ba41c2015-04-06 10:25:40 -07003622 except Exception:
Jon Hall146f1522015-03-24 15:33:24 -07003623 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003624 main.cleanAndExit()
Jon Hall146f1522015-03-24 15:33:24 -07003625
Jon Hallbe379602015-03-24 13:39:32 -07003626 def app( self, appName, option ):
3627 """
3628 Interacts with the app command for ONOS. This command manages
3629 application inventory.
3630 """
Jon Hallbe379602015-03-24 13:39:32 -07003631 try:
Jon Hallbd16b922015-03-26 17:53:15 -07003632 # Validate argument types
3633 valid = True
3634 if not isinstance( appName, types.StringType ):
3635 main.log.error( self.name + ".app(): appName must be a " +
3636 "string" )
3637 valid = False
3638 if not isinstance( option, types.StringType ):
3639 main.log.error( self.name + ".app(): option must be a string" )
3640 valid = False
3641 if not valid:
3642 return main.FALSE
3643 # Validate Option
3644 option = option.lower()
3645 # NOTE: Install may become a valid option
3646 if option == "activate":
3647 pass
3648 elif option == "deactivate":
3649 pass
3650 elif option == "uninstall":
3651 pass
3652 else:
3653 # Invalid option
3654 main.log.error( "The ONOS app command argument only takes " +
3655 "the values: (activate|deactivate|uninstall)" +
Jeremy Ronquillo82705492017-10-18 14:19:55 -07003656 "; was given '" + option + "'" )
Jon Hallbd16b922015-03-26 17:53:15 -07003657 return main.FALSE
Jon Hall146f1522015-03-24 15:33:24 -07003658 cmdStr = "onos:app " + option + " " + appName
Jon Hallbe379602015-03-24 13:39:32 -07003659 output = self.sendline( cmdStr )
You Wangb5a55f72017-03-03 12:51:05 -08003660 assert output is not None, "Error in sendline"
3661 assert "Command not found:" not in output, output
Jon Hallbe379602015-03-24 13:39:32 -07003662 if "Error executing command" in output:
3663 main.log.error( "Error in processing onos:app command: " +
3664 str( output ) )
Jon Hall146f1522015-03-24 15:33:24 -07003665 return main.FALSE
Jon Hallbe379602015-03-24 13:39:32 -07003666 elif "No such application" in output:
3667 main.log.error( "The application '" + appName +
3668 "' is not installed in ONOS" )
Jon Hall146f1522015-03-24 15:33:24 -07003669 return main.FALSE
3670 elif "Command not found:" in output:
3671 main.log.error( "Error in processing onos:app command: " +
3672 str( output ) )
3673 return main.FALSE
Jon Hallbd16b922015-03-26 17:53:15 -07003674 elif "Unsupported command:" in output:
3675 main.log.error( "Incorrect command given to 'app': " +
3676 str( output ) )
Jon Hallbe379602015-03-24 13:39:32 -07003677 # NOTE: we may need to add more checks here
Jon Hallbd16b922015-03-26 17:53:15 -07003678 # else: Command was successful
Jon Hall08f61bc2015-04-13 16:00:30 -07003679 # main.log.debug( "app response: " + repr( output ) )
Jon Hallbe379602015-03-24 13:39:32 -07003680 return main.TRUE
You Wangb5a55f72017-03-03 12:51:05 -08003681 except AssertionError:
3682 main.log.exception( self.name + ": AssertionError exception found" )
3683 return main.ERROR
Jon Hallbe379602015-03-24 13:39:32 -07003684 except TypeError:
3685 main.log.exception( self.name + ": Object not as expected" )
3686 return main.ERROR
3687 except pexpect.EOF:
3688 main.log.error( self.name + ": EOF exception found" )
3689 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003690 main.cleanAndExit()
Jon Hall77ba41c2015-04-06 10:25:40 -07003691 except Exception:
Jon Hallbe379602015-03-24 13:39:32 -07003692 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003693 main.cleanAndExit()
Jon Hall146f1522015-03-24 15:33:24 -07003694
Jon Hallbd16b922015-03-26 17:53:15 -07003695 def activateApp( self, appName, check=True ):
Jon Hall146f1522015-03-24 15:33:24 -07003696 """
3697 Activate an app that is already installed in ONOS
Jon Hallbd16b922015-03-26 17:53:15 -07003698 appName is the hierarchical app name, not the feature name
3699 If check is True, method will check the status of the app after the
3700 command is issued
Jon Hall146f1522015-03-24 15:33:24 -07003701 Returns main.TRUE if the command was successfully sent
3702 main.FALSE if the cli responded with an error or given
3703 incorrect input
3704 """
3705 try:
3706 if not isinstance( appName, types.StringType ):
3707 main.log.error( self.name + ".activateApp(): appName must be" +
3708 " a string" )
3709 return main.FALSE
3710 status = self.appStatus( appName )
3711 if status == "INSTALLED":
3712 response = self.app( appName, "activate" )
Jon Hallbd16b922015-03-26 17:53:15 -07003713 if check and response == main.TRUE:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07003714 for i in range( 10 ): # try 10 times then give up
Jon Hallbd16b922015-03-26 17:53:15 -07003715 status = self.appStatus( appName )
3716 if status == "ACTIVE":
3717 return main.TRUE
3718 else:
Jon Hall050e1bd2015-03-30 13:33:02 -07003719 main.log.debug( "The state of application " +
3720 appName + " is " + status )
Jon Hallbd16b922015-03-26 17:53:15 -07003721 time.sleep( 1 )
3722 return main.FALSE
3723 else: # not 'check' or command didn't succeed
3724 return response
Jon Hall146f1522015-03-24 15:33:24 -07003725 elif status == "ACTIVE":
3726 return main.TRUE
3727 elif status == "UNINSTALLED":
3728 main.log.error( self.name + ": Tried to activate the " +
3729 "application '" + appName + "' which is not " +
3730 "installed." )
3731 else:
3732 main.log.error( "Unexpected return value from appStatus: " +
3733 str( status ) )
3734 return main.ERROR
3735 except TypeError:
3736 main.log.exception( self.name + ": Object not as expected" )
3737 return main.ERROR
3738 except pexpect.EOF:
3739 main.log.error( self.name + ": EOF exception found" )
3740 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003741 main.cleanAndExit()
Jon Hall77ba41c2015-04-06 10:25:40 -07003742 except Exception:
Jon Hall146f1522015-03-24 15:33:24 -07003743 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003744 main.cleanAndExit()
Jon Hall146f1522015-03-24 15:33:24 -07003745
Jon Hallbd16b922015-03-26 17:53:15 -07003746 def deactivateApp( self, appName, check=True ):
Jon Hall146f1522015-03-24 15:33:24 -07003747 """
3748 Deactivate an app that is already activated in ONOS
Jon Hallbd16b922015-03-26 17:53:15 -07003749 appName is the hierarchical app name, not the feature name
3750 If check is True, method will check the status of the app after the
3751 command is issued
Jon Hall146f1522015-03-24 15:33:24 -07003752 Returns main.TRUE if the command was successfully sent
3753 main.FALSE if the cli responded with an error or given
3754 incorrect input
3755 """
3756 try:
3757 if not isinstance( appName, types.StringType ):
3758 main.log.error( self.name + ".deactivateApp(): appName must " +
3759 "be a string" )
3760 return main.FALSE
3761 status = self.appStatus( appName )
3762 if status == "INSTALLED":
3763 return main.TRUE
3764 elif status == "ACTIVE":
3765 response = self.app( appName, "deactivate" )
Jon Hallbd16b922015-03-26 17:53:15 -07003766 if check and response == main.TRUE:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07003767 for i in range( 10 ): # try 10 times then give up
Jon Hallbd16b922015-03-26 17:53:15 -07003768 status = self.appStatus( appName )
3769 if status == "INSTALLED":
3770 return main.TRUE
3771 else:
3772 time.sleep( 1 )
3773 return main.FALSE
3774 else: # not check or command didn't succeed
3775 return response
Jon Hall146f1522015-03-24 15:33:24 -07003776 elif status == "UNINSTALLED":
3777 main.log.warn( self.name + ": Tried to deactivate the " +
3778 "application '" + appName + "' which is not " +
3779 "installed." )
3780 return main.TRUE
3781 else:
3782 main.log.error( "Unexpected return value from appStatus: " +
3783 str( status ) )
3784 return main.ERROR
3785 except TypeError:
3786 main.log.exception( self.name + ": Object not as expected" )
3787 return main.ERROR
3788 except pexpect.EOF:
3789 main.log.error( self.name + ": EOF exception found" )
3790 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003791 main.cleanAndExit()
Jon Hall77ba41c2015-04-06 10:25:40 -07003792 except Exception:
Jon Hall146f1522015-03-24 15:33:24 -07003793 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003794 main.cleanAndExit()
Jon Hall146f1522015-03-24 15:33:24 -07003795
Jon Hallbd16b922015-03-26 17:53:15 -07003796 def uninstallApp( self, appName, check=True ):
Jon Hall146f1522015-03-24 15:33:24 -07003797 """
3798 Uninstall an app that is already installed in ONOS
Jon Hallbd16b922015-03-26 17:53:15 -07003799 appName is the hierarchical app name, not the feature name
3800 If check is True, method will check the status of the app after the
3801 command is issued
Jon Hall146f1522015-03-24 15:33:24 -07003802 Returns main.TRUE if the command was successfully sent
3803 main.FALSE if the cli responded with an error or given
3804 incorrect input
3805 """
3806 # TODO: check with Thomas about the state machine for apps
3807 try:
3808 if not isinstance( appName, types.StringType ):
3809 main.log.error( self.name + ".uninstallApp(): appName must " +
3810 "be a string" )
3811 return main.FALSE
3812 status = self.appStatus( appName )
3813 if status == "INSTALLED":
3814 response = self.app( appName, "uninstall" )
Jon Hallbd16b922015-03-26 17:53:15 -07003815 if check and response == main.TRUE:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07003816 for i in range( 10 ): # try 10 times then give up
Jon Hallbd16b922015-03-26 17:53:15 -07003817 status = self.appStatus( appName )
3818 if status == "UNINSTALLED":
3819 return main.TRUE
3820 else:
3821 time.sleep( 1 )
3822 return main.FALSE
3823 else: # not check or command didn't succeed
3824 return response
Jon Hall146f1522015-03-24 15:33:24 -07003825 elif status == "ACTIVE":
3826 main.log.warn( self.name + ": Tried to uninstall the " +
3827 "application '" + appName + "' which is " +
3828 "currently active." )
3829 response = self.app( appName, "uninstall" )
Jon Hallbd16b922015-03-26 17:53:15 -07003830 if check and response == main.TRUE:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07003831 for i in range( 10 ): # try 10 times then give up
Jon Hallbd16b922015-03-26 17:53:15 -07003832 status = self.appStatus( appName )
3833 if status == "UNINSTALLED":
3834 return main.TRUE
3835 else:
3836 time.sleep( 1 )
3837 return main.FALSE
3838 else: # not check or command didn't succeed
3839 return response
Jon Hall146f1522015-03-24 15:33:24 -07003840 elif status == "UNINSTALLED":
3841 return main.TRUE
3842 else:
3843 main.log.error( "Unexpected return value from appStatus: " +
3844 str( status ) )
3845 return main.ERROR
3846 except TypeError:
3847 main.log.exception( self.name + ": Object not as expected" )
3848 return main.ERROR
3849 except pexpect.EOF:
3850 main.log.error( self.name + ": EOF exception found" )
3851 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003852 main.cleanAndExit()
Jon Hall77ba41c2015-04-06 10:25:40 -07003853 except Exception:
Jon Hall146f1522015-03-24 15:33:24 -07003854 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003855 main.cleanAndExit()
Jon Hallbd16b922015-03-26 17:53:15 -07003856
3857 def appIDs( self, jsonFormat=True ):
3858 """
3859 Show the mappings between app id and app names given by the 'app-ids'
3860 cli command
3861 """
3862 try:
3863 cmdStr = "app-ids"
3864 if jsonFormat:
3865 cmdStr += " -j"
Jon Hallc6358dd2015-04-10 12:44:28 -07003866 output = self.sendline( cmdStr )
Jon Halla495f562016-05-16 18:03:26 -07003867 assert output is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08003868 assert "Command not found:" not in output, output
3869 assert "Error executing command" not in output, output
Jon Hallc6358dd2015-04-10 12:44:28 -07003870 return output
Jon Hallbd16b922015-03-26 17:53:15 -07003871 except AssertionError:
Jon Hallc6793552016-01-19 14:18:37 -08003872 main.log.exception( "Error in processing onos:app-ids command." )
Jon Hallbd16b922015-03-26 17:53:15 -07003873 return None
3874 except TypeError:
3875 main.log.exception( self.name + ": Object not as expected" )
3876 return None
3877 except pexpect.EOF:
3878 main.log.error( self.name + ": EOF exception found" )
3879 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003880 main.cleanAndExit()
Jon Hall77ba41c2015-04-06 10:25:40 -07003881 except Exception:
Jon Hallbd16b922015-03-26 17:53:15 -07003882 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003883 main.cleanAndExit()
Jon Hallbd16b922015-03-26 17:53:15 -07003884
3885 def appToIDCheck( self ):
3886 """
3887 This method will check that each application's ID listed in 'apps' is
3888 the same as the ID listed in 'app-ids'. The check will also check that
3889 there are no duplicate IDs issued. Note that an app ID should be
3890 a globaly unique numerical identifier for app/app-like features. Once
3891 an ID is registered, the ID is never freed up so that if an app is
3892 reinstalled it will have the same ID.
3893
3894 Returns: main.TRUE if the check passes and
3895 main.FALSE if the check fails or
3896 main.ERROR if there is some error in processing the test
3897 """
3898 try:
Jon Hall390696c2015-05-05 17:13:41 -07003899 bail = False
Jon Hallc6793552016-01-19 14:18:37 -08003900 rawJson = self.appIDs( jsonFormat=True )
3901 if rawJson:
3902 ids = json.loads( rawJson )
Jon Hall390696c2015-05-05 17:13:41 -07003903 else:
Jon Hallc6793552016-01-19 14:18:37 -08003904 main.log.error( "app-ids returned nothing:" + repr( rawJson ) )
Jon Hall390696c2015-05-05 17:13:41 -07003905 bail = True
Jon Hallc6793552016-01-19 14:18:37 -08003906 rawJson = self.apps( jsonFormat=True )
3907 if rawJson:
3908 apps = json.loads( rawJson )
Jon Hall390696c2015-05-05 17:13:41 -07003909 else:
Jon Hallc6793552016-01-19 14:18:37 -08003910 main.log.error( "apps returned nothing:" + repr( rawJson ) )
Jon Hall390696c2015-05-05 17:13:41 -07003911 bail = True
3912 if bail:
3913 return main.FALSE
Jon Hallbd16b922015-03-26 17:53:15 -07003914 result = main.TRUE
3915 for app in apps:
3916 appID = app.get( 'id' )
3917 if appID is None:
3918 main.log.error( "Error parsing app: " + str( app ) )
3919 result = main.FALSE
3920 appName = app.get( 'name' )
3921 if appName is None:
3922 main.log.error( "Error parsing app: " + str( app ) )
3923 result = main.FALSE
3924 # get the entry in ids that has the same appID
Jon Hall390696c2015-05-05 17:13:41 -07003925 current = filter( lambda item: item[ 'id' ] == appID, ids )
Jon Hallbd16b922015-03-26 17:53:15 -07003926 if not current: # if ids doesn't have this id
3927 result = main.FALSE
3928 main.log.error( "'app-ids' does not have the ID for " +
3929 str( appName ) + " that apps does." )
Jon Hallb9d381e2018-02-05 12:02:10 -08003930 main.log.debug( "apps command returned: " + str( app ) +
3931 "; app-ids has: " + str( ids ) )
Jon Hallbd16b922015-03-26 17:53:15 -07003932 elif len( current ) > 1:
3933 # there is more than one app with this ID
3934 result = main.FALSE
3935 # We will log this later in the method
Jeremy Ronquillo82705492017-10-18 14:19:55 -07003936 elif not current[ 0 ][ 'name' ] == appName:
3937 currentName = current[ 0 ][ 'name' ]
Jon Hallbd16b922015-03-26 17:53:15 -07003938 result = main.FALSE
3939 main.log.error( "'app-ids' has " + str( currentName ) +
3940 " registered under id:" + str( appID ) +
3941 " but 'apps' has " + str( appName ) )
3942 else:
3943 pass # id and name match!
3944 # now make sure that app-ids has no duplicates
3945 idsList = []
3946 namesList = []
3947 for item in ids:
3948 idsList.append( item[ 'id' ] )
3949 namesList.append( item[ 'name' ] )
3950 if len( idsList ) != len( set( idsList ) ) or\
3951 len( namesList ) != len( set( namesList ) ):
Jeremy Ronquillo82705492017-10-18 14:19:55 -07003952 main.log.error( "'app-ids' has some duplicate entries: \n"
3953 + json.dumps( ids,
3954 sort_keys=True,
3955 indent=4,
3956 separators=( ',', ': ' ) ) )
3957 result = main.FALSE
Jon Hallbd16b922015-03-26 17:53:15 -07003958 return result
Jon Hallc6793552016-01-19 14:18:37 -08003959 except ( TypeError, ValueError ):
3960 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, rawJson ) )
Jon Hallbd16b922015-03-26 17:53:15 -07003961 return main.ERROR
3962 except pexpect.EOF:
3963 main.log.error( self.name + ": EOF exception found" )
3964 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07003965 main.cleanAndExit()
Jon Hall77ba41c2015-04-06 10:25:40 -07003966 except Exception:
Jon Hallbd16b922015-03-26 17:53:15 -07003967 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07003968 main.cleanAndExit()
Jon Hallbd16b922015-03-26 17:53:15 -07003969
Jon Hallfb760a02015-04-13 15:35:03 -07003970 def getCfg( self, component=None, propName=None, short=False,
3971 jsonFormat=True ):
3972 """
3973 Get configuration settings from onos cli
3974 Optional arguments:
3975 component - Optionally only list configurations for a specific
3976 component. If None, all components with configurations
3977 are displayed. Case Sensitive string.
3978 propName - If component is specified, propName option will show
3979 only this specific configuration from that component.
3980 Case Sensitive string.
3981 jsonFormat - Returns output as json. Note that this will override
3982 the short option
3983 short - Short, less verbose, version of configurations.
3984 This is overridden by the json option
3985 returns:
3986 Output from cli as a string or None on error
3987 """
3988 try:
3989 baseStr = "cfg"
3990 cmdStr = " get"
3991 componentStr = ""
3992 if component:
3993 componentStr += " " + component
3994 if propName:
3995 componentStr += " " + propName
3996 if jsonFormat:
3997 baseStr += " -j"
3998 elif short:
3999 baseStr += " -s"
4000 output = self.sendline( baseStr + cmdStr + componentStr )
Jon Halla495f562016-05-16 18:03:26 -07004001 assert output is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08004002 assert "Command not found:" not in output, output
4003 assert "Error executing command" not in output, output
Jon Hallfb760a02015-04-13 15:35:03 -07004004 return output
4005 except AssertionError:
Jon Hallc6793552016-01-19 14:18:37 -08004006 main.log.exception( "Error in processing 'cfg get' command." )
Jon Hallfb760a02015-04-13 15:35:03 -07004007 return None
4008 except TypeError:
4009 main.log.exception( self.name + ": Object not as expected" )
4010 return None
4011 except pexpect.EOF:
4012 main.log.error( self.name + ": EOF exception found" )
4013 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07004014 main.cleanAndExit()
Jon Hallfb760a02015-04-13 15:35:03 -07004015 except Exception:
4016 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004017 main.cleanAndExit()
Jon Hallfb760a02015-04-13 15:35:03 -07004018
4019 def setCfg( self, component, propName, value=None, check=True ):
4020 """
4021 Set/Unset configuration settings from ONOS cli
Jon Hall390696c2015-05-05 17:13:41 -07004022 Required arguments:
Jon Hallfb760a02015-04-13 15:35:03 -07004023 component - The case sensitive name of the component whose
4024 property is to be set
4025 propName - The case sensitive name of the property to be set/unset
Jon Hall390696c2015-05-05 17:13:41 -07004026 Optional arguments:
Jon Hallfb760a02015-04-13 15:35:03 -07004027 value - The value to set the property to. If None, will unset the
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004028 property and revert it to it's default value(if applicable)
Jon Hallfb760a02015-04-13 15:35:03 -07004029 check - Boolean, Check whether the option was successfully set this
4030 only applies when a value is given.
4031 returns:
4032 main.TRUE on success or main.FALSE on failure. If check is False,
4033 will return main.TRUE unless there is an error
4034 """
4035 try:
4036 baseStr = "cfg"
4037 cmdStr = " set " + str( component ) + " " + str( propName )
4038 if value is not None:
4039 cmdStr += " " + str( value )
4040 output = self.sendline( baseStr + cmdStr )
Jon Halla495f562016-05-16 18:03:26 -07004041 assert output is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08004042 assert "Command not found:" not in output, output
4043 assert "Error executing command" not in output, output
Jon Hallfb760a02015-04-13 15:35:03 -07004044 if value and check:
4045 results = self.getCfg( component=str( component ),
4046 propName=str( propName ),
4047 jsonFormat=True )
4048 # Check if current value is what we just set
4049 try:
4050 jsonOutput = json.loads( results )
4051 current = jsonOutput[ 'value' ]
Jon Hallc6793552016-01-19 14:18:37 -08004052 except ( TypeError, ValueError ):
Jon Hallfb760a02015-04-13 15:35:03 -07004053 main.log.exception( "Error parsing cfg output" )
4054 main.log.error( "output:" + repr( results ) )
4055 return main.FALSE
4056 if current == str( value ):
4057 return main.TRUE
4058 return main.FALSE
4059 return main.TRUE
4060 except AssertionError:
Jon Hallc6793552016-01-19 14:18:37 -08004061 main.log.exception( "Error in processing 'cfg set' command." )
Jon Hallfb760a02015-04-13 15:35:03 -07004062 return main.FALSE
Jon Hallc6793552016-01-19 14:18:37 -08004063 except ( TypeError, ValueError ):
4064 main.log.exception( "{}: Object not as expected: {!r}".format( self.name, results ) )
Jon Hallfb760a02015-04-13 15:35:03 -07004065 return main.FALSE
4066 except pexpect.EOF:
4067 main.log.error( self.name + ": EOF exception found" )
4068 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07004069 main.cleanAndExit()
Jon Hallfb760a02015-04-13 15:35:03 -07004070 except Exception:
4071 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004072 main.cleanAndExit()
Jon Hallfb760a02015-04-13 15:35:03 -07004073
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004074 def distPrimitivesSend( self, cmd ):
4075 """
4076 Function to handle sending cli commands for the distributed primitives test app
4077
4078 This command will catch some exceptions and retry the command on some
4079 specific store exceptions.
4080
4081 Required arguments:
4082 cmd - The command to send to the cli
4083 returns:
4084 string containing the cli output
4085 None on Error
4086 """
4087 try:
4088 output = self.sendline( cmd )
4089 try:
4090 assert output is not None, "Error in sendline"
4091 # TODO: Maybe make this less hardcoded
4092 # ConsistentMap Exceptions
4093 assert "org.onosproject.store.service" not in output
4094 # Node not leader
4095 assert "java.lang.IllegalStateException" not in output
4096 except AssertionError:
4097 main.log.error( "Error in processing '" + cmd + "' " +
4098 "command: " + str( output ) )
4099 retryTime = 30 # Conservative time, given by Madan
4100 main.log.info( "Waiting " + str( retryTime ) +
4101 "seconds before retrying." )
4102 time.sleep( retryTime ) # Due to change in mastership
4103 output = self.sendline( cmd )
4104 assert output is not None, "Error in sendline"
4105 assert "Command not found:" not in output, output
4106 assert "Error executing command" not in output, output
4107 main.log.info( self.name + ": " + output )
4108 return output
4109 except AssertionError:
4110 main.log.exception( "Error in processing '" + cmd + "' command." )
4111 return None
4112 except TypeError:
4113 main.log.exception( self.name + ": Object not as expected" )
4114 return None
4115 except pexpect.EOF:
4116 main.log.error( self.name + ": EOF exception found" )
4117 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07004118 main.cleanAndExit()
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004119 except Exception:
4120 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004121 main.cleanAndExit()
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004122
Jon Hall390696c2015-05-05 17:13:41 -07004123 def setTestAdd( self, setName, values ):
4124 """
4125 CLI command to add elements to a distributed set.
4126 Arguments:
4127 setName - The name of the set to add to.
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004128 values - The value(s) to add to the set, space seperated.
Jon Hall390696c2015-05-05 17:13:41 -07004129 Example usages:
4130 setTestAdd( "set1", "a b c" )
4131 setTestAdd( "set2", "1" )
4132 returns:
4133 main.TRUE on success OR
4134 main.FALSE if elements were already in the set OR
4135 main.ERROR on error
4136 """
4137 try:
4138 cmdStr = "set-test-add " + str( setName ) + " " + str( values )
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004139 output = self.distPrimitivesSend( cmdStr )
Jon Hall390696c2015-05-05 17:13:41 -07004140 positiveMatch = "\[(.*)\] was added to the set " + str( setName )
4141 negativeMatch = "\[(.*)\] was already in set " + str( setName )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07004142 if re.search( positiveMatch, output ):
Jon Hall390696c2015-05-05 17:13:41 -07004143 return main.TRUE
Jeremy Ronquillo82705492017-10-18 14:19:55 -07004144 elif re.search( negativeMatch, output ):
Jon Hall390696c2015-05-05 17:13:41 -07004145 return main.FALSE
4146 else:
4147 main.log.error( self.name + ": setTestAdd did not" +
4148 " match expected output" )
Jon Hall390696c2015-05-05 17:13:41 -07004149 main.log.debug( self.name + " actual: " + repr( output ) )
4150 return main.ERROR
Jon Hall390696c2015-05-05 17:13:41 -07004151 except TypeError:
4152 main.log.exception( self.name + ": Object not as expected" )
4153 return main.ERROR
Jon Hall390696c2015-05-05 17:13:41 -07004154 except Exception:
4155 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004156 main.cleanAndExit()
Jon Hall390696c2015-05-05 17:13:41 -07004157
4158 def setTestRemove( self, setName, values, clear=False, retain=False ):
4159 """
4160 CLI command to remove elements from a distributed set.
4161 Required arguments:
4162 setName - The name of the set to remove from.
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004163 values - The value(s) to remove from the set, space seperated.
Jon Hall390696c2015-05-05 17:13:41 -07004164 Optional arguments:
4165 clear - Clear all elements from the set
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004166 retain - Retain only the given values. (intersection of the
4167 original set and the given set)
Jon Hall390696c2015-05-05 17:13:41 -07004168 returns:
4169 main.TRUE on success OR
4170 main.FALSE if the set was not changed OR
4171 main.ERROR on error
4172 """
4173 try:
4174 cmdStr = "set-test-remove "
4175 if clear:
4176 cmdStr += "-c " + str( setName )
4177 elif retain:
4178 cmdStr += "-r " + str( setName ) + " " + str( values )
4179 else:
4180 cmdStr += str( setName ) + " " + str( values )
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004181 output = self.distPrimitivesSend( cmdStr )
Jon Hall390696c2015-05-05 17:13:41 -07004182 if clear:
4183 pattern = "Set " + str( setName ) + " cleared"
4184 if re.search( pattern, output ):
4185 return main.TRUE
4186 elif retain:
4187 positivePattern = str( setName ) + " was pruned to contain " +\
4188 "only elements of set \[(.*)\]"
4189 negativePattern = str( setName ) + " was not changed by " +\
4190 "retaining only elements of the set " +\
4191 "\[(.*)\]"
4192 if re.search( positivePattern, output ):
4193 return main.TRUE
4194 elif re.search( negativePattern, output ):
4195 return main.FALSE
4196 else:
4197 positivePattern = "\[(.*)\] was removed from the set " +\
4198 str( setName )
4199 if ( len( values.split() ) == 1 ):
4200 negativePattern = "\[(.*)\] was not in set " +\
4201 str( setName )
4202 else:
4203 negativePattern = "No element of \[(.*)\] was in set " +\
4204 str( setName )
4205 if re.search( positivePattern, output ):
4206 return main.TRUE
4207 elif re.search( negativePattern, output ):
4208 return main.FALSE
4209 main.log.error( self.name + ": setTestRemove did not" +
4210 " match expected output" )
4211 main.log.debug( self.name + " expected: " + pattern )
4212 main.log.debug( self.name + " actual: " + repr( output ) )
4213 return main.ERROR
Jon Hall390696c2015-05-05 17:13:41 -07004214 except TypeError:
4215 main.log.exception( self.name + ": Object not as expected" )
4216 return main.ERROR
Jon Hall390696c2015-05-05 17:13:41 -07004217 except Exception:
4218 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004219 main.cleanAndExit()
Jon Hall390696c2015-05-05 17:13:41 -07004220
4221 def setTestGet( self, setName, values="" ):
4222 """
4223 CLI command to get the elements in a distributed set.
4224 Required arguments:
4225 setName - The name of the set to remove from.
4226 Optional arguments:
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004227 values - The value(s) to check if in the set, space seperated.
Jon Hall390696c2015-05-05 17:13:41 -07004228 returns:
4229 main.ERROR on error OR
4230 A list of elements in the set if no optional arguments are
4231 supplied OR
4232 A tuple containing the list then:
4233 main.FALSE if the given values are not in the set OR
4234 main.TRUE if the given values are in the set OR
4235 """
4236 try:
4237 values = str( values ).strip()
4238 setName = str( setName ).strip()
4239 length = len( values.split() )
4240 containsCheck = None
4241 # Patterns to match
4242 setPattern = "\[(.*)\]"
Jon Hall67253832016-12-05 09:47:13 -08004243 pattern = "Items in set " + setName + ":\r\n" + setPattern
Jon Hall390696c2015-05-05 17:13:41 -07004244 containsTrue = "Set " + setName + " contains the value " + values
4245 containsFalse = "Set " + setName + " did not contain the value " +\
4246 values
4247 containsAllTrue = "Set " + setName + " contains the the subset " +\
4248 setPattern
4249 containsAllFalse = "Set " + setName + " did not contain the the" +\
4250 " subset " + setPattern
4251
4252 cmdStr = "set-test-get "
4253 cmdStr += setName + " " + values
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004254 output = self.distPrimitivesSend( cmdStr )
Jon Hall390696c2015-05-05 17:13:41 -07004255 if length == 0:
4256 match = re.search( pattern, output )
4257 else: # if given values
4258 if length == 1: # Contains output
Jon Hall54b994f2016-12-05 10:48:59 -08004259 patternTrue = pattern + "\r\n" + containsTrue
4260 patternFalse = pattern + "\r\n" + containsFalse
Jon Hall390696c2015-05-05 17:13:41 -07004261 else: # ContainsAll output
Jon Hall54b994f2016-12-05 10:48:59 -08004262 patternTrue = pattern + "\r\n" + containsAllTrue
4263 patternFalse = pattern + "\r\n" + containsAllFalse
Jon Hall390696c2015-05-05 17:13:41 -07004264 matchTrue = re.search( patternTrue, output )
4265 matchFalse = re.search( patternFalse, output )
4266 if matchTrue:
4267 containsCheck = main.TRUE
4268 match = matchTrue
4269 elif matchFalse:
4270 containsCheck = main.FALSE
4271 match = matchFalse
4272 else:
Jon Halle0f0b342017-04-18 11:43:47 -07004273 main.log.error( self.name + " setTestGet did not match " +
Jon Hall390696c2015-05-05 17:13:41 -07004274 "expected output" )
4275 main.log.debug( self.name + " expected: " + pattern )
4276 main.log.debug( self.name + " actual: " + repr( output ) )
4277 match = None
4278 if match:
4279 setMatch = match.group( 1 )
4280 if setMatch == '':
4281 setList = []
4282 else:
4283 setList = setMatch.split( ", " )
4284 if length > 0:
4285 return ( setList, containsCheck )
4286 else:
4287 return setList
4288 else: # no match
4289 main.log.error( self.name + ": setTestGet did not" +
4290 " match expected output" )
4291 main.log.debug( self.name + " expected: " + pattern )
4292 main.log.debug( self.name + " actual: " + repr( output ) )
4293 return main.ERROR
Jon Hall390696c2015-05-05 17:13:41 -07004294 except TypeError:
4295 main.log.exception( self.name + ": Object not as expected" )
4296 return main.ERROR
Jon Hall390696c2015-05-05 17:13:41 -07004297 except Exception:
4298 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004299 main.cleanAndExit()
Jon Hall390696c2015-05-05 17:13:41 -07004300
4301 def setTestSize( self, setName ):
4302 """
4303 CLI command to get the elements in a distributed set.
4304 Required arguments:
4305 setName - The name of the set to remove from.
4306 returns:
Jon Hallfeff3082015-05-19 10:23:26 -07004307 The integer value of the size returned or
Jon Hall390696c2015-05-05 17:13:41 -07004308 None on error
4309 """
4310 try:
4311 # TODO: Should this check against the number of elements returned
4312 # and then return true/false based on that?
4313 setName = str( setName ).strip()
4314 # Patterns to match
4315 setPattern = "\[(.*)\]"
Jon Hall67253832016-12-05 09:47:13 -08004316 pattern = "There are (\d+) items in set " + setName + ":\r\n" +\
Jon Hall390696c2015-05-05 17:13:41 -07004317 setPattern
4318 cmdStr = "set-test-get -s "
4319 cmdStr += setName
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004320 output = self.distPrimitivesSend( cmdStr )
Jon Hall390696c2015-05-05 17:13:41 -07004321 match = re.search( pattern, output )
4322 if match:
4323 setSize = int( match.group( 1 ) )
4324 setMatch = match.group( 2 )
4325 if len( setMatch.split() ) == setSize:
4326 main.log.info( "The size returned by " + self.name +
4327 " matches the number of elements in " +
4328 "the returned set" )
4329 else:
4330 main.log.error( "The size returned by " + self.name +
4331 " does not match the number of " +
4332 "elements in the returned set." )
4333 return setSize
4334 else: # no match
4335 main.log.error( self.name + ": setTestGet did not" +
4336 " match expected output" )
4337 main.log.debug( self.name + " expected: " + pattern )
4338 main.log.debug( self.name + " actual: " + repr( output ) )
4339 return None
Jon Hall390696c2015-05-05 17:13:41 -07004340 except TypeError:
4341 main.log.exception( self.name + ": Object not as expected" )
4342 return None
Jon Hall390696c2015-05-05 17:13:41 -07004343 except Exception:
4344 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004345 main.cleanAndExit()
Jon Hall390696c2015-05-05 17:13:41 -07004346
Jon Hall80daded2015-05-27 16:07:00 -07004347 def counters( self, jsonFormat=True ):
Jon Hall390696c2015-05-05 17:13:41 -07004348 """
4349 Command to list the various counters in the system.
4350 returns:
Jon Hall80daded2015-05-27 16:07:00 -07004351 if jsonFormat, a string of the json object returned by the cli
4352 command
4353 if not jsonFormat, the normal string output of the cli command
Jon Hall390696c2015-05-05 17:13:41 -07004354 None on error
4355 """
Jon Hall390696c2015-05-05 17:13:41 -07004356 try:
Jon Hall390696c2015-05-05 17:13:41 -07004357 cmdStr = "counters"
Jon Hall80daded2015-05-27 16:07:00 -07004358 if jsonFormat:
4359 cmdStr += " -j"
Jon Hall390696c2015-05-05 17:13:41 -07004360 output = self.sendline( cmdStr )
Jon Halla495f562016-05-16 18:03:26 -07004361 assert output is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08004362 assert "Command not found:" not in output, output
4363 assert "Error executing command" not in output, output
Jon Hall390696c2015-05-05 17:13:41 -07004364 main.log.info( self.name + ": " + output )
Jon Hall80daded2015-05-27 16:07:00 -07004365 return output
Jon Hall390696c2015-05-05 17:13:41 -07004366 except AssertionError:
Jon Hallc6793552016-01-19 14:18:37 -08004367 main.log.exception( "Error in processing 'counters' command." )
Jon Hall80daded2015-05-27 16:07:00 -07004368 return None
Jon Hall390696c2015-05-05 17:13:41 -07004369 except TypeError:
4370 main.log.exception( self.name + ": Object not as expected" )
Jon Hall80daded2015-05-27 16:07:00 -07004371 return None
Jon Hall390696c2015-05-05 17:13:41 -07004372 except pexpect.EOF:
4373 main.log.error( self.name + ": EOF exception found" )
4374 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07004375 main.cleanAndExit()
Jon Hall390696c2015-05-05 17:13:41 -07004376 except Exception:
4377 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004378 main.cleanAndExit()
Jon Hall390696c2015-05-05 17:13:41 -07004379
Jon Hall935db192016-04-19 00:22:04 -07004380 def counterTestAddAndGet( self, counter, delta=1 ):
Jon Hall390696c2015-05-05 17:13:41 -07004381 """
Jon Halle1a3b752015-07-22 13:02:46 -07004382 CLI command to add a delta to then get a distributed counter.
Jon Hall390696c2015-05-05 17:13:41 -07004383 Required arguments:
4384 counter - The name of the counter to increment.
4385 Optional arguments:
Jon Halle1a3b752015-07-22 13:02:46 -07004386 delta - The long to add to the counter
Jon Hall390696c2015-05-05 17:13:41 -07004387 returns:
4388 integer value of the counter or
4389 None on Error
4390 """
4391 try:
4392 counter = str( counter )
Jon Halle1a3b752015-07-22 13:02:46 -07004393 delta = int( delta )
Jon Hall390696c2015-05-05 17:13:41 -07004394 cmdStr = "counter-test-increment "
Jon Hall390696c2015-05-05 17:13:41 -07004395 cmdStr += counter
Jon Halle1a3b752015-07-22 13:02:46 -07004396 if delta != 1:
4397 cmdStr += " " + str( delta )
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004398 output = self.distPrimitivesSend( cmdStr )
Jon Halle1a3b752015-07-22 13:02:46 -07004399 pattern = counter + " was updated to (-?\d+)"
Jon Hall390696c2015-05-05 17:13:41 -07004400 match = re.search( pattern, output )
4401 if match:
4402 return int( match.group( 1 ) )
4403 else:
Jon Halle1a3b752015-07-22 13:02:46 -07004404 main.log.error( self.name + ": counterTestAddAndGet did not" +
Jon Hall390696c2015-05-05 17:13:41 -07004405 " match expected output." )
4406 main.log.debug( self.name + " expected: " + pattern )
4407 main.log.debug( self.name + " actual: " + repr( output ) )
4408 return None
Jon Hall390696c2015-05-05 17:13:41 -07004409 except TypeError:
4410 main.log.exception( self.name + ": Object not as expected" )
4411 return None
Jon Hall390696c2015-05-05 17:13:41 -07004412 except Exception:
4413 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004414 main.cleanAndExit()
Jon Hall390696c2015-05-05 17:13:41 -07004415
Jon Hall935db192016-04-19 00:22:04 -07004416 def counterTestGetAndAdd( self, counter, delta=1 ):
Jon Halle1a3b752015-07-22 13:02:46 -07004417 """
4418 CLI command to get a distributed counter then add a delta to it.
4419 Required arguments:
4420 counter - The name of the counter to increment.
4421 Optional arguments:
4422 delta - The long to add to the counter
Jon Halle1a3b752015-07-22 13:02:46 -07004423 returns:
4424 integer value of the counter or
4425 None on Error
4426 """
4427 try:
4428 counter = str( counter )
4429 delta = int( delta )
4430 cmdStr = "counter-test-increment -g "
Jon Halle1a3b752015-07-22 13:02:46 -07004431 cmdStr += counter
4432 if delta != 1:
4433 cmdStr += " " + str( delta )
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004434 output = self.distPrimitivesSend( cmdStr )
Jon Halle1a3b752015-07-22 13:02:46 -07004435 pattern = counter + " was updated to (-?\d+)"
4436 match = re.search( pattern, output )
4437 if match:
4438 return int( match.group( 1 ) )
4439 else:
4440 main.log.error( self.name + ": counterTestGetAndAdd did not" +
4441 " match expected output." )
4442 main.log.debug( self.name + " expected: " + pattern )
4443 main.log.debug( self.name + " actual: " + repr( output ) )
4444 return None
Jon Halle1a3b752015-07-22 13:02:46 -07004445 except TypeError:
4446 main.log.exception( self.name + ": Object not as expected" )
4447 return None
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004448 except Exception:
4449 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004450 main.cleanAndExit()
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004451
4452 def valueTestGet( self, valueName ):
4453 """
4454 CLI command to get the value of an atomic value.
4455 Required arguments:
4456 valueName - The name of the value to get.
4457 returns:
4458 string value of the value or
4459 None on Error
4460 """
4461 try:
4462 valueName = str( valueName )
4463 cmdStr = "value-test "
4464 operation = "get"
4465 cmdStr = "value-test {} {}".format( valueName,
4466 operation )
4467 output = self.distPrimitivesSend( cmdStr )
4468 pattern = "(\w+)"
4469 match = re.search( pattern, output )
4470 if match:
4471 return match.group( 1 )
4472 else:
4473 main.log.error( self.name + ": valueTestGet did not" +
4474 " match expected output." )
4475 main.log.debug( self.name + " expected: " + pattern )
4476 main.log.debug( self.name + " actual: " + repr( output ) )
4477 return None
4478 except TypeError:
4479 main.log.exception( self.name + ": Object not as expected" )
4480 return None
4481 except Exception:
4482 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004483 main.cleanAndExit()
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004484
4485 def valueTestSet( self, valueName, newValue ):
4486 """
4487 CLI command to set the value of an atomic value.
4488 Required arguments:
4489 valueName - The name of the value to set.
4490 newValue - The value to assign to the given value.
4491 returns:
4492 main.TRUE on success or
4493 main.ERROR on Error
4494 """
4495 try:
4496 valueName = str( valueName )
4497 newValue = str( newValue )
4498 operation = "set"
4499 cmdStr = "value-test {} {} {}".format( valueName,
4500 operation,
4501 newValue )
4502 output = self.distPrimitivesSend( cmdStr )
4503 if output is not None:
4504 return main.TRUE
4505 else:
4506 return main.ERROR
4507 except TypeError:
4508 main.log.exception( self.name + ": Object not as expected" )
4509 return main.ERROR
4510 except Exception:
4511 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004512 main.cleanAndExit()
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004513
4514 def valueTestCompareAndSet( self, valueName, oldValue, newValue ):
4515 """
4516 CLI command to compareAndSet the value of an atomic value.
4517 Required arguments:
4518 valueName - The name of the value.
4519 oldValue - Compare the current value of the atomic value to this
4520 newValue - If the value equals oldValue, set the value to newValue
4521 returns:
4522 main.TRUE on success or
4523 main.FALSE on failure or
4524 main.ERROR on Error
4525 """
4526 try:
4527 valueName = str( valueName )
4528 oldValue = str( oldValue )
4529 newValue = str( newValue )
4530 operation = "compareAndSet"
4531 cmdStr = "value-test {} {} {} {}".format( valueName,
4532 operation,
4533 oldValue,
4534 newValue )
4535 output = self.distPrimitivesSend( cmdStr )
4536 pattern = "(\w+)"
4537 match = re.search( pattern, output )
4538 if match:
4539 result = match.group( 1 )
4540 if result == "true":
4541 return main.TRUE
4542 elif result == "false":
4543 return main.FALSE
4544 else:
4545 main.log.error( self.name + ": valueTestCompareAndSet did not" +
4546 " match expected output." )
4547 main.log.debug( self.name + " expected: " + pattern )
4548 main.log.debug( self.name + " actual: " + repr( output ) )
4549 return main.ERROR
4550 except TypeError:
4551 main.log.exception( self.name + ": Object not as expected" )
4552 return main.ERROR
4553 except Exception:
4554 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004555 main.cleanAndExit()
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004556
4557 def valueTestGetAndSet( self, valueName, newValue ):
4558 """
4559 CLI command to getAndSet the value of an atomic value.
4560 Required arguments:
4561 valueName - The name of the value to get.
4562 newValue - The value to assign to the given value
4563 returns:
4564 string value of the value or
4565 None on Error
4566 """
4567 try:
4568 valueName = str( valueName )
4569 cmdStr = "value-test "
4570 operation = "getAndSet"
4571 cmdStr += valueName + " " + operation
4572 cmdStr = "value-test {} {} {}".format( valueName,
4573 operation,
4574 newValue )
4575 output = self.distPrimitivesSend( cmdStr )
4576 pattern = "(\w+)"
4577 match = re.search( pattern, output )
4578 if match:
4579 return match.group( 1 )
4580 else:
4581 main.log.error( self.name + ": valueTestGetAndSet did not" +
4582 " match expected output." )
4583 main.log.debug( self.name + " expected: " + pattern )
4584 main.log.debug( self.name + " actual: " + repr( output ) )
4585 return None
4586 except TypeError:
4587 main.log.exception( self.name + ": Object not as expected" )
4588 return None
4589 except Exception:
4590 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004591 main.cleanAndExit()
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004592
4593 def valueTestDestroy( self, valueName ):
4594 """
4595 CLI command to destroy an atomic value.
4596 Required arguments:
4597 valueName - The name of the value to destroy.
4598 returns:
4599 main.TRUE on success or
4600 main.ERROR on Error
4601 """
4602 try:
4603 valueName = str( valueName )
4604 cmdStr = "value-test "
4605 operation = "destroy"
4606 cmdStr += valueName + " " + operation
4607 output = self.distPrimitivesSend( cmdStr )
4608 if output is not None:
4609 return main.TRUE
4610 else:
4611 return main.ERROR
4612 except TypeError:
4613 main.log.exception( self.name + ": Object not as expected" )
4614 return main.ERROR
Jon Halle1a3b752015-07-22 13:02:46 -07004615 except Exception:
4616 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004617 main.cleanAndExit()
Jon Halle1a3b752015-07-22 13:02:46 -07004618
YPZhangfebf7302016-05-24 16:45:56 -07004619 def summary( self, jsonFormat=True, timeout=30 ):
kelvin-onlaba297c4d2015-06-01 13:53:55 -07004620 """
4621 Description: Execute summary command in onos
4622 Returns: json object ( summary -j ), returns main.FALSE if there is
4623 no output
4624
4625 """
4626 try:
4627 cmdStr = "summary"
4628 if jsonFormat:
4629 cmdStr += " -j"
YPZhangfebf7302016-05-24 16:45:56 -07004630 handle = self.sendline( cmdStr, timeout=timeout )
Jon Halla495f562016-05-16 18:03:26 -07004631 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08004632 assert "Command not found:" not in handle, handle
Jon Hall6e709752016-02-01 13:38:46 -08004633 assert "Error:" not in handle, handle
Devin Lima7cfdbd2017-09-29 15:02:22 -07004634 assert "Error executing" not in handle, handle
kelvin-onlaba297c4d2015-06-01 13:53:55 -07004635 if not handle:
4636 main.log.error( self.name + ": There is no output in " +
4637 "summary command" )
4638 return main.FALSE
4639 return handle
Jon Hallc6793552016-01-19 14:18:37 -08004640 except AssertionError:
Jon Hall6e709752016-02-01 13:38:46 -08004641 main.log.exception( "{} Error in summary output:".format( self.name ) )
Jon Hallc6793552016-01-19 14:18:37 -08004642 return None
kelvin-onlaba297c4d2015-06-01 13:53:55 -07004643 except TypeError:
4644 main.log.exception( self.name + ": Object not as expected" )
4645 return None
4646 except pexpect.EOF:
4647 main.log.error( self.name + ": EOF exception found" )
4648 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07004649 main.cleanAndExit()
kelvin-onlaba297c4d2015-06-01 13:53:55 -07004650 except Exception:
4651 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004652 main.cleanAndExit()
Jon Hall2a5002c2015-08-21 16:49:11 -07004653
Jon Hall935db192016-04-19 00:22:04 -07004654 def transactionalMapGet( self, keyName ):
Jon Hall2a5002c2015-08-21 16:49:11 -07004655 """
4656 CLI command to get the value of a key in a consistent map using
4657 transactions. This a test function and can only get keys from the
4658 test map hard coded into the cli command
4659 Required arguments:
4660 keyName - The name of the key to get
Jon Hall2a5002c2015-08-21 16:49:11 -07004661 returns:
4662 The string value of the key or
4663 None on Error
4664 """
4665 try:
4666 keyName = str( keyName )
4667 cmdStr = "transactional-map-test-get "
Jon Hall2a5002c2015-08-21 16:49:11 -07004668 cmdStr += keyName
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004669 output = self.distPrimitivesSend( cmdStr )
Jon Hall2a5002c2015-08-21 16:49:11 -07004670 pattern = "Key-value pair \(" + keyName + ", (?P<value>.+)\) found."
4671 if "Key " + keyName + " not found." in output:
Jon Hall9bfadd22016-05-11 14:48:07 -07004672 main.log.warn( output )
Jon Hall2a5002c2015-08-21 16:49:11 -07004673 return None
4674 else:
4675 match = re.search( pattern, output )
4676 if match:
4677 return match.groupdict()[ 'value' ]
4678 else:
4679 main.log.error( self.name + ": transactionlMapGet did not" +
4680 " match expected output." )
4681 main.log.debug( self.name + " expected: " + pattern )
4682 main.log.debug( self.name + " actual: " + repr( output ) )
4683 return None
4684 except TypeError:
4685 main.log.exception( self.name + ": Object not as expected" )
4686 return None
Jon Hall2a5002c2015-08-21 16:49:11 -07004687 except Exception:
4688 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004689 main.cleanAndExit()
Jon Hall2a5002c2015-08-21 16:49:11 -07004690
Jon Hall935db192016-04-19 00:22:04 -07004691 def transactionalMapPut( self, numKeys, value ):
Jon Hall2a5002c2015-08-21 16:49:11 -07004692 """
4693 CLI command to put a value into 'numKeys' number of keys in a
4694 consistent map using transactions. This a test function and can only
4695 put into keys named 'Key#' of the test map hard coded into the cli command
4696 Required arguments:
4697 numKeys - Number of keys to add the value to
4698 value - The string value to put into the keys
Jon Hall2a5002c2015-08-21 16:49:11 -07004699 returns:
4700 A dictionary whose keys are the name of the keys put into the map
4701 and the values of the keys are dictionaries whose key-values are
4702 'value': value put into map and optionaly
4703 'oldValue': Previous value in the key or
4704 None on Error
4705
4706 Example output
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004707 { 'Key1': {'oldValue': 'oldTestValue', 'value': 'Testing'},
4708 'Key2': {'value': 'Testing'} }
Jon Hall2a5002c2015-08-21 16:49:11 -07004709 """
4710 try:
4711 numKeys = str( numKeys )
4712 value = str( value )
4713 cmdStr = "transactional-map-test-put "
Jon Hall2a5002c2015-08-21 16:49:11 -07004714 cmdStr += numKeys + " " + value
Jon Hall7a6ebfd2017-03-13 10:58:58 -07004715 output = self.distPrimitivesSend( cmdStr )
Jon Hall2a5002c2015-08-21 16:49:11 -07004716 newPattern = 'Created Key (?P<key>(\w)+) with value (?P<value>(.)+)\.'
4717 updatedPattern = "Put (?P<value>(.)+) into key (?P<key>(\w)+)\. The old value was (?P<oldValue>(.)+)\."
4718 results = {}
4719 for line in output.splitlines():
4720 new = re.search( newPattern, line )
4721 updated = re.search( updatedPattern, line )
4722 if new:
4723 results[ new.groupdict()[ 'key' ] ] = { 'value': new.groupdict()[ 'value' ] }
4724 elif updated:
4725 results[ updated.groupdict()[ 'key' ] ] = { 'value': updated.groupdict()[ 'value' ],
Jon Hallc6793552016-01-19 14:18:37 -08004726 'oldValue': updated.groupdict()[ 'oldValue' ] }
Jon Hall2a5002c2015-08-21 16:49:11 -07004727 else:
4728 main.log.error( self.name + ": transactionlMapGet did not" +
4729 " match expected output." )
Jon Hallc6793552016-01-19 14:18:37 -08004730 main.log.debug( "{} expected: {!r} or {!r}".format( self.name,
4731 newPattern,
4732 updatedPattern ) )
Jon Hall2a5002c2015-08-21 16:49:11 -07004733 main.log.debug( self.name + " actual: " + repr( output ) )
4734 return results
4735 except TypeError:
4736 main.log.exception( self.name + ": Object not as expected" )
4737 return None
Jon Hall2a5002c2015-08-21 16:49:11 -07004738 except Exception:
4739 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004740 main.cleanAndExit()
Jon Hallc6793552016-01-19 14:18:37 -08004741
acsmarsdaea66c2015-09-03 11:44:06 -07004742 def maps( self, jsonFormat=True ):
4743 """
4744 Description: Returns result of onos:maps
4745 Optional:
4746 * jsonFormat: enable json formatting of output
4747 """
4748 try:
4749 cmdStr = "maps"
4750 if jsonFormat:
4751 cmdStr += " -j"
4752 handle = self.sendline( cmdStr )
Jon Halla495f562016-05-16 18:03:26 -07004753 assert handle is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08004754 assert "Command not found:" not in handle, handle
acsmarsdaea66c2015-09-03 11:44:06 -07004755 return handle
Jon Hallc6793552016-01-19 14:18:37 -08004756 except AssertionError:
4757 main.log.exception( "" )
4758 return None
acsmarsdaea66c2015-09-03 11:44:06 -07004759 except TypeError:
4760 main.log.exception( self.name + ": Object not as expected" )
4761 return None
4762 except pexpect.EOF:
4763 main.log.error( self.name + ": EOF exception found" )
4764 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07004765 main.cleanAndExit()
acsmarsdaea66c2015-09-03 11:44:06 -07004766 except Exception:
4767 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004768 main.cleanAndExit()
GlennRC050596c2015-11-18 17:06:41 -08004769
4770 def getSwController( self, uri, jsonFormat=True ):
4771 """
4772 Descrition: Gets the controller information from the device
4773 """
4774 try:
4775 cmd = "device-controllers "
4776 if jsonFormat:
4777 cmd += "-j "
4778 response = self.sendline( cmd + uri )
Jon Halla495f562016-05-16 18:03:26 -07004779 assert response is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08004780 assert "Command not found:" not in response, response
GlennRC050596c2015-11-18 17:06:41 -08004781 return response
Jon Hallc6793552016-01-19 14:18:37 -08004782 except AssertionError:
4783 main.log.exception( "" )
4784 return None
GlennRC050596c2015-11-18 17:06:41 -08004785 except TypeError:
4786 main.log.exception( self.name + ": Object not as expected" )
4787 return None
4788 except pexpect.EOF:
4789 main.log.error( self.name + ": EOF exception found" )
4790 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07004791 main.cleanAndExit()
GlennRC050596c2015-11-18 17:06:41 -08004792 except Exception:
4793 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004794 main.cleanAndExit()
GlennRC050596c2015-11-18 17:06:41 -08004795
4796 def setSwController( self, uri, ip, proto="tcp", port="6653", jsonFormat=True ):
4797 """
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004798 Descrition: sets the controller(s) for the specified device
GlennRC050596c2015-11-18 17:06:41 -08004799
4800 Parameters:
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004801 Required: uri - String: The uri of the device(switch).
GlennRC050596c2015-11-18 17:06:41 -08004802 ip - String or List: The ip address of the controller.
4803 This parameter can be formed in a couple of different ways.
4804 VALID:
4805 10.0.0.1 - just the ip address
4806 tcp:10.0.0.1 - the protocol and the ip address
4807 tcp:10.0.0.1:6653 - the protocol and port can be specified,
4808 so that you can add controllers with different
4809 protocols and ports
4810 INVALID:
4811 10.0.0.1:6653 - this is not supported by ONOS
4812
4813 Optional: proto - The type of connection e.g. tcp, ssl. If a list of ips are given
4814 port - The port number.
4815 jsonFormat - If set ONOS will output in json NOTE: This is currently not supported
4816
4817 Returns: main.TRUE if ONOS returns without any errors, otherwise returns main.FALSE
4818 """
4819 try:
4820 cmd = "device-setcontrollers"
4821
4822 if jsonFormat:
4823 cmd += " -j"
4824 cmd += " " + uri
4825 if isinstance( ip, str ):
Jeremy Ronquillo82705492017-10-18 14:19:55 -07004826 ip = [ ip ]
GlennRC050596c2015-11-18 17:06:41 -08004827 for item in ip:
4828 if ":" in item:
4829 sitem = item.split( ":" )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07004830 if len( sitem ) == 3:
GlennRC050596c2015-11-18 17:06:41 -08004831 cmd += " " + item
Jeremy Ronquillo82705492017-10-18 14:19:55 -07004832 elif "." in sitem[ 1 ]:
4833 cmd += " {}:{}".format( item, port )
GlennRC050596c2015-11-18 17:06:41 -08004834 else:
4835 main.log.error( "Malformed entry: " + item )
4836 raise TypeError
4837 else:
4838 cmd += " {}:{}:{}".format( proto, item, port )
GlennRC050596c2015-11-18 17:06:41 -08004839 response = self.sendline( cmd )
Jon Halla495f562016-05-16 18:03:26 -07004840 assert response is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08004841 assert "Command not found:" not in response, response
GlennRC050596c2015-11-18 17:06:41 -08004842 if "Error" in response:
4843 main.log.error( response )
4844 return main.FALSE
GlennRC050596c2015-11-18 17:06:41 -08004845 return main.TRUE
Jon Hallc6793552016-01-19 14:18:37 -08004846 except AssertionError:
4847 main.log.exception( "" )
Jon Hall2c8959e2016-12-16 12:17:34 -08004848 return main.FALSE
GlennRC050596c2015-11-18 17:06:41 -08004849 except TypeError:
4850 main.log.exception( self.name + ": Object not as expected" )
4851 return main.FALSE
4852 except pexpect.EOF:
4853 main.log.error( self.name + ": EOF exception found" )
4854 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07004855 main.cleanAndExit()
GlennRC050596c2015-11-18 17:06:41 -08004856 except Exception:
4857 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004858 main.cleanAndExit()
GlennRC20fc6522015-12-23 23:26:57 -08004859
4860 def removeDevice( self, device ):
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004861 '''
GlennRC20fc6522015-12-23 23:26:57 -08004862 Description:
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004863 Remove a device from ONOS by passing the uri of the device(s).
GlennRC20fc6522015-12-23 23:26:57 -08004864 Parameters:
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004865 device - (str or list) the id or uri of the device ex. "of:0000000000000001"
GlennRC20fc6522015-12-23 23:26:57 -08004866 Returns:
4867 Returns main.FALSE if an exception is thrown or an error is present
4868 in the response. Otherwise, returns main.TRUE.
4869 NOTE:
4870 If a host cannot be removed, then this function will return main.FALSE
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004871 '''
GlennRC20fc6522015-12-23 23:26:57 -08004872 try:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07004873 if isinstance( device, str ):
You Wang823f5022016-08-18 15:24:41 -07004874 deviceStr = device
4875 device = []
4876 device.append( deviceStr )
GlennRC20fc6522015-12-23 23:26:57 -08004877
4878 for d in device:
4879 time.sleep( 1 )
4880 response = self.sendline( "device-remove {}".format( d ) )
Jon Halla495f562016-05-16 18:03:26 -07004881 assert response is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08004882 assert "Command not found:" not in response, response
GlennRC20fc6522015-12-23 23:26:57 -08004883 if "Error" in response:
4884 main.log.warn( "Error for device: {}\nResponse: {}".format( d, response ) )
4885 return main.FALSE
GlennRC20fc6522015-12-23 23:26:57 -08004886 return main.TRUE
Jon Hallc6793552016-01-19 14:18:37 -08004887 except AssertionError:
4888 main.log.exception( "" )
4889 return main.FALSE
GlennRC20fc6522015-12-23 23:26:57 -08004890 except TypeError:
4891 main.log.exception( self.name + ": Object not as expected" )
4892 return main.FALSE
4893 except pexpect.EOF:
4894 main.log.error( self.name + ": EOF exception found" )
4895 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07004896 main.cleanAndExit()
GlennRC20fc6522015-12-23 23:26:57 -08004897 except Exception:
4898 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004899 main.cleanAndExit()
GlennRC20fc6522015-12-23 23:26:57 -08004900
4901 def removeHost( self, host ):
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004902 '''
GlennRC20fc6522015-12-23 23:26:57 -08004903 Description:
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004904 Remove a host from ONOS by passing the id of the host(s)
GlennRC20fc6522015-12-23 23:26:57 -08004905 Parameters:
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004906 hostId - (str or list) the id or mac of the host ex. "00:00:00:00:00:01"
GlennRC20fc6522015-12-23 23:26:57 -08004907 Returns:
4908 Returns main.FALSE if an exception is thrown or an error is present
4909 in the response. Otherwise, returns main.TRUE.
4910 NOTE:
4911 If a host cannot be removed, then this function will return main.FALSE
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004912 '''
GlennRC20fc6522015-12-23 23:26:57 -08004913 try:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07004914 if isinstance( host, str ):
GlennRC20fc6522015-12-23 23:26:57 -08004915 host = list( host )
4916
4917 for h in host:
4918 time.sleep( 1 )
4919 response = self.sendline( "host-remove {}".format( h ) )
Jon Halla495f562016-05-16 18:03:26 -07004920 assert response is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08004921 assert "Command not found:" not in response, response
GlennRC20fc6522015-12-23 23:26:57 -08004922 if "Error" in response:
4923 main.log.warn( "Error for host: {}\nResponse: {}".format( h, response ) )
4924 return main.FALSE
GlennRC20fc6522015-12-23 23:26:57 -08004925 return main.TRUE
Jon Hallc6793552016-01-19 14:18:37 -08004926 except AssertionError:
4927 main.log.exception( "" )
Jon Hall2c8959e2016-12-16 12:17:34 -08004928 return main.FALSE
GlennRC20fc6522015-12-23 23:26:57 -08004929 except TypeError:
4930 main.log.exception( self.name + ": Object not as expected" )
4931 return main.FALSE
4932 except pexpect.EOF:
4933 main.log.error( self.name + ": EOF exception found" )
4934 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07004935 main.cleanAndExit()
GlennRC20fc6522015-12-23 23:26:57 -08004936 except Exception:
4937 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004938 main.cleanAndExit()
GlennRCed771242016-01-13 17:02:47 -08004939
YPZhangfebf7302016-05-24 16:45:56 -07004940 def link( self, begin, end, state, timeout=30, showResponse=True ):
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004941 '''
GlennRCed771242016-01-13 17:02:47 -08004942 Description:
4943 Bring link down or up in the null-provider.
4944 params:
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004945 begin - (string) One end of a device or switch.
4946 end - (string) the other end of the device or switch
GlennRCed771242016-01-13 17:02:47 -08004947 returns:
4948 main.TRUE if no exceptions were thrown and no Errors are
4949 present in the resoponse. Otherwise, returns main.FALSE
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004950 '''
GlennRCed771242016-01-13 17:02:47 -08004951 try:
Jon Halle0f0b342017-04-18 11:43:47 -07004952 cmd = "null-link null:{} null:{} {}".format( begin, end, state )
YPZhangfebf7302016-05-24 16:45:56 -07004953 response = self.sendline( cmd, showResponse=showResponse, timeout=timeout )
Jon Halla495f562016-05-16 18:03:26 -07004954 assert response is not None, "Error in sendline"
Jon Hallc6793552016-01-19 14:18:37 -08004955 assert "Command not found:" not in response, response
GlennRCed771242016-01-13 17:02:47 -08004956 if "Error" in response or "Failure" in response:
4957 main.log.error( response )
4958 return main.FALSE
GlennRCed771242016-01-13 17:02:47 -08004959 return main.TRUE
Jon Hallc6793552016-01-19 14:18:37 -08004960 except AssertionError:
4961 main.log.exception( "" )
Jon Hall2c8959e2016-12-16 12:17:34 -08004962 return main.FALSE
GlennRCed771242016-01-13 17:02:47 -08004963 except TypeError:
4964 main.log.exception( self.name + ": Object not as expected" )
4965 return main.FALSE
4966 except pexpect.EOF:
4967 main.log.error( self.name + ": EOF exception found" )
4968 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07004969 main.cleanAndExit()
GlennRCed771242016-01-13 17:02:47 -08004970 except Exception:
4971 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07004972 main.cleanAndExit()
GlennRCed771242016-01-13 17:02:47 -08004973
Jon Hall2c8959e2016-12-16 12:17:34 -08004974 def portstate( self, dpid, port, state ):
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004975 '''
Flavio Castro82ee2f62016-06-07 15:04:12 -07004976 Description:
4977 Changes the state of port in an OF switch by means of the
4978 PORTSTATUS OF messages.
4979 params:
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004980 dpid - (string) Datapath ID of the device. Ex: 'of:0000000000000102'
4981 port - (string) target port in the device. Ex: '2'
4982 state - (string) target state (enable or disable)
Flavio Castro82ee2f62016-06-07 15:04:12 -07004983 returns:
4984 main.TRUE if no exceptions were thrown and no Errors are
4985 present in the resoponse. Otherwise, returns main.FALSE
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00004986 '''
Flavio Castro82ee2f62016-06-07 15:04:12 -07004987 try:
Jon Hall2c8959e2016-12-16 12:17:34 -08004988 state = state.lower()
4989 assert state == 'enable' or state == 'disable', "Unknown state"
Jon Halle0f0b342017-04-18 11:43:47 -07004990 cmd = "portstate {} {} {}".format( dpid, port, state )
Flavio Castro82ee2f62016-06-07 15:04:12 -07004991 response = self.sendline( cmd, showResponse=True )
4992 assert response is not None, "Error in sendline"
4993 assert "Command not found:" not in response, response
4994 if "Error" in response or "Failure" in response:
4995 main.log.error( response )
4996 return main.FALSE
4997 return main.TRUE
4998 except AssertionError:
4999 main.log.exception( "" )
Jon Hall2c8959e2016-12-16 12:17:34 -08005000 return main.FALSE
Flavio Castro82ee2f62016-06-07 15:04:12 -07005001 except TypeError:
5002 main.log.exception( self.name + ": Object not as expected" )
5003 return main.FALSE
5004 except pexpect.EOF:
5005 main.log.error( self.name + ": EOF exception found" )
5006 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07005007 main.cleanAndExit()
Flavio Castro82ee2f62016-06-07 15:04:12 -07005008 except Exception:
5009 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005010 main.cleanAndExit()
Flavio Castro82ee2f62016-06-07 15:04:12 -07005011
5012 def logSet( self, level="INFO", app="org.onosproject" ):
5013 """
5014 Set the logging level to lvl for a specific app
5015 returns main.TRUE on success
5016 returns main.FALSE if Error occurred
5017 if noExit is True, TestON will not exit, but clean up
5018 Available level: DEBUG, TRACE, INFO, WARN, ERROR
5019 Level defaults to INFO
5020 """
5021 try:
Jon Halle0f0b342017-04-18 11:43:47 -07005022 self.handle.sendline( "log:set %s %s" % ( level, app ) )
Flavio Castro82ee2f62016-06-07 15:04:12 -07005023 self.handle.expect( "onos>" )
5024
5025 response = self.handle.before
5026 if re.search( "Error", response ):
5027 return main.FALSE
5028 return main.TRUE
5029 except pexpect.TIMEOUT:
5030 main.log.exception( self.name + ": TIMEOUT exception found" )
Devin Lim44075962017-08-11 10:56:37 -07005031 main.cleanAndExit()
Flavio Castro82ee2f62016-06-07 15:04:12 -07005032 except pexpect.EOF:
5033 main.log.error( self.name + ": EOF exception found" )
5034 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07005035 main.cleanAndExit()
Flavio Castro82ee2f62016-06-07 15:04:12 -07005036 except Exception:
5037 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005038 main.cleanAndExit()
You Wangdb8cd0a2016-05-26 15:19:45 -07005039
5040 def getGraphDict( self, timeout=60, includeHost=False ):
5041 """
5042 Return a dictionary which describes the latest network topology data as a
5043 graph.
5044 An example of the dictionary:
5045 { vertex1: { 'edges': ..., 'name': ..., 'protocol': ... },
5046 vertex2: { 'edges': ..., 'name': ..., 'protocol': ... } }
5047 Each vertex should at least have an 'edges' attribute which describes the
5048 adjacency information. The value of 'edges' attribute is also represented by
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00005049 a dictionary, which maps each edge (identified by the neighbor vertex) to a
You Wangdb8cd0a2016-05-26 15:19:45 -07005050 list of attributes.
5051 An example of the edges dictionary:
5052 'edges': { vertex2: { 'port': ..., 'weight': ... },
5053 vertex3: { 'port': ..., 'weight': ... } }
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00005054 If includeHost == True, all hosts (and host-switch links) will be included
You Wangdb8cd0a2016-05-26 15:19:45 -07005055 in topology data.
5056 """
5057 graphDict = {}
5058 try:
5059 links = self.links()
5060 links = json.loads( links )
5061 devices = self.devices()
5062 devices = json.loads( devices )
5063 idToDevice = {}
5064 for device in devices:
5065 idToDevice[ device[ 'id' ] ] = device
5066 if includeHost:
5067 hosts = self.hosts()
5068 # FIXME: support 'includeHost' argument
5069 for link in links:
5070 nodeA = link[ 'src' ][ 'device' ]
5071 nodeB = link[ 'dst' ][ 'device' ]
5072 assert idToDevice[ nodeA ][ 'available' ] and idToDevice[ nodeB ][ 'available' ]
Jon Halle0f0b342017-04-18 11:43:47 -07005073 if nodeA not in graphDict.keys():
5074 graphDict[ nodeA ] = { 'edges': {},
Jeremy Ronquillo82705492017-10-18 14:19:55 -07005075 'dpid': idToDevice[ nodeA ][ 'id' ][ 3: ],
Jon Halle0f0b342017-04-18 11:43:47 -07005076 'type': idToDevice[ nodeA ][ 'type' ],
5077 'available': idToDevice[ nodeA ][ 'available' ],
5078 'role': idToDevice[ nodeA ][ 'role' ],
5079 'mfr': idToDevice[ nodeA ][ 'mfr' ],
5080 'hw': idToDevice[ nodeA ][ 'hw' ],
5081 'sw': idToDevice[ nodeA ][ 'sw' ],
5082 'serial': idToDevice[ nodeA ][ 'serial' ],
5083 'chassisId': idToDevice[ nodeA ][ 'chassisId' ],
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00005084 'annotations': idToDevice[ nodeA ][ 'annotations' ]}
You Wangdb8cd0a2016-05-26 15:19:45 -07005085 else:
5086 # Assert nodeB is not connected to any current links of nodeA
5087 assert nodeB not in graphDict[ nodeA ][ 'edges' ].keys()
Jon Halle0f0b342017-04-18 11:43:47 -07005088 graphDict[ nodeA ][ 'edges' ][ nodeB ] = { 'port': link[ 'src' ][ 'port' ],
5089 'type': link[ 'type' ],
5090 'state': link[ 'state' ] }
You Wangdb8cd0a2016-05-26 15:19:45 -07005091 return graphDict
5092 except ( TypeError, ValueError ):
5093 main.log.exception( self.name + ": Object not as expected" )
5094 return None
5095 except KeyError:
5096 main.log.exception( self.name + ": KeyError exception found" )
5097 return None
5098 except AssertionError:
5099 main.log.exception( self.name + ": AssertionError exception found" )
5100 return None
5101 except pexpect.EOF:
5102 main.log.error( self.name + ": EOF exception found" )
5103 main.log.error( self.name + ": " + self.handle.before )
5104 return None
5105 except Exception:
5106 main.log.exception( self.name + ": Uncaught exception!" )
5107 return None
YPZhangcbc2a062016-07-11 10:55:44 -07005108
5109 def getIntentPerfSummary( self ):
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00005110 '''
YPZhangcbc2a062016-07-11 10:55:44 -07005111 Send command to check intent-perf summary
5112 Returns: dictionary for intent-perf summary
5113 if something wrong, function will return None
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00005114 '''
YPZhangcbc2a062016-07-11 10:55:44 -07005115 cmd = "intent-perf -s"
5116 respDic = {}
5117 resp = self.sendline( cmd )
You Wangb5a55f72017-03-03 12:51:05 -08005118 assert resp is not None, "Error in sendline"
5119 assert "Command not found:" not in resp, resp
YPZhangcbc2a062016-07-11 10:55:44 -07005120 try:
5121 # Generate the dictionary to return
5122 for l in resp.split( "\n" ):
5123 # Delete any white space in line
5124 temp = re.sub( r'\s+', '', l )
5125 temp = temp.split( ":" )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07005126 respDic[ temp[ 0 ] ] = temp[ 1 ]
YPZhangcbc2a062016-07-11 10:55:44 -07005127
Jeremy Ronquillo82705492017-10-18 14:19:55 -07005128 except ( TypeError, ValueError ):
YPZhangcbc2a062016-07-11 10:55:44 -07005129 main.log.exception( self.name + ": Object not as expected" )
5130 return None
5131 except KeyError:
5132 main.log.exception( self.name + ": KeyError exception found" )
5133 return None
5134 except AssertionError:
5135 main.log.exception( self.name + ": AssertionError exception found" )
5136 return None
5137 except pexpect.EOF:
5138 main.log.error( self.name + ": EOF exception found" )
5139 main.log.error( self.name + ": " + self.handle.before )
5140 return None
5141 except Exception:
5142 main.log.exception( self.name + ": Uncaught exception!" )
5143 return None
5144 return respDic
5145
Chiyu Chengec63bde2016-11-17 18:11:36 -08005146 def logSearch( self, mode='all', searchTerm='', startLine='', logNum=1 ):
chengchiyu08303a02016-09-08 17:40:26 -07005147 """
5148 Searches the latest ONOS log file for the given search term and
5149 return a list that contains all the lines that have the search term.
YPZhangcbc2a062016-07-11 10:55:44 -07005150
chengchiyu08303a02016-09-08 17:40:26 -07005151 Arguments:
Chiyu Chengec63bde2016-11-17 18:11:36 -08005152 searchTerm:
5153 The string to grep from the ONOS log.
5154 startLine:
5155 The term that decides which line is the start to search the searchTerm in
5156 the karaf log. For now, startTerm only works in 'first' mode.
5157 logNum:
5158 In some extreme cases, one karaf log is not big enough to contain all the
5159 information.Because of this, search mutiply logs is necessary to capture
5160 the right result. logNum is the number of karaf logs that we need to search
5161 the searchTerm.
chengchiyu08303a02016-09-08 17:40:26 -07005162 mode:
5163 all: return all the strings that contain the search term
5164 last: return the last string that contains the search term
5165 first: return the first string that contains the search term
Chiyu Chengec63bde2016-11-17 18:11:36 -08005166 num: return the number of times that the searchTerm appears in the log
5167 total: return how many lines in karaf log
chengchiyu08303a02016-09-08 17:40:26 -07005168 """
5169 try:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07005170 assert isinstance( searchTerm, str )
Jon Halle0f0b342017-04-18 11:43:47 -07005171 # Build the log paths string
Chiyu Chengec63bde2016-11-17 18:11:36 -08005172 logPath = '/opt/onos/log/karaf.log.'
5173 logPaths = '/opt/onos/log/karaf.log'
5174 for i in range( 1, logNum ):
5175 logPaths = logPath + str( i ) + " " + logPaths
5176 cmd = "cat " + logPaths
You Wang6d301d42017-04-21 10:49:33 -07005177 if startLine:
Jon Halla478b852017-12-04 15:00:15 -08005178 # 100000000 is just a extreme large number to make sure this function can
5179 # grep all the lines after startLine
You Wang6d301d42017-04-21 10:49:33 -07005180 cmd = cmd + " | grep -A 100000000 \'" + startLine + "\'"
Chiyu Chengec63bde2016-11-17 18:11:36 -08005181 if mode == 'all':
5182 cmd = cmd + " | grep \'" + searchTerm + "\'"
You Wang6d301d42017-04-21 10:49:33 -07005183 elif mode == 'last':
Chiyu Chengec63bde2016-11-17 18:11:36 -08005184 cmd = cmd + " | grep \'" + searchTerm + "\'" + " | tail -n 1"
You Wang6d301d42017-04-21 10:49:33 -07005185 elif mode == 'first':
5186 cmd = cmd + " | grep \'" + searchTerm + "\'" + " | head -n 1"
5187 elif mode == 'num':
Chiyu Chengec63bde2016-11-17 18:11:36 -08005188 cmd = cmd + " | grep -c \'" + searchTerm + "\'"
You Wang118ba582017-01-02 17:14:43 -08005189 num = self.sendline( cmd )
Chiyu Chengb8c2c842016-10-05 12:40:49 -07005190 return num
You Wang6d301d42017-04-21 10:49:33 -07005191 elif mode == 'total':
Chiyu Chengec63bde2016-11-17 18:11:36 -08005192 totalLines = self.sendline( "cat /opt/onos/log/karaf.log | wc -l" )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07005193 return int( totalLines )
You Wang6d301d42017-04-21 10:49:33 -07005194 else:
5195 main.log.error( self.name + " unsupported mode" )
5196 return main.ERROR
chengchiyu08303a02016-09-08 17:40:26 -07005197 before = self.sendline( cmd )
5198 before = before.splitlines()
5199 # make sure the returned list only contains the search term
Jeremy Ronquillo82705492017-10-18 14:19:55 -07005200 returnLines = [ line for line in before if searchTerm in line ]
chengchiyu08303a02016-09-08 17:40:26 -07005201 return returnLines
5202 except AssertionError:
5203 main.log.error( self.name + " searchTerm is not string type" )
5204 return None
5205 except pexpect.EOF:
5206 main.log.error( self.name + ": EOF exception found" )
5207 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07005208 main.cleanAndExit()
chengchiyu08303a02016-09-08 17:40:26 -07005209 except pexpect.TIMEOUT:
5210 main.log.error( self.name + ": TIMEOUT exception found" )
5211 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07005212 main.cleanAndExit()
chengchiyu08303a02016-09-08 17:40:26 -07005213 except Exception:
5214 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005215 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005216
5217 def vplsShow( self, jsonFormat=True ):
5218 """
5219 Description: Returns result of onos:vpls show, which should list the
5220 configured VPLS networks and the assigned interfaces.
5221 Optional:
5222 * jsonFormat: enable json formatting of output
5223 Returns:
5224 The output of the command or None on error.
5225 """
5226 try:
5227 cmdStr = "vpls show"
5228 if jsonFormat:
5229 raise NotImplementedError
5230 cmdStr += " -j"
5231 handle = self.sendline( cmdStr )
5232 assert handle is not None, "Error in sendline"
5233 assert "Command not found:" not in handle, handle
5234 return handle
5235 except AssertionError:
5236 main.log.exception( "" )
5237 return None
5238 except TypeError:
5239 main.log.exception( self.name + ": Object not as expected" )
5240 return None
5241 except pexpect.EOF:
5242 main.log.error( self.name + ": EOF exception found" )
5243 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07005244 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005245 except NotImplementedError:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07005246 main.log.exception( self.name + ": Json output not supported" )
Jon Hall2c8959e2016-12-16 12:17:34 -08005247 return None
5248 except Exception:
5249 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005250 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005251
5252 def parseVplsShow( self ):
5253 """
5254 Parse the cli output of 'vpls show' into json output. This is required
5255 as there is currently no json output available.
5256 """
5257 try:
5258 output = []
5259 raw = self.vplsShow( jsonFormat=False )
5260 namePat = "VPLS name: (?P<name>\w+)"
5261 interfacesPat = "Associated interfaces: \[(?P<interfaces>.*)\]"
5262 encapPat = "Encapsulation: (?P<encap>\w+)"
5263 pattern = "\s+".join( [ namePat, interfacesPat, encapPat ] )
5264 mIter = re.finditer( pattern, raw )
5265 for match in mIter:
5266 item = {}
5267 item[ 'name' ] = match.group( 'name' )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07005268 ifaces = match.group( 'interfaces' ).split( ', ' )
Jon Hall2c8959e2016-12-16 12:17:34 -08005269 if ifaces == [ "" ]:
5270 ifaces = []
5271 item[ 'interfaces' ] = ifaces
5272 encap = match.group( 'encap' )
5273 if encap != 'NONE':
5274 item[ 'encapsulation' ] = encap.lower()
5275 output.append( item )
5276 return output
5277 except Exception:
5278 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005279 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005280
5281 def vplsList( self, jsonFormat=True ):
5282 """
5283 Description: Returns result of onos:vpls list, which should list the
5284 configured VPLS networks.
5285 Optional:
5286 * jsonFormat: enable json formatting of output
5287 """
5288 try:
5289 cmdStr = "vpls list"
5290 if jsonFormat:
5291 raise NotImplementedError
5292 cmdStr += " -j"
5293 handle = self.sendline( cmdStr )
5294 assert handle is not None, "Error in sendline"
5295 assert "Command not found:" not in handle, handle
5296 return handle
5297 except AssertionError:
5298 main.log.exception( "" )
5299 return None
5300 except TypeError:
5301 main.log.exception( self.name + ": Object not as expected" )
5302 return None
5303 except pexpect.EOF:
5304 main.log.error( self.name + ": EOF exception found" )
5305 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07005306 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005307 except NotImplementedError:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07005308 main.log.exception( self.name + ": Json output not supported" )
Jon Hall2c8959e2016-12-16 12:17:34 -08005309 return None
5310 except Exception:
5311 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005312 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005313
5314 def vplsCreate( self, network ):
5315 """
5316 CLI command to create a new VPLS network.
5317 Required arguments:
5318 network - String name of the network to create.
5319 returns:
5320 main.TRUE on success and main.FALSE on failure
5321 """
5322 try:
5323 network = str( network )
5324 cmdStr = "vpls create "
5325 cmdStr += network
5326 output = self.sendline( cmdStr )
5327 assert output is not None, "Error in sendline"
5328 assert "Command not found:" not in output, output
5329 assert "Error executing command" not in output, output
5330 assert "VPLS already exists:" not in output, output
5331 return main.TRUE
5332 except AssertionError:
5333 main.log.exception( "" )
5334 return main.FALSE
5335 except TypeError:
5336 main.log.exception( self.name + ": Object not as expected" )
5337 return main.FALSE
5338 except pexpect.EOF:
5339 main.log.error( self.name + ": EOF exception found" )
5340 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07005341 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005342 except Exception:
5343 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005344 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005345
5346 def vplsDelete( self, network ):
5347 """
5348 CLI command to delete a VPLS network.
5349 Required arguments:
5350 network - Name of the network to delete.
5351 returns:
5352 main.TRUE on success and main.FALSE on failure
5353 """
5354 try:
5355 network = str( network )
5356 cmdStr = "vpls delete "
5357 cmdStr += network
5358 output = self.sendline( cmdStr )
5359 assert output is not None, "Error in sendline"
5360 assert "Command not found:" not in output, output
5361 assert "Error executing command" not in output, output
5362 assert " not found" not in output, output
Jon Hallcf97cf12017-06-06 09:37:51 -07005363 assert "still updating" not in output, output
Jon Hall2c8959e2016-12-16 12:17:34 -08005364 return main.TRUE
5365 except AssertionError:
5366 main.log.exception( "" )
5367 return main.FALSE
5368 except TypeError:
5369 main.log.exception( self.name + ": Object not as expected" )
5370 return main.FALSE
5371 except pexpect.EOF:
5372 main.log.error( self.name + ": EOF exception found" )
5373 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07005374 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005375 except Exception:
5376 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005377 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005378
5379 def vplsAddIface( self, network, iface ):
5380 """
5381 CLI command to add an interface to a VPLS network.
5382 Required arguments:
5383 network - Name of the network to add the interface to.
5384 iface - The ONOS name for an interface.
5385 returns:
5386 main.TRUE on success and main.FALSE on failure
5387 """
5388 try:
5389 network = str( network )
5390 iface = str( iface )
5391 cmdStr = "vpls add-if "
5392 cmdStr += network + " " + iface
5393 output = self.sendline( cmdStr )
5394 assert output is not None, "Error in sendline"
5395 assert "Command not found:" not in output, output
5396 assert "Error executing command" not in output, output
5397 assert "already associated to network" not in output, output
5398 assert "Interface cannot be added." not in output, output
Jon Hallcf97cf12017-06-06 09:37:51 -07005399 assert "still updating" not in output, output
Jon Hall2c8959e2016-12-16 12:17:34 -08005400 return main.TRUE
5401 except AssertionError:
5402 main.log.exception( "" )
5403 return main.FALSE
5404 except TypeError:
5405 main.log.exception( self.name + ": Object not as expected" )
5406 return main.FALSE
5407 except pexpect.EOF:
5408 main.log.error( self.name + ": EOF exception found" )
5409 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07005410 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005411 except Exception:
5412 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005413 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005414
5415 def vplsRemIface( self, network, iface ):
5416 """
5417 CLI command to remove an interface from a VPLS network.
5418 Required arguments:
5419 network - Name of the network to remove the interface from.
5420 iface - Name of the interface to remove.
5421 returns:
5422 main.TRUE on success and main.FALSE on failure
5423 """
5424 try:
5425 iface = str( iface )
5426 cmdStr = "vpls rem-if "
5427 cmdStr += network + " " + iface
5428 output = self.sendline( cmdStr )
5429 assert output is not None, "Error in sendline"
5430 assert "Command not found:" not in output, output
5431 assert "Error executing command" not in output, output
5432 assert "is not configured" not in output, output
Jon Hallcf97cf12017-06-06 09:37:51 -07005433 assert "still updating" not in output, output
Jon Hall2c8959e2016-12-16 12:17:34 -08005434 return main.TRUE
5435 except AssertionError:
5436 main.log.exception( "" )
5437 return main.FALSE
5438 except TypeError:
5439 main.log.exception( self.name + ": Object not as expected" )
5440 return main.FALSE
5441 except pexpect.EOF:
5442 main.log.error( self.name + ": EOF exception found" )
5443 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07005444 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005445 except Exception:
5446 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005447 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005448
5449 def vplsClean( self ):
5450 """
5451 Description: Clears the VPLS app configuration.
5452 Returns: main.TRUE on success and main.FALSE on failure
5453 """
5454 try:
5455 cmdStr = "vpls clean"
5456 handle = self.sendline( cmdStr )
5457 assert handle is not None, "Error in sendline"
5458 assert "Command not found:" not in handle, handle
Jon Hallcf97cf12017-06-06 09:37:51 -07005459 assert "still updating" not in handle, handle
Jon Hall2c8959e2016-12-16 12:17:34 -08005460 return handle
5461 except AssertionError:
5462 main.log.exception( "" )
5463 return main.FALSE
5464 except TypeError:
5465 main.log.exception( self.name + ": Object not as expected" )
5466 return main.FALSE
5467 except pexpect.EOF:
5468 main.log.error( self.name + ": EOF exception found" )
5469 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07005470 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005471 except Exception:
5472 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005473 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005474
5475 def vplsSetEncap( self, network, encapType ):
5476 """
5477 CLI command to add an interface to a VPLS network.
5478 Required arguments:
5479 network - Name of the network to create.
5480 encapType - Type of encapsulation.
5481 returns:
5482 main.TRUE on success and main.FALSE on failure
5483 """
5484 try:
5485 network = str( network )
5486 encapType = str( encapType ).upper()
5487 assert encapType in [ "MPLS", "VLAN", "NONE" ], "Incorrect type"
5488 cmdStr = "vpls set-encap "
5489 cmdStr += network + " " + encapType
5490 output = self.sendline( cmdStr )
5491 assert output is not None, "Error in sendline"
5492 assert "Command not found:" not in output, output
5493 assert "Error executing command" not in output, output
5494 assert "already associated to network" not in output, output
5495 assert "Encapsulation type " not in output, output
Jon Hallcf97cf12017-06-06 09:37:51 -07005496 assert "still updating" not in output, output
Jon Hall2c8959e2016-12-16 12:17:34 -08005497 return main.TRUE
5498 except AssertionError:
5499 main.log.exception( "" )
5500 return main.FALSE
5501 except TypeError:
5502 main.log.exception( self.name + ": Object not as expected" )
5503 return main.FALSE
5504 except pexpect.EOF:
5505 main.log.error( self.name + ": EOF exception found" )
5506 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07005507 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005508 except Exception:
5509 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005510 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005511
5512 def interfaces( self, jsonFormat=True ):
5513 """
5514 Description: Returns result of interfaces command.
5515 Optional:
5516 * jsonFormat: enable json formatting of output
5517 Returns:
5518 The output of the command or None on error.
5519 """
5520 try:
5521 cmdStr = "interfaces"
5522 if jsonFormat:
Jon Halle0f0b342017-04-18 11:43:47 -07005523 raise NotImplementedError
Jon Hall2c8959e2016-12-16 12:17:34 -08005524 cmdStr += " -j"
5525 handle = self.sendline( cmdStr )
5526 assert handle is not None, "Error in sendline"
5527 assert "Command not found:" not in handle, handle
5528 return handle
5529 except AssertionError:
5530 main.log.exception( "" )
5531 return None
5532 except TypeError:
5533 main.log.exception( self.name + ": Object not as expected" )
5534 return None
5535 except pexpect.EOF:
5536 main.log.error( self.name + ": EOF exception found" )
5537 main.log.error( self.name + ": " + self.handle.before )
Devin Lim44075962017-08-11 10:56:37 -07005538 main.cleanAndExit()
Jon Hall2c8959e2016-12-16 12:17:34 -08005539 except NotImplementedError:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07005540 main.log.exception( self.name + ": Json output not supported" )
Jon Hall2c8959e2016-12-16 12:17:34 -08005541 return None
5542 except Exception:
5543 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005544 main.cleanAndExit()
Chiyu Chengec63bde2016-11-17 18:11:36 -08005545
5546 def getTimeStampFromLog( self, mode, searchTerm, splitTerm_before, splitTerm_after, startLine='', logNum=1 ):
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00005547 '''
Chiyu Chengec63bde2016-11-17 18:11:36 -08005548 Get the timestamp of searchTerm from karaf log.
5549
5550 Arguments:
5551 splitTerm_before and splitTerm_after:
5552
5553 The terms that split the string that contains the timeStamp of
5554 searchTerm. For example, if that string is "xxxxxxxcreationTime =
5555 1419510501xxxxxx", then the splitTerm_before is "CreationTime = "
5556 and the splitTerm_after is "x"
5557
5558 others:
Jon Halle0f0b342017-04-18 11:43:47 -07005559 Please look at the "logsearch" Function in onosclidriver.py
Jeremy Ronquillo4d5f1d02017-10-13 20:23:57 +00005560 '''
Chiyu Chengec63bde2016-11-17 18:11:36 -08005561 if logNum < 0:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07005562 main.log.error( "Get wrong log number ")
Chiyu Chengec63bde2016-11-17 18:11:36 -08005563 return main.ERROR
5564 lines = self.logSearch( mode=mode, searchTerm=searchTerm, startLine=startLine, logNum=logNum )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07005565 if len( lines ) == 0:
Chiyu Chengec63bde2016-11-17 18:11:36 -08005566 main.log.warn( "Captured timestamp string is empty" )
5567 return main.ERROR
5568 lines = lines[ 0 ]
5569 try:
Jeremy Ronquillo82705492017-10-18 14:19:55 -07005570 assert isinstance( lines, str )
Chiyu Chengec63bde2016-11-17 18:11:36 -08005571 # get the target value
5572 line = lines.split( splitTerm_before )
5573 key = line[ 1 ].split( splitTerm_after )
5574 return int( key[ 0 ] )
5575 except IndexError:
5576 main.log.warn( "Index Error!" )
5577 return main.ERROR
5578 except AssertionError:
5579 main.log.warn( "Search Term Not Found " )
5580 return main.ERROR
Jon Halle0f0b342017-04-18 11:43:47 -07005581
5582 def workQueueAdd( self, queueName, value ):
5583 """
5584 CLI command to add a string to the specified Work Queue.
5585 This function uses the distributed primitives test app, which
5586 gives some cli access to distributed primitives for testing
5587 purposes only.
5588
5589 Required arguments:
5590 queueName - The name of the queue to add to
5591 value - The value to add to the queue
5592 returns:
5593 main.TRUE on success, main.FALSE on failure and
5594 main.ERROR on error.
5595 """
5596 try:
5597 queueName = str( queueName )
5598 value = str( value )
5599 prefix = "work-queue-test"
5600 operation = "add"
5601 cmdStr = " ".join( [ prefix, queueName, operation, value ] )
5602 output = self.distPrimitivesSend( cmdStr )
5603 if "Invalid operation name" in output:
5604 main.log.warn( output )
5605 return main.ERROR
5606 elif "Done" in output:
5607 return main.TRUE
5608 except TypeError:
5609 main.log.exception( self.name + ": Object not as expected" )
5610 return main.ERROR
5611 except Exception:
5612 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005613 main.cleanAndExit()
Jon Halle0f0b342017-04-18 11:43:47 -07005614
5615 def workQueueAddMultiple( self, queueName, value1, value2 ):
5616 """
5617 CLI command to add two strings to the specified Work Queue.
5618 This function uses the distributed primitives test app, which
5619 gives some cli access to distributed primitives for testing
5620 purposes only.
5621
5622 Required arguments:
5623 queueName - The name of the queue to add to
5624 value1 - The first value to add to the queue
5625 value2 - The second value to add to the queue
5626 returns:
5627 main.TRUE on success, main.FALSE on failure and
5628 main.ERROR on error.
5629 """
5630 try:
5631 queueName = str( queueName )
5632 value1 = str( value1 )
5633 value2 = str( value2 )
5634 prefix = "work-queue-test"
5635 operation = "addMultiple"
5636 cmdStr = " ".join( [ prefix, queueName, operation, value1, value2 ] )
5637 output = self.distPrimitivesSend( cmdStr )
5638 if "Invalid operation name" in output:
5639 main.log.warn( output )
5640 return main.ERROR
5641 elif "Done" in output:
5642 return main.TRUE
5643 except TypeError:
5644 main.log.exception( self.name + ": Object not as expected" )
5645 return main.ERROR
5646 except Exception:
5647 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005648 main.cleanAndExit()
Jon Halle0f0b342017-04-18 11:43:47 -07005649
5650 def workQueueTakeAndComplete( self, queueName, number=1 ):
5651 """
5652 CLI command to take a value from the specified Work Queue and compelte it.
5653 This function uses the distributed primitives test app, which
5654 gives some cli access to distributed primitives for testing
5655 purposes only.
5656
5657 Required arguments:
5658 queueName - The name of the queue to add to
5659 number - The number of items to take and complete
5660 returns:
5661 main.TRUE on success, main.FALSE on failure and
5662 main.ERROR on error.
5663 """
5664 try:
5665 queueName = str( queueName )
5666 number = str( int( number ) )
5667 prefix = "work-queue-test"
5668 operation = "takeAndComplete"
5669 cmdStr = " ".join( [ prefix, queueName, operation, number ] )
5670 output = self.distPrimitivesSend( cmdStr )
5671 if "Invalid operation name" in output:
5672 main.log.warn( output )
5673 return main.ERROR
5674 elif "Done" in output:
5675 return main.TRUE
5676 except TypeError:
5677 main.log.exception( self.name + ": Object not as expected" )
5678 return main.ERROR
5679 except Exception:
5680 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005681 main.cleanAndExit()
Jon Halle0f0b342017-04-18 11:43:47 -07005682
5683 def workQueueDestroy( self, queueName ):
5684 """
5685 CLI command to destroy the specified Work Queue.
5686 This function uses the distributed primitives test app, which
5687 gives some cli access to distributed primitives for testing
5688 purposes only.
5689
5690 Required arguments:
5691 queueName - The name of the queue to add to
5692 returns:
5693 main.TRUE on success, main.FALSE on failure and
5694 main.ERROR on error.
5695 """
5696 try:
5697 queueName = str( queueName )
5698 prefix = "work-queue-test"
5699 operation = "destroy"
5700 cmdStr = " ".join( [ prefix, queueName, operation ] )
5701 output = self.distPrimitivesSend( cmdStr )
5702 if "Invalid operation name" in output:
5703 main.log.warn( output )
5704 return main.ERROR
5705 return main.TRUE
5706 except TypeError:
5707 main.log.exception( self.name + ": Object not as expected" )
5708 return main.ERROR
5709 except Exception:
5710 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005711 main.cleanAndExit()
Jon Halle0f0b342017-04-18 11:43:47 -07005712
5713 def workQueueTotalPending( self, queueName ):
5714 """
5715 CLI command to get the Total Pending items of the specified Work Queue.
5716 This function uses the distributed primitives test app, which
5717 gives some cli access to distributed primitives for testing
5718 purposes only.
5719
5720 Required arguments:
5721 queueName - The name of the queue to add to
5722 returns:
5723 The number of Pending items in the specified work queue or
5724 None on error
5725 """
5726 try:
5727 queueName = str( queueName )
5728 prefix = "work-queue-test"
5729 operation = "totalPending"
5730 cmdStr = " ".join( [ prefix, queueName, operation ] )
5731 output = self.distPrimitivesSend( cmdStr )
5732 pattern = r'\d+'
5733 if "Invalid operation name" in output:
5734 main.log.warn( output )
5735 return None
5736 else:
5737 match = re.search( pattern, output )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07005738 return match.group( 0 )
Jon Halle0f0b342017-04-18 11:43:47 -07005739 except ( AttributeError, TypeError ):
5740 main.log.exception( self.name + ": Object not as expected; " + str( output ) )
5741 return None
5742 except Exception:
5743 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005744 main.cleanAndExit()
Jon Halle0f0b342017-04-18 11:43:47 -07005745
5746 def workQueueTotalCompleted( self, queueName ):
5747 """
5748 CLI command to get the Total Completed items of the specified Work Queue.
5749 This function uses the distributed primitives test app, which
5750 gives some cli access to distributed primitives for testing
5751 purposes only.
5752
5753 Required arguments:
5754 queueName - The name of the queue to add to
5755 returns:
5756 The number of complete items in the specified work queue or
5757 None on error
5758 """
5759 try:
5760 queueName = str( queueName )
5761 prefix = "work-queue-test"
5762 operation = "totalCompleted"
5763 cmdStr = " ".join( [ prefix, queueName, operation ] )
5764 output = self.distPrimitivesSend( cmdStr )
5765 pattern = r'\d+'
5766 if "Invalid operation name" in output:
5767 main.log.warn( output )
5768 return None
5769 else:
5770 match = re.search( pattern, output )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07005771 return match.group( 0 )
Jon Halle0f0b342017-04-18 11:43:47 -07005772 except ( AttributeError, TypeError ):
5773 main.log.exception( self.name + ": Object not as expected; " + str( output ) )
5774 return None
5775 except Exception:
5776 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005777 main.cleanAndExit()
Jon Halle0f0b342017-04-18 11:43:47 -07005778
5779 def workQueueTotalInProgress( self, queueName ):
5780 """
5781 CLI command to get the Total In Progress items of the specified Work Queue.
5782 This function uses the distributed primitives test app, which
5783 gives some cli access to distributed primitives for testing
5784 purposes only.
5785
5786 Required arguments:
5787 queueName - The name of the queue to add to
5788 returns:
5789 The number of In Progress items in the specified work queue or
5790 None on error
5791 """
5792 try:
5793 queueName = str( queueName )
5794 prefix = "work-queue-test"
5795 operation = "totalInProgress"
5796 cmdStr = " ".join( [ prefix, queueName, operation ] )
5797 output = self.distPrimitivesSend( cmdStr )
5798 pattern = r'\d+'
5799 if "Invalid operation name" in output:
5800 main.log.warn( output )
5801 return None
5802 else:
5803 match = re.search( pattern, output )
Jeremy Ronquillo82705492017-10-18 14:19:55 -07005804 return match.group( 0 )
Jon Halle0f0b342017-04-18 11:43:47 -07005805 except ( AttributeError, TypeError ):
5806 main.log.exception( self.name + ": Object not as expected; " + str( output ) )
5807 return None
5808 except Exception:
5809 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lim44075962017-08-11 10:56:37 -07005810 main.cleanAndExit()
Jeremy Ronquillo818bc7c2017-08-09 17:14:53 +00005811
5812 def events( self, args='-a' ):
5813 """
5814 Description: Returns events -a command output
5815 Optional:
5816 add other arguments
5817 """
5818 try:
5819 cmdStr = "events"
5820 if args:
5821 cmdStr += " " + args
5822 handle = self.sendline( cmdStr )
5823 assert handle is not None, "Error in sendline"
5824 assert "Command not found:" not in handle, handle
5825 return handle
5826 except AssertionError:
5827 main.log.exception( "" )
5828 return None
5829 except TypeError:
5830 main.log.exception( self.name + ": Object not as expected" )
5831 return None
5832 except pexpect.EOF:
5833 main.log.error( self.name + ": EOF exception found" )
5834 main.log.error( self.name + ": " + self.handle.before )
5835 main.cleanAndExit()
5836 except Exception:
5837 main.log.exception( self.name + ": Uncaught exception!" )
5838 main.cleanAndExit()
5839
5840 def getMaster( self, deviceID ):
5841 """
5842 Description: Obtains current master using "roles" command for a specific deviceID
5843 """
5844 try:
5845 return str( self.getRole( deviceID )[ 'master' ] )
5846 except AssertionError:
5847 main.log.exception( "" )
5848 return None
5849 except TypeError:
5850 main.log.exception( self.name + ": Object not as expected" )
5851 return None
5852 except pexpect.EOF:
5853 main.log.error( self.name + ": EOF exception found" )
5854 main.log.error( self.name + ": " + self.handle.before )
5855 main.cleanAndExit()
5856 except Exception:
5857 main.log.exception( self.name + ": Uncaught exception!" )
Devin Lime6fe3c42017-10-18 16:28:40 -07005858 main.cleanAndExit()
Jon Halla478b852017-12-04 15:00:15 -08005859
5860 def issu( self ):
5861 """
5862 Short summary of In-Service Software Upgrade status
5863
5864 Returns the output of the cli command or None on Error
5865 """
5866 try:
5867 cmdStr = "issu"
5868 handle = self.sendline( cmdStr )
5869 assert handle is not None, "Error in sendline"
5870 assert "Command not found:" not in handle, handle
5871 assert "Unsupported command:" not in handle, handle
5872 return handle
5873 except AssertionError:
5874 main.log.exception( "" )
5875 return None
5876 except TypeError:
5877 main.log.exception( self.name + ": Object not as expected" )
5878 return None
5879 except pexpect.EOF:
5880 main.log.error( self.name + ": EOF exception found" )
5881 main.log.error( self.name + ": " + self.handle.before )
5882 main.cleanAndExit()
5883 except Exception:
5884 main.log.exception( self.name + ": Uncaught exception!" )
5885 main.cleanAndExit()
5886
5887 def issuInit( self ):
5888 """
5889 Initiates an In-Service Software Upgrade
5890
5891 Returns main.TRUE on success, main.ERROR on error, else main.FALSE
5892 """
5893 try:
5894 cmdStr = "issu init"
5895 handle = self.sendline( cmdStr )
5896 assert handle is not None, "Error in sendline"
5897 assert "Command not found:" not in handle, handle
5898 assert "Unsupported command:" not in handle, handle
5899 if "Initialized" in handle:
5900 return main.TRUE
5901 else:
5902 return main.FALSE
5903 except AssertionError:
5904 main.log.exception( "" )
5905 return main.ERROR
5906 except TypeError:
5907 main.log.exception( self.name + ": Object not as expected" )
5908 return main.ERROR
5909 except pexpect.EOF:
5910 main.log.error( self.name + ": EOF exception found" )
5911 main.log.error( self.name + ": " + self.handle.before )
5912 main.cleanAndExit()
5913 except Exception:
5914 main.log.exception( self.name + ": Uncaught exception!" )
5915 main.cleanAndExit()
5916
5917 def issuUpgrade( self ):
5918 """
5919 Transitions stores to upgraded nodes
5920
5921 Returns main.TRUE on success, main.ERROR on error, else main.FALSE
5922 """
5923 try:
5924 cmdStr = "issu upgrade"
5925 handle = self.sendline( cmdStr )
5926 assert handle is not None, "Error in sendline"
5927 assert "Command not found:" not in handle, handle
5928 assert "Unsupported command:" not in handle, handle
5929 if "Upgraded" in handle:
5930 return main.TRUE
5931 else:
5932 return main.FALSE
5933 except AssertionError:
5934 main.log.exception( "" )
5935 return main.ERROR
5936 except TypeError:
5937 main.log.exception( self.name + ": Object not as expected" )
5938 return main.ERROR
5939 except pexpect.EOF:
5940 main.log.error( self.name + ": EOF exception found" )
5941 main.log.error( self.name + ": " + self.handle.before )
5942 main.cleanAndExit()
5943 except Exception:
5944 main.log.exception( self.name + ": Uncaught exception!" )
5945 main.cleanAndExit()
5946
5947 def issuCommit( self ):
5948 """
5949 Finalizes an In-Service Software Upgrade
5950
5951 Returns main.TRUE on success, main.ERROR on error, else main.FALSE
5952 """
5953 try:
5954 cmdStr = "issu commit"
5955 handle = self.sendline( cmdStr )
5956 assert handle is not None, "Error in sendline"
5957 assert "Command not found:" not in handle, handle
5958 assert "Unsupported command:" not in handle, handle
5959 # TODO: Check the version returned by this command
5960 if "Committed version" in handle:
5961 return main.TRUE
5962 else:
5963 return main.FALSE
5964 except AssertionError:
5965 main.log.exception( "" )
5966 return main.ERROR
5967 except TypeError:
5968 main.log.exception( self.name + ": Object not as expected" )
5969 return main.ERROR
5970 except pexpect.EOF:
5971 main.log.error( self.name + ": EOF exception found" )
5972 main.log.error( self.name + ": " + self.handle.before )
5973 main.cleanAndExit()
5974 except Exception:
5975 main.log.exception( self.name + ": Uncaught exception!" )
5976 main.cleanAndExit()
5977
5978 def issuRollback( self ):
5979 """
5980 Rolls back an In-Service Software Upgrade
5981
5982 Returns main.TRUE on success, main.ERROR on error, else main.FALSE
5983 """
5984 try:
5985 cmdStr = "issu rollback"
5986 handle = self.sendline( cmdStr )
5987 assert handle is not None, "Error in sendline"
5988 assert "Command not found:" not in handle, handle
5989 assert "Unsupported command:" not in handle, handle
5990 # TODO: Check the version returned by this command
5991 if "Rolled back to version" in handle:
5992 return main.TRUE
5993 else:
5994 return main.FALSE
5995 except AssertionError:
5996 main.log.exception( "" )
5997 return main.ERROR
5998 except TypeError:
5999 main.log.exception( self.name + ": Object not as expected" )
6000 return main.ERROR
6001 except pexpect.EOF:
6002 main.log.error( self.name + ": EOF exception found" )
6003 main.log.error( self.name + ": " + self.handle.before )
6004 main.cleanAndExit()
6005 except Exception:
6006 main.log.exception( self.name + ": Uncaught exception!" )
6007 main.cleanAndExit()
6008
6009 def issuReset( self ):
6010 """
6011 Resets the In-Service Software Upgrade status after a rollback
6012
6013 Returns main.TRUE on success, main.ERROR on error, else main.FALSE
6014 """
6015 try:
6016 cmdStr = "issu reset"
6017 handle = self.sendline( cmdStr )
6018 assert handle is not None, "Error in sendline"
6019 assert "Command not found:" not in handle, handle
6020 assert "Unsupported command:" not in handle, handle
6021 # TODO: Check the version returned by this command
6022 if "Reset version" in handle:
6023 return main.TRUE
6024 else:
6025 return main.FALSE
6026 except AssertionError:
6027 main.log.exception( "" )
6028 return main.ERROR
6029 except TypeError:
6030 main.log.exception( self.name + ": Object not as expected" )
6031 return main.ERROR
6032 except pexpect.EOF:
6033 main.log.error( self.name + ": EOF exception found" )
6034 main.log.error( self.name + ": " + self.handle.before )
6035 main.cleanAndExit()
6036 except Exception:
6037 main.log.exception( self.name + ": Uncaught exception!" )
6038 main.cleanAndExit()
6039
6040 def issuStatus( self ):
6041 """
6042 Status of an In-Service Software Upgrade
6043
6044 Returns the output of the cli command or None on Error
6045 """
6046 try:
6047 cmdStr = "issu status"
6048 handle = self.sendline( cmdStr )
6049 assert handle is not None, "Error in sendline"
6050 assert "Command not found:" not in handle, handle
6051 assert "Unsupported command:" not in handle, handle
6052 return handle
6053 except AssertionError:
6054 main.log.exception( "" )
6055 return None
6056 except TypeError:
6057 main.log.exception( self.name + ": Object not as expected" )
6058 return None
6059 except pexpect.EOF:
6060 main.log.error( self.name + ": EOF exception found" )
6061 main.log.error( self.name + ": " + self.handle.before )
6062 main.cleanAndExit()
6063 except Exception:
6064 main.log.exception( self.name + ": Uncaught exception!" )
6065 main.cleanAndExit()
6066
6067 def issuVersion( self ):
6068 """
6069 Get the version of an In-Service Software Upgrade
6070
6071 Returns the output of the cli command or None on Error
6072 """
6073 try:
6074 cmdStr = "issu version"
6075 handle = self.sendline( cmdStr )
6076 assert handle is not None, "Error in sendline"
6077 assert "Command not found:" not in handle, handle
6078 assert "Unsupported command:" not in handle, handle
6079 return handle
6080 except AssertionError:
6081 main.log.exception( "" )
6082 return None
6083 except TypeError:
6084 main.log.exception( self.name + ": Object not as expected" )
6085 return None
6086 except pexpect.EOF:
6087 main.log.error( self.name + ": EOF exception found" )
6088 main.log.error( self.name + ": " + self.handle.before )
6089 main.cleanAndExit()
6090 except Exception:
6091 main.log.exception( self.name + ": Uncaught exception!" )
6092 main.cleanAndExit()