mirror of
https://github.com/facebook/sapling.git
synced 2024-10-12 17:58:27 +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
59 lines
2.6 KiB
Python
59 lines
2.6 KiB
Python
# dirstatenonnormalcheck.py - extension to check the consistency of the
|
|
# dirstate's non-normal map
|
|
#
|
|
# For most operations on dirstate, this extensions checks that the nonnormalset
|
|
# contains the right entries.
|
|
# It compares the nonnormal file to a nonnormalset built from the map of all
|
|
# the files in the dirstate to check that they contain the same files.
|
|
|
|
from __future__ import absolute_import
|
|
|
|
from edenscm.mercurial import dirstate, extensions
|
|
|
|
|
|
def nonnormalentries(dmap):
|
|
"""Compute nonnormal entries from dirstate's dmap"""
|
|
res = set()
|
|
for f, e in dmap.iteritems():
|
|
if e[0] != "n" or e[3] == -1:
|
|
res.add(f)
|
|
return res
|
|
|
|
|
|
def checkconsistency(ui, orig, dmap, _nonnormalset, label):
|
|
"""Compute nonnormalset from dmap, check that it matches _nonnormalset"""
|
|
nonnormalcomputedmap = nonnormalentries(dmap)
|
|
if _nonnormalset != nonnormalcomputedmap:
|
|
ui.develwarn("%s call to %s\n" % (label, orig), config="dirstate")
|
|
ui.develwarn("inconsistency in nonnormalset\n", config="dirstate")
|
|
ui.develwarn("[nonnormalset] %s\n" % _nonnormalset, config="dirstate")
|
|
ui.develwarn("[map] %s\n" % nonnormalcomputedmap, config="dirstate")
|
|
|
|
|
|
def _checkdirstate(orig, self, arg):
|
|
"""Check nonnormal set consistency before and after the call to orig"""
|
|
checkconsistency(self._ui, orig, self._map, self._map.nonnormalset, "before")
|
|
r = orig(self, arg)
|
|
checkconsistency(self._ui, orig, self._map, self._map.nonnormalset, "after")
|
|
return r
|
|
|
|
|
|
def extsetup(ui):
|
|
"""Wrap functions modifying dirstate to check nonnormalset consistency"""
|
|
dirstatecl = dirstate.dirstate
|
|
devel = ui.configbool("devel", "all-warnings")
|
|
paranoid = ui.configbool("experimental", "nonnormalparanoidcheck")
|
|
if devel:
|
|
extensions.wrapfunction(dirstatecl, "_writedirstate", _checkdirstate)
|
|
if paranoid:
|
|
# We don't do all these checks when paranoid is disable as it would
|
|
# make the extension run very slowly on large repos
|
|
extensions.wrapfunction(dirstatecl, "normallookup", _checkdirstate)
|
|
extensions.wrapfunction(dirstatecl, "otherparent", _checkdirstate)
|
|
extensions.wrapfunction(dirstatecl, "normal", _checkdirstate)
|
|
extensions.wrapfunction(dirstatecl, "write", _checkdirstate)
|
|
extensions.wrapfunction(dirstatecl, "add", _checkdirstate)
|
|
extensions.wrapfunction(dirstatecl, "remove", _checkdirstate)
|
|
extensions.wrapfunction(dirstatecl, "merge", _checkdirstate)
|
|
extensions.wrapfunction(dirstatecl, "untrack", _checkdirstate)
|