def test_prepareResults_ok_multiple(self):
raw = {
"data": [{
"attributes": {
"last_analysis_stats": {
"harmless": 1,
"malicious": 0,
"suspicious": 0,
"undetected": 0,
"timeout": 0
}
}
}, {
"attributes": {
"last_analysis_stats": {
"harmless": 7,
"malicious": 0,
"suspicious": 0,
"undetected": 11,
"timeout": 0
}
}
}]
}
results = virustotal.prepareResults(raw)
self.assertEqual(results["response"], raw)
self.assertEqual(results["summary"], "harmless")
self.assertEqual(results["status"], "ok")
def test_prepareResults_error(self):
raw = {}
results = virustotal.prepareResults(raw)
self.assertEqual(results["response"], raw)
self.assertEqual(results["summary"], "internal_failure")
self.assertEqual(results["status"], "caution")