add: cve_advanced module test + functions to test attributes and objects results

pull/347/head
chrisr3d 2019-10-27 21:19:43 +01:00
parent 93858e302a
commit f15ab8162f
No known key found for this signature in database
GPG Key ID: 6BBED1B63A6D639F
1 changed files with 28 additions and 0 deletions

View File

@ -21,6 +21,14 @@ class TestExpansions(unittest.TestCase):
def misp_modules_post(self, query):
return requests.post(urljoin(self.url, "query"), json=query)
@staticmethod
def get_attribute(reponse):
data = response.json()
if not isinstance(data, dict):
print(json.dumps(data, indent=2))
return data
return data['results']['Attribute'][0]['type']
@staticmethod
def get_data(response):
data = response.json()
@ -37,6 +45,14 @@ class TestExpansions(unittest.TestCase):
return data
return data['error']
@staticmethod
def get_object(response):
data = response.json()
if not isinstance(data, dict):
print(json.dumps(data, indent=2))
return data
return data['results']['Object'][0]['name']
@staticmethod
def get_values(response):
data = response.json()
@ -75,6 +91,18 @@ class TestExpansions(unittest.TestCase):
response = self.misp_modules_post(query)
self.assertTrue(self.get_values(response).startswith("Stack-based buffer overflow in Microsoft Office XP SP3, Office 2003 SP3"))
def test_cve_advanced(self):
query = {"module": "cve_advanced",
"attribute": {"type": "vulnerability",
"value": "CVE-2010-3333",
"uuid": "ea89a33b-4ab7-4515-9f02-922a0bee333d"},
"config": {}}
response = self.misp_modules_post(query)
try:
self.assertEqual(self.get_object(response), 'vulnerability')
except Exception:
print(self.get_errors(response))
def test_dbl_spamhaus(self):
query = {"module": "dbl_spamhaus", "domain": "totalmateria.net"}
response = self.misp_modules_post(query)