( )⚙ D3510 tests: port test-wireproto.py to Python 3

This is an archive of the discontinued Mercurial Phabricator instance.

Differential D3510

tests: port test-wireproto.py to Python 3
ClosedPublic

Authored by durin42 on May 8 2018, 3:27 PM.

Details

Reviewers
pulkit
Group Reviewers
hg-reviewers
Commits
rHG40381a88bab4: tests: port test-wireproto.py to Python 3
Summary

A little bit of output churn because we now get b'' prefixes on output
in both 2 and 3, but for this test I'm more comfortable knowing that
we're using bytestrs everywhere.

Diff Detail

Repository
rHG Mercurial
Lint
Lint Skipped
Unit
Unit Tests Skipped

Event Timeline

durin42 created this revision.May 8 2018, 3:27 PM
pulkit accepted this revision.May 9 2018, 6:46 PM
Closed by commit rHG40381a88bab4: tests: port test-wireproto.py to Python 3 (authored by durin42). · Explain WhyMay 10 2018, 6:42 AM
This revision was automatically updated to reflect the committed changes.

Revision Contents
Changeset List

Diff 8578

contrib/python3-whitelist

test-username-newline.t test-username-newline.t
test-verify.t test-verify.t
test-walkrepo.py test-walkrepo.py
test-websub.t test-websub.t
test-win32text.t test-win32text.t
test-wireproto-clientreactor.py test-wireproto-clientreactor.py
test-wireproto-framing.py test-wireproto-framing.py
test-wireproto-serverreactor.py test-wireproto-serverreactor.py
test-wireproto.py
test-wsgirequest.py test-wsgirequest.py
test-xdg.t test-xdg.t

tests/test-wireproto.py

from __future__ import absolute_import, print_function from __future__ import absolute_import, print_function
import sys
from mercurial import ( from mercurial import (
error, error,
pycompat, pycompat,
ui as uimod, ui as uimod,
util, util,
wireprototypes, wireprototypes,
wireprotov1peer, wireprotov1peer,
wireprotov1server, wireprotov1server,
) )
from mercurial.utils import (
stringutil,
)
stringio = util.stringio stringio = util.stringio
class proto(object): class proto(object):
def __init__(self, args): def __init__(self, args):
self.args = args self.args = args
self.name = 'dummyproto' self.name = 'dummyproto'
def getargs(self, spec): def getargs(self, spec):
def greet(repo, proto, name): def greet(repo, proto, name):
return mangle(repo.greet(unmangle(name))) return mangle(repo.greet(unmangle(name)))
wireprotov1server.commands[b'greet'] = (greet, b'name') wireprotov1server.commands[b'greet'] = (greet, b'name')
srv = serverrepo() srv = serverrepo()
clt = clientpeer(srv, uimod.ui()) clt = clientpeer(srv, uimod.ui())
print(clt.greet(b"Foobar")) def printb(data, end=b'\n'):
out = getattr(sys.stdout, 'buffer', sys.stdout)
out.write(data + end)
out.flush()
printb(clt.greet(b"Foobar"))
with clt.commandexecutor() as e: with clt.commandexecutor() as e:
fgreet1 = e.callcommand(b'greet', {b'name': b'Fo, =;:<o'}) fgreet1 = e.callcommand(b'greet', {b'name': b'Fo, =;:<o'})
fgreet2 = e.callcommand(b'greet', {b'name': b'Bar'}) fgreet2 = e.callcommand(b'greet', {b'name': b'Bar'})
print([f.result() for f in (fgreet1, fgreet2)]) printb(stringutil.pprint([f.result() for f in (fgreet1, fgreet2)]))

tests/test-wireproto.py.out

Hello, Foobar Hello, Foobar
['Hello, Fo, =;:<o', 'Hello, Bar'] [b'Hello, Fo, =;:<o', b'Hello, Bar']