| """ |
| This file contains classes for CHOTestMonkey that are related to application event |
| Author: you@onlab.us |
| """ |
| from tests.CHOTestMonkey.dependencies.events.Event import EventType, EventStates, Event |
| from tests.CHOTestMonkey.dependencies.elements.ONOSElement import HostIntent, PointIntent |
| |
| class IntentEvent( Event ): |
| def __init__( self ): |
| Event.__init__( self ) |
| # The index of the ONOS CLI that is going to run the command |
| self.CLIIndex = 0 |
| |
| def getRandomCorrespondent( self, hostA, connected=True ): |
| import random |
| if connected: |
| candidates = hostA.correspondents |
| else: |
| candidates = [ host for host in main.hosts if host not in hostA.correspondents and host != hostA ] |
| if len( candidates ) == 0: |
| return None |
| hostB = random.sample( candidates, 1 )[0] |
| return hostB |
| |
| def getRandomHostPair( self, connected=True ): |
| import random |
| candidateDict = {} |
| with main.variableLock: |
| for host in main.hosts: |
| correspondent = self.getRandomCorrespondent( host, connected=connected ) |
| if correspondent != None: |
| candidateDict[ host ] = correspondent |
| if candidateDict == {}: |
| return None |
| hostA = random.sample( candidateDict.keys(), 1 )[0] |
| hostB = candidateDict[ hostA ] |
| return [ hostA, hostB ] |
| |
| def getIntentsByType( self, intentType ): |
| intents = [] |
| with main.variableLock: |
| for intent in main.intents: |
| if intent.type == intentType: |
| intents.append( intent ) |
| return intents |
| |
| def getRandomIntentByType( self, intentType ): |
| import random |
| intents = self.getIntentsByType( intentType ) |
| if len( intents ) == 0: |
| return None |
| intent = random.sample( intents, 1 )[ 0 ] |
| return intent |
| |
| class HostIntentEvent( IntentEvent ): |
| def __init__( self ): |
| IntentEvent.__init__( self ) |
| self.hostA = None |
| self.hostB = None |
| |
| def startHostIntentEvent( self ): |
| return EventStates().PASS |
| |
| def startEvent( self, args ): |
| with self.eventLock: |
| main.log.info( "%s - starting event" % ( self.typeString ) ) |
| if self.typeIndex == EventType().APP_INTENT_HOST_ADD or self.typeIndex == EventType().APP_INTENT_HOST_DEL: |
| if len( args ) < 3: |
| main.log.warn( "%s - Not enough arguments: %s" % ( self.typeString, args ) ) |
| return EventStates().ABORT |
| elif len( args ) > 3: |
| main.log.warn( "%s - Too many arguments: %s" % ( self.typeString, args ) ) |
| return EventStates().ABORT |
| if args[ 0 ] == 'random' or args[ 1 ] == 'random': |
| if self.typeIndex == EventType().APP_INTENT_HOST_ADD: |
| hostPairRandom = self.getRandomHostPair( connected=False ) |
| if hostPairRandom == None: |
| main.log.warn( "All host pairs are connected, aborting event" ) |
| return EventStates().ABORT |
| self.hostA = hostPairRandom[ 0 ] |
| self.hostB = hostPairRandom[ 1 ] |
| elif self.typeIndex == EventType().APP_INTENT_HOST_DEL: |
| intent = self.getRandomIntentByType( 'INTENT_HOST' ) |
| if intent == None: |
| main.log.warn( "No host intent for deletion, aborting event" ) |
| return EventStates().ABORT |
| self.hostA = intent.hostA |
| self.hostB = intent.hostB |
| elif args[ 0 ] == args[ 1 ]: |
| main.log.warn( "%s - invalid argument: %s, %s" % ( self.typeString, args[ 0 ], args[ 1 ] ) ) |
| return EventStates().ABORT |
| else: |
| for host in main.hosts: |
| if host.name == args[ 0 ]: |
| self.hostA = host |
| elif host.name == args[ 1 ]: |
| self.hostB = host |
| if self.hostA != None and self.hostB != None: |
| break |
| if self.hostA == None: |
| main.log.warn( "Host %s does not exist: " % ( args[ 0 ] ) ) |
| return EventStates().ABORT |
| if self.hostB == None: |
| main.log.warn( "Host %s does not exist: " % ( args[ 1 ] ) ) |
| return EventStates().ABORT |
| index = int( args[ 2 ] ) |
| if index < 1 or index > int( main.numCtrls ): |
| main.log.warn( "%s - invalid argument: %s" % ( self.typeString, index ) ) |
| return EventStates().ABORT |
| if not main.controllers[ index - 1 ].isUp(): |
| main.log.warn( self.typeString + " - invalid argument: onos %s is down" % ( controller.index ) ) |
| return EventStates().ABORT |
| self.CLIIndex = index |
| return self.startHostIntentEvent() |
| |
| class AddHostIntent( HostIntentEvent ): |
| """ |
| Add a host-to-host intent (bidirectional) |
| """ |
| def __init__( self ): |
| HostIntentEvent.__init__( self ) |
| self.typeString = main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeString' ] |
| self.typeIndex= int( main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeIndex' ] ) |
| |
| def startHostIntentEvent( self ): |
| try: |
| assert self.hostA != None and self.hostB != None |
| # Check whether there already exists some intent for the host pair |
| # For now we should avoid installing overlapping intents |
| for intent in main.intents: |
| if not intent.type == 'INTENT_HOST': |
| continue |
| if intent.hostA == self.hostA and intent.hostB == self.hostB or\ |
| intent.hostB == self.hostA and intent.hostA == self.hostB: |
| main.log.warn( self.typeString + " - find an exiting intent for the host pair, abort installation" ) |
| return EventStates().ABORT |
| controller = main.controllers[ self.CLIIndex - 1 ] |
| with controller.CLILock: |
| id = controller.CLI.addHostIntent( self.hostA.id, self.hostB.id ) |
| if id == None: |
| main.log.warn( self.typeString + " - add host intent failed" ) |
| return EventStates().FAIL |
| with main.variableLock: |
| newHostIntent = HostIntent( id, self.hostA, self.hostB ) |
| if self.hostA.isDown() or self.hostA.isRemoved() or self.hostB.isDown() or self.hostB.isRemoved(): |
| newHostIntent.setFailed() |
| main.intents.append( newHostIntent ) |
| # Update host connectivity status |
| # TODO: should we check whether hostA and hostB are already correspondents? |
| self.hostB.correspondents.append( self.hostA ) |
| self.hostA.correspondents.append( self.hostB ) |
| return EventStates().PASS |
| except Exception: |
| main.log.warn( "Caught exception, aborting event" ) |
| return EventStates().ABORT |
| |
| class DelHostIntent( HostIntentEvent ): |
| """ |
| Delete a host-to-host intent (bidirectional) |
| """ |
| def __init__( self ): |
| HostIntentEvent.__init__( self ) |
| self.typeString = main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeString' ] |
| self.typeIndex= int( main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeIndex' ] ) |
| |
| def startHostIntentEvent( self ): |
| try: |
| assert self.hostA != None and self.hostB != None |
| targetIntent = None |
| for intent in main.intents: |
| if not intent.type == 'INTENT_HOST': |
| continue |
| if intent.hostA == self.hostA and intent.hostB == self.hostB or\ |
| intent.hostB == self.hostA and intent.hostA == self.hostB: |
| targetIntent = intent |
| break |
| if targetIntent == None: |
| main.log.warn( self.typeString + " - intent does not exist" ) |
| return EventStates().FAIL |
| controller = main.controllers[ self.CLIIndex - 1 ] |
| with controller.CLILock: |
| result = controller.CLI.removeIntent( targetIntent.id, purge=True ) |
| if result == None or result == main.FALSE: |
| main.log.warn( self.typeString + " - delete host intent failed" ) |
| return EventStates().FAIL |
| with main.variableLock: |
| main.intents.remove( targetIntent ) |
| # Update host connectivity status |
| self.hostB.correspondents.remove( self.hostA ) |
| self.hostA.correspondents.remove( self.hostB ) |
| return EventStates().PASS |
| except Exception: |
| main.log.warn( "Caught exception, aborting event" ) |
| return EventStates().ABORT |
| |
| class PointIntentEvent( IntentEvent ): |
| def __init__( self ): |
| IntentEvent.__init__( self ) |
| self.deviceA = None |
| self.deviceB = None |
| |
| def startPointIntentEvent( self ): |
| return EventStates().PASS |
| |
| def startEvent( self, args ): |
| with self.eventLock: |
| main.log.info( "%s - starting event" % ( self.typeString ) ) |
| if self.typeIndex == EventType().APP_INTENT_POINT_ADD or self.typeIndex == EventType().APP_INTENT_POINT_DEL: |
| if len( args ) < 3: |
| main.log.warn( "%s - Not enough arguments: %s" % ( self.typeString, args ) ) |
| return EventStates().ABORT |
| elif len( args ) > 4: |
| main.log.warn( "%s - Too many arguments: %s" % ( self.typeString, args ) ) |
| return EventStates().ABORT |
| if args[ 0 ] == 'random' or args[ 1 ] == 'random': |
| if self.typeIndex == EventType().APP_INTENT_POINT_ADD: |
| hostPairRandom = self.getRandomHostPair( connected=False ) |
| if hostPairRandom == None: |
| main.log.warn( "All host pairs are connected, aborting event" ) |
| return EventStates().ABORT |
| self.deviceA = hostPairRandom[ 0 ].device |
| self.deviceB = hostPairRandom[ 1 ].device |
| elif self.typeIndex == EventType().APP_INTENT_POINT_DEL: |
| intent = self.getRandomIntentByType( 'INTENT_POINT' ) |
| if intent == None: |
| main.log.warn( "No point intent for deletion, aborting event" ) |
| return EventStates().ABORT |
| self.deviceA = intent.deviceA |
| self.deviceB = intent.deviceB |
| elif args[ 0 ] == args[ 1 ]: |
| main.log.warn( "%s - invalid argument: %s" % ( self.typeString, args[ 0 ], args[ 1 ] ) ) |
| return EventStates().ABORT |
| else: |
| for device in main.devices: |
| if device.name == args[ 0 ]: |
| self.deviceA = device |
| elif device.name == args[ 1 ]: |
| self.deviceB = device |
| if self.deviceA != None and self.deviceB != None: |
| break |
| if self.deviceA == None: |
| main.log.warn( "Device %s does not exist: " % ( args[ 0 ] ) ) |
| return EventStates().ABORT |
| if self.deviceB == None: |
| main.log.warn( "Device %s does not exist: " % ( args[ 1 ] ) ) |
| return EventStates().ABORT |
| index = int( args[ 2 ] ) |
| if index < 1 or index > int( main.numCtrls ): |
| main.log.warn( "%s - invalid argument: %s" % ( self.typeString, index ) ) |
| return EventStates().ABORT |
| if not main.controllers[ index - 1 ].isUp(): |
| main.log.warn( self.typeString + " - invalid argument: onos %s is down" % ( controller.index ) ) |
| return EventStates().ABORT |
| self.CLIIndex = index |
| if len( args ) == 4 and args[ 3 ] == 'bidirectional': |
| # Install point intents for both directions |
| resultA = self.startPointIntentEvent() |
| [ self.deviceA, self.deviceB ] = [ self.deviceB, self.deviceA ] |
| resultB = self.startPointIntentEvent() |
| if resultA == EventStates().PASS and resultB == EventStates().PASS: |
| return EventStates().PASS |
| else: |
| return EventStates().FAIL |
| else: |
| return self.startPointIntentEvent() |
| |
| class AddPointIntent( PointIntentEvent ): |
| """ |
| Add a point-to-point intent |
| """ |
| def __init__( self ): |
| PointIntentEvent.__init__( self ) |
| self.typeString = main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeString' ] |
| self.typeIndex= int( main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeIndex' ] ) |
| |
| def startPointIntentEvent( self ): |
| try: |
| assert self.deviceA != None and self.deviceB != None |
| controller = main.controllers[ self.CLIIndex - 1 ] |
| # TODO: the following check only work when we use default port number for point intents |
| # Check whether there already exists some intent for the device pair |
| # For now we should avoid installing overlapping intents |
| for intent in main.intents: |
| if not intent.type == 'INTENT_POINT': |
| continue |
| if intent.deviceA == self.deviceA and intent.deviceB == self.deviceB: |
| main.log.warn( self.typeString + " - find an exiting intent for the device pair, abort installation" ) |
| return EventStates().ABORT |
| controller = main.controllers[ self.CLIIndex - 1 ] |
| with controller.CLILock: |
| # TODO: handle the case that multiple hosts attach to one device |
| srcMAC = "" |
| dstMAC = "" |
| if len( self.deviceA.hosts ) > 0: |
| srcMAC = self.deviceA.hosts[ 0 ].mac |
| if len( self.deviceB.hosts ) > 0: |
| dstMAC = self.deviceB.hosts[ 0 ].mac |
| id = controller.CLI.addPointIntent( self.deviceA.dpid, self.deviceB.dpid, |
| 1, 1, '', srcMAC, dstMAC ) |
| if id == None: |
| main.log.warn( self.typeString + " - add point intent failed" ) |
| return EventStates().FAIL |
| with main.variableLock: |
| newPointIntent = PointIntent( id, self.deviceA, self.deviceB ) |
| if self.deviceA.isDown() or self.deviceB.isDown() or self.deviceA.isRemoved() or self.deviceB.isRemoved(): |
| newPointIntent.setFailed() |
| main.intents.append( newPointIntent ) |
| # Update host connectivity status |
| for hostA in self.deviceA.hosts: |
| for hostB in self.deviceB.hosts: |
| hostA.correspondents.append( hostB ) |
| return EventStates().PASS |
| except Exception: |
| main.log.warn( "Caught exception, aborting event" ) |
| return EventStates().ABORT |
| |
| class DelPointIntent( PointIntentEvent ): |
| """ |
| Delete a point-to-point intent |
| """ |
| def __init__( self ): |
| PointIntentEvent.__init__( self ) |
| self.typeString = main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeString' ] |
| self.typeIndex= int( main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeIndex' ] ) |
| |
| def startPointIntentEvent( self ): |
| try: |
| assert self.deviceA != None and self.deviceB != None |
| targetIntent = None |
| for intent in main.intents: |
| if not intent.type == 'INTENT_POINT': |
| continue |
| if intent.deviceA == self.deviceA and intent.deviceB == self.deviceB: |
| targetIntent = intent |
| break |
| if targetIntent == None: |
| main.log.warn( self.typeString + " - intent does not exist" ) |
| return EventStates().FAIL |
| controller = main.controllers[ self.CLIIndex - 1 ] |
| with controller.CLILock: |
| result = controller.CLI.removeIntent( targetIntent.id, purge=True ) |
| if result == None or result == main.FALSE: |
| main.log.warn( self.typeString + " - delete point intent failed" ) |
| return EventStates().FAIL |
| with main.variableLock: |
| main.intents.remove( targetIntent ) |
| # Update host connectivity status |
| for hostA in self.deviceA.hosts: |
| for hostB in self.deviceB.hosts: |
| hostA.correspondents.remove( hostB ) |
| return EventStates().PASS |
| except Exception: |
| main.log.warn( "Caught exception, aborting event" ) |
| return EventStates().ABORT |