tests: restored tests

Author: PascalDR

pymdoccbor/tests/pkey.py

@@ -0,0 +1,9 @@
+import os
+
+PKEY = {
+    'KTY': 'EC2',
+    'CURVE': 'P_256',
+    'ALG': 'ES256',
+    'D': b"<\xe5\xbc;\x08\xadF\x1d\xc5\x0czR'T&\xbb\x91\xac\x84\xdc\x9ce\xbf\x0b,\x00\xcb\xdd\xbf\xec\xa2\xa5",
+    'KID': b"demo-kid"
+}

pymdoccbor/tests/test_03_mdoc_issuer.py

@@ -0,0 +1,73 @@
+from pycose.keys import EC2Key
+from pymdoccbor.mdoc.issuer import MdocCborIssuer
+from pymdoccbor.tests.micov_data import MICOV_DATA
+from pymdoccbor.tests.pid_data import PID_DATA
+from pymdoccbor.tests.pkey import PKEY
+
+mdoc = MdocCborIssuer(
+    private_key=PKEY,
+    alg="ES256",
+)
+
+def test_MdocCborIssuer_creation():
+    assert mdoc.version == '1.0'
+    assert mdoc.status == 0
+
+def test_mdoc_without_private_key_must_fail():
+    try:
+        MdocCborIssuer(None)
+    except Exception as e:
+        assert str(e) == "You must provide a private key"
+
+def test_MdocCborIssuer_new_single():
+    mdoc.new(
+        data=MICOV_DATA,
+        #devicekeyinfo=PKEY,  # TODO
+        doctype="org.micov.medical.1",
+        validity={
+            "issuance_date": "2024-12-31",
+            "expiry_date": "2050-12-31"
+        },
+    )
+    assert mdoc.signed['version'] == '1.0'
+    assert mdoc.signed['status'] == 0
+    assert mdoc.signed['documents'][0]['docType'] == 'org.micov.medical.1'
+    assert mdoc.signed['documents'][0]['issuerSigned']['nameSpaces']['org.micov.medical.1'][0].tag == 24
+
+# TODO: restore multiple documents support
+"""
+def test_MdocCborIssuer_new_multiple():
+    micov_data = {"doctype": "org.micov.medical.1", "data": MICOV_DATA}
+    pid_data = {"doctype": "eu.europa.ec.eudiw.pid.1", "data": PID_DATA}
+
+    mdoc.new(
+        #TODO: fix the doctype handling
+        doctype="org.micov.medical.1",
+        data=[micov_data, pid_data],
+        validity={
+            "issuance_date": "2024-12-31",
+            "expiry_date": "2050-12-31"
+        },
+        #devicekeyinfo=PKEY  # TODO
+    )
+    assert mdoc.signed['version'] == '1.0'
+    assert mdoc.signed['status'] == 0
+    assert mdoc.signed['documents'][0]['docType'] == 'org.micov.medical.1'
+    assert mdoc.signed['documents'][0]['issuerSigned']['nameSpaces']['org.micov.medical.1'][0].tag == 24
+    assert mdoc.signed['documents'][1]['docType'] == 'eu.europa.ec.eudiw.pid.1'
+    assert mdoc.signed['documents'][1]['issuerSigned']['nameSpaces']['eu.europa.ec.eudiw.pid.1'][0].tag == 24
+"""
+
+def test_MdocCborIssuer_dump():
+    dump = mdoc.dump()
+    
+    assert dump
+    assert isinstance(dump, bytes)
+    assert len(dump) > 0
+
+def test_MdocCborIssuer_dumps():
+    dumps = mdoc.dumps()
+    
+    assert dumps
+    assert isinstance(dumps, bytes)
+    assert len(dumps) > 0

pymdoccbor/tests/test_04_issuer_signed.py

@@ -0,0 +1,52 @@
+from pycose.keys import EC2Key
+from pymdoccbor.mdoc.issuersigned import IssuerSigned
+from pymdoccbor.mdoc.issuer import MdocCborIssuer
+from pymdoccbor.tests.micov_data import MICOV_DATA
+from pymdoccbor.tests.pkey import PKEY
+
+
+mdoc = MdocCborIssuer(
+    private_key=PKEY,
+    alg="ES256",
+)
+mdoc.new(
+    data=MICOV_DATA,
+    #devicekeyinfo=PKEY,  # TODO
+    doctype="org.micov.medical.1",
+    validity={
+        "issuance_date": "2024-12-31",
+        "expiry_date": "2050-12-31"
+    },
+)
+issuerAuth = mdoc.signed["documents"][0]["issuerSigned"]
+issuer_signed = IssuerSigned(**issuerAuth)
+
+def test_issuer_signed_fail():
+    try:
+        IssuerSigned(None, None)
+    except Exception as e:
+        assert str(e) == "issuerAuth must be provided"
+
+def test_issuer_signed_creation():
+    assert issuer_signed.namespaces
+    assert issuer_signed.issuer_auth
+
+def test_issuer_signed_dump():
+    issuerAuth = mdoc.signed["documents"][0]["issuerSigned"]
+
+    issuer_signed = IssuerSigned(**issuerAuth)
+
+    dump = issuer_signed.dump()
+    assert dump
+    assert dump["nameSpaces"] == issuer_signed.namespaces
+    assert dump["issuerAuth"] == issuer_signed.issuer_auth
+
+def test_issuer_signed_dumps():
+    issuerAuth = mdoc.signed["documents"][0]["issuerSigned"]
+
+    issuer_signed = IssuerSigned(**issuerAuth)
+
+    dumps = issuer_signed.dumps()
+    assert dumps
+    assert isinstance(dumps, bytes)
+    assert len(dumps) > 0

pymdoccbor/tests/test_05_mdoc_verifier.py

@@ -0,0 +1,107 @@
+from pycose.keys import EC2Key
+from pymdoccbor.mdoc.verifier import MobileDocument
+from pymdoccbor.mdoc.issuer import MdocCborIssuer
+from pymdoccbor.tests.micov_data import MICOV_DATA
+from pymdoccbor.tests.pkey import PKEY
+
+def test_verifier_must_fail_document_type():
+    try:
+        MobileDocument(None, None)
+    except Exception as e:
+        assert str(e) == "You must provide a document type"
+
+def test_verifier_must_fail_issuer_signed():
+    try:
+        MobileDocument("org.micov.medical.1", None)
+    except Exception as e:
+        assert str(e) == "You must provide a signed document"
+
+def test_mobile_document():
+    mdoc = MdocCborIssuer(
+        private_key=PKEY,
+        alg="ES256",
+    )
+    mdoc.new(
+        data=MICOV_DATA,
+        #devicekeyinfo=PKEY,  # TODO
+        doctype="org.micov.medical.1",
+        validity={
+            "issuance_date": "2024-12-31",
+            "expiry_date": "2050-12-31"
+        },
+    )
+
+
+    document = mdoc.signed["documents"][0]
+    doc = MobileDocument(**document)
+
+    assert doc.doctype == "org.micov.medical.1"
+    assert doc.issuersigned
+
+def test_mobile_document_dump():
+    mdoc = MdocCborIssuer(
+        private_key=PKEY,
+        alg="ES256"
+    )
+    mdoc.new(
+        data=MICOV_DATA,
+        #devicekeyinfo=PKEY,  # TODO
+        doctype="org.micov.medical.1",
+        validity={
+            "issuance_date": "2024-12-31",
+            "expiry_date": "2050-12-31"
+        },
+    )
+
+
+    document = mdoc.signed["documents"][0]
+    doc = MobileDocument(**document)
+
+    dump = doc.dump()
+    assert dump
+    assert isinstance(dump, bytes)
+    assert len(dump) > 0
+
+def test_mobile_document_dumps():
+    mdoc = MdocCborIssuer(
+        private_key=PKEY,
+        alg="ES256"
+    )
+    mdoc.new(
+        data=MICOV_DATA,
+        #devicekeyinfo=PKEY,  # TODO
+        doctype="org.micov.medical.1",
+        validity={
+            "issuance_date": "2024-12-31",
+            "expiry_date": "2050-12-31"
+        },
+    )
+
+
+    document = mdoc.signed["documents"][0]
+    doc = MobileDocument(**document)
+
+    dumps = doc.dumps()
+    assert dumps
+    assert isinstance(dumps, bytes)
+    assert len(dumps) > 0
+
+def test_mobile_document_verify():
+    mdoc = MdocCborIssuer(
+        private_key=PKEY,
+        alg="ES256"
+    )
+    mdoc.new(
+        data=MICOV_DATA,
+        #devicekeyinfo=PKEY,  # TODO
+        doctype="org.micov.medical.1",
+        validity={
+            "issuance_date": "2024-12-31",
+            "expiry_date": "2050-12-31"
+        },
+    )
+
+    document = mdoc.signed["documents"][0]
+    doc = MobileDocument(**document)
+
+    assert doc.verify()

pymdoccbor/tests/test_06_mso_issuer.py

@@ -0,0 +1,43 @@
+from pycose.keys import EC2Key
+from pycose.messages import CoseMessage
+from pymdoccbor.mso.issuer import MsoIssuer
+from pymdoccbor.tests.micov_data import MICOV_DATA
+from pymdoccbor.tests.pkey import PKEY
+
+
+def test_mso_issuer_fail():
+    try:
+        MsoIssuer(None, None)
+    except Exception as e:
+        assert str(e) == "MSO Writer requires a valid private key"
+
+def test_mso_issuer_creation():
+    msoi = MsoIssuer(
+        data=MICOV_DATA,
+        private_key=PKEY,
+        validity={
+            "issuance_date": "2024-12-31",
+            "expiry_date": "2050-12-31"
+        },
+        alg="ES256"
+    )
+
+    assert msoi.private_key
+    assert msoi.data
+    assert msoi.hash_map
+    assert list(msoi.hash_map.keys())[0] == 'org.micov.medical.1'
+    assert msoi.disclosure_map['org.micov.medical.1']
+
+def test_mso_issuer_sign():
+    msoi = MsoIssuer(
+        data=MICOV_DATA,
+        private_key=PKEY,
+        validity={
+            "issuance_date": "2024-12-31",
+            "expiry_date": "2050-12-31"
+        },
+        alg="ES256"
+    )
+
+    mso = msoi.sign()
+    assert isinstance(mso, CoseMessage)

pymdoccbor/tests/test_07_mso_verifier.py

@@ -0,0 +1,67 @@
+
+import os
+from pycose.keys import CoseKey, EC2Key
+from pymdoccbor.mso.verifier import MsoVerifier
+from pymdoccbor.mdoc.issuer import MdocCborIssuer
+from pymdoccbor.tests.micov_data import MICOV_DATA
+from pycose.messages import CoseMessage
+from pymdoccbor.tests.pkey import PKEY
+
+
+mdoc = MdocCborIssuer(
+    private_key=PKEY,
+    alg="ES256",
+)
+
+mdoc.new(
+    data=MICOV_DATA,
+    #devicekeyinfo=PKEY,  # TODO
+    doctype="org.micov.medical.1",
+    validity={
+        "issuance_date": "2024-12-31",
+        "expiry_date": "2050-12-31"
+    },
+)
+
+def test_mso_verifier_fail():
+    try:
+        MsoVerifier(None)
+    except Exception as e:
+        assert str(e) == "MsoParser only supports raw bytes and list, a <class 'NoneType'> was provided"
+
+def test_mso_verifier_creation():
+    issuerAuth = mdoc.signed["documents"][0]["issuerSigned"]["issuerAuth"]
+
+    msov = MsoVerifier(issuerAuth)
+
+    assert isinstance(msov.object, CoseMessage)
+
+def test_mso_verifier_verify_signatures():
+    issuerAuth = mdoc.signed["documents"][0]["issuerSigned"]["issuerAuth"]
+
+    msov = MsoVerifier(issuerAuth)
+
+    assert msov.verify_signature()
+
+def test_mso_verifier_payload_as_cbor():
+    issuerAuth = mdoc.signed["documents"][0]["issuerSigned"]["issuerAuth"]
+
+    msov = MsoVerifier(issuerAuth)
+
+    cbor = msov.payload_as_dict
+
+    assert cbor
+    assert cbor["version"] == "1.0"
+    assert cbor["digestAlgorithm"] == "SHA-256"
+    assert cbor["valueDigests"]["org.micov.medical.1"]
+
+def test_payload_as_raw():
+    issuerAuth = mdoc.signed["documents"][0]["issuerSigned"]["issuerAuth"]
+
+    msov = MsoVerifier(issuerAuth)
+
+    raw = msov.payload_as_raw
+
+    assert raw
+    assert isinstance(raw, bytes)
+    assert len(raw) > 0

pymdoccbor/tests/test_08_mdoc_cbor.py

@@ -3,14 +3,7 @@
 from pymdoccbor.mdoc.issuer import MdocCborIssuer
 from pymdoccbor.tests.micov_data import MICOV_DATA
 from pymdoccbor.mdoc.verifier import MdocCbor
-
-PKEY = {
-    'KTY': 'EC2',
-    'CURVE': 'P_256',
-    'ALG': 'ES256',
-    'D': os.urandom(32),
-    'KID': b"demo-kid"
-}
+from pymdoccbor.tests.pkey import PKEY
 
 def test_mdoc_cbor_creation():
     mdoci = MdocCborIssuer(