blob: 688324c335df8499c8ebf23b8e2b2b3cb30cfb35 [file] [log] [blame]
"""
Copyright 2016 Open Networking Foundation ( ONF )
Please refer questions to either the onos test mailing list at <onos-test@onosproject.org>,
the System Testing Plans and Results wiki page at <https://wiki.onosproject.org/x/voMg>,
or the System Testing Guide page at <https://wiki.onosproject.org/x/WYQg>
TestON is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
( at your option ) any later version.
TestON is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with TestON. If not, see <http://www.gnu.org/licenses/>.
"""
"""
This file contains intent class for CHOTestMonkey
Author: you@onlab.us
"""
from threading import Lock
class Controller:
def __init__( self, index ):
self.default = ''
self.index = index
self.ip = main.Cluster.getIps()[ index - 1 ]
self.CLI = None
self.CLILock = Lock()
self.status = 'up'
def setCLI( self, CLI ):
self.CLI = CLI
def startCLI( self ):
return self.CLI.startOnosCli( self.ip )
def isUp( self ):
return self.status == 'up'
def bringDown( self ):
self.status = 'down'
def bringUp( self ):
self.status = 'up'
class Intent:
def __init__( self, id ):
self.default = ''
self.type = 'INTENT'
self.id = id
self.expectedState = 'UNKNOWN'
def isHostIntent( self ):
return self.type == 'INTENT_HOST'
def isPointIntent( self ):
return self.type == 'INTENT_POINT'
def isFailed( self ):
return self.expectedState == 'FAILED'
def isInstalled( self ):
return self.expectedState == 'INSTALLED'
class HostIntent( Intent ):
def __init__( self, id, hostA, hostB ):
Intent.__init__( self, id )
self.type = 'INTENT_HOST'
self.hostA = hostA
self.hostB = hostB
self.deviceA = hostA.device
self.deviceB = hostB.device
def setWithdrawn( self ):
self.expectedState = 'WITHDRAWN'
if self.hostB in self.hostA.correspondents:
self.hostA.correspondents.remove( self.hostB )
if self.hostA in self.hostB.correspondents:
self.hostB.correspondents.remove( self.hostA )
def setFailed( self ):
self.expectedState = 'FAILED'
def setInstalled( self ):
if self.expectedState == 'UNKNOWN':
self.hostA.correspondents.append( self.hostB )
self.hostB.correspondents.append( self.hostA )
self.expectedState = 'INSTALLED'
def __str__( self ):
return "ID: " + self.id
class PointIntent( Intent ):
def __init__( self, id, deviceA, deviceB ):
Intent.__init__( self, id )
self.type = 'INTENT_POINT'
self.deviceA = deviceA
self.deviceB = deviceB
def setWithdrawn( self ):
self.expectedState = 'WITHDRAWN'
for hostA in self.deviceA.hosts:
for hostB in self.deviceB.hosts:
if hostB in hostA.correspondents:
hostA.correspondents.remove( hostB )
def setFailed( self ):
self.expectedState = 'FAILED'
def setInstalled( self ):
if self.expectedState == 'UNKNOWN':
for hostA in self.deviceA.hosts:
for hostB in self.deviceB.hosts:
hostA.correspondents.append( hostB )
self.expectedState = 'INSTALLED'
def __str__( self ):
return "ID: " + self.id