aljaz/python-teos
1
0
Fork 0
mirror of https://github.com/aljazceru/python-teos.git synced 2025-12-17 22:24:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

Improves test appointment and add missing tests

Browse Source
This commit is contained in:
Sergi Delgado Segura
2019-11-07 18:16:14 +00:00
parent c384bc78c2
commit f4f77fb39a
1 changed files with 82 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

112
test/unit/test_appointment.py
View File

@@ -20,7 +20,15 @@ def appointment_data():
cipher = "AES-GCM-128" cipher = "AES-GCM-128"
hash_function = "SHA256" hash_function = "SHA256"
return locator, start_time, end_time, dispute_delta, encrypted_blob_data, cipher, hash_function return {
"locator": locator,
"start_time": start_time,
"end_time": end_time,
"dispute_delta": dispute_delta,
"encrypted_blob": encrypted_blob_data,
"cipher": cipher,
"hash_function": hash_function,
}
def test_init_appointment(appointment_data): def test_init_appointment(appointment_data):
@@ -28,51 +36,95 @@ def test_init_appointment(appointment_data):
# creating appointments. # creating appointments.
# DISCUSS: whether this makes sense by design or checks should be ported from the inspector to the appointment # DISCUSS: whether this makes sense by design or checks should be ported from the inspector to the appointment
# 35-appointment-checks # 35-appointment-checks
appointment = Appointment(
locator, start_time, end_time, dispute_delta, encrypted_blob_data, cipher, hash_function = appointment_data appointment_data["locator"],
appointment_data["start_time"],
appointment = Appointment(locator, start_time, end_time, dispute_delta, encrypted_blob_data, cipher, hash_function) appointment_data["end_time"],
appointment_data["dispute_delta"],
appointment_data["encrypted_blob"],
appointment_data["cipher"],
appointment_data["hash_function"],
)
assert ( assert (
locator == appointment.locator appointment_data["locator"] == appointment.locator
and start_time == appointment.start_time and appointment_data["start_time"] == appointment.start_time
and end_time == appointment.end_time and appointment_data["end_time"] == appointment.end_time
and EncryptedBlob(encrypted_blob_data) == appointment.encrypted_blob and appointment_data["dispute_delta"] == appointment.dispute_delta
and cipher == appointment.cipher and EncryptedBlob(appointment_data["encrypted_blob"]) == appointment.encrypted_blob
and dispute_delta == appointment.dispute_delta and appointment_data["cipher"] == appointment.cipher
and hash_function == appointment.hash_function and appointment_data["hash_function"] == appointment.hash_function
) )
def test_to_dict(appointment_data): def test_to_dict(appointment_data):
locator, start_time, end_time, dispute_delta, encrypted_blob_data, cipher, hash_function = appointment_data appointment = Appointment(
appointment = Appointment(locator, start_time, end_time, dispute_delta, encrypted_blob_data, cipher, hash_function) appointment_data["locator"],
appointment_data["start_time"],
appointment_data["end_time"],
appointment_data["dispute_delta"],
appointment_data["encrypted_blob"],
appointment_data["cipher"],
appointment_data["hash_function"],
)
dict_appointment = appointment.to_dict() dict_appointment = appointment.to_dict()
assert ( assert (
locator == dict_appointment.get("locator") appointment_data["locator"] == dict_appointment["locator"]
and start_time == dict_appointment.get("start_time") and appointment_data["start_time"] == dict_appointment["start_time"]
and end_time == dict_appointment.get("end_time") and appointment_data["end_time"] == dict_appointment["end_time"]
and dispute_delta == dict_appointment.get("dispute_delta") and appointment_data["dispute_delta"] == dict_appointment["dispute_delta"]
and cipher == dict_appointment.get("cipher") and EncryptedBlob(appointment_data["encrypted_blob"]) == EncryptedBlob(dict_appointment["encrypted_blob"])
and hash_function == dict_appointment.get("hash_function") and appointment_data["cipher"] == dict_appointment["cipher"]
and encrypted_blob_data == dict_appointment.get("encrypted_blob") and appointment_data["hash_function"] == dict_appointment["hash_function"]
) )
def test_to_json(appointment_data): def test_to_json(appointment_data):
locator, start_time, end_time, dispute_delta, encrypted_blob_data, cipher, hash_function = appointment_data appointment = Appointment(
appointment = Appointment(locator, start_time, end_time, dispute_delta, encrypted_blob_data, cipher, hash_function) appointment_data["locator"],
appointment_data["start_time"],
appointment_data["end_time"],
appointment_data["dispute_delta"],
appointment_data["encrypted_blob"],
appointment_data["cipher"],
appointment_data["hash_function"],
)
dict_appointment = json.loads(appointment.to_json()) dict_appointment = json.loads(appointment.to_json())
assert ( assert (
locator == dict_appointment.get("locator") appointment_data["locator"] == dict_appointment["locator"]
and start_time == dict_appointment.get("start_time") and appointment_data["start_time"] == dict_appointment["start_time"]
and end_time == dict_appointment.get("end_time") and appointment_data["end_time"] == dict_appointment["end_time"]
and dispute_delta == dict_appointment.get("dispute_delta") and appointment_data["dispute_delta"] == dict_appointment["dispute_delta"]
and cipher == dict_appointment.get("cipher") and EncryptedBlob(appointment_data["encrypted_blob"]) == EncryptedBlob(dict_appointment["encrypted_blob"])
and hash_function == dict_appointment.get("hash_function") and appointment_data["cipher"] == dict_appointment["cipher"]
and encrypted_blob_data == dict_appointment.get("encrypted_blob") and appointment_data["hash_function"] == dict_appointment["hash_function"]
) )
def test_from_dict(appointment_data):
# The appointment should be build if we don't miss any field
appointment = Appointment.from_dict(appointment_data)
assert isinstance(appointment, Appointment)
# Otherwise it should fail
appointment_data["hash_function"] = None
try:
Appointment.from_dict(appointment_data)
assert False
except ValueError:
assert True
def test_serialize(appointment_data):
appointment = Appointment.from_dict(appointment_data)
assert appointment.triggered is False
ser_appointment = appointment.serialize()
assert ser_appointment == json.dumps(appointment_data, sort_keys=True, separators=(",", ":")).encode("utf-8")