[P] Refactor testing code

This commit is contained in:
2026-03-23 16:10:09 +02:00
parent fe6bb6ab3a
commit 60a846d8b8
3 changed files with 6 additions and 4 deletions

View File

@@ -13,7 +13,7 @@ class StoreApiAdapter(StoreGateway):
def __init__(self, api_base_url):
self.api_base_url = api_base_url
def processed_agent_data_batch_to_json(self, processed_agent_data_batch: List[ProcessedAgentData]):
def processed_agent_data_batch_to_payload(self, processed_agent_data_batch: List[ProcessedAgentData]):
if not processed_agent_data_batch:
return False
@@ -28,7 +28,7 @@ class StoreApiAdapter(StoreGateway):
return payload
def save_data(self, processed_agent_data_batch: List[ProcessedAgentData]):
payload = self.processed_agent_data_batch_to_json(processed_agent_data_batch)
payload = self.processed_agent_data_batch_to_payload(processed_agent_data_batch)
if payload == False:
return False

View File

@@ -2,7 +2,7 @@ from app.adapters.store_api_adapter import StoreApiAdapter
from app.entities.agent_data import AccelerometerData, AgentData, GpsData
from app.entities.processed_agent_data import ProcessedAgentData
def _test_processed_agent_data_batch_to_json():
def _test_processed_agent_data_batch_to_payload():
processed_data_batch = [
ProcessedAgentData(road_state = "normal",
agent_data = AgentData(user_id = 1,
@@ -24,12 +24,14 @@ def _test_processed_agent_data_batch_to_json():
),
]
res = StoreApiAdapter(None).processed_agent_data_batch_to_json(processed_data_batch)
res = StoreApiAdapter(None).processed_agent_data_batch_to_payload(processed_data_batch)
assert res["data"][0]["agent_data"]["user_id"] == 1
assert res["data"][1]["agent_data"]["user_id"] == 2
assert res["data"][2]["agent_data"]["user_id"] == 3
assert StoreApiAdapter(None).processed_agent_data_batch_to_payload([]) == False
if __name__ == "__main__":
test_functions = [i for i in dir() if i.startswith('_test_')]