[P] Refactor testing code
This commit is contained in:
@@ -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
|
||||
|
||||
@@ -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_')]
|
||||
|
||||
Reference in New Issue
Block a user