diff -r 57875cf423c9 -r 2372284d9457 tests/testlib/ext-sidedata.py --- a/tests/testlib/ext-sidedata.py Sat Oct 05 10:29:34 2019 -0400 +++ b/tests/testlib/ext-sidedata.py Sun Oct 06 09:45:02 2019 -0400 @@ -16,12 +16,12 @@ revlog, ) -from mercurial.revlogutils import ( - sidedata, -) +from mercurial.revlogutils import sidedata + -def wrapaddrevision(orig, self, text, transaction, link, p1, p2, *args, - **kwargs): +def wrapaddrevision( + orig, self, text, transaction, link, p1, p2, *args, **kwargs +): if kwargs.get('sidedata') is None: kwargs['sidedata'] = {} sd = kwargs['sidedata'] @@ -33,6 +33,7 @@ sd[sidedata.SD_TEST2] = struct.pack('>32s', sha256) return orig(self, text, transaction, link, p1, p2, *args, **kwargs) + def wraprevision(orig, self, nodeorrev, *args, **kwargs): text = orig(self, nodeorrev, *args, **kwargs) if nodeorrev != node.nullrev and nodeorrev != node.nullid: @@ -45,6 +46,7 @@ raise RuntimeError('sha256 mismatch') return text + def extsetup(ui): extensions.wrapfunction(revlog.revlog, 'addrevision', wrapaddrevision) extensions.wrapfunction(revlog.revlog, 'revision', wraprevision)