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
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
|
#!/usr/bin/env python
# Copyright 2013 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
# Run build_server so that files needed by tests are copied to the local
# third_party directory.
import build_server
build_server.main()
import logging
import optparse
import os
import sys
import time
import unittest
from handler import Handler
from local_renderer import LocalRenderer
from test_util import DisableLogging
# Arguments set up if __main__ specifies them.
_BASE_PATH = os.path.join(
os.path.abspath(os.path.dirname(__file__)), os.pardir, os.pardir)
_EXPLICIT_TEST_FILES = None
def _GetPublicFiles():
'''Gets all public files mapped to their contents.
'''
public_path = os.path.join(_BASE_PATH, 'docs', 'templates', 'public', '')
public_files = {}
for path, dirs, files in os.walk(public_path, topdown=True):
dirs[:] = [d for d in dirs if d != '.svn']
relative_path = path[len(public_path):]
for filename in files:
with open(os.path.join(path, filename), 'r') as f:
public_files[os.path.join(relative_path, filename)] = f.read()
return public_files
class IntegrationTest(unittest.TestCase):
def setUp(self):
self._renderer = LocalRenderer(_BASE_PATH)
def testCronAndPublicFiles(self):
'''Runs cron then requests every public file. Cron needs to be run first
because the public file requests are offline.
'''
if _EXPLICIT_TEST_FILES is not None:
return
print('Running cron...')
start_time = time.time()
try:
render_content, render_status, _ = self._renderer.Render('/cron/stable')
self.assertEqual(200, render_status)
self.assertEqual('Success', render_content)
finally:
print('Took %s seconds' % (time.time() - start_time))
public_files = _GetPublicFiles()
print('Rendering %s public files...' % len(public_files.keys()))
start_time = time.time()
try:
for path, content in _GetPublicFiles().iteritems():
def check_result(render_content, render_status, _):
self.assertEqual(200, render_status,
'Got %s when rendering %s' % (render_status, path))
# This is reaaaaally rough since usually these will be tiny templates
# that render large files. At least it'll catch zero-length responses.
self.assertTrue(len(render_content) >= len(content),
'Content was "%s" when rendering %s' % (render_content, path))
check_result(*self._renderer.Render(path))
# Samples are internationalized, test some locales.
if path.endswith('/samples.html'):
for lang in ['en-US', 'es', 'ar']:
check_result(*self._renderer.Render(
path, headers={'Accept-Language': '%s;q=0.8' % lang}))
finally:
print('Took %s seconds' % (time.time() - start_time))
def testExplicitFiles(self):
'''Tests just the files in _EXPLICIT_TEST_FILES.
'''
if _EXPLICIT_TEST_FILES is None:
return
for filename in _EXPLICIT_TEST_FILES:
print('Rendering %s...' % filename)
start_time = time.time()
try:
render_content, render_status, _ = self._renderer.Render(
filename, always_online=True)
self.assertEqual(200, render_status)
self.assertTrue(render_content != '')
finally:
print('Took %s seconds' % (time.time() - start_time))
@DisableLogging('warning')
def testFileNotFound(self):
render_content, render_status, _ = self._renderer.Render(
'/extensions/notfound.html', always_online=True)
self.assertEqual(404, render_status)
if __name__ == '__main__':
parser = optparse.OptionParser()
parser.add_option('-p', '--path', default=None)
parser.add_option('-a', '--all', action='store_true', default=False)
(opts, args) = parser.parse_args()
if not opts.all:
_EXPLICIT_TEST_FILES = args
if opts.path is not None:
_BASE_PATH = opts.path
# Kill sys.argv because we have our own flags.
sys.argv = [sys.argv[0]]
unittest.main()
|
