Mercurial > public > mercurial-scm > hg
diff tests/flagprocessorext.py @ 43076:2372284d9457
formatting: blacken the codebase
This is using my patch to black
(https://github.com/psf/black/pull/826) so we don't un-wrap collection
literals.
Done with:
hg files 'set:**.py - mercurial/thirdparty/** - "contrib/python-zstandard/**"' | xargs black -S
# skip-blame mass-reformatting only
# no-check-commit reformats foo_bar functions
Differential Revision: https://phab.mercurial-scm.org/D6971
| author | Augie Fackler <augie@google.com> |
|---|---|
| date | Sun, 06 Oct 2019 09:45:02 -0400 |
| parents | f4caf910669e |
| children | 74271829ddc0 |
line wrap: on
line diff
--- a/tests/flagprocessorext.py Sat Oct 05 10:29:34 2019 -0400 +++ b/tests/flagprocessorext.py Sun Oct 06 09:45:02 2019 -0400 @@ -12,42 +12,49 @@ revlog, util, ) -from mercurial.revlogutils import ( - flagutil, -) +from mercurial.revlogutils import flagutil # Test only: These flags are defined here only in the context of testing the # behavior of the flag processor. The canonical way to add flags is to get in # touch with the community and make them known in revlog. -REVIDX_NOOP = (1 << 3) -REVIDX_BASE64 = (1 << 2) -REVIDX_GZIP = (1 << 1) +REVIDX_NOOP = 1 << 3 +REVIDX_BASE64 = 1 << 2 +REVIDX_GZIP = 1 << 1 REVIDX_FAIL = 1 + def validatehash(self, text): return True + def bypass(self, text): return False + def noopdonothing(self, text, sidedata): return (text, True) + def noopdonothingread(self, text): return (text, True, {}) + def b64encode(self, text, sidedata): return (base64.b64encode(text), False) + def b64decode(self, text): return (base64.b64decode(text), True, {}) + def gzipcompress(self, text, sidedata): return (zlib.compress(text), False) + def gzipdecompress(self, text): return (zlib.decompress(text), True, {}) + def supportedoutgoingversions(orig, repo): versions = orig(repo) versions.discard(b'01') @@ -55,16 +62,26 @@ versions.add(b'03') return versions + def allsupportedversions(orig, ui): versions = orig(ui) versions.add(b'03') return versions + def makewrappedfile(obj): class wrappedfile(obj.__class__): - def addrevision(self, text, transaction, link, p1, p2, - cachedelta=None, node=None, - flags=flagutil.REVIDX_DEFAULT_FLAGS): + def addrevision( + self, + text, + transaction, + link, + p1, + p2, + cachedelta=None, + node=None, + flags=flagutil.REVIDX_DEFAULT_FLAGS, + ): if b'[NOOP]' in text: flags |= REVIDX_NOOP @@ -79,14 +96,20 @@ if b'[FAIL]' in text: flags |= REVIDX_FAIL - return super(wrappedfile, self).addrevision(text, transaction, link, - p1, p2, - cachedelta=cachedelta, - node=node, - flags=flags) + return super(wrappedfile, self).addrevision( + text, + transaction, + link, + p1, + p2, + cachedelta=cachedelta, + node=node, + flags=flags, + ) obj.__class__ = wrappedfile + def reposetup(ui, repo): class wrappingflagprocessorrepo(repo.__class__): def file(self, f): @@ -96,15 +119,14 @@ repo.__class__ = wrappingflagprocessorrepo + def extsetup(ui): # Enable changegroup3 for flags to be sent over the wire wrapfunction = extensions.wrapfunction - wrapfunction(changegroup, - 'supportedoutgoingversions', - supportedoutgoingversions) - wrapfunction(changegroup, - 'allsupportedversions', - allsupportedversions) + wrapfunction( + changegroup, 'supportedoutgoingversions', supportedoutgoingversions + ) + wrapfunction(changegroup, 'allsupportedversions', allsupportedversions) # Teach revlog about our test flags flags = [REVIDX_NOOP, REVIDX_BASE64, REVIDX_GZIP, REVIDX_FAIL] @@ -117,26 +139,11 @@ # Register flag processors for each extension flagutil.addflagprocessor( - REVIDX_NOOP, - ( - noopdonothingread, - noopdonothing, - validatehash, - ) + REVIDX_NOOP, (noopdonothingread, noopdonothing, validatehash,) ) flagutil.addflagprocessor( - REVIDX_BASE64, - ( - b64decode, - b64encode, - bypass, - ), + REVIDX_BASE64, (b64decode, b64encode, bypass,), ) flagutil.addflagprocessor( - REVIDX_GZIP, - ( - gzipdecompress, - gzipcompress, - bypass - ) + REVIDX_GZIP, (gzipdecompress, gzipcompress, bypass) )