python-teos/test/unit/test_appointment.py

import json
from pytest import fixture

from pisa import c_logger
from pisa.appointment import Appointment
from pisa.encrypted_blob import EncryptedBlob
from test.unit.conftest import get_random_value_hex


c_logger.disabled = True

# Not much to test here, adding it for completeness
@fixture
def appointment_data():
    locator = get_random_value_hex(32)
    start_time = 100
    end_time = 120
    dispute_delta = 20
    encrypted_blob_data = get_random_value_hex(100)
    cipher = "AES-GCM-128"
    hash_function = "SHA256"

    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):
    # The appointment has no checks whatsoever, since the inspector is the one taking care or that, and the only one
    # creating appointments.
    # DISCUSS: whether this makes sense by design or checks should be ported from the inspector to the appointment
    #          35-appointment-checks
    appointment = Appointment(
        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"],
    )

    assert (
        appointment_data["locator"] == appointment.locator
        and appointment_data["start_time"] == appointment.start_time
        and appointment_data["end_time"] == appointment.end_time
        and appointment_data["dispute_delta"] == appointment.dispute_delta
        and EncryptedBlob(appointment_data["encrypted_blob"]) == appointment.encrypted_blob
        and appointment_data["cipher"] == appointment.cipher
        and appointment_data["hash_function"] == appointment.hash_function
    )


def test_to_dict(appointment_data):
    appointment = Appointment(
        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()

    assert (
        appointment_data["locator"] == dict_appointment["locator"]
        and appointment_data["start_time"] == dict_appointment["start_time"]
        and appointment_data["end_time"] == dict_appointment["end_time"]
        and appointment_data["dispute_delta"] == dict_appointment["dispute_delta"]
        and EncryptedBlob(appointment_data["encrypted_blob"]) == EncryptedBlob(dict_appointment["encrypted_blob"])
        and appointment_data["cipher"] == dict_appointment["cipher"]
        and appointment_data["hash_function"] == dict_appointment["hash_function"]
    )


def test_to_json(appointment_data):
    appointment = Appointment(
        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())

    assert (
        appointment_data["locator"] == dict_appointment["locator"]
        and appointment_data["start_time"] == dict_appointment["start_time"]
        and appointment_data["end_time"] == dict_appointment["end_time"]
        and appointment_data["dispute_delta"] == dict_appointment["dispute_delta"]
        and EncryptedBlob(appointment_data["encrypted_blob"]) == EncryptedBlob(dict_appointment["encrypted_blob"])
        and appointment_data["cipher"] == dict_appointment["cipher"]
        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")