idds/tests/test_client.py

0001 #!/usr/bin/env python
0002 #
0003 # Licensed under the Apache License, Version 2.0 (the "License");
0004 # You may not use this file except in compliance with the License.
0005 # You may obtain a copy of the License at
0006 # http://www.apache.org/licenses/LICENSE-2.0OA
0007 #
0008 # Authors:
0009 # - Wen Guan, <wen.guan@cern.ch>, 2021
0010
0011
0012 """
0013 Test client.
0014 """
0015
0016
0017 from idds.client.clientmanager import ClientManager
0018 from idds.common.utils import json_dumps  # noqa F401
0019 from idds.rest.v1.utils import convert_old_req_2_workflow_req   # noqa F401
0020 from idds.common.utils import setup_logging
0021
0022
0023 setup_logging("idds.log")
0024
0025
0026 def test():
0027     # 72533, 72569, 72733, 72769, 72783, 72939, 73351, 73983, 74545,
0028     # 74567, 74569, 74573
0029     # dev
0030     dev_host = 'https://aipanda160.cern.ch:443/idds'    # noqa F841
0031     # doma
0032     doma_host = 'https://aipanda015.cern.ch:443/idds'   # noqa F841
0033     # atlas
0034     atlas_host = 'https://aipanda181.cern.ch:443/idds'  # noqa F841
0035     # doma google
0036     doma_google_host = 'https://34.133.138.229:443/idds'  # noqa F841
0037
0038     cm1 = ClientManager(host=atlas_host)
0039     cm1 = ClientManager(host=doma_host)
0040     cm1 = ClientManager(host=dev_host)
0041     request_id = 414
0042
0043     ret = cm1.get_requests(request_id, with_detail=True)
0044     print(json_dumps(ret, sort_keys=True, indent=4))
0045
0046     cm1.setup_json_outputs()
0047     ret = cm1.get_requests(request_id, with_detail=True)
0048     print(json_dumps(ret, sort_keys=True, indent=4))
0049
0050     ret = cm1.get_request_id_by_name(name='test_workflow.idds.1672836584.9900262.test')
0051     print(json_dumps(ret, sort_keys=True, indent=4))
0052
0053     ret = cm1.get_request_id_by_name(name='test_workflow.idds.1672836584.9900262.test1')
0054     print(json_dumps(ret, sort_keys=True, indent=4))
0055
0056     ret = cm1.get_request_id_by_name(name='test_workflow.idds*')
0057     print(json_dumps(ret, sort_keys=True, indent=4))
0058
0059     ret = cm1.get_contents_output_ext(request_id=request_id)
0060     print(json_dumps(ret, sort_keys=True, indent=4))
0061
0062     ret = cm1.get_contents_output_ext(request_id=request_id, group_by_jedi_task_id=True)
0063     print(json_dumps(ret, sort_keys=True, indent=4))
0064
0065     ret = cm1.get_contents_output_ext(request_id=None)
0066     print(json_dumps(ret, sort_keys=True, indent=4))
0067
0068     ret = cm1.get_contents_output_ext(request_id=99999)
0069     print(json_dumps(ret, sort_keys=True, indent=4))
0070
0071
0072 if __name__ == '__main__':
0073     test()