mirror of
https://github.com/facebook/sapling.git
synced 2024-10-07 15:27:13 +03:00
e2aeb2930a
1. The module named 'wireproto' is tested. 2. There already is a test-wireproto.t for related tests on shell level.
46 lines
1.1 KiB
Python
46 lines
1.1 KiB
Python
from mercurial import wireproto
|
|
|
|
class proto(object):
|
|
def __init__(self, args):
|
|
self.args = args
|
|
def getargs(self, spec):
|
|
args = self.args
|
|
args.setdefault('*', {})
|
|
names = spec.split()
|
|
return [args[n] for n in names]
|
|
|
|
class clientrepo(wireproto.wirerepository):
|
|
def __init__(self, serverrepo):
|
|
self.serverrepo = serverrepo
|
|
def _call(self, cmd, **args):
|
|
return wireproto.dispatch(self.serverrepo, proto(args), cmd)
|
|
|
|
@wireproto.batchable
|
|
def greet(self, name):
|
|
f = wireproto.future()
|
|
yield wireproto.todict(name=mangle(name)), f
|
|
yield unmangle(f.value)
|
|
|
|
class serverrepo(object):
|
|
def greet(self, name):
|
|
return "Hello, " + name
|
|
|
|
def mangle(s):
|
|
return ''.join(chr(ord(c) + 1) for c in s)
|
|
def unmangle(s):
|
|
return ''.join(chr(ord(c) - 1) for c in s)
|
|
|
|
def greet(repo, proto, name):
|
|
return mangle(repo.greet(unmangle(name)))
|
|
|
|
wireproto.commands['greet'] = (greet, 'name',)
|
|
|
|
srv = serverrepo()
|
|
clt = clientrepo(srv)
|
|
|
|
print clt.greet("Foobar")
|
|
b = clt.batch()
|
|
fs = [b.greet(s) for s in ["Fo, =;o", "Bar"]]
|
|
b.submit()
|
|
print [f.value for f in fs]
|