sapling/eden/scm/tests/test-linkrevcache-linkrevdb.py
Adam Simpkins ab3a7cb21f Move fb-mercurial sources into an eden/scm subdirectory.
Summary:
In preparation for merging fb-mercurial sources to the Eden repository,
move everything from the top-level directory into an `eden/scm`
subdirectory.
2019-11-13 16:04:48 -08:00

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()