TestON/tests/CHOTestMonkey/dependencies/events/TestEvent.py

"""
This file contains classes for CHOTestMonkey that are related to check event
Author: you@onlab.us
"""
from tests.CHOTestMonkey.dependencies.events.Event import EventType, EventStates, Event


class TestEvent( Event ):

    def __init__( self ):
        Event.__init__( self )

    def startTestEvent( self ):
        return EventStates().PASS

    def startEvent( self, args ):
        with self.eventLock:
            main.log.info( "%s - starting event" % ( self.typeString ) )
            result = self.startTestEvent( args )
            return result


class TestPause( TestEvent ):

    def __init__( self ):
        TestEvent.__init__( self )
        self.typeString = main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeString' ]
        self.typeIndex = int( main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeIndex' ] )

    def startTestEvent( self, args=None ):
        result = EventStates().PASS
        main.eventScheduler.setRunningState( False )
        return result


class TestResume( TestEvent ):

    def __init__( self ):
        TestEvent.__init__( self )
        self.typeString = main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeString' ]
        self.typeIndex = int( main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeIndex' ] )

    def startTestEvent( self, args=None ):
        result = EventStates().PASS
        main.eventScheduler.setRunningState( True )
        return result


class TestSleep( TestEvent ):

    def __init__( self ):
        TestEvent.__init__( self )
        self.typeString = main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeString' ]
        self.typeIndex = int( main.params[ 'EVENT' ][ self.__class__.__name__ ][ 'typeIndex' ] )

    def startTestEvent( self, args ):
        import time
        result = EventStates().PASS
        if len( args ) < 1:
            main.log.warn( "%s - Not enough arguments: %s" % ( self.typeString, args ) )
            result = EventStates().ABORT
        elif len( args ) > 1:
            main.log.warn( "%s - Too many arguments: %s" % ( self.typeString, args ) )
            result = EventStates().ABORT
        sleepTime = int( args[ 0 ] )
        time.sleep( sleepTime )
        return result