CSEC_PUBLIC/securityonion
1
0
Fork 0
mirror of https://github.com/Security-Onion-Solutions/securityonion.git synced 2025-12-06 17:22:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add function/test for non-existent VT api_key

Browse Source
This commit is contained in:
Wes Lambert
2022-04-21 17:33:24 +00:00
parent 9287d6adf7
commit b2db32a2c7
2 changed files with 15 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
salt/sensoroni/files/analyzers/virustotal/virustotal.py
View File

@@ -3,6 +3,14 @@ import requests
import argparse import argparse
import helpers import helpers
import os import os
import sys
def checkConfigRequirements(conf):
if "api_key" not in conf or len(conf['api_key']) == 0:
sys.exit(126)
else:
return True
def buildHeaders(conf): def buildHeaders(conf):
@@ -58,6 +66,7 @@ def prepareResults(raw):
def analyze(conf, input): def analyze(conf, input):
checkConfigRequirements(conf)
meta = helpers.loadMetadata(__file__) meta = helpers.loadMetadata(__file__)
data = helpers.parseArtifact(input) data = helpers.parseArtifact(input)
helpers.checkSupportedType(meta, data["artifactType"]) helpers.checkSupportedType(meta, data["artifactType"])

6
salt/sensoroni/files/analyzers/virustotal/virustotal_test.py
View File

@@ -25,6 +25,12 @@ class TestVirusTotalMethods(unittest.TestCase):
self.assertEqual(mock_stdout.getvalue(), expected) self.assertEqual(mock_stdout.getvalue(), expected)
mock.assert_called_once() mock.assert_called_once()
def checkConfigRequirements(self):
conf = {"not_a_key": "abcd12345"}
with self.assertRaises(SystemExit) as cm:
virustotal.checkConfigRequirements(conf)
self.assertEqual(cm.exception.code, 126)
def test_buildHeaders(self): def test_buildHeaders(self):
result = virustotal.buildHeaders({"api_key": "xyz"}) result = virustotal.buildHeaders({"api_key": "xyz"})
self.assertEqual("xyz", result["x-apikey"]) self.assertEqual("xyz", result["x-apikey"])