[PATCH 1 of 8 V2] tests: add extension to emulate invoking dirstate.write at the specific time
FUJIWARA Katsunori
foozy at lares.dti.ne.jp
Wed Jul 8 08:08:37 UTC 2015
# HG changeset patch
# User FUJIWARA Katsunori <foozy at lares.dti.ne.jp>
# Date 1436342469 -32400
# Wed Jul 08 17:01:09 2015 +0900
# Node ID f4ecdb438dc85be64267f9efe68c62bde5d6bb07
# Parent 3948cb4d0ae70e7257e47e2fd9f657c0c1af7c2b
tests: add extension to emulate invoking dirstate.write at the specific time
This extension fakes `now` for `parsers.pack_dirstate()` to emulate
invoking `dirstate.write()` at the specific time, only when
`dirstate.write()` is invoked via functions below:
- `workingctx._checklookup()` (= `repo.status()`)
- `committablectx.markcommitted()`
This is useful to reproduce timing critical issues fixed in subsequent
patches.
diff --git a/tests/fakedirstatewritetime.py b/tests/fakedirstatewritetime.py
new file mode 100644
--- /dev/null
+++ b/tests/fakedirstatewritetime.py
@@ -0,0 +1,58 @@
+# extension to emulate invoking `dirstate.write()` at the time
+# specified by `[fakedirstatewritetime] fakenow`, only when
+# `dirstate.write()` is invoked via functions below:
+#
+# - `workingctx._checklookup()` (= `repo.status()`)
+# - `committablectx.markcommitted()`
+
+from mercurial import context, extensions, parsers, util
+
+def pack_dirstate(fakenow, orig, dmap, copymap, pl, now):
+ # execute what original parsers.pack_dirstate should do actually
+ # for consistency
+ actualnow = int(now)
+ for f, e in dmap.iteritems():
+ if e[0] == 'n' and e[3] == actualnow:
+ e = parsers.dirstatetuple(e[0], e[1], e[2], -1)
+ dmap[f] = e
+
+ return orig(dmap, copymap, pl, fakenow)
+
+def fakewrite(ui, func):
+ # fake "now" of `pack_dirstate` only if it is invoked while `func`
+
+ fakenow = ui.config('fakedirstatewritetime', 'fakenow')
+ if not fakenow:
+ # Execute original one, if fakenow isn't configured. This is
+ # usefull to prevent subrepos from exeucting replaced one,
+ # because replacing `parsers.pack_dirstate` is also effective
+ # in subrepos.
+ return func()
+
+ # parsing `fakenow` in YYYYmmddHHMM format makes comparison between
+ # `fakenow` value and `touch -t YYYYmmddHHMM` argument easy
+ timestamp = util.parsedate(fakenow, ['%Y%m%d%H%M'])[0]
+ fakenow = float(timestamp)
+
+ orig_pack_dirstate = parsers.pack_dirstate
+ wrapper = lambda *args: pack_dirstate(fakenow, orig_pack_dirstate, *args)
+
+ parsers.pack_dirstate = wrapper
+ try:
+ return func()
+ finally:
+ parsers.pack_dirstate = orig_pack_dirstate
+
+def _checklookup(orig, workingctx, files):
+ ui = workingctx.repo().ui
+ return fakewrite(ui, lambda : orig(workingctx, files))
+
+def markcommitted(orig, committablectx, node):
+ ui = committablectx.repo().ui
+ return fakewrite(ui, lambda : orig(committablectx, node))
+
+def extsetup(ui):
+ extensions.wrapfunction(context.workingctx, '_checklookup',
+ _checklookup)
+ extensions.wrapfunction(context.committablectx, 'markcommitted',
+ markcommitted)
More information about the Mercurial-devel
mailing list