def test_wait_post_reboot_inform_transition(self):
"""Testing if SM steps in and out of BaicellsQRTBWaitInformRebootState as per state map"""
acs_state_machine = provision_clean_sm(state='wait_reboot')
acs_state_machine.handle_tr069_message(models.RebootResponse())
self.assertIsInstance(
acs_state_machine.state,
BaicellsQRTBWaitInformRebootState,
)
msg = Tr069MessageBuilder.get_inform(event_codes=['1 BOOT'])
acs_state_machine.handle_tr069_message(msg)
self.assertIsInstance(
acs_state_machine.state,
BaicellsQRTBQueuedEventsWaitState,
)
def get_reboot_response(cls) -> models.RebootResponse:
return models.RebootResponse()
def _get_reboot_response(self) -> models.RebootResponse:
return models.RebootResponse()