import json import multicodec import multiformats import nacl.encoding from pyvckit.did import generate_keys, generate_did, gen_did_document from pyvckit.sign import sign from pyvckit.sign_vp import get_presentation from pyvckit.verify import verify_vc from pyvckit.verify import verify_vp from pyvckit.utils import now def test_generated_did_key(): key = generate_keys() key_d = json.loads(key) did = generate_did(key) _did = did.split("#")[0] pub = _did.split(":")[-1] mc = multiformats.multibase.decode(pub) public_key_bytes = multicodec.remove_prefix(mc) x = nacl.encoding.URLSafeBase64Encoder.encode(public_key_bytes).decode('utf-8') k_x = key_d.get('x', '') missing_padding = len(k_x) % 4 if missing_padding: k_x += '=' * (4 - missing_padding) assert key_d.get('kty') == 'OKP' assert key_d.get('crv') == 'Ed25519' assert key_d.get('kid') == 'Generated' assert k_x == x assert key_d.get('d') is not None assert did.split(":")[:-1] == ['did', 'key'] def test_generated_did_web(): key = generate_keys() key_d = json.loads(key) url = "https://localhost/did-registry" did = generate_did(key, url) _did = did.split("#")[0] pub = _did.split(":")[-1] mc = multiformats.multibase.decode(pub) public_key_bytes = multicodec.remove_prefix(mc) x = nacl.encoding.URLSafeBase64Encoder.encode(public_key_bytes).decode('utf-8') k_x = key_d.get('x', '') missing_padding = len(k_x) % 4 if missing_padding: k_x += '=' * (4 - missing_padding) assert key_d.get('kty') == 'OKP' assert key_d.get('crv') == 'Ed25519' assert key_d.get('kid') == 'Generated' assert k_x == x assert key_d.get('d') is not None assert did.split(":")[:-1] == ['did', 'web', 'localhost', 'did-registry'] def test_generated_did_document(): key = generate_keys() key_d = json.loads(key) url = "https://localhost/did-registry" did = generate_did(key, url) definitive_url, document = gen_did_document(did, key_d) pubkey = did.split(":")[-1] doc_id = json.loads(document)["id"] assert doc_id == did assert definitive_url == f"{url}/{pubkey}/did.json" def test_credential(): key = generate_keys() did = generate_did(key) credential = { "@context": "https://www.w3.org/2018/credentials/v1", "id": "http://example.org/credentials/3731", "type": ["VerifiableCredential"], "credentialSubject": { "id": "did:key:z6MkgGXSJoacuuNdwU1rGfPpFH72GACnzykKTxzCCTZs6Z2M", }, "issuer": { "id": did, "name": "" }, "issuanceDate": now() } cred = json.dumps(credential) vc = sign(cred, key, did) header = 'eyJhbGciOiJFZERTQSIsImNyaXQiOlsiYjY0Il0sImI2NCI6ZmFsc2V9' assert vc.get('proof', {}).get('jws') is not None assert header in vc.get('proof', {}).get('jws') assert did in vc.get('proof', {}).get('verificationMethod') def test_presentation(): key = generate_keys() did = generate_did(key) credential = { "@context": "https://www.w3.org/2018/credentials/v1", "id": "http://example.org/credentials/3731", "type": ["VerifiableCredential"], "credentialSubject": { "id": "did:key:z6MkgGXSJoacuuNdwU1rGfPpFH72GACnzykKTxzCCTZs6Z2M", }, "issuer": { "id": did, "name": "" }, "issuanceDate": now() } cred = json.dumps(credential) vc = sign(cred, key, did) vc_json = json.dumps(vc) holder_key = generate_keys() holder_did = generate_did(holder_key) unsigned_vp = get_presentation(vc_json, holder_did) vp = sign(unsigned_vp, holder_key, holder_did) header = 'eyJhbGciOiJFZERTQSIsImNyaXQiOlsiYjY0Il0sImI2NCI6ZmFsc2V9' assert vp.get('proof', {}).get('jws') is not None assert header in vp.get('proof', {}).get('jws') assert holder_did in vp.get('proof', {}).get('verificationMethod') def test_verifiable_credential(): key = generate_keys() did = generate_did(key) credential = { "@context": "https://www.w3.org/2018/credentials/v1", "id": "http://example.org/credentials/3731", "type": ["VerifiableCredential"], "credentialSubject": { "id": "did:key:z6MkgGXSJoacuuNdwU1rGfPpFH72GACnzykKTxzCCTZs6Z2M", }, "issuer": { "id": did, "name": "" }, "issuanceDate": now() } cred = json.dumps(credential) vc = sign(cred, key, did) verified = verify_vc(json.dumps(vc)) assert verified def test_verifiable_presentation(): key = generate_keys() did = generate_did(key) credential = { "@context": "https://www.w3.org/2018/credentials/v1", "id": "http://example.org/credentials/3732", "type": ["VerifiableCredential"], "credentialSubject": { "id": "did:key:z6MkgGXSJoacuuNdwU1rGfPpFH72GACnzykKTxzCCTZs6Z2M", }, "issuer": { "id": did, "name": "" }, "issuanceDate": now() } cred = json.dumps(credential) vc = sign(cred, key, did) vc_json = json.dumps(vc) assert verify_vc(json.dumps(vc)) holder_key = generate_keys() holder_did = generate_did(holder_key) unsigned_vp = get_presentation(vc_json, holder_did) vp = sign(unsigned_vp, holder_key, holder_did) verified = verify_vp(json.dumps(vp)) assert verified def test_verifiable_credential_fail(): key = generate_keys() did = generate_did(key) credential = { "@context": "https://www.w3.org/2018/credentials/v1", "id": "http://example.org/credentials/3731", "type": ["VerifiableCredential"], "credentialSubject": { "id": "did:key:z6MkgGXSJoacuuNdwU1rGfPpFH72GACnzykKTxzCCTZs6Z2M", }, "issuer": { "id": did, "name": "" }, "issuanceDate": now() } cred = json.dumps(credential) vc = sign(cred, key, did) vc["id"] = "bar" verified = verify_vc(json.dumps(vc)) assert not verified def test_verifiable_presentation_fail1(): key = generate_keys() did = generate_did(key) credential = { "@context": "https://www.w3.org/2018/credentials/v1", "id": "http://example.org/credentials/3731", "type": ["VerifiableCredential"], "credentialSubject": { "id": "did:key:z6MkgGXSJoacuuNdwU1rGfPpFH72GACnzykKTxzCCTZs6Z2M", }, "issuer": { "id": did, "name": "" }, "issuanceDate": now() } cred = json.dumps(credential) vc = sign(cred, key, did) vc_json = json.dumps(vc) holder_key = generate_keys() holder_did = generate_did(holder_key) unsigned_vp = get_presentation(vc_json, holder_did) vp = sign(unsigned_vp, holder_key, holder_did) vp["verifiableCredential"][0]["id"] = "bar" verified = verify_vp(json.dumps(vp)) assert not verified def test_verifiable_presentation_fail2(): key = generate_keys() did = generate_did(key) credential = { "@context": "https://www.w3.org/2018/credentials/v1", "id": "http://example.org/credentials/3731", "type": ["VerifiableCredential"], "credentialSubject": { "id": "did:key:z6MkgGXSJoacuuNdwU1rGfPpFH72GACnzykKTxzCCTZs6Z2M", }, "issuer": { "id": did, "name": "" }, "issuanceDate": now() } cred = json.dumps(credential) vc = sign(cred, key, did) vc_json = json.dumps(vc) holder_key = generate_keys() holder_did = generate_did(holder_key) unsigned_vp = get_presentation(vc_json, holder_did) vp = sign(unsigned_vp, holder_key, holder_did) vp["id"] = "http://example.org/presentations/3732" verified = verify_vp(json.dumps(vp)) assert not verified