# 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