feat: add gateways property (#34)

* feat: add gateways property * refactor: update dict name
noverde · Nov 30, 2023 · 6d82508 · 6d82508
1 parent ae2b33a
commit 6d82508
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/src/cloudformation.py b/src/cloudformation.py
@@ -63,11 +63,18 @@ def construct_getatt(node: yaml.nodes.Node) -> List[Any]:
 class Template:
  def __init__(self, template: Optional[str] = None) -> None:
  self._template = self.load(template)
+ self._gateways = None
 
  @property
  def template(self):
  return self._template
 
+ @property
+ def gateways(self):
+ if not self._gateways:
+ self._gateways = self.find_nodes(self._template["Resources"], NodeType.API_GATEWAY)
+ return self._gateways
+
  def load(self, template: Optional[str] = None) -> Dict[str, Any]:
  path: Optional[Path] = None
 

diff --git a/tests/test_cloudformation.py b/tests/test_cloudformation.py
@@ -89,6 +89,23 @@ def test_load_raises_exception(self):
  with self.assertRaisesRegex(cf.CFTemplateNotFound, regex):
  Template(template)
 
+ def test_list_gateways(self):
+ cloudformation = Template("tests/fixtures/templates/example2.yml")
+
+ expected_gateways = [
+ {
+ "ApiGateway": {
+ "Type": "AWS::Serverless::Api",
+ "Properties": {
+ "Name": "sam-api",
+ "StageName": "v1",
+ },
+ },
+ },
+ ]
+
+ self.assertEqual(cloudformation.gateways, expected_gateways)
+
  def test_find_nodes(self):
  cloudformation = Template("tests/fixtures/templates/example1.yml")
  tree = cloudformation.template