def test_close_client_error(self, mock_abort, mock_executor,
mock_logger_warning, mock_session):
ce = ClientError(MOCK_CLIENT_ERROR_MESSAGE, MOCK_ERROR_MESSAGE)
mock_abort.side_effect = ce
transaction = Transaction(mock_session, MOCK_READ_AHEAD, MOCK_ID,
mock_executor)
transaction._close()
mock_abort.assert_called_once_with()
mock_logger_warning.assert_called_once()
def test_close(self, mock_abort, mock_executor, mock_session):
transaction = Transaction(mock_session, MOCK_READ_AHEAD, MOCK_ID,
mock_executor)
transaction._close()
mock_abort.assert_called_once_with()