def test_must_get_body_directly(self, parse_mock): body = {'this': 'swagger'} reader = SwaggerReader(definition_body=body) parse_mock.return_value = None # No location is returned from aws_include parser actual = reader._read_from_definition_body() self.assertEquals(actual, body)
def test_must_work_with_include_transform(self, parse_mock): body = {'this': 'swagger'} expected = {'k': 'v'} location = "some location" reader = SwaggerReader(definition_body=body) reader._download_swagger = Mock() reader._download_swagger.return_value = expected parse_mock.return_value = location actual = reader._read_from_definition_body() self.assertEquals(actual, expected) parse_mock.assert_called_with(body)
def test_must_read_first_from_definition_body(self): body = {"this is": "swagger"} uri = "./file.txt" expected = {"some": "value"} reader = SwaggerReader(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 = SwaggerReader(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)