diff --git a/lib/tfw/networking/server/tfw_server.py b/lib/tfw/networking/server/tfw_server.py index d6e6889..b18dc06 100644 --- a/lib/tfw/networking/server/tfw_server.py +++ b/lib/tfw/networking/server/tfw_server.py @@ -38,13 +38,16 @@ class TFWServer: return self._fsm_manager def make_response(self, message): + self.trigger_fsm(message) + message['FSMUpdate'] = self._fsm_updater.get_fsm_state_and_transitions() + return message + + def trigger_fsm(self, message): trigger = message.get('trigger', '') try: self._fsm_manager.trigger(trigger, message) except AttributeError: LOG.debug('FSM failed to execute nonexistent trigger: "%s"', trigger) - message['FSMUpdate'] = self._fsm_updater.get_fsm_state_and_transitions() - return message def proxy_filter(self, message): # pylint: disable=unused-argument,no-self-use