Fixed up test_db_api

This commit is contained in:
2024-01-13 19:02:55 +01:00
parent e7ec0a593c
commit a51e94fef3
21 changed files with 309 additions and 377 deletions

View File

@@ -13,7 +13,7 @@ from openapi_client.models import (
Eventmessage,
EventmessageCreate,
Machine,
Roles,
Role,
ServiceCreate,
Status,
)
@@ -46,7 +46,7 @@ def create_entities(num: int = 10, role: str = "entity") -> list[EntityCreate]:
name=f"C{i}",
ip=f"{host}:{port_client_base+i}",
network="255.255.0.0",
role=Roles("service_prosumer"),
roles=[Role("service_prosumer")],
visible=True,
other={},
)
@@ -56,7 +56,7 @@ def create_entities(num: int = 10, role: str = "entity") -> list[EntityCreate]:
name="DLG",
ip=f"{host}:{port_dlg}/health",
network="255.255.0.0",
role=Roles("DLG"),
roles=[Role("DLG")],
visible=True,
other={},
)
@@ -66,7 +66,7 @@ def create_entities(num: int = 10, role: str = "entity") -> list[EntityCreate]:
name="AP",
ip=f"{host}:{port_ap}/health",
network="255.255.0.0",
role=Roles("AP"),
roles=[Role("AP")],
visible=True,
other={},
)
@@ -116,7 +116,6 @@ def create_eventmessages(num: int = 2) -> list[EventmessageCreate]:
for i in range(num):
group_id = i % 5 + random.getrandbits(6)
em_req_send = EventmessageCreate(
id=random.getrandbits(18),
timestamp=starttime + i * 10,
group=i % 5,
group_id=group_id,
@@ -127,7 +126,6 @@ def create_eventmessages(num: int = 2) -> list[EventmessageCreate]:
)
res.append(em_req_send)
em_req_rec = EventmessageCreate(
id=random.getrandbits(18),
timestamp=starttime + (i * 10) + 2,
group=i % 5,
group_id=group_id,
@@ -139,7 +137,6 @@ def create_eventmessages(num: int = 2) -> list[EventmessageCreate]:
res.append(em_req_rec)
group_id = i % 5 + random.getrandbits(6)
em_res_send = EventmessageCreate(
id=random.getrandbits(18),
timestamp=starttime + i * 10 + 4,
group=i % 5,
group_id=group_id,
@@ -150,7 +147,6 @@ def create_eventmessages(num: int = 2) -> list[EventmessageCreate]:
)
res.append(em_res_send)
em_res_rec = EventmessageCreate(
id=random.getrandbits(6),
timestamp=starttime + (i * 10) + 8,
group=i % 5,
group_id=group_id,
@@ -166,8 +162,10 @@ def create_eventmessages(num: int = 2) -> list[EventmessageCreate]:
def test_create_eventmessages(api_client: ApiClient) -> None:
api = EventmessagesApi(api_client=api_client)
assert [] == api.get_all_eventmessages()
for own_eventmsg in create_eventmessages():
for idx, own_eventmsg in enumerate(create_eventmessages()):
res: Eventmessage = api.create_eventmessage(own_eventmsg)
# breakpoint()
assert res.id == own_eventmsg.id
assert res.msg == own_eventmsg.msg
assert res.src_did == own_eventmsg.src_did
assert res.des_did == own_eventmsg.des_did
assert [] != api.get_all_eventmessages()