Rietveld Code Review Tool
Help | Bug tracker | Discussion group | Source code

Unified Diff: tests/test_page_outputs.py

Issue 29495555: Fixes 5343 - add global function: get_canonical_url (Closed)
Patch Set: Implement get_canonical_url (according to updated ticket) Created July 28, 2017, 10:07 a.m.
Use n/p to move between diff chunks; N/P to move between comments.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « tests/expected_output/sitemap ('k') | tests/test_site/pages/get_page_url.tmpl » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: tests/test_page_outputs.py
===================================================================
--- a/tests/test_page_outputs.py
+++ b/tests/test_page_outputs.py
@@ -9,25 +9,34 @@
from conftest import ROOTPATH
def get_dir_contents(path):
return_data = {}
for dirpath, dirnames, filenames in os.walk(path):
for output_file in filenames:
with open(os.path.join(dirpath, output_file)) as f:
- return_data[output_file] = f.read()
+ return_data[output_file] = f.read().strip()
return return_data
-def get_expected_outputs():
+def get_expected_outputs(test_type):
expected_out_path = os.path.join(ROOTPATH, 'tests', 'expected_output')
- return get_dir_contents(expected_out_path).items()
+ outputs = get_dir_contents(expected_out_path)
+ for filename in list(outputs):
+ if filename.endswith('@' + test_type):
+ realname = filename.split('@')[0]
+ outputs[realname] = outputs[filename]
+ if '@' in filename:
+ del outputs[filename]
+ return outputs.items()
-expected_outputs = get_expected_outputs()
+
+static_expected_outputs = get_expected_outputs('static')
+dynamic_expected_outputs = get_expected_outputs('dynamic')
@pytest.fixture(scope='session', params=['master', None])
def revision(request):
return request.param
@pytest.fixture(scope='session')
@@ -45,33 +54,33 @@
def dynamic_server(temp_site):
args = ['python', 'runserver.py', temp_site]
# Werkzeug is a dependency of flask which we are using for the mock api
# however there is an issue with Werkzeug that prevents it from properly
# handling the SIGTERM sent by p.kill() or terminate()
# Issue: https://github.com/pallets/werkzeug/issues/58
p = subprocess.Popen(args, stdout=subprocess.PIPE, preexec_fn=os.setsid)
time.sleep(0.5)
- yield 'http://localhost:5000/root/'
+ yield 'http://localhost:5000/en/'
os.killpg(os.getpgid(p.pid), signal.SIGTERM)
@pytest.fixture(scope='session')
def output_pages(static_output):
return get_dir_contents(static_output)
-@pytest.mark.parametrize('filename,expected_output', expected_outputs)
+@pytest.mark.parametrize('filename,expected_output', static_expected_outputs)
def test_static(output_pages, filename, expected_output):
assert output_pages[filename] == expected_output
-@pytest.mark.parametrize('filename,expected_output', expected_outputs)
+@pytest.mark.parametrize('filename,expected_output', dynamic_expected_outputs)
def test_dynamic(dynamic_server, filename, expected_output):
response = urllib2.urlopen(dynamic_server + filename)
- assert response.read() == expected_output
+ assert response.read().strip() == expected_output
def test_revision_arg(revision, output_pages):
if revision is None:
assert 'bar' in output_pages
else:
assert 'bar' not in output_pages
« no previous file with comments | « tests/expected_output/sitemap ('k') | tests/test_site/pages/get_page_url.tmpl » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld