Fix missing resources

Author: dylanhellems

terraform_compliance/extensions/terraform.py

@@ -76,8 +76,6 @@ def _parse_resources(self):
         :return: none
         '''
 
-        #TODO: Consider about using 'resource_changes' instead of 'resources'
-
         # Resources ( exists in Plan )
         for findings in seek_key_in_dict(self.raw.get('planned_values', {}).get('root_module', {}), 'resources'):
             for resource in findings.get('resources', []):
@@ -110,6 +108,18 @@ def _parse_resources(self):
                 else:
                     self.resources[resource['address']] = resource
 
+        # Resource Changes ( exists in Plan )
+        for finding in self.raw.get('resource_changes', {}):
+            resource = deepcopy(finding)
+            resource['values'] = resource.get('change', {}).get('after', {})
+            if 'change' in resource:
+                del resource['change']
+
+            if resource['address'].startswith('data'):
+                self.data[resource['address']] = resource
+            else:
+                self.resources[resource['address']] = resource
+
     def _parse_configurations(self):
         '''
         Assigns all configuration related data defined in the terraform plan. This is mostly used for

tests/terraform_compliance/extensions/test_terraform.py

@@ -154,6 +154,45 @@ def test_parse_resources_child_resources_exists_in_the_state_resource(self, *arg
         obj._parse_resources()
         self.assertEqual(obj.resources['something'], {'address': 'something'})
 
+    @patch.object(TerraformParser, '_read_file', return_value={})
+    def test_parse_resources_resources_exists_in_the_resource_changes_data(self, *args):
+        obj = TerraformParser('somefile', parse_it=False)
+        obj.raw['resource_changes'] = [
+            {
+                'address': 'data_something',
+                'change': {
+                    'actions': ['update'],
+                    'before': {
+                        'key': 'foo'
+                    },
+                    'after': {
+                        'key': 'bar'
+                    }
+                }
+            }
+        ]
+        obj._parse_resources()
+        self.assertEqual(obj.data['data_something'], {'address': 'data_something', 'values': {'key': 'bar'}})
+
+    @patch.object(TerraformParser, '_read_file', return_value={})
+    def test_parse_resources_resources_exists_in_the_resource_changes_resource(self, *args):
+        obj = TerraformParser('somefile', parse_it=False)
+        obj.raw['resource_changes'] = [
+            {
+                'address': 'something',
+                'change': {
+                    'actions': ['update'],
+                    'before': {
+                        'key': 'foo'
+                    },
+                    'after': {
+                        'key': 'bar'
+                    }
+                }
+            }
+        ]
+        obj._parse_resources()
+        self.assertEqual(obj.resources['something'], {'address': 'something', 'values': {'key': 'bar'}})
 
     @patch.object(TerraformParser, '_read_file', return_value={})
     def test_parse_configurations_resources(self, *args):