|
16 | 16 | from ocpp.v16.datatypes import MeterValue, SampledValue |
17 | 17 | from ocpp.v16.enums import Action, RegistrationStatus |
18 | 18 | from ocpp.v201 import ChargePoint as cp_201 |
| 19 | +from ocpp.v201 import call_result, datatypes, enums |
19 | 20 | from ocpp.v201.call import GetVariables as v201GetVariables |
20 | 21 | from ocpp.v201.call import SetNetworkProfile as v201SetNetworkProfile |
21 | 22 | from ocpp.v201.datatypes import ( |
@@ -312,6 +313,67 @@ def test_serialize_as_dict(): |
312 | 313 | assert serialize_as_dict(payload) == expected |
313 | 314 |
|
314 | 315 |
|
| 316 | +def test_serialization_of_collection_of_multiple_elements(): |
| 317 | + """This test validates that bug #635 is fixed. |
| 318 | + That bug incorrectly serialized payloads that contain a collection |
| 319 | + of elements. |
| 320 | +
|
| 321 | + This test serializes a call_result.SetVariables that contains 2 |
| 322 | + SetVariableResultTypes. |
| 323 | +
|
| 324 | + https://github.com/mobilityhouse/ocpp/issues/635 |
| 325 | + """ |
| 326 | + payload = call_result.SetVariables( |
| 327 | + set_variable_result=[ |
| 328 | + datatypes.SetVariableResultType( |
| 329 | + attribute_status=enums.SetVariableStatusType.accepted, |
| 330 | + component={ |
| 331 | + "name": "TemperatureSensor", |
| 332 | + "instance": "First", |
| 333 | + "evse": {"id": 1, "connector_id": 1}, |
| 334 | + }, |
| 335 | + variable={"name": "DisplayUnit", "instance": "Main"}, |
| 336 | + attribute_type="Actual", |
| 337 | + attribute_status_info=None, |
| 338 | + ), |
| 339 | + datatypes.SetVariableResultType( |
| 340 | + attribute_status="Accepted", |
| 341 | + component={"name": "TxCtrlr"}, |
| 342 | + variable={"name": "TxStopPoint"}, |
| 343 | + attribute_type="Actual", |
| 344 | + attribute_status_info=None, |
| 345 | + ), |
| 346 | + ], |
| 347 | + custom_data=None, |
| 348 | + ) |
| 349 | + |
| 350 | + expected = { |
| 351 | + "custom_data": None, |
| 352 | + "set_variable_result": [ |
| 353 | + { |
| 354 | + "attribute_status": "Accepted", |
| 355 | + "attribute_status_info": None, |
| 356 | + "attribute_type": "Actual", |
| 357 | + "component": { |
| 358 | + "evse": {"connector_id": 1, "id": 1}, |
| 359 | + "instance": "First", |
| 360 | + "name": "TemperatureSensor", |
| 361 | + }, |
| 362 | + "variable": {"instance": "Main", "name": "DisplayUnit"}, |
| 363 | + }, |
| 364 | + { |
| 365 | + "attribute_status": "Accepted", |
| 366 | + "attribute_status_info": None, |
| 367 | + "attribute_type": "Actual", |
| 368 | + "component": {"name": "TxCtrlr"}, |
| 369 | + "variable": {"name": "TxStopPoint"}, |
| 370 | + }, |
| 371 | + ], |
| 372 | + } |
| 373 | + # Execute / Assert |
| 374 | + assert serialize_as_dict(payload) == expected |
| 375 | + |
| 376 | + |
315 | 377 | @pytest.mark.asyncio |
316 | 378 | async def test_call_unique_id_added_to_handler_args_correctly(connection): |
317 | 379 | """ |
|
0 commit comments