| Index: tests/test_page_outputs.py |
| =================================================================== |
| --- a/tests/test_page_outputs.py |
| +++ b/tests/test_page_outputs.py |
| @@ -12,16 +12,19 @@ |
| expected_out_path = os.path.join(ROOTPATH, 'tests', 'expected_output') |
| outputs = get_dir_contents(expected_out_path) |
| for filename in list(outputs): |
| # Move test-type-specific expected outputs (e.g. "xyz@static" -> "xyz") |
| # and remove the expected outputs that don't apply for this test type. |
| if filename.endswith('@' + test_type): |
| realname = filename.split('@')[0] |
| outputs[realname] = outputs[filename] |
| + if '+' in filename: |
| + realname = ''.join(filename.split('@' + test_type)) |
| + outputs[realname] = outputs[filename] |
| if '@' in filename: |
| del outputs[filename] |
| return outputs.items() |
| static_expected_outputs = get_expected_outputs('static') |
| dynamic_expected_outputs = get_expected_outputs('dynamic') |
| @@ -49,20 +52,23 @@ |
| @pytest.fixture(scope='session') |
| def output_pages(static_output): |
| return get_dir_contents(static_output) |
| @pytest.mark.parametrize('filename,expected_output', static_expected_outputs) |
| -def test_static(output_pages, filename, expected_output): |
| +def test_static(revision, output_pages, filename, expected_output): |
| if expected_output.startswith('## MISSING'): |
| assert filename not in output_pages |
| - else: |
| + elif revision and '+' + revision in filename: |
| + filename = filename.split('+')[0] |
| + assert expected_output == output_pages[filename] |
| + elif not revision and '+' not in filename: |
| assert expected_output == output_pages[filename] |
| @pytest.mark.parametrize('filename,expected_output', dynamic_expected_outputs) |
| def test_dynamic(dynamic_server, filename, expected_output): |
| response = urllib2.urlopen(dynamic_server + filename) |
| assert expected_output == response.read().strip() |