def test_rollback_production_one(self, mock_logger):
"""CLI - Can't rollback a function at version 1"""
mock_client = mock.MagicMock()
mock_client.get_alias.return_value = {'FunctionVersion': '1'}
rollback._rollback_production(mock_client, 'test_function')
mock_logger.warning.assert_called_once()
mock_client.update_alias.assert_not_called()
def test_rollback_production_latest(self, mock_logger):
"""CLI - Can't rollback a function at $LATEST"""
mock_client = mock.MagicMock()
mock_client.get_alias.return_value = {'FunctionVersion': '$LATEST'}
rollback._rollback_production(mock_client, 'test_function')
mock_logger.error.assert_called_once()
mock_client.update_alias.assert_not_called()
def test_rollback_production_error(self, mock_logger):
"""CLI - Exception when rolling back alias"""
mock_client = mock.MagicMock()
mock_client.get_alias.return_value = {'FunctionVersion': '3'}
mock_client.update_alias.side_effect = ClientError({}, None)
rollback._rollback_production(mock_client, 'test_function')
mock_logger.assert_has_calls([
mock.call.info('Rolling back %s:production from version %d => %d',
'test_function', 3, 2),
mock.call.exception('version not updated')
])