#!/usr/bin/env python

# encoding: utf-8

"""

BLIPConnectionTest.py

Created by Jens Alfke on 2008-06-04.

This source file is test/example code, and is in the public domain.

"""

from BLIP import Connection, OutgoingRequest, kOpening

import asyncore

from cStringIO import StringIO

from datetime import datetime

import logging

import random

import unittest

kSendInterval = 0.2

kNBatchedMessages = 10

kUrgentEvery = 4

def randbool():

    return random.randint(0,1) == 1

class BLIPConnectionTest(unittest.TestCase):

    def setUp(self):

        self.connection = Connection( ('localhost',46353) )

        self.nRepliesPending = 0

    def sendRequest(self):

        size = random.randint(0,32767)

        io = StringIO()

        for i in xrange(0,size):

            io.write( chr(i % 256) )

        body = io.getvalue()

        io.close

        req = OutgoingRequest(self.connection, body,{'Content-Type': 'application/octet-stream',

                                                     'User-Agent':  'PyBLIP',

                                                     'Date': datetime.now(),

                                                     'Size': size})

        req.compressed = randbool()

        req.urgent     = (random.randint(0,kUrgentEvery-1)==0)

        req.response.onComplete = self.gotResponse

        return req.send()

    def gotResponse(self, response):

        self.nRepliesPending -= 1

        logging.info("Got response!: %s (%i pending)",response,self.nRepliesPending)

        request = response.request

        assert response.body == request.body

    def testClient(self):

        lastReqTime = None

        nIterations = 0

        while nIterations < 10:

            asyncore.loop(timeout=kSendInterval,count=1)

            now = datetime.now()

            if self.connection.status!=kOpening and (not lastReqTime or (now-lastReqTime).microseconds >= kSendInterval*1.0e6):

                lastReqTime = now

                for i in xrange(0,kNBatchedMessages):

                    if not self.sendRequest():

                        logging.warn("Couldn't send request (connection is probably closed)")

                        break;

                    self.nRepliesPending += 1

                nIterations += 1

    def tearDown(self):

        self.connection.close()

        asyncore.loop() # got to give it time to negotiate close; this call should exit eventually

if __name__ == '__main__':

    logging.basicConfig(level=logging.DEBUG)

    unittest.main()