view tests/testlib/crash_transaction_late.py @ 46074:31ecf715efe2

tests: conditionalize the progress timestamp for Windows

It looks like for py2 on Windows, the start date is 1970.  It matches the other
platforms for py3, so I'm just going to match the tests and move on, given that
py2 is on the way out.

Differential Revision: https://phab.mercurial-scm.org/D9541

# tiny extension to abort a transaction very late during test
#
# Copyright 2020 Pierre-Yves David <[e-mail]>
#
# 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