sapling/eden/scm/tests/test-hgsql-metrics-t.py
John Reese 9fd86a4fae apply upgraded black 21.4b2 formatting to fbsource
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
2021-05-04 22:16:51 -07:00

106 lines
2.9 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 or any later version.
from __future__ import absolute_import
from testutil.dott import feature, sh, testtmp # noqa: F401
feature.require(["py2"])
sh % ". '$TESTDIR/hgsql/library.sh'"
sh % "initdb"
sh % "setconfig 'extensions.treemanifest=!'"
# Populate the db with an initial commit
sh % "initclient client"
sh % "cd client"
sh % "echo x" > "x"
sh % "hg commit -qAm x"
sh % "cd .."
sh % "initserver master masterrepo"
(
sh % "cat"
<< r"""
from edenscm.mercurial import extensions
from edenscm.mercurial import ui as uimod
def uisetup(ui):
extensions.wrapfunction(uimod.ui, 'log', mylog)
def mylog(orig, self, service, *msg, **opts):
if service in ['sqllock']:
kwstr = ", ".join("%s=%s" % (k, v) for k, v in
sorted(opts.items()))
msgstr = msg[0] % msg[1:]
self.warn('%s: %s (%s)\n' % (service, msgstr, kwstr))
return orig(self, service, *msg, **opts)
"""
>> "$TESTTMP/uilog.py"
)
(
sh % "cat"
<< r"""
[extensions]
uilog=$TESTTMP/uilog.py
"""
>> "master/.hg/hgrc"
)
# Verify timeouts are logged
(
sh % "cat"
<< r"""
from edenscm.mercurial import error, extensions
def uisetup(ui):
hgsql = extensions.find('hgsql')
extensions.wrapfunction(hgsql.sqlcontext, '__enter__', fakeenter)
def fakeenter(orig, self):
if self.dbwritable:
extensions.wrapfunction(self.repo.__class__, '_sqllock', lockthrow)
return orig(self)
def lockthrow(*args, **kwargs):
raise error.Abort("fake timeout")
"""
>> "$TESTTMP/forcetimeout.py"
)
sh % "cp master/.hg/hgrc $TESTTMP/orighgrc"
(
sh % "cat"
<< r"""
[extensions]
forcetimeout=$TESTTMP/forcetimeout.py
"""
>> "master/.hg/hgrc"
)
sh % "cd client"
sh % "hg push 'ssh://user@dummy/master'" == r"""
pushing to ssh://user@dummy/master
searching for changes
remote: sqllock: failed to get sql lock after * seconds (glob)
remote: (elapsed=*, repository=$TESTTMP/master, success=false, valuetype=lockwait) (glob)
remote: abort: fake timeout
abort: not a Mercurial bundle
[255]"""
sh % "cd .."
sh % "cp $TESTTMP/orighgrc master/.hg/hgrc"
# Verify sqllock times are logged
sh % "cd client"
sh % "hg push 'ssh://user@dummy/master'" == r"""
pushing to ssh://user@dummy/master
searching for changes
remote: sqllock: waited for sql lock for * seconds (read 1 rows) (glob)
remote: (elapsed=*, repository=$TESTTMP/master, success=true, valuetype=lockwait) (glob)
remote: adding changesets
remote: adding manifests
remote: adding file changes
remote: added 1 changesets with 1 changes to 1 files
remote: sqllock: held sql lock for * seconds (read 5 rows; write 5 rows) (glob)
remote: (elapsed=*, readrows=5, repository=$TESTTMP/master, valuetype=lockheld, writerows=5) (glob)"""