tests/testlib/badserverext.py

48632:caa6694dac45

    def write(self, data):
        remaining = object.__getattribute__(self, '_cond').remaining_send_bytes

        # No byte limit on this operation. Call original function.
        if not remaining:
            self._writelog(b'write(%d) -> %s' % (len(data), data))
            result = object.__getattribute__(self, '_orig').write(data)
            return result

        if len(data) > remaining:
            newdata = data[0:remaining]
        else:
            newdata = data

        remaining -= len(newdata)

        self._writelog(
            b'write(%d from %d) -> (%d) %s'
            % (len(newdata), len(data), remaining, newdata)
        )

        result = object.__getattribute__(self, '_orig').write(newdata)

        object.__getattribute__(self, '_cond').remaining_send_bytes = remaining

        if remaining <= 0:
            self._writelog(b'write limit reached; closing socket')

48633:f91f98e9834a

    def write(self, data):
        remaining = object.__getattribute__(self, '_cond').remaining_send_bytes

        # No byte limit on this operation. Call original function.
        if not remaining:
            result = object.__getattribute__(self, '_orig').write(data)
            self._writelog(b'write(%d) -> %s' % (len(data), data))
            return result

        if len(data) > remaining:
            newdata = data[0:remaining]
        else:
            newdata = data

        remaining -= len(newdata)

        result = object.__getattribute__(self, '_orig').write(newdata)

        self._writelog(
            b'write(%d from %d) -> (%d) %s'
            % (len(newdata), len(data), remaining, newdata)
        )

        object.__getattribute__(self, '_cond').remaining_send_bytes = remaining

        if remaining <= 0:
            self._writelog(b'write limit reached; closing socket')