tests/test-wireproto.py
changeset 17192 1ac628cd7113
parent 14765 08ef6b5f3715
child 18278 753acee7d6dd
equal deleted inserted replaced
17191:5884812686f7 17192:1ac628cd7113
     7         args = self.args
     7         args = self.args
     8         args.setdefault('*', {})
     8         args.setdefault('*', {})
     9         names = spec.split()
     9         names = spec.split()
    10         return [args[n] for n in names]
    10         return [args[n] for n in names]
    11 
    11 
    12 class clientrepo(wireproto.wirerepository):
    12 class clientpeer(wireproto.wirepeer):
    13     def __init__(self, serverrepo):
    13     def __init__(self, serverrepo):
    14         self.serverrepo = serverrepo
    14         self.serverrepo = serverrepo
    15     def _call(self, cmd, **args):
    15     def _call(self, cmd, **args):
    16         return wireproto.dispatch(self.serverrepo, proto(args), cmd)
    16         return wireproto.dispatch(self.serverrepo, proto(args), cmd)
    17 
    17 
    34     return mangle(repo.greet(unmangle(name)))
    34     return mangle(repo.greet(unmangle(name)))
    35 
    35 
    36 wireproto.commands['greet'] = (greet, 'name',)
    36 wireproto.commands['greet'] = (greet, 'name',)
    37 
    37 
    38 srv = serverrepo()
    38 srv = serverrepo()
    39 clt = clientrepo(srv)
    39 clt = clientpeer(srv)
    40 
    40 
    41 print clt.greet("Foobar")
    41 print clt.greet("Foobar")
    42 b = clt.batch()
    42 b = clt.batch()
    43 fs = [b.greet(s) for s in ["Fo, =;o", "Bar"]]
    43 fs = [b.greet(s) for s in ["Fo, =;o", "Bar"]]
    44 b.submit()
    44 b.submit()