mirror of
https://github.com/facebook/sapling.git
synced 2024-10-11 09:17:30 +03:00
9dc21f8d0b
Summary: D13853115 adds `edenscm/` to `sys.path` and code still uses `import mercurial`. That has nasty problems if both `import mercurial` and `import edenscm.mercurial` are used, because Python would think `mercurial.foo` and `edenscm.mercurial.foo` are different modules so code like `try: ... except mercurial.error.Foo: ...`, or `isinstance(x, mercurial.foo.Bar)` would fail to handle the `edenscm.mercurial` version. There are also some module-level states (ex. `extensions._extensions`) that would cause trouble if they have multiple versions in a single process. Change imports to use the `edenscm` so ideally the `mercurial` is no longer imported at all. Add checks in extensions.py to catch unexpected extensions importing modules from the old (wrong) locations when running tests. Reviewed By: phillco Differential Revision: D13868981 fbshipit-source-id: f4e2513766957fd81d85407994f7521a08e4de48
76 lines
2.8 KiB
Python
76 lines
2.8 KiB
Python
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")
|
|
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__)
|