p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Now run monkey works

This commit is contained in:
Shay Nehmad 2019-08-29 18:48:35 +03:00
parent ebd2628516
commit 5afded480e
2 changed files with 31 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
envs/monkey_zoo/blackbox/monkey_island_client.py
View File

@ -1,3 +1,5 @@
import json
import requests
# SHA3-512 of '1234567890!@#$%^&*()_nothing_up_my_sleeve_1234567890!@#$%^&*()'
@ -15,13 +17,35 @@ class MonkeyIslandClient(object):
verify=False)
return resp.json()["access_token"]
def request_get(self, url):
return requests.get(
self.addr + url,
headers={"Authorization": "JWT " + self.token},
verify=False
)
def request_post(self, url, data):
return requests.post(
self.addr + url,
data=data,
headers={"Authorization": "JWT " + self.token},
verify=False
)
def request_json(self, url, dict_data):
return requests.post(
self.addr + url,
json=dict_data,
headers={"Authorization": "JWT " + self.token},
verify=False
)
def get_api_status(self):
return requests.get(self.addr + "api", headers={"Authorization": "JWT " + self.token}, verify=False)
return self.request_get("api")
def import_config(self, config_contents):
resp = requests.post(
self.addr + "api/configuration/island",
headers={"Authorization": "JWT " + self.token},
data=config_contents,
verify=False)
_ = self.request_post("api/configuration/island", data=config_contents)
def run_monkey_local(self):
resp = self.request_json("api/local-monkey", dict_data={"action": "run"})
print(resp.text)

2
envs/monkey_zoo/blackbox/test_blackbox.py
View File

@ -9,7 +9,7 @@ from envs.monkey_zoo.blackbox.monkey_island_client import MonkeyIslandClient
def generic_blackbox_test_case(client, config_file_path, analyzers):
with open(config_file_path, "r") as config_file:
client.import_config(config_file.read())
# run_local_monkey_on_island()
client.run_monkey_local()
for analyzer in analyzers:
assert analyzer.analyze_test_results()