updating tests

requirements/common.txt

@@ -1 +1 @@
-requests==2.20.0
+requests==2.20.0

tests/test_advisory.py

@@ -1,4 +1,7 @@
 import unittest
+from unittest import mock
+import requests
+
 from openVulnQuery import advisory
 from openVulnQuery import constants
 
@@ -148,7 +151,7 @@ def test_advisory_advisory_factory_cvrf_with_key_cvrf_url_and_ips(self):
             adv_map[v] = k
         self.assertTrue(advisory.advisory_factory(
                         adv_map,
-                        constants.CVRF_ADVISORY_FORMAT_TOKEN,
+                        constants.DEFAULT_ADVISORY_FORMAT_TOKEN,
                         MockLogger()))
 
     def test_advisory_advisory_factory_oval_with_key_oval_url_and_ips(self):

tests/test_authorization.py

@@ -1,7 +1,7 @@
 import unittest
-
-import mock
 import requests
+from unittest import mock
+
 from openVulnQuery import authorization
 from openVulnQuery import config
 

tests/test_cli_api.py

@@ -2,6 +2,7 @@
 import unittest
 import os
 import inspect
+import requests
 
 from openVulnQuery import config
 from openVulnQuery import cli_api

tests/test_config.py

@@ -2,6 +2,8 @@
 
 import os
 import unittest
+import requests
+from unittest import mock
 
 from openVulnQuery import config
 

tests/test_constants.py

@@ -1,4 +1,6 @@
 import unittest
+import requests
+from unittest import mock
 
 from openVulnQuery import constants
 
@@ -41,13 +43,11 @@
 JSON_OUTPUT_FORMAT_TOKEN = 'json'
 CSV_OUTPUT_FORMAT_TOKEN = 'csv'
 
-CVRF_ADVISORY_FORMAT_TOKEN = 'cvrf'
-OVAL_ADVISORY_FORMAT_TOKEN = 'oval'
+DEFAULT_ADVISORY_FORMAT_TOKEN = 'default'
 IOS_ADVISORY_FORMAT_TOKEN = 'ios'
 
 ADVISORY_FORMAT_TOKENS = (
-    CVRF_ADVISORY_FORMAT_TOKEN,
-    OVAL_ADVISORY_FORMAT_TOKEN,
+    DEFAULT_ADVISORY_FORMAT_TOKEN,
     IOS_ADVISORY_FORMAT_TOKEN,
 )
 
@@ -73,11 +73,8 @@ def test_constants_unchanged_csv_format_token(self):
 
     def test_constants_unchanged_cvrf_advisory_format_token(self):
         self.assertEqual(
-            constants.CVRF_ADVISORY_FORMAT_TOKEN, CVRF_ADVISORY_FORMAT_TOKEN)
+            constants.DEFAULT_ADVISORY_FORMAT_TOKEN, DEFAULT_ADVISORY_FORMAT_TOKEN)
 
-    def test_constants_unchanged_oval_advisory_format_token(self):
-        self.assertEqual(
-            constants.OVAL_ADVISORY_FORMAT_TOKEN, OVAL_ADVISORY_FORMAT_TOKEN)
 
     def test_constants_unchanged_ios_advisory_format_token(self):
         self.assertEqual(

tests/test_main.py

@@ -1,4 +1,6 @@
 import unittest
+from unittest import mock
+import requests
 
 from openVulnQuery import advisory
 from openVulnQuery import config

tests/test_query_client.py

@@ -1,7 +1,9 @@
 import unittest
-import mock
 import requests
 import json
+
+from unittest import mock
+
 from openVulnQuery import query_client
 from openVulnQuery import constants
 from openVulnQuery import config

tests/test_query_client_cvrf.py

@@ -2,7 +2,8 @@
 import unittest
 import pytest
 
-import mock
+from unittest import mock
+
 from openVulnQuery import advisory
 from openVulnQuery import config
 from openVulnQuery import query_client

tests/test_utils.py

@@ -1,5 +1,4 @@
 import unittest
-import mock
 import sys
 import json
 import pytest