def run_with_fake_auth(): def get_project_ids(role="_member_", project_ids=None): from gdcdatamodel import models as md if project_ids is None: project_ids = [] if not project_ids: with current_app.db.session_scope(): project_ids += [ "{}-{}".format(p.programs[0].name, p.code) for p in current_app.db.nodes(md.Project).all() ] return project_ids with patch( "sheepdog.auth.FederatedUser.roles", new_callable=PropertyMock, return_value=roles, ), patch( "sheepdog.auth.FederatedUser.logged_in", new_callable=PropertyMock, return_value=lambda: True, ), patch( "sheepdog.auth.FederatedUser.get_project_ids", new_callable=PropertyMock, return_value=get_project_ids, ), patch("sheepdog.auth.verify_hmac", new=set_user): run_for_development(debug=debug, threaded=True)
def run_with_fake_auth(): def get_project_ids(role='_member_', project_ids=None): if project_ids is None: project_ids = [] if not project_ids: with current_app.db.session_scope(): project_ids += [ '{}-{}'.format(p.programs[0].name, p.code) for p in current_app.db.nodes(md.Project).all() ] return project_ids with patch('sheepdog.api.AuthDriver', spec=True) as _, patch( 'sheepdog.auth.FederatedUser.roles', new_callable=PropertyMock, return_value=roles, ), patch( 'sheepdog.auth.FederatedUser.logged_in', new_callable=PropertyMock, return_value=lambda: True, ), patch( 'sheepdog.auth.FederatedUser.get_project_ids', new_callable=PropertyMock, return_value=get_project_ids, ), patch( 'sheepdog.auth.verify_hmac', new=set_user, ): run_for_development(debug=debug, threaded=True)
def run_with_fake_download(): with patch("sheepdog.download.get_nodes", fake_get_nodes): with patch.multiple("sheepdog.download", key_for=fake_key_for, key_for_node=fake_key_for_node, urls_from_signpost=fake_urls_from_signpost): if os.environ.get("GDC_FAKE_AUTH"): run_with_fake_auth() else: run_for_development(debug=debug, threaded=True)
def run_with_fake_authz(): """ Mocks arborist calls. """ authorized = True # modify this to mock authorized/unauthorized with patch( 'gen3authz.client.arborist.client.ArboristClient.create_resource', new_callable=PropertyMock, ), patch( 'gen3authz.client.arborist.client.ArboristClient.auth_request', new_callable=PropertyMock, return_value=lambda jwt, service, methods, resources: authorized, ): run_for_development(debug=debug, threaded=True)
return_value=get_project_ids, ), patch( 'sheepdog.auth.verify_hmac', new=set_user, ): run_for_development(debug=debug, threaded=True) def run_with_fake_download(): with patch("sheepdog.download.get_nodes", fake_get_nodes): with patch.multiple("sheepdog.download", key_for=fake_key_for, key_for_node=fake_key_for_node, urls_from_signpost=fake_urls_from_signpost): if os.environ.get("GDC_FAKE_AUTH"): run_with_fake_auth() else: run_for_development(debug=debug, threaded=True) if __name__ == '__main__': debug = bool(os.environ.get('SHEEPDOG_DEBUG', True)) if os.environ.get("GDC_FAKE_DOWNLOAD") == 'True': run_with_fake_download() else: if os.environ.get("GDC_FAKE_AUTH") == 'True': run_with_fake_auth() else: run_for_development(debug=debug, threaded=True)