fix formatting

felixhammerl · Oct 16, 2023 · 435e243 · 435e243
1 parent 891e8a9
commit 435e243
Show file tree

Hide file tree

Showing 3 changed files with 29 additions and 14 deletions.
diff --git a/steps/gather/src/handler/gather.py b/steps/gather/src/handler/gather.py
@@ -34,14 +34,18 @@ def gather_results(event, context):
         )
 
     master_report = deepcopy(account_reports[0]["report"])
-    master_report = list(unique_everseen([
-        {
-            k: v
-            for k, v in elem.items()
-            if k in ["plugin", "category", "title", "description"]
-        }
-        for elem in master_report
-    ]))
+    master_report = list(
+        unique_everseen(
+            [
+                {
+                    k: v
+                    for k, v in elem.items()
+                    if k in ["plugin", "category", "title", "description"]
+                }
+                for elem in master_report
+            ]
+        )
+    )
     for plugin in master_report:
         plugin["reports"] = {
             report["account"]: [
@@ -56,8 +60,14 @@ def gather_results(event, context):
             for report in account_reports
         }
 
-    master_report_by_category = {k:list(v) for k,v in groupby(master_report, key=itemgetter("category"))}
+    master_report_by_category = {
+        k: list(v) for k, v in groupby(master_report, key=itemgetter("category"))
+    }
 
-    result = s3.put_object(Bucket=bucket, Key=f"{date}/report.json", Body=json.dumps(master_report_by_category).encode())
+    result = s3.put_object(
+        Bucket=bucket,
+        Key=f"{date}/report.json",
+        Body=json.dumps(master_report_by_category).encode(),
+    )
 
     return master_report_by_category
diff --git a/steps/gather/test/fixture/__init__.py b/steps/gather/test/fixture/__init__.py
@@ -1,5 +1,6 @@
 from pathlib import Path
 
+
 def read_test_scan_reports():
     report_files = ["123123123123.json", "456456456456.json", "789789789789.json"]
 

diff --git a/steps/gather/test/unit/handler/test_gather.py b/steps/gather/test/unit/handler/test_gather.py
@@ -44,12 +44,16 @@ def test_should_create_master_report():
         assert_that(category, is_(instance_of(str)))
         for plugin in plugins:
             for key in plugin.keys():
-                assert_that(key, is_in(["plugin", "category", "title", "description", "reports"]))
+                assert_that(
+                    key,
+                    is_in(["plugin", "category", "title", "description", "reports"]),
+                )
 
             for account, findings in plugin["reports"].items():
                 assert_that(account, is_(instance_of(str)))
                 for finding in findings:
                     for finding_property in finding.keys():
-                        assert_that(finding_property, is_in(["resource", "region", "status", "message"]))
-
-
+                        assert_that(
+                            finding_property,
+                            is_in(["resource", "region", "status", "message"]),
+                        )