tests/sshprotoext.py
changeset 52643 5cc8deb96b48
parent 51569 13c004b54cbe
equal deleted inserted replaced
52642:73ab542565e0 52643:5cc8deb96b48
    30 
    30 
    31     def serve_forever(self):
    31     def serve_forever(self):
    32         for i in range(10):
    32         for i in range(10):
    33             self._ui.fout.write(b'banner: line %d\n' % i)
    33             self._ui.fout.write(b'banner: line %d\n' % i)
    34 
    34 
    35         super(bannerserver, self).serve_forever()
    35         super().serve_forever()
    36 
    36 
    37 
    37 
    38 class prehelloserver(wireprotoserver.sshserver):
    38 class prehelloserver(wireprotoserver.sshserver):
    39     """Tests behavior when connecting to <0.9.1 servers.
    39     """Tests behavior when connecting to <0.9.1 servers.
    40 
    40 
    54         assert l == b'between\n'
    54         assert l == b'between\n'
    55         proto = wireprotoserver.sshv1protocolhandler(ui, ui.fin, ui.fout)
    55         proto = wireprotoserver.sshv1protocolhandler(ui, ui.fin, ui.fout)
    56         rsp = wireprotov1server.dispatch(self._repo, proto, b'between')
    56         rsp = wireprotov1server.dispatch(self._repo, proto, b'between')
    57         wireprotoserver._sshv1respondbytes(ui.fout, rsp.data)
    57         wireprotoserver._sshv1respondbytes(ui.fout, rsp.data)
    58 
    58 
    59         super(prehelloserver, self).serve_forever()
    59         super().serve_forever()
    60 
    60 
    61 
    61 
    62 def performhandshake(orig, ui, stdin, stdout, stderr):
    62 def performhandshake(orig, ui, stdin, stdout, stderr):
    63     """Wrapped version of sshpeer._performhandshake to send extra commands."""
    63     """Wrapped version of sshpeer._performhandshake to send extra commands."""
    64     mode = ui.config(b'sshpeer', b'handshake-mode')
    64     mode = ui.config(b'sshpeer', b'handshake-mode')