"""Throughput test."""

import time, sys
from twisted.internet import reactor, protocol
from twisted.python import log

TIMES = 10000
S = "0123456789" * 1240

toReceive = len(S) * TIMES

class Sender(protocol.Protocol):

    def connectionMade(self):
        start()
        self.numSent = 0
        self.received = 0
        self.transport.registerProducer(self, 0)

    def stopProducing(self):
        pass

    def pauseProducing(self):
        pass
    
    def resumeProducing(self):
        self.numSent += 1
        self.transport.write(S)
        if self.numSent == TIMES:
            self.transport.unregisterProducer()
            self.transport.loseConnection()

    def connectionLost(self, reason):
        shutdown(self.numSent == TIMES)


started = None

def start():
    global started
    started = time.time()

def shutdown(success):
    if not success:
        raise SystemExit, "failure or something"
    passed = time.time() - started
    print "Throughput (send): %s kbytes/sec" % ((toReceive / passed) / 1024)
    reactor.stop()


def main():
    f = protocol.ClientFactory()
    f.protocol = Sender
    reactor.connectTCP(sys.argv[1], int(sys.argv[2]), f)
    reactor.run()


if __name__ == '__main__':
    #log.startLogging(sys.stdout)
    main()
