def test_multiple_element_return():
inp = {"_id": 1, "_type": "domain", "domain": "bad.local"}
lineage = [{"_id": 1, "_type": "domain", "domain": "bad.local"}]
plugin = plugin_runner("cat examples/tool_multiple_output.json")
res = list(plugin(lineage, inp))
expected_output = [[{
"_id": 1,
"_type": "domain",
"domain": "bad.local"
}, {
"_id": 2,
"_type": "ip",
"ip": "127.0.0.1",
"_cmd": "cat examples/tool_multiple_output.json"
}, {
"_id": 212281217781803245129106443529066261547,
"_type": "vulnerability",
"_cmd": "cat examples/tool_multiple_output.json",
"cve": "CVE-2014-9999999",
"cwe": "CWE-245",
"description": "we are doomed!"
}]]
print("res")
pprint(res)
print("exp")
pprint(expected_output)
assert res == expected_output
def test_call_plugin():
inp = {"_id": 1, "_type": "ip", "ip": "192.168.1.1"}
lineage = [{"_id": 1, "_type": "ip", "ip": "192.168.1.1"}]
plugin = plugin_runner("cat examples/tool_output.json")
res = list(plugin(lineage, inp))
expected_output = [[{
"_id": 1,
"_type": "ip",
"ip": "192.168.1.1"
}, {
"_id": "123",
"_type": "port",
"_cmd": "cat examples/tool_output.json",
"port": 80
}],
[{
"_id": 1,
"_type": "ip",
"ip": "192.168.1.1"
}, {
"_id": "321",
"_type": "port",
"_cmd": "cat examples/tool_output.json",
"port": 81
}]]
print("res")
pprint(res)
print("exp")
pprint(expected_output)
assert res == expected_output
def test_multiple_lineage():
inp = {"_id": 1, "_type": "domain", "domain": "bad.local"}
lineage = [{
"_id": 1,
"_type": "domain",
"domain": "bad.local"
}, {
"_id": 2,
"_type": "ip",
"ip": "127.0.0.1",
"_cmd": "cat examples/tool_multiple_output.json"
}]
plugin = plugin_runner("cat examples/tool_output.json")
res = list(plugin(lineage, inp))
expected_output = [[{
"_id": 1,
"_type": "domain",
"domain": "bad.local"
}, {
"_id": 2,
"_type": "ip",
"ip": "127.0.0.1",
"_cmd": "cat examples/tool_multiple_output.json"
}, {
"_id": "123",
"_type": "port",
"_cmd": "cat examples/tool_output.json",
"port": 80
}],
[{
"_id": 1,
"_type": "domain",
"domain": "bad.local"
}, {
"_id": 2,
"_type": "ip",
"ip": "127.0.0.1",
"_cmd": "cat examples/tool_multiple_output.json"
}, {
"_id": "321",
"_type": "port",
"_cmd": "cat examples/tool_output.json",
"port": 81
}]]
print("res")
pprint(res)
print("exp")
pprint(expected_output)
assert res == expected_output
def test_fail_plugin():
inp = {"_id": 1, "_type": "ip", "ip": "192.168.1.1"}
lineage = [{"_id": 1, "_type": "ip", "ip": "192.168.1.1"}]
plugin = plugin_runner("echo 'Error!' 1>&2; exit 64")
res = list(plugin(lineage, inp))
expected_output = [[{
"_id": 1,
"_type": "ip",
"ip": "192.168.1.1"
}, {
'_id': 212281217781803245129106443529066261547,
'_type': 'error',
'_cmd': "echo 'Error!' 1>&2; exit 64",
'error': 'Error!\n'
}]]
print("res")
pprint(res)
print("exp")
pprint(expected_output)
assert res == expected_output
def launch_task(self, task):
runner = plugin_runner(task["command"])
lineage = unfreeze(task["lineages"])
stdin = unfreeze(task["stdin"])
self.tasks.append(runner(lineage, stdin))