from store_api_adapter import StoreApiAdapter
from app.entities.agent_data import AccelerometerData, AgentData, GpsData

def test_processed_agent_data_batch_to_json():
    processed_data_batch = [
            ProcessedAgentData(road_state = "normal",
                               agent_data = AgentData(user_id = 1, accelerometer = AccelerometerData(x = 0.1, y = 0.2, z = 0.3)),
                               gps = GpsData(latitude = 10.123, longitude = 20.456),
                               timestamp = "2023-07-21T12:34:56Z"),
            ProcessedAgentData(road_state = "normal",
                               agent_data = AgentData(user_id = 2, accelerometer = AccelerometerData(x = 0.1, y = 0.2, z = 0.3)),
                               gps = GpsData(latitude = 10.123, longitude = 20.456),
                               timestamp = "2023-07-21T12:34:56Z"),
            ProcessedAgentData(road_state = "normal",
                               agent_data = AgentData(user_id = 3, accelerometer = AccelerometerData(x = 0.1, y = 0.2, z = 0.3)),
                               gps = GpsData(latitude = 10.123, longitude = 20.456),
                               timestamp = "2023-07-21T12:34:56Z"),
    ]

    res = json.loads(StoreApiAdapter(None).processed_agent_data_batch_to_json(processed_data_batch))

    assert res[0].agent_data.user_id == 1
    assert res[1].agent_data.user_id == 2
    assert res[2].agent_data.user_id == 3

if __name__ == "__main__":
    test_processed_agent_data_batch_to_json()