PySafetyBear: Remove use of generic no cover

Add `test_cve_key_checker` Closes coala#2773
Leciurr0 · Dec 1, 2018 · 0f1cd3b · 0f1cd3b
1 parent cacfe6f
commit 0f1cd3b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/bears/python/requirements/PySafetyBear.py b/bears/python/requirements/PySafetyBear.py
@@ -14,7 +14,7 @@
 
 
 # It was for old versions of safety and those versions will be allow in future.
-def cve_key_checker(vulnerability):  # pragma: no cover
+def cve_key_checker(vulnerability):
     if 'cve' in vulnerability.data:
         if vulnerability.data['cve'] is None:
             return None

diff --git a/tests/python/requirements/PySafetyBearTest.py b/tests/python/requirements/PySafetyBearTest.py
@@ -17,6 +17,12 @@ class PySafetyBearTest(LocalBearTestHelper):
     def setUp(self):
         self.uut = PySafetyBear(Section('name'), Queue())
 
+    def test_cve_key_checker(self):
+        # avoid pragma: no cover
+        assert cve_key_checker(mock.Mock(data={'cve': None})) is None
+        assert cve_key_checker(mock.Mock(data={'cve': True}))
+        assert cve_key_checker(mock.Mock(data={})) is None
+
     def test_without_vulnerability(self):
         with mock.patch(
             'bears.python.requirements.PySafetyBear.safety.check',