Fixing tests.

SCH227 · Jan 9, 2021 · 1940339 · 1940339
1 parent cc20b41
commit 1940339
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/safety/safety.py b/safety/safety.py
@@ -170,7 +170,7 @@ def check(packages, key, db_mirror, cached, ignore_ids, proxy):
                                     advisory=data.get("advisory"),
                                     vuln_id=vuln_id,
                                     cvssv2=cve_meta.get("cvssv2", None),
-                                    cvssv3=cve_meta.get("cvssv3", None)
+                                    cvssv3=cve_meta.get("cvssv3", None),
                                 )
                             )
     return vulnerable

diff --git a/tests/test_safety.py b/tests/test_safety.py
@@ -121,6 +121,8 @@ def test_full_report(self):
                          + ' blah' * 15 + '.\r\n\r\n'
                          + 'All users are urged to upgrade please.\r\n',
                 vuln_id=1234,
+                cvssv2=None,
+                cvssv3=None,
             ),
         ]
         full_report = formatter.SheetReport.render(