tests/test-wireproto.py
changeset 41840 d6569f1e9b37
parent 37942 32bc3815efae
child 43076 2372284d9457
equal deleted inserted replaced
41839:38de3300414f 41840:d6569f1e9b37
    76         f = wireprotov1peer.future()
    76         f = wireprotov1peer.future()
    77         yield {b'name': mangle(name)}, f
    77         yield {b'name': mangle(name)}, f
    78         yield unmangle(f.value)
    78         yield unmangle(f.value)
    79 
    79 
    80 class serverrepo(object):
    80 class serverrepo(object):
       
    81     def __init__(self, ui):
       
    82         self.ui = ui
       
    83 
    81     def greet(self, name):
    84     def greet(self, name):
    82         return b"Hello, " + name
    85         return b"Hello, " + name
    83 
    86 
    84     def filtered(self, name):
    87     def filtered(self, name):
    85         return self
    88         return self
    92 def greet(repo, proto, name):
    95 def greet(repo, proto, name):
    93     return mangle(repo.greet(unmangle(name)))
    96     return mangle(repo.greet(unmangle(name)))
    94 
    97 
    95 wireprotov1server.commands[b'greet'] = (greet, b'name')
    98 wireprotov1server.commands[b'greet'] = (greet, b'name')
    96 
    99 
    97 srv = serverrepo()
   100 srv = serverrepo(uimod.ui())
    98 clt = clientpeer(srv, uimod.ui())
   101 clt = clientpeer(srv, uimod.ui())
    99 
   102 
   100 def printb(data, end=b'\n'):
   103 def printb(data, end=b'\n'):
   101     out = getattr(sys.stdout, 'buffer', sys.stdout)
   104     out = getattr(sys.stdout, 'buffer', sys.stdout)
   102     out.write(data + end)
   105     out.write(data + end)