mirror of
https://github.com/facebook/sapling.git
synced 2024-10-08 15:57:43 +03:00
656bf8e5fb
Summary: In an upcoming diff we'll be making hgsubversion pull use a single transaction. We need the lastpulled file to be part of that transaction. The easiest way to do that is to store it in the database. Reviewed By: quark-zju Differential Revision: D13433558 fbshipit-source-id: 013ce09aa7a7382008979a7f424860fc98dc8b5d
200 lines
7.3 KiB
Python
200 lines
7.3 KiB
Python
# no-check-code -- see T24862348
|
|
|
|
import os
|
|
import sys
|
|
|
|
import test_hgsubversion_util
|
|
from hgext.hgsubversion import maps, svncommands, svnmeta, util
|
|
from mercurial import context, extensions, hg, localrepo, util as hgutil
|
|
|
|
|
|
# These test repositories have harmless skew in rebuildmeta for the
|
|
# last-pulled-rev because the last rev in svn causes absolutely no
|
|
# changes in hg.
|
|
expect_youngest_skew = [
|
|
("file_mixed_with_branches.svndump", False, False),
|
|
("file_mixed_with_branches.svndump", True, False),
|
|
("unrelatedbranch.svndump", False, False),
|
|
("unrelatedbranch.svndump", True, False),
|
|
]
|
|
|
|
|
|
def _do_case(self, name, layout):
|
|
subdir = test_hgsubversion_util.subdir.get(name, "")
|
|
single = layout == "single"
|
|
u = test_hgsubversion_util.testui()
|
|
config = {}
|
|
if layout == "custom":
|
|
for branch, path in test_hgsubversion_util.custom.get(name, {}).iteritems():
|
|
config["hgsubversionbranch.%s" % branch] = path
|
|
u.setconfig("hgsubversionbranch", branch, path)
|
|
repo, repo_path = self.load_and_fetch(name, subdir=subdir, layout=layout)
|
|
assert test_hgsubversion_util.repolen(self.repo) > 0
|
|
wc2_path = self.wc_path + "_clone"
|
|
src, dest = test_hgsubversion_util.hgclone(u, self.wc_path, wc2_path, update=False)
|
|
src = test_hgsubversion_util.getlocalpeer(src)
|
|
dest = test_hgsubversion_util.getlocalpeer(dest)
|
|
|
|
# insert a wrapper that prevents calling changectx.children()
|
|
def failfn(orig, ctx):
|
|
self.fail(
|
|
"calling %s is forbidden; it can cause massive slowdowns "
|
|
"when rebuilding large repositories" % orig
|
|
)
|
|
|
|
origchildren = getattr(context.changectx, "children")
|
|
extensions.wrapfunction(context.changectx, "children", failfn)
|
|
|
|
try:
|
|
svncommands.rebuildmeta(
|
|
u, dest, args=[test_hgsubversion_util.fileurl(repo_path + subdir)]
|
|
)
|
|
finally:
|
|
# remove the wrapper
|
|
context.changectx.children = origchildren
|
|
|
|
self._run_assertions(name, single, src, dest, u)
|
|
|
|
wc3_path = self.wc_path + "_partial"
|
|
src, dest = test_hgsubversion_util.hgclone(
|
|
u, self.wc_path, wc3_path, update=False, rev=[0]
|
|
)
|
|
srcrepo = test_hgsubversion_util.getlocalpeer(src)
|
|
dest = test_hgsubversion_util.getlocalpeer(dest)
|
|
|
|
# insert a wrapper that prevents calling changectx.children()
|
|
extensions.wrapfunction(context.changectx, "children", failfn)
|
|
|
|
try:
|
|
svncommands.rebuildmeta(
|
|
u, dest, args=[test_hgsubversion_util.fileurl(repo_path + subdir)]
|
|
)
|
|
finally:
|
|
# remove the wrapper
|
|
context.changectx.children = origchildren
|
|
|
|
if hgutil.safehasattr(localrepo.localrepository, "pull"):
|
|
dest.pull(src)
|
|
else:
|
|
# Mercurial >= 3.2
|
|
from mercurial import exchange
|
|
|
|
exchange.pull(dest, src)
|
|
|
|
# insert a wrapper that prevents calling changectx.children()
|
|
extensions.wrapfunction(context.changectx, "children", failfn)
|
|
try:
|
|
svncommands.updatemeta(
|
|
u, dest, args=[test_hgsubversion_util.fileurl(repo_path + subdir)]
|
|
)
|
|
finally:
|
|
# remove the wrapper
|
|
context.changectx.children = origchildren
|
|
|
|
self._run_assertions(name, single, srcrepo, dest, u)
|
|
|
|
|
|
def _run_assertions(self, name, single, src, dest, u):
|
|
|
|
self.assertTrue(
|
|
os.path.isdir(os.path.join(src.path, "svn")),
|
|
"no .hg/svn directory in the source!",
|
|
)
|
|
self.assertTrue(
|
|
os.path.isdir(os.path.join(dest.path, "svn")),
|
|
"no .hg/svn directory in the destination!",
|
|
)
|
|
dest = hg.repository(u, os.path.dirname(dest.path))
|
|
for tf in ("lastpulled", "rev_map", "uuid", "tagmap", "layout", "subdir"):
|
|
if tf == "lastpulled" and isinstance(src.svnmeta().revmap, maps.SqliteRevMap):
|
|
self.assertEqual(
|
|
src.svnmeta().revmap.lastpulled, dest.svnmeta().revmap.lastpulled
|
|
)
|
|
continue
|
|
|
|
stf = os.path.join(src.path, "svn", tf)
|
|
# the generation of tagmap is lazy so it doesn't strictly need to exist
|
|
# if it's not being used
|
|
if not stf.endswith("tagmap"):
|
|
self.assertTrue(os.path.isfile(stf), "%r is missing!" % stf)
|
|
dtf = os.path.join(dest.path, "svn", tf)
|
|
old, new = None, None
|
|
if not dtf.endswith("tagmap"):
|
|
self.assertTrue(os.path.isfile(dtf), "%r is missing!" % tf)
|
|
if os.path.isfile(stf) and os.path.isfile(dtf):
|
|
old, new = util.load(stf, resave=False), util.load(dtf, resave=False)
|
|
if tf == "lastpulled" and (name, self.stupid, single) in expect_youngest_skew:
|
|
self.assertNotEqual(
|
|
old,
|
|
new,
|
|
"rebuildmeta unexpected match on lastpulled: "
|
|
"old %d new %d, case %r %r %r" % (old, new, name, self.stupid, single),
|
|
)
|
|
continue
|
|
self.assertEqual(old, new, "%s differs old: %r new %r" % (tf, old, new))
|
|
try:
|
|
self.assertEqual(src.branchmap(), dest.branchmap())
|
|
except AttributeError:
|
|
# hg 2.8 and earlier
|
|
self.assertEqual(src.branchtags(), dest.branchtags())
|
|
srcbi = util.load(os.path.join(src.path, "svn", "branch_info"))
|
|
destbi = util.load(os.path.join(dest.path, "svn", "branch_info"))
|
|
self.assertEqual(sorted(srcbi.keys()), sorted(destbi.keys()))
|
|
revmap = svnmeta.SVNMeta(dest).revmap
|
|
# revmap disables __iter__ intentionally to avoid possible slow code
|
|
# (not using database index in SqliteRevMap)
|
|
# we need to fetch all keys so enable it by setting _allowiter
|
|
revmap._allowiter = True
|
|
revkeys = revmap.keys()
|
|
for branch in destbi:
|
|
srcinfo = srcbi[branch]
|
|
destinfo = destbi[branch]
|
|
if srcinfo[:2] == (None, 0) or destinfo[:2] == (None, 0):
|
|
self.assertTrue(
|
|
srcinfo[2] <= destinfo[2],
|
|
"Latest revision for %s decreased from %d to %d!"
|
|
% (branch or "default", srcinfo[2], destinfo[2]),
|
|
)
|
|
self.assertEqual(srcinfo[0], destinfo[0])
|
|
else:
|
|
pr = sorted(
|
|
filter(lambda x: x[1] == srcinfo[0] and x[0] <= srcinfo[1], revkeys),
|
|
reverse=True,
|
|
)[0][0]
|
|
self.assertEqual(pr, destinfo[1])
|
|
self.assertEqual(srcinfo[2], destinfo[2])
|
|
|
|
|
|
def buildmethod(case, name, layout):
|
|
m = lambda self: self._do_case(case, layout)
|
|
m.__name__ = name
|
|
m.__doc__ = "Test rebuildmeta on %s (%s)" % (case, layout)
|
|
return m
|
|
|
|
|
|
skip = set(["project_root_not_repo_root.svndump", "corrupt.svndump"])
|
|
|
|
attrs = {
|
|
"_do_case": _do_case,
|
|
"_run_assertions": _run_assertions,
|
|
"stupid_mode_tests": True,
|
|
}
|
|
for case in [
|
|
f for f in os.listdir(test_hgsubversion_util.FIXTURES) if f.endswith(".svndump")
|
|
]:
|
|
# this fixture results in an empty repository, don't use it
|
|
if case in skip:
|
|
continue
|
|
bname = "test_" + case[: -len(".svndump")]
|
|
attrs[bname] = buildmethod(case, bname, "auto")
|
|
attrs[bname + "_single"] = buildmethod(case, bname + "_single", "single")
|
|
if case in test_hgsubversion_util.custom:
|
|
attrs[bname + "_custom"] = buildmethod(case, bname + "_custom", "single")
|
|
|
|
RebuildMetaTests = type("RebuildMetaTests", (test_hgsubversion_util.TestBase,), attrs)
|
|
|
|
if __name__ == "__main__":
|
|
import silenttestrunner
|
|
|
|
silenttestrunner.main(__name__)
|