Add json_only keyword to allow serving only swagger.json and not static files

Add a test

Author: ukch

aiohttp_swagger/__init__.py

@@ -46,7 +46,8 @@ def setup_swagger(app: web.Application,
                   contact: str = "",
                   swagger_home_decor: FunctionType = None,
                   swagger_def_decor: FunctionType = None,
-                  swagger_info: dict = None):
+                  swagger_info: dict = None,
+                  json_only: bool = False):
     _swagger_url = ("/{}".format(swagger_url)
                     if not swagger_url.startswith("/")
                     else swagger_url)
@@ -74,9 +75,15 @@ def setup_swagger(app: web.Application,
         _swagger_def_func = swagger_def_decor(_swagger_def)
 
     # Add API routes
+    app.router.add_route('GET', _swagger_def_url, _swagger_def_func)
+
+    app["SWAGGER_DEF_CONTENT"] = swagger_info
+
+    if json_only:
+        return
+
     app.router.add_route('GET', _swagger_url, _swagger_home_func)
     app.router.add_route('GET', "{}/".format(_swagger_url), _swagger_home_func)
-    app.router.add_route('GET', _swagger_def_url, _swagger_def_func)
 
     # Set statics
     statics_path = '{}/swagger_static'.format(_swagger_url)
@@ -85,7 +92,6 @@ def setup_swagger(app: web.Application,
     # --------------------------------------------------------------------------
     # Build templates
     # --------------------------------------------------------------------------
-    app["SWAGGER_DEF_CONTENT"] = swagger_info
     with open(join(STATIC_PATH, "index.html"), "r") as f:
         app["SWAGGER_TEMPLATE_CONTENT"] = (
             f.read()

setup.py

@@ -1,3 +1,4 @@
+#!/usr/bin/env python
 # -*- coding: utf-8 -*-
 #
 # aiohttp-swagger

tests/test_swagger.py

@@ -1,5 +1,6 @@
 import asyncio
 import json
+import pathlib
 import pytest
 import yaml
 from os.path import join, dirname, abspath
@@ -50,7 +51,8 @@ def test_swagger_file_url(test_client, loop):
 
     app = web.Application(loop=loop)
     setup_swagger(app,
-                  swagger_from_file=TESTS_PATH + "/data/example_swagger.yaml")
+                  swagger_from_file=TESTS_PATH + "/data/example_swagger.yaml",
+                  json_only=True)
 
     client = yield from test_client(app)
     resp1 = yield from client.get('/api/doc/swagger.json')
@@ -66,7 +68,7 @@ def test_swagger_file_url(test_client, loop):
 def test_partial_swagger_file(test_client, loop):
     app = web.Application(loop=loop)
     app.router.add_route('GET', "/ping-partial", ping_partial)
-    setup_swagger(app)
+    setup_swagger(app, json_only=True)
 
     client = yield from test_client(app)
     resp1 = yield from client.get('/api/doc/swagger.json')
@@ -86,7 +88,8 @@ def test_custom_swagger(test_client, loop):
                   description=description,
                   title="Test Custom Title",
                   api_version="1.0.0",
-                  contact="my.custom.contact@example.com")
+                  contact="my.custom.contact@example.com",
+                  json_only=True)
 
     client = yield from test_client(app)
     resp1 = yield from client.get('/api/v1/doc/swagger.json')
@@ -108,7 +111,8 @@ def test_swagger_home_decorator(test_client, loop):
                   title="Test Custom Title",
                   api_version="1.0.0",
                   contact="my.custom.contact@example.com",
-                  swagger_home_decor=lambda x: x)
+                  swagger_home_decor=lambda x: x,
+                  json_only=True)
 
     client = yield from test_client(app)
     resp1 = yield from client.get('/api/v1/doc/swagger.json')
@@ -130,7 +134,8 @@ def test_swagger_def_decorator(test_client, loop):
                   title="Test Custom Title",
                   api_version="1.0.0",
                   contact="my.custom.contact@example.com",
-                  swagger_def_decor=lambda x: x)
+                  swagger_def_decor=lambda x: x,
+                  json_only=True)
 
     client = yield from test_client(app)
     resp1 = yield from client.get('/api/v1/doc/swagger.json')
@@ -154,7 +159,8 @@ def test_swagger_info(test_client, loop, swagger_info):
     description = "Test Custom Swagger"
     setup_swagger(app,
                   swagger_url="/api/v1/doc",
-                  swagger_info=swagger_info)
+                  swagger_info=swagger_info,
+                  json_only=True)
 
     client = yield from test_client(app)
     resp1 = yield from client.get('/api/v1/doc/swagger.json')
@@ -164,3 +170,22 @@ def test_swagger_info(test_client, loop, swagger_info):
     assert '/example1' in result['paths']
     assert '/example2' in result['paths']
     assert 'API Title' in result['info']['title']
+
+
+@asyncio.coroutine
+def test_swagger_defined_paths(test_client, loop):
+    app1 = web.Application(loop=loop)
+    setup_swagger(app1, json_only=True)
+    urls1 = [r.get_info() for r in app1.router.resources()]
+    assert urls1 == [{'path': '/api/doc/swagger.json'}]
+
+    app2 = web.Application(loop=loop)
+    setup_swagger(app2)
+    urls2 = [r.get_info() for r in app2.router.resources()]
+    assert urls2 == [
+        {'path': '/api/doc/swagger.json'},
+        {'path': '/api/doc'},
+        {'path': '/api/doc/'},
+        {'directory': pathlib.Path('aiohttp_swagger/swagger_ui').absolute(),
+         'prefix': '/api/doc/swagger_static'},
+    ]