misp-modules/tests/test.py

57 lines
1.8 KiB
Python
Raw Normal View History

2016-06-18 07:53:26 +02:00
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import unittest
import requests
import base64
import json
2016-08-17 14:01:11 +02:00
import os
2016-06-18 07:53:26 +02:00
class TestModules(unittest.TestCase):
def setUp(self):
self.maxDiff = None
self.headers = {'Content-Type': 'application/json'}
self.url = "http://127.0.0.1:6666/"
2016-06-18 07:53:26 +02:00
def test_introspection(self):
response = requests.get(self.url + "modules")
2016-06-18 07:53:26 +02:00
print(response.json())
def test_cve(self):
with open('tests/bodycve.json', 'r') as f:
response = requests.post(self.url + "query", data=f.read())
2016-06-18 07:53:26 +02:00
print(response.json())
def test_dns(self):
with open('tests/body.json', 'r') as f:
response = requests.post(self.url + "query", data=f.read())
print(response.json())
2016-08-25 17:36:28 +02:00
with open('tests/body_timeout.json', 'r') as f:
response = requests.post(self.url + "query", data=f.read())
print(response.json())
def test_stix(self):
with open("tests/stix.xml", "r") as f:
data = json.dumps({"module":"stiximport",
"data":str(base64.b64encode(bytes(f.read(), 'utf-8')), 'utf-8'),
"config": {"max_size": "15000"},
})
response = requests.post(self.url + "query", data=data)
2016-06-18 07:53:26 +02:00
print(response.json())
2016-08-17 14:01:11 +02:00
def test_virustotal(self):
# This can't actually be tested without disclosing a private
# API key. This will attempt to run with a .gitignored keyfile
# and pass if it can't find one
if not os.path.exists("tests/bodyvirustotal.json"):
return
with open("tests/bodyvirustotal.json", "r") as f:
response = requests.post(self.url + "query", data=f.read()).json()
assert(response)
if __name__ == '__main__':
unittest.main()