Mercurial > public > mercurial-scm > hg-stable
annotate tests/testlib/ext-phase-report.py @ 36082:3b4d14beac3d
py3: port ext-phase-report.py extension
The custom module importer doesn't run on Python files in the
tests directory. So we need the source to be compatible with
both Python 2 and 3.
Differential Revision: https://phab.mercurial-scm.org/D2145
author | Gregory Szorc <gregory.szorc@gmail.com> |
---|---|
date | Sun, 11 Feb 2018 13:32:18 -0800 |
parents | 67a3204c83c1 |
children | 2372284d9457 |
rev | line source |
---|---|
33459
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
1 # tiny extension to report phase changes during transaction |
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
2 |
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
3 from __future__ import absolute_import |
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
4 |
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
5 def reposetup(ui, repo): |
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
6 |
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
7 def reportphasemove(tr): |
36082
3b4d14beac3d
py3: port ext-phase-report.py extension
Gregory Szorc <gregory.szorc@gmail.com>
parents:
33459
diff
changeset
|
8 for rev, move in sorted(tr.changes[b'phases'].items()): |
33459
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
9 if move[0] is None: |
36082
3b4d14beac3d
py3: port ext-phase-report.py extension
Gregory Szorc <gregory.szorc@gmail.com>
parents:
33459
diff
changeset
|
10 ui.write((b'test-debug-phase: new rev %d: x -> %d\n' |
33459
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
11 % (rev, move[1]))) |
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
12 else: |
36082
3b4d14beac3d
py3: port ext-phase-report.py extension
Gregory Szorc <gregory.szorc@gmail.com>
parents:
33459
diff
changeset
|
13 ui.write((b'test-debug-phase: move rev %d: %d -> %d\n' |
33459
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
14 % (rev, move[0], move[1]))) |
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
15 |
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
16 class reportphaserepo(repo.__class__): |
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
17 def transaction(self, *args, **kwargs): |
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
18 tr = super(reportphaserepo, self).transaction(*args, **kwargs) |
36082
3b4d14beac3d
py3: port ext-phase-report.py extension
Gregory Szorc <gregory.szorc@gmail.com>
parents:
33459
diff
changeset
|
19 tr.addpostclose(b'report-phase', reportphasemove) |
33459
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
20 return tr |
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
21 |
67a3204c83c1
phases: test phases tracking at the transaction level
Boris Feld <boris.feld@octobus.net>
parents:
diff
changeset
|
22 repo.__class__ = reportphaserepo |