[CORD-1136] Added test for swagger url
Change-Id: I751fa89d330dd4fee3b7e25de28820e4408831b8
diff --git a/tests/utests/chameleon/protoc_plugins/swagger_template_test.py b/tests/utests/chameleon/protoc_plugins/swagger_template_test.py
index 1344430..577d070 100644
--- a/tests/utests/chameleon/protoc_plugins/swagger_template_test.py
+++ b/tests/utests/chameleon/protoc_plugins/swagger_template_test.py
@@ -16,6 +16,7 @@
import json
import os
from unittest import TestCase
+from mock import Mock
from chameleon.protoc_plugins.descriptor_parser import DescriptorParser
from chameleon.protoc_plugins.swagger_template \
@@ -38,6 +39,22 @@
swagger = native_descriptors_to_swagger(native_data)
return swagger
+ def test_swagger_url(self):
+
+ grpc = {
+ '_channel': {
+ '_Rendezvous': {}
+ }
+ }
+ from chameleon.web_server.web_server import WebServer
+ server = yield WebServer(9101, '/', '/swagger', grpc)
+ server.app = 'app'
+
+ server.add_swagger_routes = Mock()
+
+ self.assertEqual(server.add_swagger_routes.call_count, 1)
+ server.add_swagger_routes.assert_called_once_with('app', '/swagger')
+
def test_empty_def(self):
proto = unindent("""