[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): def __init__(self, api_base_url):
self.api_base_url = 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: if not processed_agent_data_batch:
return False return False
@@ -28,7 +28,7 @@ class StoreApiAdapter(StoreGateway):
return payload return payload
def save_data(self, processed_agent_data_batch: List[ProcessedAgentData]): 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: if payload == False:
return 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.agent_data import AccelerometerData, AgentData, GpsData
from app.entities.processed_agent_data import ProcessedAgentData 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 = [ processed_data_batch = [
ProcessedAgentData(road_state = "normal", ProcessedAgentData(road_state = "normal",
agent_data = AgentData(user_id = 1, 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"][0]["agent_data"]["user_id"] == 1
assert res["data"][1]["agent_data"]["user_id"] == 2 assert res["data"][1]["agent_data"]["user_id"] == 2
assert res["data"][2]["agent_data"]["user_id"] == 3 assert res["data"][2]["agent_data"]["user_id"] == 3
assert StoreApiAdapter(None).processed_agent_data_batch_to_payload([]) == False
if __name__ == "__main__": if __name__ == "__main__":
test_functions = [i for i in dir() if i.startswith('_test_')] test_functions = [i for i in dir() if i.startswith('_test_')]