def test_submit_file(self, s, r):
c = triage.Client("token")
c.submit_sample_file("test", io.StringIO("file"))
r.assert_called_with("POST",
"/v0/samples",
b=BytesArg(),
headers=MultiPartArg())
def test_zip(self, s, r):
c = triage.Client("token")
c.sample_archive_zip("test1")
r.assert_called_with(
"GET",
"/v0/samples/test1/archive.zip",
)
def test_file(self, s, r):
c = triage.Client("token")
c.sample_task_file("sample1", "task1", "file1")
r.assert_called_with(
"GET",
"/v0/samples/sample1/task1/file1",
)
def test_profile_automatic(self, s, r):
c = triage.Client("token")
c.set_sample_profile_automatically("sample1")
r.assert_called_with("POST",
"/v0/samples/sample1/profile", {
'auto': True,
'pick': []
},
headers={'Content-Type': 'application/json'})
def test_public_samples(self, s, r):
c = triage.Client("token")
m = Mock()
m.json = Mock(return_value={"data": [{"name": "test"}], "next": None})
s.return_value = m
for i in c.public_samples():
assert i["name"] == "test"
m.json.assert_called_once()
r.assert_called_with("GET", "/v0/samples?subset=public&limit=20", None)
def test_no_data(self, s, r):
c = triage.Client("token")
m = Mock()
m.json = Mock(return_value={"data": None, "next": None})
s.return_value = m
p = c.profiles()
with pytest.raises(StopIteration):
next(p)
r.assert_called_with("GET", "/v0/profiles?limit=20", None)
def test_submit_url(self, s, r):
c = triage.Client("token")
c.submit_sample_url("http://9gag.com")
r.assert_called_with("POST",
"/v0/samples", {
'kind': 'url',
'url': 'http://9gag.com',
'interactive': False,
'profiles': []
},
headers={'Content-Type': 'application/json'})
def test_create_profile(self, s, r):
c = triage.Client("token")
c.create_profile("meme", ["yes1", "yes2"], "drop", 30)
r.assert_called_with("POST",
"/v0/profiles", {
"name": "meme",
"tags": ["yes1", "yes2"],
"network": "drop",
"timeout": 30
},
headers={'Content-Type': 'application/json'})
def test_list_profiles(self, s, r):
c = triage.Client("token")
m = Mock()
m.json = Mock(return_value={"data": [{"name": "test"}], "next": None})
s.return_value = m
for x in c.profiles():
assert x["name"] == "test"
m.json.assert_called_once()
r.assert_called_with("GET", "/v0/profiles?limit=20", None)
def test_next(self, s, r):
c = triage.Client("token")
m = Mock()
m.json = Mock(return_value={"data": [{"name": "test"}], "next": 1})
s.return_value = m
p = c.profiles()
assert next(p)["name"] == "test"
with pytest.raises(StopIteration):
next(p)
r.assert_called_with("GET", "/v0/profiles?limit=20&offset=1", None)
def test_search_pagination(self, s, r):
c = triage.Client("token")
m = Mock()
m.json = Mock(return_value={"data": [{"name": "test"}], "next": None})
s.return_value = m
for i in c.search("NOT family:emotet", 1000):
assert i["name"] == "test"
m.json.assert_called_once()
r.assert_called_with(
"GET", "/v0/search?query=NOT%20family%3Aemotet&limit=200", None)
# Copyright (C) 2021 Hatching B.V
# All rights reserved.
import io
import sys
import triage
url = "https://api.tria.ge"
token = "<YOUR-APIKEY-HERE>"
c = triage.Client(token, root_url=url)
f = open(sys.argv[1], "rb")
r = c.submit_sample_file("test.zip", f, password="******")
print(r)
def test_delete_profile(self, s, r):
c = triage.Client("token")
c.delete_profile("meme")
r.assert_called_with("DELETE", "/v0/profiles/meme", None)
def test_sample(self, s, r):
c = triage.Client("token")
c.sample_by_id("sample1")
r.assert_called_with("GET", "/v0/samples/sample1", None)
def test_task(self, s, r):
c = triage.Client("token")
c.task_report("sample1", "task1")
r.assert_called_with("GET",
"/v0/samples/sample1/task1/report_triage.json",
None)
def test_overview(self, s, r):
c = triage.Client("token")
c.overview_report("sample1")
r.assert_called_with("GET", "/v1/samples/sample1/overview.json", None)
def test_static(self, s, r):
c = triage.Client("token")
c.static_report("sample1")
r.assert_called_with("GET", "/v0/samples/sample1/reports/static", None)
def test_delete(self, s, r):
c = triage.Client("token")
c.delete_sample("sample1")
r.assert_called_with("DELETE", "/v0/samples/sample1", None)