Skip to content

Commit

Permalink
moving to Blueapi rest client
Browse files Browse the repository at this point in the history
  • Loading branch information
@ZohebShaikh
ZohebShaikh committed Sep 4, 2024
1 parent 991fd62 commit b9bb6dd
Showing 1 changed file with 111 additions and 161 deletions.
272 changes: 111 additions & 161 deletions tests/system_tests/test_api_endpoints.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5,30 +5,23 @@
from fastapi import status
from pydantic import TypeAdapter

from blueapi.config import RestConfig
from blueapi.client.rest import BlueapiRestClient
from blueapi.service.interface import get_devices, get_plans
from blueapi.service.model import (
DeviceModel,
DeviceResponse,
EnvironmentResponse,
PlanModel,
PlanResponse,
StateChangeRequest,
TaskResponse,
TasksListResponse,
WorkerTask,
)
from blueapi.worker.event import WorkerState
from blueapi.worker.task import Task
from blueapi.worker.task_worker import TrackableTask

ENDPOINT = f"{RestConfig().protocol}://{RestConfig().host}:{RestConfig().port}"

_SIMPLE_TASK = Task(name="sleep", params={"time": 0.0})
_LONG_TASK = Task(name="sleep", params={"time": 1.0})

TASKS: list[Task] = [_SIMPLE_TASK, _LONG_TASK]

rest = BlueapiRestClient()


def get_response(
url: str, BaseModel: Any, status_code: int = status.HTTP_200_OK
Expand Down Expand Up @@ -69,68 +62,40 @@ def put_response(
return TypeAdapter(BaseModel).validate_python(put_response.json())


def test_server_up():
response = requests.get(ENDPOINT + "/docs")
assert response.status_code == status.HTTP_200_OK


def test_get_current_state_of_environment():
environment = get_response(ENDPOINT + "/environment", EnvironmentResponse)

assert isinstance(
environment, EnvironmentResponse
) and environment == EnvironmentResponse(initialized=True)
assert rest.get_environment() == EnvironmentResponse(initialized=True)


def test_get_plans():
get_plans_response = get_response(ENDPOINT + "/plans", PlanResponse)
assert get_plans_response == PlanResponse(plans=get_plans())
assert rest.get_plans() == PlanResponse(plans=get_plans())


def test_get_plans_by_name():
for plan in get_plans():
plan_response = get_response(f"{ENDPOINT}/plans/{plan.name}", PlanModel)
assert plan_response == plan
assert rest.get_plan(plan.name) == plan


def test_get_devices():
get_device_response = get_response(ENDPOINT + "/devices", DeviceResponse)
assert get_device_response == DeviceResponse(devices=get_devices())
assert rest.get_devices() == DeviceResponse(devices=get_devices())


def test_get_device_by_name():
for device in get_devices():
device_response = get_response(f"{ENDPOINT}/devices/{device.name}", DeviceModel)
assert device_response == device
assert rest.get_device(device.name) == device


def test_post_task_and_delete_task_by_id():
created_task = post_response(
url=ENDPOINT + "/tasks",
json=_SIMPLE_TASK.model_dump(),
BaseModel=TaskResponse,
)

assert isinstance(created_task, TaskResponse)

delete_response(
url=f"{ENDPOINT}/tasks/{created_task.task_id}",
BaseModel=TaskResponse,
)
create_task = rest.create_task(_SIMPLE_TASK)
rest.clear_task(create_task.task_id)


def test_get_tasks():
created_tasks = []
for task in TASKS:
created_task = post_response(
url=ENDPOINT + "/tasks",
json=task.model_dump(),
BaseModel=TaskResponse,
)
assert isinstance(created_task, TaskResponse)
created_task = rest.create_task(task)
created_tasks.append(created_task)

task_list = get_response(ENDPOINT + "/tasks", TasksListResponse)
task_list = get_response(rest._url("/tasks"), TasksListResponse)

assert isinstance(task_list, TasksListResponse)
task_ids = [task.task_id for task in created_tasks]
Expand All @@ -139,121 +104,106 @@ def test_get_tasks():
assert task.is_complete is False and task.is_pending is True

for task in created_tasks:
delete_response(
url=f"{ENDPOINT}/tasks/{task.task_id}",
BaseModel=TaskResponse,
)


def test_get_task_by_id():
created_task = post_response(
url=ENDPOINT + "/tasks",
json=_SIMPLE_TASK.model_dump(),
BaseModel=TaskResponse,
)

assert isinstance(created_task, TaskResponse)

get_task = get_response(
url=f"{ENDPOINT}/tasks/{created_task.task_id}", BaseModel=TrackableTask
)

assert isinstance(get_task, TrackableTask)
assert (
get_task.task_id == created_task.task_id
and get_task.is_pending
and not get_task.is_complete
and len(get_task.errors) == 0
)
delete_response(
url=f"{ENDPOINT}/tasks/{created_task.task_id}",
BaseModel=TaskResponse,
)


def test_get_worker_task_by_id():
created_task = post_response(
url=ENDPOINT + "/tasks",
json=_SIMPLE_TASK.model_dump(),
BaseModel=TaskResponse,
)
assert isinstance(created_task, TaskResponse)

get_worker_info = get_response(
f"{ENDPOINT}/tasks/{created_task.task_id}", TrackableTask
)

assert (
isinstance(get_worker_info, TrackableTask)
and get_worker_info.task_id == created_task.task_id
and get_worker_info.is_complete is False
and get_worker_info.is_pending is True
and get_worker_info.errors == []
)

delete_response(
url=f"{ENDPOINT}/tasks/{created_task.task_id}",
BaseModel=TaskResponse,
status_code=status.HTTP_200_OK,
)


def test_put_worker_task():
created_task = post_response(
url=ENDPOINT + "/tasks",
json=_SIMPLE_TASK.model_dump(),
BaseModel=TaskResponse,
status_code=status.HTTP_201_CREATED,
)

worker_task = put_response(
url=ENDPOINT + "/worker/task",
data=WorkerTask(task_id=created_task.task_id).model_dump_json(),
BaseModel=WorkerTask,
)
assert isinstance(worker_task, WorkerTask)
delete_response(
url=f"{ENDPOINT}/tasks/{created_task.task_id}",
BaseModel=TaskResponse,
status_code=status.HTTP_200_OK,
)


def test_get_worker_state():
created_task = post_response(
url=ENDPOINT + "/tasks",
json=_SIMPLE_TASK.model_dump(),
BaseModel=TaskResponse,
status_code=status.HTTP_201_CREATED,
)
assert isinstance(created_task, TaskResponse)
get_worker_state = get_response(ENDPOINT + "/worker/state", WorkerState)

assert isinstance(get_worker_state, WorkerState)
delete_response(
url=f"{ENDPOINT}/tasks/{created_task.task_id}",
BaseModel=TaskResponse,
status_code=status.HTTP_200_OK,
)


def test_put_worker_state():
put_task = put_response(
url=ENDPOINT + "/worker/state",
data=StateChangeRequest(new_state=WorkerState.RUNNING).model_dump_json(),
BaseModel=WorkerState,
status_code=status.HTTP_400_BAD_REQUEST,
)
assert isinstance(put_task, WorkerState)
rest.clear_task(task.task_id)


# def test_get_task_by_id():
# created_task = rest.create_task(_SIMPLE_TASK)

# get_task = get_response(
# url=f"{ENDPOINT}/tasks/{created_task.task_id}", BaseModel=TrackableTask
# )

# assert isinstance(get_task, TrackableTask)
# assert (
# get_task.task_id == created_task.task_id
# and get_task.is_pending
# and not get_task.is_complete
# and len(get_task.errors) == 0
# )
# delete_response(
# url=f"{ENDPOINT}/tasks/{created_task.task_id}",
# BaseModel=TaskResponse,
# )


# def test_get_worker_task_by_id():
# created_task = post_response(
# url=ENDPOINT + "/tasks",
# json=_SIMPLE_TASK.model_dump(),
# BaseModel=TaskResponse,
# )
# assert isinstance(created_task, TaskResponse)

# get_worker_info = get_response(
# f"{ENDPOINT}/tasks/{created_task.task_id}", TrackableTask
# )

# assert (
# isinstance(get_worker_info, TrackableTask)
# and get_worker_info.task_id == created_task.task_id
# and get_worker_info.is_complete is False
# and get_worker_info.is_pending is True
# and get_worker_info.errors == []
# )

# delete_response(
# url=f"{ENDPOINT}/tasks/{created_task.task_id}",
# BaseModel=TaskResponse,
# status_code=status.HTTP_200_OK,
# )


# def test_put_worker_task():
# created_task = post_response(
# url=ENDPOINT + "/tasks",
# json=_SIMPLE_TASK.model_dump(),
# BaseModel=TaskResponse,
# status_code=status.HTTP_201_CREATED,
# )

# worker_task = put_response(
# url=ENDPOINT + "/worker/task",
# data=WorkerTask(task_id=created_task.task_id).model_dump_json(),
# BaseModel=WorkerTask,
# )
# assert isinstance(worker_task, WorkerTask)
# delete_response(
# url=f"{ENDPOINT}/tasks/{created_task.task_id}",
# BaseModel=TaskResponse,
# status_code=status.HTTP_200_OK,
# )


# def test_get_worker_state():
# created_task = post_response(
# url=ENDPOINT + "/tasks",
# json=_SIMPLE_TASK.model_dump(),
# BaseModel=TaskResponse,
# status_code=status.HTTP_201_CREATED,
# )
# assert isinstance(created_task, TaskResponse)
# get_worker_state = get_response(ENDPOINT + "/worker/state", WorkerState)

# assert isinstance(get_worker_state, WorkerState)
# delete_response(
# url=f"{ENDPOINT}/tasks/{created_task.task_id}",
# BaseModel=TaskResponse,
# status_code=status.HTTP_200_OK,
# )


# def test_put_worker_state():
# put_task = put_response(
# url=ENDPOINT + "/worker/state",
# data=StateChangeRequest(new_state=WorkerState.RUNNING).model_dump_json(),
# BaseModel=WorkerState,
# status_code=status.HTTP_400_BAD_REQUEST,
# )
# assert isinstance(put_task, WorkerState)


def test_delete_current_environment():
delete_response = requests.delete(ENDPOINT + "/environment")
assert delete_response.status_code == status.HTTP_200_OK
assert EnvironmentResponse(initialized=False).model_dump() == delete_response.json()
rest.delete_environment()
time.sleep(5)
get_response = requests.get(ENDPOINT + "/environment")
assert get_response.status_code == status.HTTP_200_OK
assert TypeAdapter(EnvironmentResponse).validate_python(
get_response.json()
) == EnvironmentResponse(initialized=True)
assert rest.get_environment() == EnvironmentResponse(initialized=True)

0 comments on commit b9bb6dd

Please sign in to comment.