From 502f507b2170ae18b0435c182cd4afa1ed89c498 Mon Sep 17 00:00:00 2001 From: Sergi Delgado Segura Date: Tue, 31 Mar 2020 15:00:30 +0200 Subject: [PATCH] Adds missing size field in Builder when loading data from the database --- teos/builder.py | 6 +++++- test/teos/unit/test_builder.py | 1 + 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/teos/builder.py b/teos/builder.py index 379bac8..6d087fa 100644 --- a/teos/builder.py +++ b/teos/builder.py @@ -26,7 +26,11 @@ class Builder: locator_uuid_map = {} for uuid, data in appointments_data.items(): - appointments[uuid] = {"locator": data.get("locator"), "end_time": data.get("end_time")} + appointments[uuid] = { + "locator": data.get("locator"), + "end_time": data.get("end_time"), + "size": len(data.get("encrypted_blob")), + } if data.get("locator") in locator_uuid_map: locator_uuid_map[data.get("locator")].append(uuid) diff --git a/test/teos/unit/test_builder.py b/test/teos/unit/test_builder.py index 014d797..756cc5e 100644 --- a/test/teos/unit/test_builder.py +++ b/test/teos/unit/test_builder.py @@ -46,6 +46,7 @@ def test_build_appointments(): assert uuid in appointments_data.keys() assert appointments_data[uuid].get("locator") == appointment.get("locator") assert appointments_data[uuid].get("end_time") == appointment.get("end_time") + assert len(appointments_data[uuid].get("encrypted_blob")) == appointment.get("size") assert uuid in locator_uuid_map[appointment.get("locator")]