def test_delete_project_handles_503(requests_mock): requests_mock.delete('http://testing-es-url/api/v1/projects/pid', text='', status_code=503) with pytest.raises(ServiceError): rest_client.project_delete('http://testing-es-url', 'pid', 'mykey') assert requests_mock.last_request.headers['Authorization'] == 'mykey'
def _delete_created_projects(cls): for project in cls._created_projects: try: rest_client.project_delete(cls.url, project['project_id'], project['result_token']) except ServiceError: # probably already deleted in the test pass
def test_project_run(self): p = self._create_project() p_id = p['project_id'] upload_response_1 = rest_client.project_upload_clks( self.url, p_id, p['update_tokens'][0], self.clk_data_1) upload_response_2 = rest_client.project_upload_clks( self.url, p_id, p['update_tokens'][1], self.clk_data_2) run_response = rest_client.run_create(self.url, p_id, p['result_token'], 0.999, name='clkhash rest client test') assert 'run_id' in run_response r_id = run_response['run_id'] with pytest.raises(ServiceError): rest_client.run_get_status(self.url, p_id, 'invalid-run-id', p['result_token']) with pytest.raises(ServiceError): rest_client.run_get_status(self.url, p_id, r_id, 'invalid-token') status1 = rest_client.run_get_status(self.url, p_id, r_id, p['result_token']) assert 'state' in status1 assert 'stages' in status1 print(rest_client.format_run_status(status1)) # Check we can watch the run progress this will raise if not # completed in 10 seconds for status_update in rest_client.watch_run_status( self.url, p_id, r_id, p['result_token'], 10, 0.5): print(rest_client.format_run_status(status_update)) # Check that we can still "wait" on a completed run and get a valid # status status2 = rest_client.wait_for_run(self.url, p_id, r_id, p['result_token'], 10) assert status2['state'] == 'completed' coord_result_raw = rest_client.run_get_result_text( self.url, p_id, r_id, p['result_token']) coord_result = json.loads(coord_result_raw) assert 'mask' in coord_result assert len(coord_result['mask']) == 1000 assert coord_result['mask'].count(1) > 10 result_a = json.loads( rest_client.run_get_result_text( self.url, p_id, r_id, upload_response_1['receipt_token'])) result_b = json.loads( rest_client.run_get_result_text( self.url, p_id, r_id, upload_response_2['receipt_token'])) assert 'permutation' in result_a assert 'rows' in result_a assert 1000 == result_b['rows'] == result_a['rows'] rest_client.run_delete(self.url, p_id, r_id, p['result_token']) rest_client.project_delete(self.url, p_id, p['result_token'])
def delete_resources(config, credentials, run): try: if run is not None and 'run_id' in run: rest_client.run_delete(config['server'], credentials['project_id'], run['run_id'], credentials['result_token']) rest_client.project_delete(config['server'], credentials['project_id'], credentials['result_token']) except Exception as e: logger.warning('Error while deleting resources... {}'.format(e))
def delete_created_projects(self): for project in self._created_projects: try: rest_client.project_delete(self.url, project['project_id'], project['result_token']) except KeyError: pass except ServiceError: # probably already deleted pass
def delete_project(server, project, apikey, verbose): """Delete a project on an entity matching server. """ if verbose: log("Entity Matching Server: {}".format(server)) try: project_delete(server, project, apikey) except ServiceError as e: log("Unexpected response with status {}".format(e.status_code)) log(e.text) else: log("Project deleted")