Cleaned up mess with parametrize decorators, added two more tests

Author: grazor

tests/app.py

@@ -1,4 +1,4 @@
-from functools import partial
+import pytest
 
 try:
     from urllib import urlencode
@@ -44,3 +44,12 @@ def url_string(uri='/graphql', **url_params):
 
     return string
 
+
+def parametrize_sync_async_app_test(arg, **extra_options):
+    def decorator(test):
+        apps = []
+        for ae in [False, True]:
+            apps.append(create_app(async_executor=ae, **extra_options))
+
+        return pytest.mark.parametrize('app', apps)(test)
+    return decorator

tests/test_graphiqlview.py

@@ -2,7 +2,8 @@
 
 from jinja2 import Environment
 
-from .app import create_app, url_string
+from .app import create_app, url_string, parametrize_sync_async_app_test
+from .schema import AsyncSchema
 
 
 @pytest.fixture
@@ -16,49 +17,61 @@ def pretty_response():
     ).replace('\"','\\\"').replace('\n', '\\n')
 
 
-@pytest.mark.parametrize('app', [
-    (create_app(async_executor=False, graphiql=True)),
-    (create_app(async_executor=True, graphiql=True)),
-])
+@parametrize_sync_async_app_test('app', graphiql=True)
 def test_graphiql_is_enabled(app):
     _, response = app.client.get( uri=url_string(), headers={'Accept': 'text/html'})
     assert response.status == 200
 
 
-@pytest.mark.parametrize('app', [
-    (create_app(async_executor=False, graphiql=True)),
-    (create_app(async_executor=True, graphiql=True)),
-])
+@parametrize_sync_async_app_test('app', graphiql=True)
 def test_graphiql_simple_renderer(app, pretty_response):
     _, response = app.client.get(uri=url_string(query='{test}'), headers={'Accept': 'text/html'})
     assert response.status == 200
     assert pretty_response in response.body.decode('utf-8')
 
 
-@pytest.mark.parametrize('app', [
-    (create_app(async_executor=False, graphiql=True, jinja_env=Environment())),
-    (create_app(async_executor=True, graphiql=True, jinja_env=Environment())),
-])
+@parametrize_sync_async_app_test('app', graphiql=True, jinja_env=Environment())
 def test_graphiql_jinja_renderer(app, pretty_response):
     _, response = app.client.get(uri=url_string(query='{test}'), headers={'Accept': 'text/html'})
     assert response.status == 200
     assert pretty_response in response.body.decode('utf-8')
 
 
-@pytest.mark.parametrize('app', [
-    (create_app(async_executor=False, graphiql=True, jinja_env=Environment(enable_async=True))),
-    (create_app(async_executor=True, graphiql=True, jinja_env=Environment(enable_async=True))),
-])
+@parametrize_sync_async_app_test('app', graphiql=True, jinja_env=Environment(enable_async=True))
 def test_graphiql_jinja_async_renderer(app, pretty_response):
     _, response = app.client.get(uri=url_string(query='{test}'), headers={'Accept': 'text/html'})
     assert response.status == 200
     assert pretty_response in response.body.decode('utf-8')
 
 
-@pytest.mark.parametrize('app', [
-    (create_app(async_executor=False, graphiql=True, jinja_env=Environment())),
-    (create_app(async_executor=True, graphiql=True, jinja_env=Environment())),
-])
+@parametrize_sync_async_app_test('app', graphiql=True)
 def test_graphiql_html_is_not_accepted(app):
     _, response = app.client.get(uri=url_string(), headers={'Accept': 'application/json'})
     assert response.status == 400
+
+
+@parametrize_sync_async_app_test('app', graphiql=True)
+def test_graphiql_get_mutation(app, pretty_response):
+    query = 'mutation TestMutation { writeTest { test } }'
+    _, response = app.client.get(uri=url_string(query=query), headers={'Accept': 'text/html'})
+    assert response.status == 200
+    assert 'response: null' in response.body.decode('utf-8')
+
+
+@pytest.mark.parametrize('app', [create_app(async_executor=True, graphiql=True, schema=AsyncSchema)])
+def test_graphiql_asyncio_schema(app):
+    query = '{a,b,c}'
+    _, response = app.client.get(uri=url_string(query=query), headers={'Accept': 'text/html'})
+
+    expected_response = (
+        '{\n'
+        '  "data": {\n'
+        '    "a": "hey",\n'
+        '    "b": "hey2",\n'
+        '    "c": "hey3"\n'
+        '  }\n'
+        '}'
+    ).replace('\"','\\\"').replace('\n', '\\n')
+
+    assert response.status == 200
+    assert expected_response in response.body.decode('utf-8')