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)
