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)