comparison tests/sshprotoext.py @ 36064:5767664d39a5

wireprotoserver: extract SSH response handling functions

The lookup/dispatch table was cute. But it isn't needed. Future
refactors will benefit from the handlers for individual response
types living outside the class.

As part of this, I snuck in a change that changes a type compare
from str to bytes. This has no effect on Python 2. But it might
make Python 3 a bit happier.

Differential Revision: https://phab.mercurial-scm.org/D2091

36063:5a53af7d09aa

    """
    def serve_forever(self):
        l = self._fin.readline()
        assert l == b'hello\n'
        # Respond to unknown commands with an empty reply.
        self._sendresponse(b'')
        l = self._fin.readline()
        assert l == b'between\n'
        rsp = wireproto.dispatch(self._repo, self, b'between')
        self._handlers[rsp.__class__](self, rsp)

        super(prehelloserver, self).serve_forever()

class upgradev2server(wireprotoserver.sshserver):
    """Tests behavior for clients that issue upgrade to version 2."""

36064:5767664d39a5

    """
    def serve_forever(self):
        l = self._fin.readline()
        assert l == b'hello\n'
        # Respond to unknown commands with an empty reply.
        wireprotoserver._sshv1respondbytes(self._fout, b'')
        l = self._fin.readline()
        assert l == b'between\n'
        rsp = wireproto.dispatch(self._repo, self, b'between')
        wireprotoserver._sshv1respondbytes(self._fout, rsp)

        super(prehelloserver, self).serve_forever()

class upgradev2server(wireprotoserver.sshserver):
    """Tests behavior for clients that issue upgrade to version 2."""