def test_read_from_definition_uri(self):
uri = "./file.txt"
expected = {"some": "value"}
reader = SamSwaggerReader(definition_uri=uri)
reader._download_swagger = Mock()
reader._download_swagger.return_value = expected
actual = reader.read()
self.assertEquals(actual, expected)
reader._download_swagger.assert_called_with(uri)
def test_read_from_definition_uri(self):
uri = "./file.txt"
expected = {"some": "value"}
reader = SamSwaggerReader(definition_uri=uri)
reader._download_swagger = Mock()
reader._download_swagger.return_value = expected
actual = reader.read()
self.assertEquals(actual, expected)
reader._download_swagger.assert_called_with(uri)
def _extract_from_serverless_api(self, logical_id, api_resource,
collector):
"""
Extract APIs from AWS::Serverless::Api resource by reading and parsing Swagger documents. The result is added
to the collector.
Parameters
----------
logical_id : str
Logical ID of the resource
api_resource : dict
Resource definition, including its properties
collector : ApiCollector
Instance of the API collector that where we will save the API information
"""
properties = api_resource.get("Properties", {})
body = properties.get("DefinitionBody")
uri = properties.get("DefinitionUri")
binary_media = properties.get("BinaryMediaTypes", [])
stage_name = properties.get("StageName")
stage_variables = properties.get("Variables")
if not body and not uri:
# Swagger is not found anywhere.
LOG.debug(
"Skipping resource '%s'. Swagger document not found in DefinitionBody and DefinitionUri",
logical_id)
return
reader = SamSwaggerReader(definition_body=body,
definition_uri=uri,
working_dir=self.cwd)
swagger = reader.read()
parser = SwaggerParser(swagger)
apis = parser.get_apis()
LOG.debug("Found '%s' APIs in resource '%s'", len(apis), logical_id)
collector.add_apis(logical_id, apis)
collector.add_binary_media_types(
logical_id,
parser.get_binary_media_types()) # Binary media from swagger
collector.add_binary_media_types(
logical_id,
binary_media) # Binary media specified on resource in template
collector.add_stage_name(logical_id, stage_name)
collector.add_stage_variables(logical_id, stage_variables)
def test_must_read_first_from_definition_body(self):
body = {"this is": "swagger"}
uri = "./file.txt"
expected = {"some": "value"}
reader = SamSwaggerReader(definition_body=body, definition_uri=uri)
reader._download_swagger = Mock()
reader._read_from_definition_body = Mock()
reader._read_from_definition_body.return_value = expected
actual = reader.read()
self.assertEquals(actual, expected)
reader._read_from_definition_body.assert_called_with()
reader._download_swagger.assert_not_called()
def test_must_read_first_from_definition_body(self):
body = {"this is": "swagger"}
uri = "./file.txt"
expected = {"some": "value"}
reader = SamSwaggerReader(definition_body=body, definition_uri=uri)
reader._download_swagger = Mock()
reader._read_from_definition_body = Mock()
reader._read_from_definition_body.return_value = expected
actual = reader.read()
self.assertEquals(actual, expected)
reader._read_from_definition_body.assert_called_with()
reader._download_swagger.assert_not_called()
def test_must_use_definition_uri_if_body_does_not_exist(self):
body = {"this is": "swagger"}
uri = "./file.txt"
expected = {"some": "value"}
reader = SamSwaggerReader(definition_body=body, definition_uri=uri)
reader._download_swagger = Mock()
reader._download_swagger.return_value = expected
# Set the output of reading the definition body to be None
reader._read_from_definition_body = Mock()
reader._read_from_definition_body.return_value = None
actual = reader.read()
self.assertEquals(actual, expected)
reader._read_from_definition_body.assert_called_with()
reader._download_swagger.assert_called_with(uri)
def test_must_use_definition_uri_if_body_does_not_exist(self):
body = {"this is": "swagger"}
uri = "./file.txt"
expected = {"some": "value"}
reader = SamSwaggerReader(definition_body=body, definition_uri=uri)
reader._download_swagger = Mock()
reader._download_swagger.return_value = expected
# Set the output of reading the definition body to be None
reader._read_from_definition_body = Mock()
reader._read_from_definition_body.return_value = None
actual = reader.read()
self.assertEquals(actual, expected)
reader._read_from_definition_body.assert_called_with()
reader._download_swagger.assert_called_with(uri)