refactor: changing find resources to find nodes (#23)

src/cloudformation.py

@@ -24,9 +24,10 @@ class CFLoader(yaml.SafeLoader):
  pass
 
 
-class ResourceType(Enum):
+class NodeType(Enum):
  API_GATEWAY = "AWS::Serverless::Api"
  LAMBDA = "AWS::Serverless::Function"
+ API = "Api"
 
 
 def multi_constructor(loader: CFLoader, tag_suffix: str, node: yaml.nodes.Node) -> Dict[str, Any]:
@@ -77,11 +78,11 @@ def load(template: Optional[str] = None) -> Dict[str, Any]:
  return yaml.load(fp, CFLoader)
 
 
-def find_resources(template: Dict[str, Any], resource_type: ResourceType) -> List[Dict[str, Any]]:
- resources = []
+def find_nodes(tree: Dict[str, Any], node_type: NodeType) -> List[Dict[str, Any]]:
+ nodes = []
 
- for id_, properties in template["Resources"].items():
- if properties["Type"] == resource_type.value:
- resources.append({id_: properties})
+ for id_, properties in tree.items():
+ if properties["Type"] == node_type.value:
+ nodes.append({id_: properties})
 
- return resources
+ return nodes

tests/test_cloudformation.py

@@ -86,11 +86,11 @@ def test_load_raises_exception(self):
  with self.assertRaisesRegex(cf.CFTemplateNotFound, regex):
  cf.load(template)
 
- def test_find_resources(self):
- template = cf.load("tests/fixtures/templates/example1.yml")
- resources = cf.find_resources(template, cf.ResourceType.LAMBDA)
+ def test_find_nodes(self):
+ tree = cf.load("tests/fixtures/templates/example1.yml")
+ nodes = cf.find_nodes(tree["Resources"], cf.NodeType.LAMBDA)
 
- expected_resources = [
+ expected_nodes = [
  {
  "HelloWorldFunction": {
  "Type": "AWS::Serverless::Function",
@@ -110,4 +110,4 @@ def test_find_resources(self):
  }
  ]
 
- self.assertEqual(resources, expected_resources)
+ self.assertEqual(nodes, expected_nodes)