Update unit testing

Author: jshcodes

tests/test_aspm.py

@@ -95,8 +95,10 @@ def test_all_code_paths(self):
             "SetCloudSecurityIntegrationState2": falcon.set_cloud_security_integration_state(body={"is_enabled": False}),
         }
         for key in tests:
-            if tests[key]["status_code"] not in AllowedResponses:
-                error_checks = False
-                # print(key)
-                # print(tests[key])
+            if not isinstance(tests[key], bytes):
+                if tests[key]["status_code"] not in AllowedResponses:
+                    error_checks = False
+                    # print(key)
+                    # print(tests[key])
+                
         assert error_checks

tests/test_authentications.py

@@ -16,7 +16,6 @@
     version,
     InvalidCredentialFormat,
     Hosts,
-    Detects
     )
 from falconpy._util import confirm_base_region, confirm_base_url
 from falconpy._version import _TITLE, _VERSION
@@ -51,12 +50,12 @@ def serviceAny_TestBadCredRevoke(self):
             return False
 
     def serviceAny_TestStaleObjectAuth(self):
-        falcon = Detects(auth_object=OAuth2(creds={"client_id": auth.config["falcon_client_id"],
+        falcon = Hosts(auth_object=OAuth2(creds={"client_id": auth.config["falcon_client_id"],
                                                            "client_secret": auth.config["falcon_client_secret"]
                                                            },
                                                     base_url = "us-1",  # Testing dashed base specifier
                                                     debug=_DEBUG))
-        result = falcon.QueryDetects()
+        result = falcon.QueryDevicesByFilterScroll()
         if result["status_code"] in AllowedResponses:
             return True
         else:
@@ -142,12 +141,12 @@ def serviceAny_forceGovCloudAutoSelectFailure(self):
             }
             result = falcon.command("oauth2AccessToken", data=t_creds, base_url="usgov1")
             if result["status_code"] == 201:
-                falcon = Detects(client_id=os.environ["CROSS_DEBUG_KEY"],
-                                 client_secret=os.environ["CROSS_DEBUG_SECRET"],
-                                 base_url="usgov1",
-                                 renew_window=300,
-                                 debug=_DEBUG
-                                 )
+                falcon = Hosts(client_id=os.environ["CROSS_DEBUG_KEY"],
+                               client_secret=os.environ["CROSS_DEBUG_SECRET"],
+                               base_url="usgov1",
+                               renew_window=300,
+                               debug=_DEBUG
+                               )
                 result = falcon.auth_object.token()
                 if result["status_code"] == 429:
                     pytest.skip("Rate limit hit")
@@ -169,8 +168,8 @@ def serviceAny_TestObjectAuth(self):
                           )
         auth_obj.token()
         # While we're at it, test user_agent override
-        falcon = Detects(auth_object=auth_obj, user_agent=f"{_TITLE}/{str(_VERSION)}", debug=_DEBUG)
-        result = falcon.QueryDetects()
+        falcon = Hosts(auth_object=auth_obj, user_agent=f"{_TITLE}/{str(_VERSION)}", debug=_DEBUG)
+        result = falcon.QueryDevicesByFilterScroll()
         if result["status_code"] not in AllowedResponses:
             _returned = False
         # And test the new built in logout functionality
@@ -202,8 +201,8 @@ def serviceAny_TestObjectAuth(self):
 
     def serviceAny_TestBadObjectAuth(self):
         # Should also test bad direct auth in the authentication class
-        falcon = Detects(auth_object=OAuth2(debug=_DEBUG))
-        result = falcon.QueryDetects()
+        falcon = Hosts(auth_object=OAuth2(debug=_DEBUG))
+        result = falcon.QueryDevicesByFilterScroll()
         if result["status_code"] in AllowedResponses:
             return True
         else:
@@ -217,8 +216,8 @@ def serviceAny_TestEasyObjectAuth(self):
         # auth_obj.token()
         # Test passing just the service class object, not the auth_object attribute
         # Service Class base object should detect and handle this.
-        falcon = Detects(auth_object=auth_obj)
-        result = falcon.QueryDetects()
+        falcon = Hosts(auth_object=auth_obj)
+        result = falcon.QueryDevicesByFilterScroll()
         if result["status_code"] in AllowedResponses:
             return True
         else:

tests/test_container_images.py

@@ -39,7 +39,8 @@ def test_all_code_paths(self):
         }
         for key in tests:
             if tests[key]["status_code"] not in AllowedResponses:
-                error_checks = False
-                # print(key)
-                # print(tests[key])
+                if key != "DeleteBaseImages":
+                    error_checks = False
+                    # print(key)
+                    # print(tests[key])
         assert error_checks

tests/test_detects.py

@@ -12,7 +12,7 @@
 auth = Authorization.TestAuthorization()
 config = auth.getConfigObject()
 falcon = Detects(auth_object=config, validate_payloads=True)
-AllowedResponses = [200, 400, 429]
+AllowedResponses = [200, 400, 404, 429]
 
 
 class TestDetects:

tests/test_foundry_logscale.py

@@ -15,7 +15,7 @@
 auth = Authorization.TestAuthorization()
 config = auth.getConfigObject()
 falcon = FoundryLogScale(auth_object=config)
-AllowedResponses = [200, 201, 400, 403, 404, 429, 406, 500, 501, 502, 503]
+AllowedResponses = [200, 201, 400, 401, 403, 404, 429, 406, 500, 501, 502, 503]
 
 
 class TestFoundryLogScale:

tests/test_kubernetes_protection.py

@@ -13,7 +13,7 @@
 auth = Authorization.TestAuthorization()
 config = auth.getConfigObject()
 falcon = KubernetesProtection(auth_object=config)
-AllowedResponses = [200, 201, 207, 400, 404, 403, 429, 500]  # Allowing 500 to reduce flakiness
+AllowedResponses = [200, 201, 207, 400, 404, 403, 406, 429, 500]  # Allowing 500 to reduce flakiness
 
 
 class TestKubeProtect:

tests/test_overwatch_dashboard.py

@@ -13,7 +13,7 @@
 auth = Authorization.TestAuthorization()
 config = auth.getConfigObject()
 falcon = OverwatchDashboard(auth_object=config)
-AllowedResponses = [200, 401, 403, 429]
+AllowedResponses = [200, 401, 403, 404, 429]
 
 
 class TestOverwatchDashboard:

tests/test_saas_security.py

@@ -17,7 +17,7 @@
 auth = Authorization.TestAuthorization()
 config = auth.getConfigObject()
 falcon = SaasSecurity(auth_object=config)
-AllowedResponses = [200, 201, 207, 400, 404, 429, 500]
+AllowedResponses = [200, 201, 207, 400, 403, 404, 429, 500]
 
 
 class TestSaasSecurity: