tests/testlib/crash_transaction_late.py
author Gregory Szorc <gregory.szorc@gmail.com>
Sun, 19 Apr 2020 13:29:50 -0700
branchstable
changeset 44739 828d3277618a
parent 44634 01b0805534bb
child 48685 21ac6aedd5e5
permissions -rw-r--r--
automation: always use latest Windows AMI The old AMI isn't available any more. We seem to run into this problem every few months. Amazon (or Microsoft) appears to be removing the old AMIs when they are superseded or something. Let's give up on tracking known images and switch the image selection logic to use the latest published image. Differential Revision: https://phab.mercurial-scm.org/D8465

# tiny extension to abort a transaction very late during test
#
# Copyright 2020 Pierre-Yves David <pierre-yves.david@octobus.net>
#
# This software may be used and distributed according to the terms of the
# GNU General Public License version 2 or any later version.

from __future__ import absolute_import

from mercurial import (
    error,
    transaction,
)


def abort(fp):
    raise error.Abort(b"This is a late abort")


def reposetup(ui, repo):

    transaction.postfinalizegenerators.add(b'late-abort')

    class LateAbortRepo(repo.__class__):
        def transaction(self, *args, **kwargs):
            tr = super(LateAbortRepo, self).transaction(*args, **kwargs)
            tr.addfilegenerator(
                b'late-abort', [b'late-abort'], abort, order=9999999
            )
            return tr

    repo.__class__ = LateAbortRepo