def testWarnAboutDestructiveActionRejected(self, promptWithTimeoutMock,
_modelCheckpointMgrModuleMock,
_messageBusConnectorClassMock,
_amqpModuleMock,
_repositoryMock):
# NOTE: rejection can be anything other than the expected input of
# "Yes-<randint>"
promptWithTimeoutMock.return_value = "No"
with self.assertRaises(
reset_all_taurus_engine_data.UserAbortedOperation):
reset_all_taurus_engine_data._warnAboutDestructiveAction()
def testWarnAboutDestructiveActionRejected(self,
promptWithTimeoutMock,
_modelCheckpointMgrModuleMock,
_messageBusConnectorClassMock,
_amqpModuleMock,
_repositoryMock):
# NOTE: rejection can be anything other than the expected input of
# "Yes-<randint>"
promptWithTimeoutMock.return_value = "No"
with self.assertRaises(reset_all_taurus_engine_data.UserAbortedOperation):
reset_all_taurus_engine_data._warnAboutDestructiveAction()
def testWarnAboutDestructiveActionTimedOut(self, promptWithTimeoutMock,
_modelCheckpointMgrModuleMock,
_messageBusConnectorClassMock,
_amqpModuleMock,
_repositoryMock):
# NOTE: py.test by default captures console output and patches console input
# such that raw_input fails. Although not ideal, we have to patch
# raw_input with something else that blocks and will be interrupted by
# SIGINT.
promptWithTimeoutMock.side_effect = nta.utils.prompt_utils.PromptTimeout
with self.assertRaises(nta.utils.prompt_utils.PromptTimeout):
reset_all_taurus_engine_data._warnAboutDestructiveAction()
def testWarnAboutDestructiveActionTimedOut(self,
promptWithTimeoutMock,
_modelCheckpointMgrModuleMock,
_messageBusConnectorClassMock,
_amqpModuleMock,
_repositoryMock):
# NOTE: py.test by default captures console output and patches console input
# such that raw_input fails. Although not ideal, we have to patch
# raw_input with something else that blocks and will be interrupted by
# SIGINT.
promptWithTimeoutMock.side_effect = nta.utils.prompt_utils.PromptTimeout
with self.assertRaises(nta.utils.prompt_utils.PromptTimeout):
reset_all_taurus_engine_data._warnAboutDestructiveAction()
def testWarnAboutDestructiveActionConfirmed(self, # pylint: disable=R0201
randintMock,
promptWithTimeoutMock,
_modelCheckpointMgrModuleMock,
_messageBusConnectorClassMock,
_amqpModuleMock,
_repositoryMock):
randintMock.return_value = 1
promptWithTimeoutMock.return_value = "Yes-1"
reset_all_taurus_engine_data._warnAboutDestructiveAction()
promptWithTimeoutMock.assert_called_once_with(
promptText=ANY,
timeout=reset_all_taurus_engine_data._WARNING_PROMPT_TIMEOUT_SEC)
def testWarnAboutDestructiveActionConfirmed(
self, # pylint: disable=R0201
randintMock,
promptWithTimeoutMock,
_modelCheckpointMgrModuleMock,
_messageBusConnectorClassMock,
_amqpModuleMock,
_repositoryMock):
randintMock.return_value = 1
promptWithTimeoutMock.return_value = "Yes-1"
reset_all_taurus_engine_data._warnAboutDestructiveAction()
promptWithTimeoutMock.assert_called_once_with(
promptText=ANY,
timeout=reset_all_taurus_engine_data._WARNING_PROMPT_TIMEOUT_SEC)