tests/test-wireproto.py
changeset 37614 a81d02ea65db
parent 37321 e826fe7a08c7
child 37633 33a6eee08db2
equal deleted inserted replaced
37613:96d735601ca1 37614:a81d02ea65db
     5     pycompat,
     5     pycompat,
     6     ui as uimod,
     6     ui as uimod,
     7     util,
     7     util,
     8     wireproto,
     8     wireproto,
     9     wireprototypes,
     9     wireprototypes,
       
    10     wireprotov1peer,
    10 )
    11 )
    11 stringio = util.stringio
    12 stringio = util.stringio
    12 
    13 
    13 class proto(object):
    14 class proto(object):
    14     def __init__(self, args):
    15     def __init__(self, args):
    27 wireprototypes.TRANSPORTS['dummyproto'] = {
    28 wireprototypes.TRANSPORTS['dummyproto'] = {
    28     'transport': 'dummy',
    29     'transport': 'dummy',
    29     'version': 1,
    30     'version': 1,
    30 }
    31 }
    31 
    32 
    32 class clientpeer(wireproto.wirepeer):
    33 class clientpeer(wireprotov1peer.wirepeer):
    33     def __init__(self, serverrepo, ui):
    34     def __init__(self, serverrepo, ui):
    34         self.serverrepo = serverrepo
    35         self.serverrepo = serverrepo
    35         self.ui = ui
    36         self.ui = ui
    36 
    37 
    37     def url(self):
    38     def url(self):
    63             raise error.Abort('dummy client does not support response type')
    64             raise error.Abort('dummy client does not support response type')
    64 
    65 
    65     def _callstream(self, cmd, **args):
    66     def _callstream(self, cmd, **args):
    66         return stringio(self._call(cmd, **args))
    67         return stringio(self._call(cmd, **args))
    67 
    68 
    68     @wireproto.batchable
    69     @wireprotov1peer.batchable
    69     def greet(self, name):
    70     def greet(self, name):
    70         f = wireproto.future()
    71         f = wireprotov1peer.future()
    71         yield {b'name': mangle(name)}, f
    72         yield {b'name': mangle(name)}, f
    72         yield unmangle(f.value)
    73         yield unmangle(f.value)
    73 
    74 
    74 class serverrepo(object):
    75 class serverrepo(object):
    75     def greet(self, name):
    76     def greet(self, name):