def _post_json_data(url, data): th_collection = data[jm.project] OAuthCredentials.set_credentials( SampleData.get_credentials() ) credentials = OAuthCredentials.get_credentials(jm.project) tr = TreeherderRequest( protocol='http', host='localhost', project=jm.project, oauth_key=credentials['consumer_key'], oauth_secret=credentials['consumer_secret'] ) signed_uri = tr.oauth_client.get_signed_uri( th_collection.to_json(), tr.get_uri(th_collection.endpoint_base), "POST" ) response = TestApp(application).post_json( str(signed_uri), params=th_collection.get_collection_data() ) response.getcode = lambda: response.status_int return response
def post_job_data(project, uri, data, status=None, expect_errors=False): # Since the uri is passed in it's not generated by the # treeherder request or collection and is missing the protocol # and host. Add those missing elements here. uri = 'http://localhost{0}'.format(uri) # Set the credentials OAuthCredentials.set_credentials(SampleData.get_credentials()) credentials = OAuthCredentials.get_credentials(project) tr = TreeherderRequest(protocol='http', host='localhost', project=project, oauth_key=credentials['consumer_key'], oauth_secret=credentials['consumer_secret']) signed_uri = tr.get_signed_uri(json.dumps(data), uri) response = TestApp(application).post_json(str(signed_uri), params=data, status=status, expect_errors=expect_errors) return response
def post_collection( project, th_collection, status=None, expect_errors=False, consumer_key=None, consumer_secret=None): # Set the credentials OAuthCredentials.set_credentials(SampleData.get_credentials()) credentials = OAuthCredentials.get_credentials(project) # The only time the credentials should be overridden are when # a client needs to test authentication failure confirmation consumer_key = consumer_key or credentials['consumer_key'] consumer_secret = consumer_secret or credentials['consumer_secret'] auth = TreeherderAuth(consumer_key, consumer_secret, project) client = TreeherderClient(protocol='http', host='localhost', auth=auth) uri = client._get_project_uri(project, th_collection.endpoint_base) req = Request('POST', uri, json=th_collection.get_collection_data(), auth=auth) prepped_request = req.prepare() response = TestApp(application).post_json( prepped_request.url, params=th_collection.get_collection_data(), status=status ) return response
def post_collection( project, th_collection, status=None, expect_errors=False, consumer_key=None, consumer_secret=None): # Set the credentials OAuthCredentials.set_credentials(SampleData.get_credentials()) credentials = OAuthCredentials.get_credentials(project) # The only time the credentials should be overridden are when # a client needs to test authentication failure confirmation if consumer_key: credentials['consumer_key'] = consumer_key if consumer_secret: credentials['consumer_secret'] = consumer_secret cli = TreeherderClient( protocol='http', host='localhost', ) jsondata = th_collection.to_json() signed_uri = cli._get_uri(project, th_collection.endpoint_base, data=jsondata, oauth_key=credentials['consumer_key'], oauth_secret=credentials['consumer_secret'], method='POST') response = TestApp(application).post_json( str(signed_uri), params=th_collection.get_collection_data(), status=status ) return response
def post_job_data( project, uri, data, status=None, expect_errors=False): # Since the uri is passed in it's not generated by the # treeherder request or collection and is missing the protocol # and host. Add those missing elements here. uri = 'http://localhost{0}'.format(uri) # Set the credentials OAuthCredentials.set_credentials(SampleData.get_credentials()) credentials = OAuthCredentials.get_credentials(project) tr = TreeherderRequest( protocol='http', host='localhost', project=project, oauth_key=credentials['consumer_key'], oauth_secret=credentials['consumer_secret'] ) signed_uri = tr.get_signed_uri( json.dumps(data), uri ) response = TestApp(application).post_json( str(signed_uri), params=data, status=status, expect_errors=expect_errors ) return response
def post_collection(project, th_collection, status=None, expect_errors=False, consumer_key=None, consumer_secret=None): # Set the credentials OAuthCredentials.set_credentials(SampleData.get_credentials()) credentials = OAuthCredentials.get_credentials(project) # The only time the credentials should be overridden are when # a client needs to test authentication failure confirmation if consumer_key: credentials["consumer_key"] = consumer_key if consumer_secret: credentials["consumer_secret"] = consumer_secret tr = TreeherderRequest( protocol="http", host="localhost", project=project, oauth_key=credentials["consumer_key"], oauth_secret=credentials["consumer_secret"], ) signed_uri = tr.oauth_client.get_signed_uri( th_collection.to_json(), tr.get_uri(th_collection.endpoint_base), "POST" ) response = TestApp(application).post_json( str(signed_uri), params=th_collection.get_collection_data(), status=status ) return response
def _send(th_request, th_collection): OAuthCredentials.set_credentials(SampleData.get_credentials()) credentials = OAuthCredentials.get_credentials(jm.project) th_request.oauth_key = credentials['consumer_key'] th_request.oauth_secret = credentials['consumer_secret'] signed_uri = th_request.get_signed_uri( th_collection.to_json(), th_request.get_uri(th_collection) ) response = TestApp(application).post_json( str(signed_uri), params=th_collection.get_collection_data() ) response.getcode = lambda: response.status_int return response
def _send(th_request, endpoint, method=None, data=None): OAuthCredentials.set_credentials(SampleData.get_credentials()) credentials = OAuthCredentials.get_credentials(jm.project) th_request.oauth_key = credentials['consumer_key'] th_request.oauth_secret = credentials['consumer_secret'] if data and not isinstance(data, str): data = json.dumps(data) signed_uri = th_request.oauth_client.get_signed_uri( data, th_request.get_uri(endpoint), method) response = getattr(TestApp(application), method.lower())(str(signed_uri), params=data, content_type='application/json') response.getcode = lambda: response.status_int return response
def post_collection(project, th_collection, status=None, expect_errors=False, consumer_key=None, consumer_secret=None): # Set the credentials OAuthCredentials.set_credentials(SampleData.get_credentials()) credentials = OAuthCredentials.get_credentials(project) # The only time the credentials should be overridden are when # a client needs to test authentication failure confirmation if consumer_key: credentials['consumer_key'] = consumer_key if consumer_secret: credentials['consumer_secret'] = consumer_secret cli = TreeherderClient( protocol='http', host='localhost', ) jsondata = th_collection.to_json() signed_uri = cli._get_project_uri( project, th_collection.endpoint_base, data=jsondata, oauth_key=credentials['consumer_key'], oauth_secret=credentials['consumer_secret'], method='POST') response = TestApp(application).post_json( str(signed_uri), params=th_collection.get_collection_data(), status=status) return response
def _send(th_request, endpoint, method=None, data=None): OAuthCredentials.set_credentials(SampleData.get_credentials()) credentials = OAuthCredentials.get_credentials(jm.project) th_request.oauth_key = credentials['consumer_key'] th_request.oauth_secret = credentials['consumer_secret'] if data and not isinstance(data, str): data = json.dumps(data) signed_uri = th_request.oauth_client.get_signed_uri( data, th_request.get_uri(endpoint), method ) response = getattr(TestApp(application), method.lower())( str(signed_uri), params=data, content_type='application/json' ) response.getcode = lambda: response.status_int return response
def set_oauth_credentials(): OAuthCredentials.set_credentials(SampleData.get_credentials())