mirror of
https://github.com/facebook/sapling.git
synced 2024-10-07 07:17:55 +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
1.3 KiB
Python
59 lines
1.3 KiB
Python
from __future__ import absolute_import, print_function
|
|
|
|
import tempfile
|
|
|
|
from edenscm.hgext import linkrevcache
|
|
|
|
|
|
try:
|
|
xrange(0)
|
|
except NameError:
|
|
xrange = range
|
|
|
|
|
|
def genhsh(i):
|
|
return chr(i) + b"\0" * 19
|
|
|
|
|
|
def ensure(condition):
|
|
if not condition:
|
|
raise RuntimeError("Unexpected")
|
|
|
|
|
|
def testbasicreadwrite():
|
|
path = tempfile.mkdtemp()
|
|
db = linkrevcache.linkrevdb(path, write=True)
|
|
|
|
ensure(db.getlastrev() == 0)
|
|
for i in xrange(25):
|
|
fname = str(i % 5)
|
|
fnode = genhsh(i // 5)
|
|
ensure(db.getlinkrevs(fname, fnode) == [])
|
|
db.appendlinkrev(fname, fnode, i)
|
|
ensure(db.getlinkrevs(fname, fnode) == [i])
|
|
db.appendlinkrev(fname, fnode, i)
|
|
db.appendlinkrev(fname, fnode, i + 1)
|
|
db.appendlinkrev(fname, fnode, i)
|
|
ensure(db.getlinkrevs(fname, fnode) == [i, i + 1])
|
|
|
|
db.setlastrev(i)
|
|
ensure(db.getlastrev() == i)
|
|
|
|
db.close()
|
|
|
|
# re-open for reading
|
|
db = linkrevcache.linkrevdb(path)
|
|
ensure(db.getlastrev() == 24)
|
|
for i in xrange(25):
|
|
fname = str(i % 5)
|
|
fnode = genhsh(i // 5)
|
|
ensure(db.getlinkrevs(fname, fnode) == [i, i + 1])
|
|
|
|
for i in xrange(26, 50):
|
|
fname = str(i % 5)
|
|
fnode = genhsh(i // 5)
|
|
ensure(db.getlinkrevs(fname, fnode) == [])
|
|
|
|
|
|
testbasicreadwrite()
|