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