#!/usr/bin/env python

"""Server environment for testing simplewebserver (command line web server)
This uses the basic environment from test_create and runs a web server in a separate thread
The top-level tests are then in test_client / test_ie / test_mozilla etc
These can then be layered on top of this server"""

from Pootle import test_create
from jToolkit.web import simplewebserver
from jToolkit import errors
import urllib2
import time

class TestCmdlineServer(test_create.NoReuse):
    # this is called from setup_class in test_create.TestCreate
    def setup_webserver(self):
        """setup the webserver that will be used for the tests"""
        print "setting up web server"
        webserverclass = simplewebserver.jToolkitHTTPServer(simplewebserver.ThreadedHTTPServer)
        options = simplewebserver.WebOptionParser().parse_args([])[0]
        options.port = 0
        options.servertype = 'standard'
        errorhandler = errors.ConsoleErrorHandler()
        webserver = webserverclass(options, errorhandler)
        return webserver

    def setup_method(self, method):
        """starts a new simplewebserver in a separate thread"""
        try:
            test_create.NoReuse.setup_method(self, method)
        except Exception, e:
            print "exception in test_create setup_method:", e
            test_create.NoReuse.teardown_method(self, method)
            raise
        print "finished setup_method phase 1"
        # self.webserver.options.port = self.port
        self.baseaddress = "http://%s:%d/" % (self.webserver.hostname, self.webserver.port)
        ThreadClass = self.webserver.ThreadClass
        self.webserverthread = ThreadClass(target = simplewebserver.run, name="webserver", args=(self.server, self.webserver.options))
        self.webserverthread.start()
        # wait until it actually is started
        waited = 0.0
        sleep = 0.001
        maxwait = 3.0
        while not hasattr(self.webserver, "stop"):
            time.sleep(sleep)
            waited += sleep
            if sleep > maxwait:
                raise RuntimeError("webserver failed to start in %0.1f seconds" % maxwait)

    def teardown_method(self, method):
        """close the web server for this method"""
        try:
            self.server.sessioncache.clear()
            self.webserver.setstop(True)
            # ping the webserver to make it stop
            waited = 0.0
            sleep = 0.05
            max_wait = 5.0
            while self.webserverthread.isAlive():
                time.sleep(sleep)
                if not self.webserverthread.isAlive():
                    break
                # if the webserver is sleeping a request should wake it up...
                # try:
                #     x = urllib2.urlopen(self.baseaddress)
                # except Exception, e:
                #     pass
                waited += sleep
                sleep += 0.05
                if waited >= max_wait:
                    raise RuntimeError("webserver failed to stop in %0.1f seconds" % (waited))
	finally:
            test_create.NoReuse.teardown_method(self, method)