Hooks for lettuce to run a test pylons server

terrain.py

'''Hooks for lettuce to run a test pylons server'''

import threading
from os import getcwd

import pylons.test

from paste.deploy.loadwsgi import loadapp
from paste.httpserver import serve
from paste.script.appinstall import SetupCommand

from routes.util import URLGenerator

from lettuce import *


@before.all
def start_test_server():
    '''Start a test server

    Adds the following to world:
    - app: The WSGI application defined in test.ini
    - environ:
    - url: A routes url generator for the app
    - server: The paste web server
    - server_thread: The thread from which the paste server is started
    '''

    if not hasattr(world, 'server_thread'):
        # create the server first, so we can give the app a baseurl hint
        # (needed to configure the cas plugin to talk to the cas stub server)
        world.server = serve(None, port=0, start_loop=False)
        world.environ = {}
        world.environ.update(SERVER_NAME=str(world.server.server_name))
        world.environ.update(SERVER_PORT=str(world.server.server_port))
        if not hasattr(world, 'app'):
            if pylons.test.pylonsapp:
                print "Using pylons.test.pylonsapp"
                world.app = pylons.test.pylonsapp
            else:
                print "Loading test app from test.ini"
                global_conf = {'test.baseurl':'http://%s:%s/' %
                        (world.server.server_name, world.server.server_port)}
                world.app = loadapp('config:test.ini', relative_to=getcwd(),
                        global_conf=global_conf)
                # setup-app looks here for an app before loading it, we need to
                # use the same app otherwise we won't be using the same in
                # memory database
                pylons.test.pylonsapp = world.app
            world.server.wsgi_application = world.app
            SetupCommand('setup-app').run([world.app.config['__file__']])
            config = world.app.config
            world.url = URLGenerator(config['routes.map'], world.environ)

        world.server_thread = threading.Thread(target=world.server.serve_forever, name=repr(world.server))
        print "Server starting"
        world.server_thread.start()


@after.all
def stop_test_server(total):
    if hasattr(world, 'server_thread'):
        print "Server stopping"
        world.server.server_close()
        world.server_thread.join()
        del world.server
        del world.server_thread

This code tries to make everything automatic. It uses a random available port for the server and I used a sqlite:///:memory: database, this allows you to have more than one set of tests running at once.