sapling/eden/scm/tests/test-context.py
Durham Goode 4409ebb103 py3: fix test-context.py
Summary:
It now works with python 3.  I dropped some test output.  In one case
it was confirming that various path encoding work, but we only support utf8 now.
I also dropped some bit that enabled debug output, since the output order was
not the same between python 2 and 3 and it seemed non-trivial to fix.

Reviewed By: xavierd

Differential Revision: D22101132

fbshipit-source-id: 682e7c61f1dd18cce34bdc014c85d560f883bd18
2020-06-19 13:40:17 -07:00

221 lines
6.0 KiB
Python

from __future__ import absolute_import, print_function
import os
from edenscm.mercurial import context, encoding, hg, pycompat, scmutil, ui as uimod
from edenscm.mercurial.node import hex
from hghave import require
u = uimod.ui.load()
u.setconfig("extensions", "treemanifest", "!")
repo = hg.repository(u, "test1", create=1)
os.chdir("test1")
# create 'foo' with fixed time stamp
f = open("foo", "wb")
f.write(b"foo\n")
f.close()
os.utime("foo", (1000, 1000))
# add+commit 'foo'
repo[None].add(["foo"])
repo.commit(text="commit1", date="0 0")
d = repo[None]["foo"].date()
if os.name == "nt":
d = d[:2]
print("workingfilectx.date = (%d, %d)" % d)
# test memctx with non-ASCII commit message
def filectxfn(repo, memctx, path):
return context.memfilectx(repo, memctx, "foo", b"")
ctx = context.memctx(
repo, ["tip", None], encoding.tolocal("Gr\xc3\xbcezi!"), ["foo"], filectxfn
)
ctx.commit()
# test performing a status
def getfilectx(repo, memctx, f):
fctx = memctx.parents()[0][f]
data, flags = fctx.data(), fctx.flags()
if f == "foo":
data += b"bar\n"
return context.memfilectx(repo, memctx, f, data, "l" in flags, "x" in flags)
ctxa = repo.changectx(0)
ctxb = context.memctx(
repo,
[ctxa.node(), None],
"test diff",
["foo"],
getfilectx,
ctxa.user(),
ctxa.date(),
)
print(ctxb.status(ctxa))
# test performing a diff on a memctx
for d in ctxb.diff(ctxa, git=True):
print(pycompat.decodeutf8(d), end="")
# test safeness and correctness of "ctx.status()"
print("= checking context.status():")
# ancestor "wcctx ~ 2"
actx2 = repo["."]
repo.wwrite("bar-m", b"bar-m\n", "")
repo.wwrite("bar-r", b"bar-r\n", "")
repo[None].add(["bar-m", "bar-r"])
repo.commit(text="add bar-m, bar-r", date="0 0")
# ancestor "wcctx ~ 1"
actx1 = repo["."]
repo.wwrite("bar-m", b"bar-m bar-m\n", "")
repo.wwrite("bar-a", b"bar-a\n", "")
repo[None].add(["bar-a"])
repo[None].forget(["bar-r"])
# status at this point:
# M bar-m
# A bar-a
# R bar-r
# C foo
print("== checking workingctx.status:")
wctx = repo[None]
print("wctx._status=%s" % (str(wctx._status)))
print('=== with "pattern match":')
print(actx1.status(other=wctx, match=scmutil.matchfiles(repo, ["bar-m", "foo"])))
print("wctx._status=%s" % (str(wctx._status)))
print(actx2.status(other=wctx, match=scmutil.matchfiles(repo, ["bar-m", "foo"])))
print("wctx._status=%s" % (str(wctx._status)))
print('=== with "always match" and "listclean=True":')
print(actx1.status(other=wctx, listclean=True))
print("wctx._status=%s" % (str(wctx._status)))
print(actx2.status(other=wctx, listclean=True))
print("wctx._status=%s" % (str(wctx._status)))
print("== checking workingcommitctx.status:")
wcctx = context.workingcommitctx(
repo, scmutil.status(["bar-m"], ["bar-a"], [], [], [], [], []), text="", date="0 0"
)
print("wcctx._status=%s" % (str(wcctx._status)))
print('=== with "always match":')
print(actx1.status(other=wcctx))
print("wcctx._status=%s" % (str(wcctx._status)))
print(actx2.status(other=wcctx))
print("wcctx._status=%s" % (str(wcctx._status)))
print('=== with "always match" and "listclean=True":')
print(actx1.status(other=wcctx, listclean=True))
print("wcctx._status=%s" % (str(wcctx._status)))
print(actx2.status(other=wcctx, listclean=True))
print("wcctx._status=%s" % (str(wcctx._status)))
print('=== with "pattern match":')
print(actx1.status(other=wcctx, match=scmutil.matchfiles(repo, ["bar-m", "foo"])))
print("wcctx._status=%s" % (str(wcctx._status)))
print(actx2.status(other=wcctx, match=scmutil.matchfiles(repo, ["bar-m", "foo"])))
print("wcctx._status=%s" % (str(wcctx._status)))
print('=== with "pattern match" and "listclean=True":')
print(
actx1.status(
other=wcctx, match=scmutil.matchfiles(repo, ["bar-r", "foo"]), listclean=True
)
)
print("wcctx._status=%s" % (str(wcctx._status)))
print(
actx2.status(
other=wcctx, match=scmutil.matchfiles(repo, ["bar-r", "foo"]), listclean=True
)
)
print("wcctx._status=%s" % (str(wcctx._status)))
os.chdir("..")
# test manifestlog being changed
print("== commit with manifestlog invalidated")
repo = hg.repository(u, "test2", create=1)
os.chdir("test2")
# make some commits
for i in ["1", "2", "3"]:
with open(i, "wb") as f:
f.write(pycompat.encodeutf8(i))
status = scmutil.status([], [i], [], [], [], [], [])
ctx = context.workingcommitctx(
repo, status, text=i, user="test@test.com", date=(0, 0)
)
ctx.p1().manifest() # side effect: cache manifestctx
n = repo.commitctx(ctx)
print("commit %s: %s" % (i, hex(n)))
# touch 00manifest.i mtime so storecache could expire.
# repo.__dict__['manifestlog'] is deleted by transaction releasefn.
st = repo.svfs.stat("00manifest.i")
repo.svfs.utime("00manifest.i", (st.st_mtime + 1, st.st_mtime + 1))
# read the file just committed
try:
if pycompat.decodeutf8(repo[n][i].data()) != i:
print("data mismatch")
except Exception as ex:
print("cannot read data: %r" % ex)
with repo.wlock(), repo.lock(), repo.transaction("test"):
with open(b"4", "wb") as f:
f.write(b"4")
repo.dirstate.normal("4")
repo.commit("4")
revsbefore = len(repo.changelog)
repo.invalidate(clearfilecache=True)
revsafter = len(repo.changelog)
if revsbefore != revsafter:
print("changeset lost by repo.invalidate()")
# Copy filectx from repo to newrepo using overlayfilectx and memctx
# overlayfilectx implements rawdata, rawflags, and a fast path would
# be used to skip calculating hash.
print("=== filelog rawdata reuse ===")
os.chdir(os.getenv("TESTTMP"))
newrepo = hg.repository(u, "test3", create=1)
def copyctx(newrepo, ctx):
cl = newrepo.changelog
p1 = cl.node(len(cl) - 1)
files = ctx.files()
desc = "copied: %s" % ctx.description()
def getfctx(repo, memctx, path):
if path not in ctx:
return None
return context.overlayfilectx(ctx[path])
return context.memctx(newrepo, [p1, None], desc, files, getfctx)
for rev in repo:
print("copying rev %d from test1 to test3" % rev)
copyctx(newrepo, repo[rev]).commit()