Mercurial > public > mercurial-scm > hg-stable
diff tests/testlib/ext-sidedata.py @ 43040:ba4072c0a911
sidedata: test we can successfully write sidedata
For now we just write them and rejoice on the lack of crashes.
Differential Revision: https://phab.mercurial-scm.org/D6896
author | Pierre-Yves David <pierre-yves.david@octobus.net> |
---|---|
date | Mon, 09 Sep 2019 21:38:29 +0200 |
parents | |
children | 03e769278ef3 |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/tests/testlib/ext-sidedata.py Mon Sep 09 21:38:29 2019 +0200 @@ -0,0 +1,36 @@ +# ext-sidedata.py - small extension to test the sidedata logic +# +# Copyright 2019 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 + +import hashlib +import struct + +from mercurial import ( + extensions, + revlog, +) + +from mercurial.revlogutils import ( + sidedata, +) + +def wrapaddrevision(orig, self, text, transaction, link, p1, p2, *args, + **kwargs): + if kwargs.get('sidedata') is None: + kwargs['sidedata'] = {} + sd = kwargs['sidedata'] + ## let's store some arbitrary data just for testing + # text length + sd[sidedata.SD_TEST1] = struct.pack('>I', len(text)) + # and sha2 hashes + sha256 = hashlib.sha256(text).digest() + sd[sidedata.SD_TEST2] = struct.pack('>32s', sha256) + return orig(self, text, transaction, link, p1, p2, *args, **kwargs) + +def extsetup(ui): + extensions.wrapfunction(revlog.revlog, 'addrevision', wrapaddrevision)