def serviceCCAWS_GetAWSSettings(self): auth = Authorization.TestAuthorization() auth.serviceAuth() falcon = FalconAWS.Cloud_Connect_AWS(access_token=auth.token) if falcon.GetAWSSettings()["status_code"] > 0: auth.serviceRevoke() return True else: auth.serviceRevoke() return False
# test_spotlight_vulnerabilities.py # This class tests the spotlight_vulnerabilities service class import os import sys import pytest # Authentication via the test_authorization.py from tests import test_authorization as Authorization # Classes to test - manually imported from sibling folder from falconpy import SpotlightVulnerabilities # Import our sibling src folder into the path sys.path.append(os.path.abspath('src')) auth = Authorization.TestAuthorization() config = auth.getConfigObject() falcon = SpotlightVulnerabilities(auth_object=config) AllowedResponses = [200, 400, 403, 429] class TestSpotlight: def spotlight_queryVulnerabilities(self): if falcon.queryVulnerabilities( parameters={ "limit": 1, "filter": "created_timestamp:>'2021-01-01T00:00:01Z'" })["status_code"] in AllowedResponses: return True else: return False def spotlight_query_vulnerabilities_combined(self):