def test_with_inline_swagger_apis(self):
template = {
"Resources": {
"Api1": {"Type": "AWS::ApiGateway::RestApi", "Properties": {"Body": make_swagger(self.input_routes)}}
}
}
provider = ApiProvider(template)
self.assertCountEqual(self.input_routes, provider.routes)
def test_body_with_swagger_as_local_file_expect_fail(self):
with tempfile.NamedTemporaryFile(mode="w", delete=False) as fp:
filename = fp.name
swagger = make_swagger(self.input_routes)
json.dump(swagger, fp)
fp.flush()
template = {"Resources": {"Api1": {"Type": "AWS::ApiGateway::RestApi", "Properties": {"Body": filename}}}}
self.assertRaises(Exception, ApiProvider, template)
def test_with_swagger_as_local_file(self):
with tempfile.NamedTemporaryFile(mode="w", delete=False) as fp:
filename = fp.name
swagger = make_swagger(self.input_routes)
json.dump(swagger, fp)
fp.flush()
template = {
"Resources": {"Api1": {"Type": "AWS::ApiGateway::RestApi", "Properties": {"BodyS3Location": filename}}}
}
provider = ApiProvider(template)
self.assertCountEqual(self.input_routes, provider.routes)
def test_with_swagger_as_both_body_and_uri_called(self, SwaggerReaderMock):
body = {"some": "body"}
filename = "somefile.txt"
template = {
"Resources": {
"Api1": {"Type": "AWS::ApiGateway::RestApi", "Properties": {"BodyS3Location": filename, "Body": body}}
}
}
SwaggerReaderMock.return_value.read.return_value = make_swagger(self.input_routes)
cwd = "foo"
provider = ApiProvider(template, cwd=cwd)
self.assertCountEqual(self.input_routes, provider.routes)
SwaggerReaderMock.assert_called_with(definition_body=body, definition_uri=filename, working_dir=cwd)
def test_with_binary_media_types(self):
template = {
"Resources": {
"Api1": {
"Type": "AWS::ApiGateway::RestApi",
"Properties": {"Body": make_swagger(self.input_routes, binary_media_types=self.binary_types)},
}
}
}
expected_binary_types = sorted(self.binary_types)
expected_apis = [
Route(path="/path1", methods=["GET", "POST"], function_name="SamFunc1"),
Route(path="/path2", methods=["PUT", "GET"], function_name="SamFunc1"),
Route(path="/path3", methods=["DELETE"], function_name="SamFunc1"),
]
provider = ApiProvider(template)
self.assertCountEqual(expected_apis, provider.routes)
self.assertCountEqual(provider.api.binary_media_types, expected_binary_types)
def test_with_binary_media_types_in_swagger_and_on_resource(self):
input_routes = [Route(path="/path", methods=["OPTIONS"], function_name="SamFunc1")]
extra_binary_types = ["text/html"]
template = {
"Resources": {
"Api1": {
"Type": "AWS::ApiGateway::RestApi",
"Properties": {
"BinaryMediaTypes": extra_binary_types,
"Body": make_swagger(input_routes, binary_media_types=self.binary_types),
},
}
}
}
expected_binary_types = sorted(self.binary_types + extra_binary_types)
expected_routes = [Route(path="/path", methods=["OPTIONS"], function_name="SamFunc1")]
provider = ApiProvider(template)
self.assertCountEqual(expected_routes, provider.routes)
self.assertCountEqual(provider.api.binary_media_types, expected_binary_types)
def test_swagger_with_any_method(self):
routes = [Route(path="/path", methods=["any"], function_name="SamFunc1")]
expected_routes = [
Route(
path="/path",
methods=["GET", "DELETE", "PUT", "POST", "HEAD", "OPTIONS", "PATCH"],
function_name="SamFunc1",
)
]
template = {
"Resources": {"Api1": {"Type": "AWS::ApiGateway::RestApi", "Properties": {"Body": make_swagger(routes)}}}
}
provider = ApiProvider(template)
self.assertCountEqual(expected_routes, provider.routes)