root / trunk / sandbox / jhb / oc2 / testutils.py

"""This file is needed by documentation.py and testissuer.py"""

import BaseHTTPServer, threading
import issuer, mint, transports, urllib


class Handler(BaseHTTPServer.BaseHTTPRequestHandler):

    def do_POST(self):
        #print self.server
        if self.path == '/stop':
            raise 'foobar'            
        length = self.headers.get('Content-Length')
        data = self.rfile.read(int(length))
        data = urllib.unquote(data)
        if not data.startswith('['):
            return
        message = transports.createMessage(data)
        if message.header == 'AskLatestCDD':
            answer = self.issuer.giveLatestCDD(message)
        elif message.header == 'FetchMintKeys':
            answer = self.issuer.giveMintKeys(message)
        elif message.header == 'TransferRequest':
            answer = self.issuer.handleTransferRequest(self.mint,self.authorizer,message)
        elif message.header == 'TransferResume':
            answer = self.issuer.resumeTransfer(message)
 
        self.send_response(200)
        self.send_header("Content-type", "text/plain")
        self.wfile.write('\r\n')
        self.wfile.write(answer.toString(True))

    #disable dns lookup
    def address_string(self):
        return str(self.client_address[0])


class TestingHandler(Handler):
    """Supress output to stderr"""

    def log_message(self, format, *args):
        pass

def run_once(port,issuer=None,mint=None,authorizer=None):
    import time
    time.sleep(0.001)
    Handler.issuer = issuer
    Handler.mint = mint
    Handler.authorizer = authorizer
    httpd = BaseHTTPServer.HTTPServer(("", port), TestingHandler)
    import threading
    t = threading.Thread(target=httpd.handle_request)     
    t.start()