def test_runFromStepFunction(self):
target = "infosec.mozilla.org"
partial_stepf_event = {"target": target}
test_context = None
try:
a_key = os.environ["TIOA"]
s_key = os.environ["TIOS"]
except Exception:
assert False
tenableio_scan_handler = TIOScanHandler()
response = tenableio_scan_handler.runFromStepFunction(
partial_stepf_event, test_context)
assert type(response) is dict
# The next few lines are for cleaning up,
# we should abort the scan just launched
scan_id = response['id']
scanner = TIOScanner(access_key=a_key, secret_key=s_key)
client = scanner._TIOScanner__createClient()
scan_ref = client.scan_helper.id(scan_id)
scan_ref.stop(wait=False)
test_invalid_event = {"TEST": "TEST"}
invalid_response = tenableio_scan_handler.runFromStepFunction(
test_invalid_event, test_context)
assert invalid_response is False
def run_tenableioscan(event, context):
tenableio_scan_handler = TIOScanHandler(sqs_client=SQS_CLIENT, logger=logger)
response = tenableio_scan_handler.runFromStepFunction(event, context)
return response