#!/usr/bin/python #-*- coding: utf-8 -*- # ====================================================================== # Copyright 2016 Julien LE CLEACH # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. # ====================================================================== import sys import unittest class TypesTest(unittest.TestCase): """ Test case for the types module. """ def test_AddressStates(self): """ Test the AddressStates enumeration. """ from supvisors.ttypes import AddressStates self.assertEqual('UNKNOWN', AddressStates._to_string(AddressStates.UNKNOWN)) self.assertEqual('RUNNING', AddressStates._to_string(AddressStates.RUNNING)) self.assertEqual('SILENT', AddressStates._to_string(AddressStates.SILENT)) self.assertEqual('ISOLATING', AddressStates._to_string(AddressStates.ISOLATING)) self.assertEqual('ISOLATED', AddressStates._to_string(AddressStates.ISOLATED)) def test_ApplicationStates(self): """ Test the ApplicationStates enumeration. """ from supvisors.ttypes import ApplicationStates self.assertEqual('STOPPED', ApplicationStates._to_string(ApplicationStates.STOPPED)) self.assertEqual('STARTING', ApplicationStates._to_string(ApplicationStates.STARTING)) self.assertEqual('RUNNING', ApplicationStates._to_string(ApplicationStates.RUNNING)) self.assertEqual('STOPPING', ApplicationStates._to_string(ApplicationStates.STOPPING)) def test_StartingStrategies(self): """ Test the StartingStrategies enumeration. """ from supvisors.ttypes import StartingStrategies self.assertEqual('CONFIG', StartingStrategies._to_string(StartingStrategies.CONFIG)) self.assertEqual('LESS_LOADED', StartingStrategies._to_string(StartingStrategies.LESS_LOADED)) self.assertEqual('MOST_LOADED', StartingStrategies._to_string(StartingStrategies.MOST_LOADED)) def test_ConciliationStrategies(self): """ Test the ConciliationStrategies enumeration. """ from supvisors.ttypes import ConciliationStrategies self.assertEqual('SENICIDE', ConciliationStrategies._to_string(ConciliationStrategies.SENICIDE)) self.assertEqual('INFANTICIDE', ConciliationStrategies._to_string(ConciliationStrategies.INFANTICIDE)) self.assertEqual('USER', ConciliationStrategies._to_string(ConciliationStrategies.USER)) self.assertEqual('STOP', ConciliationStrategies._to_string(ConciliationStrategies.STOP)) self.assertEqual('RESTART', ConciliationStrategies._to_string(ConciliationStrategies.RESTART)) self.assertEqual('RUNNING_FAILURE', ConciliationStrategies._to_string(ConciliationStrategies.RUNNING_FAILURE)) def test_StartingFailureStrategies(self): """ Test the StartingFailureStrategies enumeration. """ from supvisors.ttypes import StartingFailureStrategies self.assertEqual('ABORT', StartingFailureStrategies._to_string(StartingFailureStrategies.ABORT)) self.assertEqual('CONTINUE', StartingFailureStrategies._to_string(StartingFailureStrategies.CONTINUE)) def test_RunningFailureStrategies(self): """ Test the RunningFailureStrategies enumeration. """ from supvisors.ttypes import RunningFailureStrategies self.assertEqual('CONTINUE', RunningFailureStrategies._to_string(RunningFailureStrategies.CONTINUE)) self.assertEqual('RESTART_PROCESS', RunningFailureStrategies._to_string(RunningFailureStrategies.RESTART_PROCESS)) self.assertEqual('STOP_APPLICATION', RunningFailureStrategies._to_string(RunningFailureStrategies.STOP_APPLICATION)) self.assertEqual('RESTART_APPLICATION', RunningFailureStrategies._to_string(RunningFailureStrategies.RESTART_APPLICATION)) def test_SupvisorsStates(self): """ Test the SupvisorsStates enumeration. """ from supvisors.ttypes import SupvisorsStates self.assertEqual('INITIALIZATION', SupvisorsStates._to_string(SupvisorsStates.INITIALIZATION)) self.assertEqual('DEPLOYMENT', SupvisorsStates._to_string(SupvisorsStates.DEPLOYMENT)) self.assertEqual('OPERATION', SupvisorsStates._to_string(SupvisorsStates.OPERATION)) self.assertEqual('CONCILIATION', SupvisorsStates._to_string(SupvisorsStates.CONCILIATION)) self.assertEqual('RESTARTING', SupvisorsStates._to_string(SupvisorsStates.RESTARTING)) self.assertEqual('SHUTTING_DOWN', SupvisorsStates._to_string(SupvisorsStates.SHUTTING_DOWN)) self.assertEqual('SHUTDOWN', SupvisorsStates._to_string(SupvisorsStates.SHUTDOWN)) def test_exception(self): """ Test the exception InvalidTransition. """ from supvisors.ttypes import InvalidTransition # test with unknown attributes with self.assertRaises(InvalidTransition) as exc: raise InvalidTransition('invalid transition') self.assertEqual('invalid transition', str(exc.exception)) def test_suite(): return unittest.findTestCases(sys.modules[__name__]) if __name__ == '__main__': unittest.main(defaultTest='test_suite')