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
206 lines
4.8 KiB
Python
206 lines
4.8 KiB
Python
from __future__ import absolute_import, print_function
|
|
|
|
import os
|
|
import tempfile
|
|
|
|
from edenscm.hgext.fastannotate import error, revmap
|
|
|
|
|
|
try:
|
|
xrange(0)
|
|
except NameError:
|
|
xrange = range
|
|
|
|
|
|
def genhsh(i):
|
|
return chr(i) + b"\0" * 19
|
|
|
|
|
|
def gettemppath():
|
|
fd, path = tempfile.mkstemp()
|
|
os.unlink(path)
|
|
os.close(fd)
|
|
return path
|
|
|
|
|
|
def ensure(condition):
|
|
if not condition:
|
|
raise RuntimeError("Unexpected")
|
|
|
|
|
|
def testbasicreadwrite():
|
|
path = gettemppath()
|
|
|
|
rm = revmap.revmap(path)
|
|
ensure(rm.maxrev == 0)
|
|
for i in xrange(5):
|
|
ensure(rm.rev2hsh(i) is None)
|
|
ensure(rm.hsh2rev(b"\0" * 20) is None)
|
|
|
|
paths = ["", "a", None, "b", "b", "c", "c", None, "a", "b", "a", "a"]
|
|
for i in xrange(1, 5):
|
|
ensure(rm.append(genhsh(i), sidebranch=(i & 1), path=paths[i]) == i)
|
|
|
|
ensure(rm.maxrev == 4)
|
|
for i in xrange(1, 5):
|
|
ensure(rm.hsh2rev(genhsh(i)) == i)
|
|
ensure(rm.rev2hsh(i) == genhsh(i))
|
|
|
|
# re-load and verify
|
|
rm.flush()
|
|
rm = revmap.revmap(path)
|
|
ensure(rm.maxrev == 4)
|
|
for i in xrange(1, 5):
|
|
ensure(rm.hsh2rev(genhsh(i)) == i)
|
|
ensure(rm.rev2hsh(i) == genhsh(i))
|
|
ensure(bool(rm.rev2flag(i) & revmap.sidebranchflag) == bool(i & 1))
|
|
|
|
# append without calling save() explicitly
|
|
for i in xrange(5, 12):
|
|
ensure(rm.append(genhsh(i), sidebranch=(i & 1), path=paths[i], flush=True) == i)
|
|
|
|
# re-load and verify
|
|
rm = revmap.revmap(path)
|
|
ensure(rm.maxrev == 11)
|
|
for i in xrange(1, 12):
|
|
ensure(rm.hsh2rev(genhsh(i)) == i)
|
|
ensure(rm.rev2hsh(i) == genhsh(i))
|
|
ensure(rm.rev2path(i) == paths[i] or paths[i - 1])
|
|
ensure(bool(rm.rev2flag(i) & revmap.sidebranchflag) == bool(i & 1))
|
|
|
|
os.unlink(path)
|
|
|
|
# missing keys
|
|
ensure(rm.rev2hsh(12) is None)
|
|
ensure(rm.rev2hsh(0) is None)
|
|
ensure(rm.rev2hsh(-1) is None)
|
|
ensure(rm.rev2flag(12) is None)
|
|
ensure(rm.rev2path(12) is None)
|
|
ensure(rm.hsh2rev(b"\1" * 20) is None)
|
|
|
|
# illformed hash (not 20 bytes)
|
|
try:
|
|
rm.append(b"\0")
|
|
ensure(False)
|
|
except Exception:
|
|
pass
|
|
|
|
|
|
def testcorruptformat():
|
|
path = gettemppath()
|
|
|
|
# incorrect header
|
|
with open(path, "w") as f:
|
|
f.write(b"NOT A VALID HEADER")
|
|
try:
|
|
revmap.revmap(path)
|
|
ensure(False)
|
|
except error.CorruptedFileError:
|
|
pass
|
|
|
|
# rewrite the file
|
|
os.unlink(path)
|
|
rm = revmap.revmap(path)
|
|
rm.append(genhsh(0), flush=True)
|
|
|
|
rm = revmap.revmap(path)
|
|
ensure(rm.maxrev == 1)
|
|
|
|
# corrupt the file by appending a byte
|
|
size = os.stat(path).st_size
|
|
with open(path, "a") as f:
|
|
f.write("\xff")
|
|
try:
|
|
revmap.revmap(path)
|
|
ensure(False)
|
|
except error.CorruptedFileError:
|
|
pass
|
|
|
|
# corrupt the file by removing the last byte
|
|
ensure(size > 0)
|
|
with open(path, "w") as f:
|
|
f.truncate(size - 1)
|
|
try:
|
|
revmap.revmap(path)
|
|
ensure(False)
|
|
except error.CorruptedFileError:
|
|
pass
|
|
|
|
os.unlink(path)
|
|
|
|
|
|
def testcopyfrom():
|
|
path = gettemppath()
|
|
rm = revmap.revmap(path)
|
|
for i in xrange(1, 10):
|
|
ensure(rm.append(genhsh(i), sidebranch=(i & 1), path=str(i // 3)) == i)
|
|
rm.flush()
|
|
|
|
# copy rm to rm2
|
|
rm2 = revmap.revmap()
|
|
rm2.copyfrom(rm)
|
|
path2 = gettemppath()
|
|
rm2.path = path2
|
|
rm2.flush()
|
|
|
|
# two files should be the same
|
|
ensure(len(set(open(p).read() for p in [path, path2])) == 1)
|
|
|
|
os.unlink(path)
|
|
os.unlink(path2)
|
|
|
|
|
|
class fakefctx(object):
|
|
def __init__(self, node, path=None):
|
|
self._node = node
|
|
self._path = path
|
|
|
|
def node(self):
|
|
return self._node
|
|
|
|
def path(self):
|
|
return self._path
|
|
|
|
|
|
def testcontains():
|
|
path = gettemppath()
|
|
|
|
rm = revmap.revmap(path)
|
|
for i in xrange(1, 5):
|
|
ensure(rm.append(genhsh(i), sidebranch=(i & 1)) == i)
|
|
|
|
for i in xrange(1, 5):
|
|
ensure(((genhsh(i), None) in rm) == ((i & 1) == 0))
|
|
ensure((fakefctx(genhsh(i)) in rm) == ((i & 1) == 0))
|
|
for i in xrange(5, 10):
|
|
ensure(fakefctx(genhsh(i)) not in rm)
|
|
ensure((genhsh(i), None) not in rm)
|
|
|
|
# "contains" checks paths
|
|
rm = revmap.revmap()
|
|
for i in xrange(1, 5):
|
|
ensure(rm.append(genhsh(i), path=str(i // 2)) == i)
|
|
for i in xrange(1, 5):
|
|
ensure(fakefctx(genhsh(i), path=str(i // 2)) in rm)
|
|
ensure(fakefctx(genhsh(i), path="a") not in rm)
|
|
|
|
|
|
def testlastnode():
|
|
path = gettemppath()
|
|
ensure(revmap.getlastnode(path) is None)
|
|
rm = revmap.revmap(path)
|
|
ensure(revmap.getlastnode(path) is None)
|
|
for i in xrange(1, 10):
|
|
hsh = genhsh(i)
|
|
rm.append(hsh, path=str(i // 2), flush=True)
|
|
ensure(revmap.getlastnode(path) == hsh)
|
|
rm2 = revmap.revmap(path)
|
|
ensure(rm2.rev2hsh(rm2.maxrev) == hsh)
|
|
|
|
|
|
testbasicreadwrite()
|
|
testcorruptformat()
|
|
testcopyfrom()
|
|
testcontains()
|
|
testlastnode()
|