--- 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)