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

Unified Diff: cms/tests/test_page_outputs.py

Issue 29345468: Issue 4045 - Add Test Suite To CMS (Closed)
Patch Set: Add tox.ini patch and change dyncamic server to yield_fixture Created July 13, 2016, 11:17 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
Index: cms/tests/test_page_outputs.py
===================================================================
new file mode 100644
--- /dev/null
+++ b/cms/tests/test_page_outputs.py
@@ -0,0 +1,70 @@
+import os
+import sys
+import shutil
+import time
+import runpy
+import pytest
+import urllib2
+import subprocess
+
+ROOTPATH = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+
+
+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()
Vasily Kuznetsov 2016/07/14 12:14:00 Is it intentional that we ignore the directory her
Jon Sonesen 2016/07/18 14:48:28 The reason we are ignoring the directories is beca
+ return return_data
+
+
+def get_expected_outputs():
+ expected_out_path = os.path.join(ROOTPATH, 'tests', 'expected_output')
+ return get_dir_contents(expected_out_path).items()
+
+expected_outputs = get_expected_outputs()
+
+
+@pytest.fixture(scope='session')
+def temp_site(tmpdir_factory):
+ out_dir = tmpdir_factory.mktemp('temp_out')
+ site_dir = out_dir.join('test_site').strpath
+
+ shutil.copytree(os.path.join(ROOTPATH, 'tests', 'test_site'), site_dir)
+ subprocess.check_call(['hg', 'init', site_dir])
+ subprocess.check_call(['hg', '-R', site_dir, 'commit', '-A', '-m', 'foo'])
+ return site_dir
+
+
+@pytest.fixture(scope='session')
+def static_output(request, temp_site):
+ static_out_path = os.path.join(temp_site, 'static_out')
+ sys.argv = ['filler', temp_site, static_out_path]
+
+ runpy.run_module('cms.bin.generate_static_pages', run_name='__main__')
+ return static_out_path
+
+
+@pytest.yield_fixture(scope='session')
+def dynamic_server(temp_site):
+ p = subprocess.Popen(['python', 'runserver.py', temp_site])
+ time.sleep(0.5)
+ yield 'http://localhost:5000/root/'
+ p.terminate()
+
+
+@pytest.fixture(scope='session')
+def output_pages(static_output):
+ return get_dir_contents(static_output)
+
+
+@pytest.mark.parametrize('filename,expected_output', expected_outputs)
+def test_static(output_pages, filename, expected_output):
+ assert output_pages[filename] == expected_output
+
+
+@pytest.mark.parametrize('filename,expected_output', expected_outputs)
+def test_dynamic(dynamic_server, filename, expected_output):
+ response = urllib2.urlopen(dynamic_server + filename)
+ assert response.read() == expected_output

Powered by Google App Engine
This is Rietveld