generated from Luis/nextjs-python-web-template
Fixed up test_db_api
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user