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 state(self, newState): if self._state != newState: if self.check_transition(newState): self._state = newState self.logger.info('Address {} is {}'.format( self.address_name, self.state_string())) else: raise InvalidTransition('Address: transition rejected {} to {}'. format(self.state_string(), AddressStates._to_string(newState)))
def state(self, newState): if self._state != newState: if self.check_transition(newState): self._state = newState self.logger.info('Address {} is {}'.format( self.address_name, self.state_string())) else: raise InvalidTransition( 'Address: transition rejected {} to {}'.format( self.state_string(), AddressStates._to_string(newState)))
def test_timer_event(self): """ Test the handling of a timer event. """ from supvisors.context import Context from supvisors.ttypes import AddressStates context = Context(self.supvisors) with patch.object(self.supvisors.zmq.publisher, 'send_address_status') as mocked_send: # test address states excepting RUNNING: nothing happens for state in [x for x in AddressStates._values() if x != AddressStates.RUNNING]: context.on_timer_event() for address in context.addresses.values(): self.assertEqual(AddressStates.UNKNOWN, address.state) self.assertEqual(0, mocked_send.call_count) # test RUNNING address state with recent local_time test_addresses = ['10.0.0.1', '10.0.0.3', '10.0.0.5'] for address_name in test_addresses: address = context.addresses[address_name] address._state = AddressStates.RUNNING address.local_time = time.time() context.on_timer_event() for address_name in test_addresses: self.assertEqual(AddressStates.RUNNING, context.addresses[address_name].state) for address_name in [x for x in context.addresses.keys() if x not in test_addresses]: self.assertEqual(AddressStates.UNKNOWN, context.addresses[address_name].state) self.assertEqual(0, mocked_send.call_count) # test RUNNING address state with one recent local_time and with auto_fence activated address1 = context.addresses['10.0.0.3'] address1.local_time = time.time() - 100 context.on_timer_event() self.assertEqual(AddressStates.ISOLATING, address1.state) for address_name in [x for x in test_addresses if x != '10.0.0.3']: self.assertEqual(AddressStates.RUNNING, context.addresses[address_name].state) for address_name in [x for x in context.addresses.keys() if x not in test_addresses]: self.assertEqual(AddressStates.UNKNOWN, context.addresses[address_name].state) self.assertEqual(call(address1), mocked_send.call_args) # test with one other recent local_time and with auto_fence deactivated with patch.object(self.supvisors.options, 'auto_fence', False): mocked_send.reset_mock() address2 = context.addresses['10.0.0.5'] address2.local_time = time.time() - 100 address3 = context.addresses['10.0.0.1'] address3.local_time = time.time() - 100 context.on_timer_event() self.assertEqual(AddressStates.SILENT, address2.state) self.assertEqual(AddressStates.SILENT, address3.state) self.assertEqual(AddressStates.ISOLATING, address1.state) for address_name in [x for x in context.addresses.keys() if x not in test_addresses]: self.assertEqual(AddressStates.UNKNOWN, context.addresses[address_name].state) self.assertItemsEqual([call(address2), call(address3)], mocked_send.call_args_list)
def test_transitions(self): """ Test the state transitions of AddressStatus. """ from supvisors.address import AddressStatus from supvisors.ttypes import AddressStates, InvalidTransition status = AddressStatus('10.0.0.1', self.supvisors.logger) for state1 in self.all_states: for state2 in self.all_states: # check all possible transitions from each state status._state = state1 if state2 in status._Transitions[state1]: status.state = state2 self.assertEqual(state2, status.state) self.assertEqual(AddressStates._to_string(state2), status.state_string()) elif state1 == state2: self.assertEqual(state1, status.state) else: with self.assertRaises(InvalidTransition): status.state = state2
def setUp(self): """ Create a logger that stores log traces. """ self.supvisors = DummySupvisors() from supvisors.ttypes import AddressStates self.all_states = AddressStates._values()
def setUp(self): """ Create a logger that stores log traces. """ self.supvisors = MockedSupvisors() from supvisors.ttypes import AddressStates self.all_states = AddressStates._values()
def state_string(self): """ Return the application state as a string. """ return AddressStates._to_string(self.state)
def state_string(self): """ Return the application state as a string. """ return AddressStates._to_string(self.state)
def test_timer_event(self): """ Test the handling of a timer event. """ from supvisors.context import Context from supvisors.ttypes import AddressStates context = Context(self.supvisors) with patch.object(self.supvisors.zmq.publisher, 'send_address_status') as mocked_send: # test address states excepting RUNNING: nothing happens for state in [ x for x in AddressStates._values() if x != AddressStates.RUNNING ]: context.on_timer_event() for address in context.addresses.values(): self.assertEqual(AddressStates.UNKNOWN, address.state) self.assertEqual(0, mocked_send.call_count) # test RUNNING address state with recent local_time test_addresses = ['10.0.0.1', '10.0.0.3', '10.0.0.5'] for address_name in test_addresses: address = context.addresses[address_name] address._state = AddressStates.RUNNING address.local_time = time.time() context.on_timer_event() for address_name in test_addresses: self.assertEqual(AddressStates.RUNNING, context.addresses[address_name].state) for address_name in [ x for x in context.addresses.keys() if x not in test_addresses ]: self.assertEqual(AddressStates.UNKNOWN, context.addresses[address_name].state) self.assertEqual(0, mocked_send.call_count) # test RUNNING address state with one recent local_time and with auto_fence activated address1 = context.addresses['10.0.0.3'] address1.local_time = time.time() - 100 context.on_timer_event() self.assertEqual(AddressStates.ISOLATING, address1.state) for address_name in [x for x in test_addresses if x != '10.0.0.3']: self.assertEqual(AddressStates.RUNNING, context.addresses[address_name].state) for address_name in [ x for x in context.addresses.keys() if x not in test_addresses ]: self.assertEqual(AddressStates.UNKNOWN, context.addresses[address_name].state) self.assertEqual(call(address1), mocked_send.call_args) # test with one other recent local_time and with auto_fence deactivated with patch.object(self.supvisors.options, 'auto_fence', False): mocked_send.reset_mock() address2 = context.addresses['10.0.0.5'] address2.local_time = time.time() - 100 address3 = context.addresses['10.0.0.1'] address3.local_time = time.time() - 100 context.on_timer_event() self.assertEqual(AddressStates.SILENT, address2.state) self.assertEqual(AddressStates.SILENT, address3.state) self.assertEqual(AddressStates.ISOLATING, address1.state) for address_name in [ x for x in context.addresses.keys() if x not in test_addresses ]: self.assertEqual(AddressStates.UNKNOWN, context.addresses[address_name].state) self.assertItemsEqual( [call(address2), call(address3)], mocked_send.call_args_list)