def test_live_hunt_list_json(self):
with patch('polyswarm_api.api.PolyswarmAPI.live_list',
return_value=[mock_polyswarm_api_results.hunts(self)[0]]):
result = self._run_cli(['--output-format', 'json', 'live', 'list'])
self._assert_json_result(
result,
expected_output=mock_polyswarm_api_results.hunts(self)[0].json,
expected_return_code=0,
)
def test_live_hunt_create_json(self):
with patch('polyswarm_api.api.PolyswarmAPI.live_create',
return_value=mock_polyswarm_api_results.hunts(self)[0]):
result = self._run_cli([
'--output-format', 'json', 'live', 'create',
self._get_test_resource_file_path('eicar.yara')
])
self._assert_json_result(
result,
expected_output=mock_polyswarm_api_results.hunts(self)[0].json,
expected_return_code=0,
)
def test_historical_hunt_delete_json(self):
with patch('polyswarm_api.api.PolyswarmAPI.historical_delete',
return_value=mock_polyswarm_api_results.hunts(self)[0]):
result = self._run_cli([
'--output-format', 'json', 'historical', 'delete',
'61210404295535902'
])
self._assert_json_result(
result,
expected_output=mock_polyswarm_api_results.hunts(self)[0].json,
expected_return_code=0,
)
def test_historical_hunt_list_text(self):
with patch('polyswarm_api.api.PolyswarmAPI.historical_list',
return_value=[mock_polyswarm_api_results.hunts(self)[0]]):
result = self._run_cli(
['--output-format', 'text', 'historical', 'list'])
self._assert_text_result(
result,
expected_output=mock_polyswarm_api_results.text_hunts()[0],
expected_return_code=0,
)
def test_live_hunt_list(self):
url = 'https://api.polyswarm.network/v2/hunt/live/list'
responses.add(responses.GET, url, body=self.mock_hunt_response_page1)
result = self._run_cli(['--output-format', 'json', 'live', 'list'])
self._assert_json_result(
result,
expected_output=mock_polyswarm_api_results.hunts(self)[0].json,
expected_return_code=0,
)
def test_historical_hunt_create_text(self):
with patch('polyswarm_api.api.PolyswarmAPI.historical_create',
return_value=mock_polyswarm_api_results.hunts(self)[0]):
result = self._run_cli([
'--output-format', 'text', 'historical', 'start',
self._get_test_resource_file_path('eicar.yara')
])
self._assert_text_result(
result,
expected_output=mock_polyswarm_api_results.text_hunts()[0],
expected_return_code=0,
)
def test_live_hunt_delete_text(self):
with patch('polyswarm_api.api.PolyswarmAPI.live_delete',
return_value=mock_polyswarm_api_results.hunts(self)[0]):
result = self._run_cli([
'--output-format', 'text', 'live', 'delete',
'61210404295535902'
])
self._assert_text_result(
result,
expected_output=mock_polyswarm_api_results.text_detele_hunts()[0],
expected_return_code=0,
)
def test_live_hunt_delete(self):
url = 'https://api.polyswarm.network/v2/hunt/live?id=63433636835291189'
responses.add(responses.DELETE, url, body=self.mock_hunt_response)
result = self._run_cli(
['--output-format', 'json', 'live', 'delete', self.test_hunt_id])
self._assert_json_result(
result,
expected_output=mock_polyswarm_api_results.hunts(self)[0].json,
expected_return_code=0,
)
def test_historical_hunt_start(self):
url = 'https://api.polyswarm.network/v2/hunt/historical'
responses.add(responses.POST, url, body=self.mock_hunt_response)
yara_file = self._get_test_resource_file_path('eicar.yara')
result = self._run_cli(
['--output-format', 'json', 'historical', 'start', yara_file])
self._assert_json_result(
result,
expected_output=mock_polyswarm_api_results.hunts(self)[0].json,
expected_return_code=0,
)
def __init__(self, *args, **kwargs):
super(IntegrationTest, self).__init__(*args, **kwargs)
self.mock_search_response_page1 = self._create_response(
[mock_polyswarm_api_results.instances(self)[0].json],
offset=1,
limit=1,
has_more=False)
self.mock_metadata_search_response = self._create_response(
[mock_polyswarm_api_results.metadata(self)[0].json],
offset=1,
limit=1,
has_more=False)
self.mock_submission_response = \
self._create_response(mock_polyswarm_api_results.instances(self)[0].json)
self.mock_hunt_live_results_response_page1 = \
self._create_response([mock_polyswarm_api_results.live_results(self)[0].json], offset=1, limit=1, has_more=False)
self.mock_hunt_historical_results_response_page1 = \
self._create_response([mock_polyswarm_api_results.historical_results(self)[0].json], offset=1, limit=1, has_more=False)
self.mock_hunt_response = self._create_response(
mock_polyswarm_api_results.hunts(self)[0].json)
self.mock_hunt_response_page1 = self._create_response(
[mock_polyswarm_api_results.hunts(self)[0].json],
offset=1,
limit=1,
has_more=False)
self.mock_stream_response_page1 = self._create_response([
mock_polyswarm_api_results.stream_results(self.test_s3_file_url)[0]
],
offset=1,
limit=1,
has_more=False)
