1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
|
# 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 mercurial import (
error,
)
def abort(fp):
raise error.Abort(b"This is a late abort")
def reposetup(ui, repo):
class LateAbortRepo(repo.__class__):
def transaction(self, *args, **kwargs):
tr = super().transaction(*args, **kwargs)
tr.addfilegenerator(
b'late-abort',
[b'late-abort'],
abort,
order=9999999,
post_finalize=True,
)
return tr
repo.__class__ = LateAbortRepo
|