tests/testlib/ext-sidedata.py
changeset 47012 d55b71393907
parent 46718 ba8e508a8e69
child 47084 27f1191b1305
equal deleted inserted replaced
46992:5fa019ceb499 47012:d55b71393907
     8 from __future__ import absolute_import
     8 from __future__ import absolute_import
     9 
     9 
    10 import hashlib
    10 import hashlib
    11 import struct
    11 import struct
    12 
    12 
    13 from mercurial.node import (
    13 from mercurial.node import nullrev
    14     nullid,
       
    15     nullrev,
       
    16 )
       
    17 from mercurial import (
    14 from mercurial import (
    18     extensions,
    15     extensions,
    19     requirements,
    16     requirements,
    20     revlog,
    17     revlog,
    21 )
    18 )
    44     text, sd = orig(self, nodeorrev, *args, **kwargs)
    41     text, sd = orig(self, nodeorrev, *args, **kwargs)
    45     if getattr(self, 'sidedatanocheck', False):
    42     if getattr(self, 'sidedatanocheck', False):
    46         return text, sd
    43         return text, sd
    47     if self.version & 0xFFFF != 2:
    44     if self.version & 0xFFFF != 2:
    48         return text, sd
    45         return text, sd
    49     if nodeorrev != nullrev and nodeorrev != nullid:
    46     if nodeorrev != nullrev and nodeorrev != self.nullid:
    50         cat1 = sd.get(sidedata.SD_TEST1)
    47         cat1 = sd.get(sidedata.SD_TEST1)
    51         if cat1 is not None and len(text) != struct.unpack('>I', cat1)[0]:
    48         if cat1 is not None and len(text) != struct.unpack('>I', cat1)[0]:
    52             raise RuntimeError('text size mismatch')
    49             raise RuntimeError('text size mismatch')
    53         expected = sd.get(sidedata.SD_TEST2)
    50         expected = sd.get(sidedata.SD_TEST2)
    54         got = hashlib.sha256(text).digest()
    51         got = hashlib.sha256(text).digest()