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

tests/test_page_outputs.py

 import os
 import sys
 import runpy
 
 import mock
 import pytest
 import urllib2
 
 from .conftest import ROOTPATH
 from .utils import get_dir_contents, run_test_server
 from cms.sources import FileSource
+from cms.bin.test_server import DynamicServerHandler
 
 
 def get_expected_outputs(test_type):
     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")
         # There are cases where we need to test outputs which differ depending
         # on how they are generated; either statically or dynamically
         if filename.endswith('@' + test_type):
@@ skipping 11 matching lines @@
 
 @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]
     with mock.patch('cms.sources.FileSource.version', 1):
         runpy.run_module('cms.bin.generate_static_pages', run_name='__main__')
     return static_out_path
 
 
-@pytest.fixture(scope='module')
-def dynamic_server(temp_site):
-    with run_test_server(temp_site) as ts:
+@pytest.fixture(scope='function')
+def dynamic_server_werkzeug(temp_site):
+    with run_test_server(str(temp_site)) as ts:
         yield ts
 
 
+@pytest.fixture(scope='function')
+def dynamic_server_builtins(temp_site, tmpdir):
+    werkzeug_dir = tmpdir.mkdir('werkzeug')
+    werkzeug_dir.join('__init__.py').write('raise ImportError')
+    werkzeug_dir.join('serving.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.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):
     if expected_output.startswith('## MISSING'):
         assert filename not in output_pages
     else:
         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()
-
-
 def test_cache(output_pages):
     source = FileSource(os.path.join('test_site'))
     assert source.get_cache_dir() == os.path.join('test_site', 'cache')
+
+
+@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'
+
+    with pytest.raises(urllib2.HTTPError) as err:
+        urllib2.urlopen(dynamic_server_werkzeug + filename)
+
+    assert exp_msg in str(err.value)
+
+
+@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'
+
+    with pytest.raises(urllib2.HTTPError) as err:
+        urllib2.urlopen(dynamic_server_builtins + filename)
+
+    assert exp_msg in str(err.value)
+
+
+@pytest.mark.parametrize('filename,expected_output', dynamic_expected_outputs)
+def test_dynamic_server_handler(filename, expected_output, temp_site):
+    def cleanup(page):
+        return page.replace(os.linesep, '').strip()
+
+    handler = DynamicServerHandler('localhost', 5000, str(temp_site))
+    environ = {'PATH_INFO': filename}
+
+    generated_page = handler(environ, lambda x, y: None)
+
+    assert cleanup(expected_output) == cleanup(generated_page[0])
+
+
+@pytest.mark.parametrize('page', ['en/translate', '/en/translate'])
+def test_dynamic_server_handler_with_conflicts(page, temp_site_with_conflicts):
+    handler = DynamicServerHandler('localhost', 5000,
+                                   str(temp_site_with_conflicts))
+    environ = {'PATH_INFO': page}
+
+    with pytest.raises(Exception) as err:
+        handler(environ, lambda x, y: None)
+
+    assert str(err.value) == 'The requested page conflicts with another page.'