1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86
|
import sys
import httplib
import os
import pkg_resources
import shutil
import subprocess
import tempfile
import time
import signal
class TemplateTest(object):
def make_venv(self, directory): # pragma: no cover
import virtualenv
import sys
from virtualenv import Logger
logger = Logger([(Logger.level_for_integer(2), sys.stdout)])
virtualenv.logger = logger
virtualenv.create_environment(directory,
site_packages=False,
clear=False,
unzip_setuptools=True,
use_distribute=False)
def install(self, tmpl_name): # pragma: no cover
try:
self.old_cwd = os.getcwd()
self.directory = tempfile.mkdtemp()
self.make_venv(self.directory)
os.chdir(pkg_resources.get_distribution('pyramid').location)
subprocess.check_call(
[os.path.join(self.directory, 'bin', 'python'),
'setup.py', 'develop'])
os.chdir(self.directory)
subprocess.check_call(['bin/paster', 'create', '-t', tmpl_name,
'Dingle'])
os.chdir('Dingle')
py = os.path.join(self.directory, 'bin', 'python')
subprocess.check_call([py, 'setup.py', 'install'])
subprocess.check_call([py, 'setup.py', 'test'])
paster = os.path.join(self.directory, 'bin', 'paster')
for ininame, hastoolbar in (('development.ini', True),
('production.ini', False)):
proc = subprocess.Popen([paster, 'serve', ininame])
try:
time.sleep(5)
proc.poll()
if proc.returncode is not None:
raise RuntimeError('%s didnt start' % ininame)
conn = httplib.HTTPConnection('localhost:6543')
conn.request('GET', '/')
resp = conn.getresponse()
assert resp.status == 200, ininame
data = resp.read()
toolbarchunk = '<div id="pDebug"'
if hastoolbar:
assert toolbarchunk in data, ininame
else:
assert not toolbarchunk in data, ininame
finally:
if hasattr(proc, 'terminate'):
# 2.6+
proc.terminate()
else:
# 2.5
os.kill(proc.pid, signal.SIGTERM)
finally:
shutil.rmtree(self.directory)
os.chdir(self.old_cwd)
if __name__ == '__main__': # pragma: no cover
if not hasattr(subprocess, 'check_call'):
# 2.4
def check_call(*arg, **kw):
returncode = subprocess.call(*arg, **kw)
if returncode:
raise ValueError(returncode)
subprocess.check_call = check_call
templates = ['pyramid_starter', 'pyramid_alchemy', 'pyramid_routesalchemy',]
if sys.version_info >= (2, 5):
templates.append('pyramid_zodb')
for name in templates:
test = TemplateTest()
test.install(name)
|