Issue 7019 - [CMS] Refactor `test_server.py`

tests/test_dynamic_server.py

+import pytest
+import urllib2
+import os
+
+from tests.test_page_outputs import dynamic_expected_outputs
+from tests.utils import run_test_server, exception_test
+
+
+@pytest.fixture(scope='function')
+def dynamic_server_werkzeug(temp_site):
+    """Test server run using the `werkzeug` module."""
+    with run_test_server(str(temp_site)) as ts:
+        yield ts
+
+
+@pytest.fixture(scope='function')
+def dynamic_server_builtins(temp_site, tmpdir):
+    """Test server run using builtin modules."""
+    # Creating an invalid `werkzeug` module and add it to the head of the
+    # PYTHONPATH. This would cause the import of `werkzeug` to fail when
+    # running the test server and force it to use the builtin modules.
+    werkzeug_dir = tmpdir.mkdir('werkzeug')
+    werkzeug_dir.join('__init__.py').write('raise ImportError')
+    new_env = dict(os.environ)
+    new_env['PYTHONPATH'] = os.pathsep.join([str(tmpdir),
+                                             os.getenv('PYTHONPATH', '')])
+
+    with run_test_server(str(temp_site), new_env) as ts:
+        yield ts
+
+
+@pytest.mark.slowtest
+def test_dynamic_werkzeug_good_page(dynamic_server_werkzeug):
+    filename, expected_output = dynamic_expected_outputs[0]
+    response = urllib2.urlopen(dynamic_server_werkzeug + filename)
+
+    assert expected_output in response.read().strip()
+
+
+@pytest.mark.slowtest
+def test_dynamic_werkzeug_not_found(dynamic_server_werkzeug):
+    filename = 'en/no-page-here'
+    exp_msg = 'Not Found'
+
+    exception_test(urllib2.urlopen, urllib2.HTTPError, exp_msg,
+                   dynamic_server_werkzeug + filename)
+
+
+@pytest.mark.slowtest
+def test_dynamic_builtins_good_page(dynamic_server_builtins):
+    filename, expected_output = dynamic_expected_outputs[0]
+    response = urllib2.urlopen(dynamic_server_builtins + filename)
+
+    assert expected_output in response.read().strip()
+
+
+@pytest.mark.slowtest
+def test_dynamic_builtins_not_found(dynamic_server_builtins):
+    filename = 'en/no-page-here'
+    exp_msg = 'Not Found'
+
+    exception_test(urllib2.urlopen, urllib2.HTTPError, exp_msg,
+                   dynamic_server_builtins + filename)