tests: add b'' prefixes to flagprocessorext.py

Otherwise a part of test-flavprocessor.t fails due to not
setting the proper key.

Differential Revision: https://phab.mercurial-scm.org/D5724

tests/flagprocessorext.py

@@ -1,136 +1,136
 # coding=UTF-8
 
 from __future__ import absolute_import
 
 import base64
 import zlib
 
 from mercurial import (
     changegroup,
     exchange,
     extensions,
     revlog,
     util,
 )
 
 # 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_FAIL = 1
 
 def validatehash(self, text):
     return True
 
 def bypass(self, text):
     return False
 
 def noopdonothing(self, text):
     return (text, True)
 
 def b64encode(self, text):
     return (base64.b64encode(text), False)
 
 def b64decode(self, text):
     return (base64.b64decode(text), True)
 
 def gzipcompress(self, text):
     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')
     versions.discard(b'02')
     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=revlog.REVIDX_DEFAULT_FLAGS):
             if b'[NOOP]' in text:
                 flags |= REVIDX_NOOP
 
             if b'[BASE64]' in text:
                 flags |= REVIDX_BASE64
 
             if b'[GZIP]' in text:
                 flags |= REVIDX_GZIP
 
             # This addrevision wrapper is meant to add a flag we will not have
             # transforms registered for, ensuring we handle this error case.
             if b'[FAIL]' in text:
                 flags |= REVIDX_FAIL
 
             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):
             orig = super(wrappingflagprocessorrepo, self).file(f)
             makewrappedfile(orig)
             return orig
 
     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)
 
     # Teach revlog about our test flags
     flags = [REVIDX_NOOP, REVIDX_BASE64, REVIDX_GZIP, REVIDX_FAIL]
     revlog.REVIDX_KNOWN_FLAGS |= util.bitsfrom(flags)
     revlog.REVIDX_FLAGS_ORDER.extend(flags)
 
     # Teach exchange to use changegroup 3
     for k in exchange._bundlespeccontentopts.keys():
-        exchange._bundlespeccontentopts[k]["cg.version"] = "03"
+        exchange._bundlespeccontentopts[k][b"cg.version"] = b"03"
 
     # Register flag processors for each extension
     revlog.addflagprocessor(
         REVIDX_NOOP,
         (
             noopdonothing,
             noopdonothing,
             validatehash,
         )
     )
     revlog.addflagprocessor(
         REVIDX_BASE64,
         (
             b64decode,
             b64encode,
             bypass,
         ),
     )
     revlog.addflagprocessor(
         REVIDX_GZIP,
         (
             gzipdecompress,
             gzipcompress,
             bypass
         )
     )