1 from __future__ import absolute_import

     2 import unittest

     3 

     4 try:

     5     from mercurial import rustext

     6 

     7     rustext.__name__  # trigger immediate actual import

     8 except ImportError:

     9     rustext = None

    10 else:

    11     from mercurial.rustext import revlog

    12 

    13 from mercurial.testing import revlog as revlogtesting

    14 

    15 

    16 @unittest.skipIf(

    17     rustext is None, "rustext module revlog relies on is not available",

    18 )

    19 class RustRevlogIndexTest(revlogtesting.RevlogBasedTestBase):

    20     def test_heads(self):

    21         idx = self.parseindex()

    22         rustidx = revlog.MixedIndex(idx)

    23         self.assertEqual(rustidx.headrevs(), idx.headrevs())

    24 

    25     def test_len(self):

    26         idx = self.parseindex()

    27         rustidx = revlog.MixedIndex(idx)

    28         self.assertEqual(len(rustidx), len(idx))

    29 

    30 

    31 if __name__ == '__main__':

    32     import silenttestrunner

    33 

    34     silenttestrunner.main(__name__)