def mocked_requests_get(*args, **kwargs): if args[0] == "{backend}{namespace}/projects?{query}".format( backend=BACKEND, namespace=NAMESPACE, query=QUERY_STR ): return MockResponse( { "projects": [ { "id": "id", "name": "name", "description": "description", "source_storage_id": "source_storage_id", "source_path": "source_path", "destination_storage_id": "destination_storage_id", "destination_path": "destination_path", "user_id": "user_id", "creation_timestamp": "creation_timestamp", "project_type_id": "project_type_id", } ] }, 200, ) return MockResponse(None, 404)
def mocked_requests_delete(*args, **kwargs): if args[0] == "{backend}{namespace}/projects/{id}".format( backend=BACKEND, namespace=NAMESPACE, id=PROJECT_ID ): return MockResponse(True, 200) return MockResponse(None, 404)
def mocked_requests_post(*args, **kwargs): if args[0] == "{backend}{namespace}/projects".format( backend=BACKEND, namespace=NAMESPACE, query=QUERY_STR ): return MockResponse( { "project": { "id": "id", "name": "name", "description": "description", "source_storage_id": "source_storage_id", "source_path": "source_path", "destination_storage_id": "destination_storage_id", "destination_path": "destination_path", "user_id": "user_id", "creation_timestamp": "creation_timestamp", "mission_type_id": "mission_type_id", } }, 200, ) elif args[0] == "{backend}{namespace}/projects/{project_id}/files".format( backend=BACKEND, namespace=NAMESPACE, project_id=PROJECT_ID ): return MockResponse(True, 200) elif args[0] == "{backend}{namespace}/projects/{project_id}/jobs".format( backend=BACKEND, namespace=NAMESPACE, project_id=PROJECT_ID ): return MockResponse( { "job": { "jobid": "jobid", "receiverid": "receiverid", "mission_id": "mission_id", "time_created": TIMESTAMP, "last_updated": TIMESTAMP, "status": "uploaded", "cpu_count": 1, "gpu_count": 0, "memory_amount": 0, "enable_tunnel": False, "tunnel_port": 8080, "tunnel_url": "https://example.com", "name": "name", "stationid": "stationid", "userid": "userid", "state": "state", "pay_status": "pay_status", "pay_interval": 1, "total_runtime": 10000, "archived": False, "container": "container", "oaid": "oaid", "status_history": [{"timestamp": TIMESTAMP, "status": "uploaded"}], } }, 200, ) return MockResponse(None, 404)
def mocked_requests_get(*args, **kwargs): if args[0] == "{backend}{namespace}/stations".format( backend=BACKEND, namespace=NAMESPACE ): return MockResponse( { "stations": [ { "name": "name", "stationid": "stationid", "description": "description", "users": [ { "stationuserid": "stationuserid", "userid": "userid", "status": "OWNER", "creation_timestamp": "date", "updated_timestamp": "date", "station_id": "station_id", "role_id": "role_id", } ], "mids": ["mid"], "volumes": [ { "stationid": "stationid", "name": NAME, "mount_point": "mount_point", "access": "rw", "host_paths": [], "volumeid": "volumeid", } ], "creation_timestamp": "creation_timestamp", "updated_timestamp": "updated_timestamp", "organization_id": "organization_id", "status": "stable", } for _ in range(5) ] }, 200, ) return MockResponse(None, 404)
def mocked_requests_get(*args, **kwargs): if args[0] == "{backend}{namespace}/machines/{mid}".format( backend=BACKEND, namespace=NAMESPACE, mid=MID): x = 1 return MockResponse( { "machine": { "mid": str(x), "gpu": str(x), "cpu": str(x), "arch": str(x), "memory": str(x), "name": str(x), "os": str(x), "running_jobs_limit": x, "status": "online", "userid": str(x), } }, 200, ) elif args[0] == "{backend}{namespace}/machines".format( backend=BACKEND, namespace=NAMESPACE): return MockResponse( { "machines": [{ "mid": str(x), "gpu": str(x), "cpu": str(x), "arch": str(x), "memory": str(x), "name": str(x), "os": str(x), "running_jobs_limit": x, "status": "online", "userid": str(x), } for x in range(5)] }, 200, ) return MockResponse(None, 404)
def mocked_requests_delete(*args, **kwargs): if args[0] == "{backend}{namespace}/station/{station_id}/user/{user_id}".format( backend=BACKEND, namespace=NAMESPACE, station_id=STATION_ID, user_id=USER_ID ): return MockResponse(True, 200) elif args[0] == "{backend}{namespace}/station/{station_id}".format( backend=BACKEND, namespace=NAMESPACE, station_id=STATION_ID ): return MockResponse(True, 200) elif args[0] == "{backend}{namespace}/station/{station_id}/machines".format( backend=BACKEND, namespace=NAMESPACE, station_id=STATION_ID ): return MockResponse(True, 200) elif args[ 0 ] == "{backend}{namespace}/station/{station_id}/volumes/{volumes_id}/host_paths/{host_path}".format( backend=BACKEND, namespace=NAMESPACE, station_id=STATION_ID, volumes_id=VOLUMES_ID, host_path=HOST_PATH, ): return MockResponse(True, 200) elif args[ 0 ] == "{backend}{namespace}/station/{station_id}/volumes/{volumes_id}".format( backend=BACKEND, namespace=NAMESPACE, station_id=STATION_ID, volumes_id=VOLUMES_ID, ): return MockResponse(True, 200) return MockResponse(None, 404)
def mocked_requests_get(*args, **kwargs): if args[0] == "{backend}{namespace}/job/upload_request".format( backend=BACKEND, namespace=NAMESPACE): return MockResponse({"location": LOCATION, "filename": FILENAME}, 200) elif args[ 0] == "{backend}{namespace}/jobs/{job_id}/results/location".format( backend=BACKEND, namespace=NAMESPACE, job_id=JOB_ID): return MockResponse({"location": LOCATION, "filename": FILENAME}, 200) elif args[0] == "{backend}{namespace}/jobs/{job_id}/results".format( backend=BACKEND, namespace=NAMESPACE, job_id=JOB_ID): return MockResponse( {"files": [{ "path": LOCATION, "filename": FILENAME }]}, 200) elif args[0] == "{backend}{namespace}/jobs/{job_id}/top".format( backend=BACKEND, namespace=NAMESPACE, job_id=JOB_ID): return MockResponse( { "top": { "Processes": [ ["process11", "process12"], ["process21", "process22"], ], "Titles": ["title1", "title2"], } }, 200, ) elif args[0] == "{backend}{namespace}/jobs/{job_id}/logs".format( backend=BACKEND, namespace=NAMESPACE, job_id=JOB_ID): return MockResponse({"logs": "logs"}, 200) elif args[0] == "{backend}{namespace}/jobs".format(backend=BACKEND, namespace=NAMESPACE): return MockResponse({"jobs": [job]}, 200) return MockResponse(None, 404)
def mocked_requests_put(*args, **kwargs): if args[0] == "{backend}{namespace}/station/{station_id}/users/accept".format( backend=BACKEND, namespace=NAMESPACE, station_id=STATION_ID ): return MockResponse(True, 200) elif args[0] == "{backend}{namespace}/station/{station_id}/users/reject".format( backend=BACKEND, namespace=NAMESPACE, station_id=STATION_ID ): return MockResponse(True, 200) elif args[0] == "{backend}{namespace}/station/{station_id}/requests/approve".format( backend=BACKEND, namespace=NAMESPACE, station_id=STATION_ID ): return MockResponse(True, 200) elif args[0] == "{backend}{namespace}/station/{station_id}/requests/reject".format( backend=BACKEND, namespace=NAMESPACE, station_id=STATION_ID ): return MockResponse(True, 200) elif args[0] == "{backend}{namespace}/station/{station_id}/user/withdraw".format( backend=BACKEND, namespace=NAMESPACE, station_id=STATION_ID ): return MockResponse(True, 200) return MockResponse(None, 404)
def mocked_requests_post(*args, **kwargs): if args[0] == "{backend}{namespace}/station".format( backend=BACKEND, namespace=NAMESPACE ): return MockResponse( { "station": { "name": kwargs["json"]["name"], "stationid": "stationid", "description": kwargs["json"]["description"], "users": [ { "userid": kwargs["json"]["user_ids"][0], "stationuserid": "stationuserid", "status": "OWNER", "creation_timestamp": "date", "updated_timestamp": "date", "station_id": "station_id", "role_id": "role_id", }, { "userid": kwargs["json"]["user_ids"][1], "stationuserid": "stationuserid", "status": "OWNER", "creation_timestamp": "date", "updated_timestamp": "date", "station_id": "station_id", "role_id": "role_id", }, ], "mids": ["mid"], "volumes": [ { "stationid": "stationid", "name": NAME, "mount_point": "mount_point", "access": "rw", "host_paths": [], "volumeid": "volumeid", } ], "creation_timestamp": "creation_timestamp", "updated_timestamp": "updated_timestamp", "organization_id": "organization_id", "status": "stable", } }, 200, ) elif args[0] == "{backend}{namespace}/station/{station_id}/users/invite".format( backend=BACKEND, namespace=NAMESPACE, station_id=STATION_ID ): return MockResponse(True, 200) elif args[0] == "{backend}{namespace}/station/{station_id}/requests".format( backend=BACKEND, namespace=NAMESPACE, station_id=STATION_ID ): return MockResponse(True, 200) elif args[0] == "{backend}{namespace}/station/{station_id}/machines".format( backend=BACKEND, namespace=NAMESPACE, station_id=STATION_ID ): return MockResponse(True, 200) elif args[0] == "{backend}{namespace}/station/{station_id}/volumes".format( backend=BACKEND, namespace=NAMESPACE, station_id=STATION_ID ): return MockResponse( { "volumes": { "stationid": "stationid", "name": NAME, "mount_point": "mount_point", "access": "rw", "host_paths": [], "volumeid": "volumeid", } }, 200, ) elif args[ 0 ] == "{backend}{namespace}/station/{station_id}/volumes/{volumes_id}/host_paths".format( backend=BACKEND, namespace=NAMESPACE, station_id=STATION_ID, volumes_id=VOLUMES_ID, ): return MockResponse( { "volume": { "stationid": "stationid", "name": NAME, "mount_point": "mount_point", "access": "rw", "host_paths": [], "volumeid": "volumeid", } }, 200, ) return MockResponse(None, 404)