steps file to go with the feature

gistfile1.py

from lettuce import *
from fabric import api
from fabric.api import env
import socket


servers = {
        'xxx.local': {
            'ip': ' xxxx',
            'roles': ['production']
        }
}

# world is the lettuce singleton
world.hostname = socket.gethostname()
world.installed = {'postfix': False }

# configure fabric to point at localhost
# and to supress output
env.host_string = 'localhost'
env.warn_only = True
import fabric.state
for k in  fabric.state.output:
    fabric.state.output[k] = False


@step('I am a (\w+) server')
def has_a_role(step, role):
    hostname = api.run('hostname')
    return role in servers[hostname]['roles']

@step('I am not a (\w+) server')
def doesnt_have_a_role(step, role):
    return not has_a_role(step, role)

@step('I should have (\w+) installed')
def check_installed(step, package):
    # should call out to apt/rpm
    return world.installed.get(package, False)

@step('I should have a mail relay to (\w+)')
def check_mail_relay(step, package):
    result = api.run('grep foo /etc/passwd')
    assert result, 'relay not set up'


@step('(\w+) should be running')
def check_service_running(step, service):
    return True

@step('I should be listening on port (\d+)')
def check_service_running(step, port):
    return True