mirror of
https://github.com/facebook/sapling.git
synced 2024-10-10 08:47:12 +03:00
9fd86a4fae
Summary: This applies the formatting changes from black v21.4b2 to all covered projects in fbsource. Most changes are to single line docstrings, as black will now remove leading and trailing whitespace to match PEP8. Any other formatting changes are likely due to files that landed without formatting, or files that previously triggered errors in black. Any changes to code should be AST identical. Any test failures are likely due to bad tests, or testing against the output of pyfmt. Reviewed By: thatch Differential Revision: D28204910 fbshipit-source-id: 804725bcd14f763e90c5ddff1d0418117c15809a
150 lines
3.4 KiB
Python
150 lines
3.4 KiB
Python
# Copyright (c) Facebook, Inc. and its affiliates.
|
|
#
|
|
# This software may be used and distributed according to the terms of the
|
|
# GNU General Public License version 2.
|
|
|
|
# tracing-level: debug
|
|
|
|
from __future__ import absolute_import
|
|
|
|
import os
|
|
|
|
from bindings import tracing
|
|
from testutil.autofix import eq
|
|
from testutil.dott import feature, sh, testtmp # noqa: F401
|
|
|
|
|
|
idtopath = {}
|
|
|
|
|
|
def getidtopath():
|
|
"""Return a dict mapping from id (in hex form) to path"""
|
|
output = sh.hg("debugmanifestdirs", "-rall()")
|
|
# debugmanifestdirs prints "<id> <path>" per line
|
|
result = dict(l.split() for l in output.splitlines())
|
|
return result
|
|
|
|
|
|
def collectprefetch(command):
|
|
"""Updating to commit, check prefetched paths"""
|
|
d = tracing.tracingdata()
|
|
|
|
with d:
|
|
(sh % command).output
|
|
|
|
ids = []
|
|
for spans in d.treespans().values():
|
|
for span in spans.flatten():
|
|
name = span.get("name")
|
|
if name == "tree::store::prefetch":
|
|
ids += span["ids"].split()
|
|
elif name == "tree::store::get":
|
|
ids.append(span["id"])
|
|
idtopath.update(getidtopath())
|
|
paths = set(idtopath[id] for id in set(ids)) - {"/"}
|
|
|
|
# Translate ids to paths
|
|
return sorted(filter(None, paths))
|
|
|
|
|
|
# Use some production settings. They avoid expensive paths.
|
|
sh % "setconfig experimental.copytrace=off copytrace.fastcopytrace=true perftweaks.disablecasecheck=true"
|
|
sh % "enable sparse treemanifest rebase copytrace"
|
|
|
|
# flatcompat calls '.text()' which invalidates fast paths. So disable it.
|
|
sh % "setconfig treemanifest.flatcompat=0"
|
|
|
|
sh % "newrepo"
|
|
(
|
|
sh % "drawdag"
|
|
<< r"""
|
|
B # B/x/x/y/z=B1
|
|
| # B/y/x/y/z=B2
|
|
|
|
|
|
|
|
| D # D/d/d/d/d=D1
|
|
| |
|
|
| C # C/c/c/c/c=C1
|
|
|/
|
|
A # A/x/x/y/z=A1
|
|
# A/y/x/y/z=A2
|
|
# A/z/x/y/z=A3
|
|
"""
|
|
)
|
|
|
|
sh % "hg sparse include x"
|
|
|
|
# Good: Updating to A should avoid downloading y/ or z/
|
|
eq(collectprefetch("hg update -q $A"), ["x", "x/x", "x/x/y"])
|
|
|
|
# Good: Updating to B should avoid downloading y/
|
|
eq(collectprefetch("hg update -q $B"), ["x", "x/x", "x/x/y"])
|
|
|
|
|
|
sh % "hg update -q $D"
|
|
|
|
|
|
# Good: Rebasing B to D should avoid downloading d/ or c/, or z/.
|
|
# (This is optimized by "rebase: use matcher to optimize manifestmerge",
|
|
# https://www.mercurial-scm.org/repo/hg/rev/4d504e541d3d,
|
|
# fbsource-hg: 94ad1b49ede1f8e5897c7c9381304785746fa460)
|
|
eq(
|
|
collectprefetch("hg rebase -r $B -d $D -q"),
|
|
["x", "x/x", "x/x/y", "y", "y/x", "y/x/y"],
|
|
)
|
|
|
|
# Good: Changing sparse profile should not download everything.
|
|
eq(collectprefetch("hg sparse exclude y"), ["x", "x/x", "x/x/y"])
|
|
|
|
|
|
# Test sparse profile change.
|
|
|
|
sh % "newrepo"
|
|
(
|
|
sh % "drawdag"
|
|
<< r"""
|
|
# B/profile=[include]\nx\ny
|
|
B # B/x/x/x=2
|
|
| # B/y/y/y=2
|
|
| # B/z/z/z=2
|
|
|
|
|
A # A/profile=[include]\nx
|
|
# A/x/x/x=1
|
|
# A/y/y/y=1
|
|
# A/z/z/z=1
|
|
"""
|
|
)
|
|
|
|
idtopath = getidtopath()
|
|
|
|
eq(collectprefetch("hg sparse enable profile"), [])
|
|
|
|
# Good: Updating to A should avoid downloading y/ or z/
|
|
eq(collectprefetch("hg update -q $A"), ["x", "x/x"])
|
|
|
|
# Good: Updating to B should avoid downloading z/
|
|
eq(collectprefetch("hg update -q $B"), ["x", "x/x", "y", "y/y"])
|
|
|
|
|
|
# Test 'status'.
|
|
|
|
sh % "newrepo"
|
|
(
|
|
sh % "drawdag"
|
|
<< r"""
|
|
A # A/x/x/x=1
|
|
# A/y/y/y=1
|
|
# A/z/z/z=1
|
|
"""
|
|
)
|
|
|
|
eq(collectprefetch("hg sparse include x"), [])
|
|
sh % "hg up -q $A"
|
|
open("y", "w").write("2")
|
|
os.mkdir("z")
|
|
open("z/1", "w").write("2")
|
|
open("z/z", "w").write("2")
|
|
|
|
# Good: 'status' should avoid downloading y/ or z/.
|
|
eq(sorted(set(collectprefetch("hg status")) - {"x", "x/x"}), [])
|