def test_export_rest(self): """ EXPORT (REST): Test the export of data.""" mw = [] headers1 = { 'X-Rucio-Account': 'root', 'X-Rucio-Username': '******', 'X-Rucio-Password': '******' } r1 = TestApp(auth_app.wsgifunc(*mw)).get('/userpass', headers=headers1, expect_errors=True) token = str(r1.header('X-Rucio-Auth-Token')) headers2 = { 'X-Rucio-Type': 'user', 'X-Rucio-Account': 'root', 'X-Rucio-Auth-Token': str(token) } r2 = TestApp(export_app.wsgifunc(*mw)).get('/', headers=headers2, expect_errors=True) rses = [export_rse(rse['rse']) for rse in list_rses()] assert_equal(r2.status, 200) assert_equal( r2.body, render_json(**{ 'rses': rses, 'distances': export_distances() }))
def test_export_client(self): """ EXPORT (CLIENT): Test the export of data.""" export_client = ExportClient() data = export_client.export_data() rses = {} for rse in list_rses(): rse_name = rse['rse'] rses[rse_name] = export_rse(rse_name) assert_equal(data['rses'], parse_response(render_json(**rses))) assert_equal(data['distances'], parse_response(render_json(**export_distances())))
def export_rses(session=None): """ Export RSE data. :param session: database session in use. """ data = {} for rse in rse_module.list_rses(session=session): rse_name = rse['rse'] data[rse_name] = rse_module.export_rse(rse_name, session=session) return data
def export_rses(vo='def', session=None): """ Export RSE data. :param vo: The VO to export. :param session: database session in use. """ data = {} for rse in rse_module.list_rses(filters={'vo': vo}, session=session): rse_id = rse['id'] data[rse_id] = rse_module.export_rse(rse_id, session=session) return data
def export_data(session=None): """ Export data. :param session: database session in use. """ data = { 'rses': [ rse_module.export_rse(rse['rse'], session=session) for rse in rse_module.list_rses(session=session) ], 'distances': distance_module.export_distances(session=session) } return data
def test_export_core(self): """ EXPORT (CORE): Test the export of data.""" data = export_data() assert_equal(data['rses'], [export_rse(rse['rse']) for rse in list_rses()]) assert_equal(data['distances'], export_distances())