mirror of
https://github.com/facebook/sapling.git
synced 2024-10-11 17:27:53 +03:00
435bb980f7
Summary: At Facebook, we only use the sqlite revmap implementation. Lets run the test in sqlite. Reviewed By: farnz Differential Revision: D16181057 fbshipit-source-id: f4e74f05419d614c7d82772d5ac3e39443e84599
80 lines
3.0 KiB
Python
80 lines
3.0 KiB
Python
from __future__ import absolute_import
|
|
|
|
import test_hgsubversion_util
|
|
from edenscm.hgext.hgsubversion import maps, svnmeta
|
|
|
|
|
|
class TestRevMapMigrate(test_hgsubversion_util.TestBase):
|
|
def tearDown(self):
|
|
# revert changes to defaultrevmapclass
|
|
svnmeta.SVNMeta._defaultrevmapclass = maps.RevMap
|
|
|
|
def _test_revmap_migrate(self, fromclass, toclass):
|
|
# revmap interfaces to test
|
|
getters = [
|
|
lambda x: x.branchedits("the_branch", 3),
|
|
lambda x: x.branchedits("the_branch", 4),
|
|
lambda x: x.branchedits("the_branch", 5),
|
|
lambda x: x.branchedits("the_branch", 6),
|
|
lambda x: x.branchedits(None, 5),
|
|
lambda x: x.branchedits("non_existed", 10),
|
|
lambda x: x.branchmaxrevnum("the_branch", 3),
|
|
lambda x: x.branchmaxrevnum("the_branch", 4),
|
|
lambda x: x.branchmaxrevnum("the_branch", 5),
|
|
lambda x: x.branchmaxrevnum("the_branch", 6),
|
|
lambda x: x.branchmaxrevnum(None, 5),
|
|
lambda x: x.branchmaxrevnum("non_existed", 10),
|
|
lambda x: list(x.revhashes(3)),
|
|
lambda x: list(x.revhashes(4)),
|
|
lambda x: list(x.revhashes(42)),
|
|
lambda x: list(x.revhashes(105)),
|
|
lambda x: x.firstpulled,
|
|
lambda x: x.lastpulled,
|
|
lambda x: x.lasthash,
|
|
]
|
|
|
|
svnmeta.SVNMeta._defaultrevmapclass = fromclass
|
|
repo = self._load_fixture_and_fetch("two_heads.svndump")
|
|
# set to None to the default revmapclass will be used
|
|
repo.ui.setconfig("hgsubversion", "revmapimpl", None)
|
|
meta = svnmeta.SVNMeta(repo)
|
|
self.assertEqual(meta.revmap.__class__, fromclass)
|
|
origrevmap = meta.revmap
|
|
|
|
# insert fake special (duplicated, with '\0') data
|
|
origrevmap[103, None] = b"\0" * 20
|
|
origrevmap[104, None] = b"\0" * 18 + b"cd"
|
|
origrevmap[105, None] = b"ab\0cdefghijklmnopqrs"
|
|
origrevmap[104, None] = b"\0" * 18 + b"\xff\0"
|
|
origrevmap[105, "ab"] = origrevmap[105, None]
|
|
|
|
origvalues = [f(meta.revmap) for f in getters]
|
|
|
|
# migrate to another format (transparently)
|
|
svnmeta.SVNMeta._defaultrevmapclass = toclass
|
|
meta = svnmeta.SVNMeta(repo)
|
|
self.assertEqual(meta.revmap.__class__, toclass)
|
|
|
|
# enable iteration otherwise we cannot use iteritems
|
|
origrevmap._allowiter = True
|
|
for k, v in origrevmap.iteritems():
|
|
newv = meta.revmap[k]
|
|
self.assertEqual(newv, v)
|
|
self.assertEqual(len(newv), 20)
|
|
self.assertEqual(meta.revmap[meta.revmap.hashes()[v]], v)
|
|
|
|
newvalues = [f(meta.revmap) for f in getters]
|
|
self.assertEqual(origvalues, newvalues)
|
|
|
|
def test_revmap_migrate_up(self):
|
|
self._test_revmap_migrate(maps.RevMap, maps.SqliteRevMap)
|
|
|
|
def test_revmap_migrate_down(self):
|
|
self._test_revmap_migrate(maps.SqliteRevMap, maps.RevMap)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import silenttestrunner
|
|
|
|
silenttestrunner.main(__name__)
|