0fd03cfb02
Refactored stub services for style and DRY Added unit tests for stub implementations Updated acceptance tests that depend on stubs. Updated Studio acceptance tests to use YouTube stub server; fixed failing tests in devstack.
38 lines
979 B
Python
38 lines
979 B
Python
"""
|
|
Initialize and teardown fake HTTP services for use in acceptance tests.
|
|
"""
|
|
|
|
from lettuce import before, after, world
|
|
from django.conf import settings
|
|
from terrain.stubs.youtube import StubYouTubeService
|
|
from terrain.stubs.xqueue import StubXQueueService
|
|
|
|
|
|
USAGE = "USAGE: python -m fakes.start SERVICE_NAME PORT_NUM"
|
|
|
|
SERVICES = {
|
|
"youtube": {"port": settings.YOUTUBE_PORT, "class": StubYouTubeService},
|
|
"xqueue": {"port": settings.XQUEUE_PORT, "class": StubXQueueService},
|
|
}
|
|
|
|
|
|
@before.all
|
|
def start_stubs():
|
|
"""
|
|
Start each stub service running on a local port.
|
|
"""
|
|
for name, service in SERVICES.iteritems():
|
|
fake_server = service['class'](port_num=service['port'])
|
|
setattr(world, name, fake_server)
|
|
|
|
|
|
@after.all
|
|
def stop_stubs(_):
|
|
"""
|
|
Shut down each stub service.
|
|
"""
|
|
for name in SERVICES.keys():
|
|
stub_server = getattr(world, name, None)
|
|
if stub_server is not None:
|
|
stub_server.shutdown()
|