262 lines
7.7 KiB
Python
262 lines
7.7 KiB
Python
import json
|
|
import multicodec
|
|
import multiformats
|
|
import nacl.encoding
|
|
|
|
from pyvckit.did import generate_keys, generate_did, get_signing_key, gen_did_document
|
|
from pyvckit.sign_vc import sign
|
|
from pyvckit.sign_vp import sign_vp
|
|
from pyvckit.verify import verify_vc
|
|
from pyvckit.verify_vp 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)
|
|
signing_key = get_signing_key(key)
|
|
|
|
credential = {
|
|
"@context": "https://www.w3.org/2018/credentials/v1",
|
|
"id": "http://example.org/credentials/3731",
|
|
"type": ["VerifiableCredential"],
|
|
"credentialSubject": {
|
|
"id": "did:key:z6MkgGXSJoacuuNdwU1rGfPpFH72GACnzykKTxzCCTZs6Z2M",
|
|
},
|
|
"issuer": did,
|
|
"issuanceDate": now()
|
|
}
|
|
|
|
cred = json.dumps(credential)
|
|
|
|
vc = sign(cred, signing_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)
|
|
signing_key = get_signing_key(key)
|
|
|
|
credential = {
|
|
"@context": "https://www.w3.org/2018/credentials/v1",
|
|
"id": "http://example.org/credentials/3731",
|
|
"type": ["VerifiableCredential"],
|
|
"credentialSubject": {
|
|
"id": "did:key:z6MkgGXSJoacuuNdwU1rGfPpFH72GACnzykKTxzCCTZs6Z2M",
|
|
},
|
|
"issuer": did,
|
|
"issuanceDate": now()
|
|
}
|
|
|
|
cred = json.dumps(credential)
|
|
|
|
vc = sign(cred, signing_key, did)
|
|
vc_json = json.dumps(vc)
|
|
|
|
holder_key = generate_keys()
|
|
holder_did = generate_did(holder_key)
|
|
holder_signing_key = get_signing_key(holder_key)
|
|
vp = sign_vp(holder_signing_key, holder_did, vc_json)
|
|
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)
|
|
signing_key = get_signing_key(key)
|
|
|
|
credential = {
|
|
"@context": "https://www.w3.org/2018/credentials/v1",
|
|
"id": "http://example.org/credentials/3731",
|
|
"type": ["VerifiableCredential"],
|
|
"credentialSubject": {
|
|
"id": "did:key:z6MkgGXSJoacuuNdwU1rGfPpFH72GACnzykKTxzCCTZs6Z2M",
|
|
},
|
|
"issuer": did,
|
|
"issuanceDate": now()
|
|
}
|
|
|
|
cred = json.dumps(credential)
|
|
|
|
vc = sign(cred, signing_key, did)
|
|
verified = verify_vc(json.dumps(vc))
|
|
assert verified
|
|
|
|
|
|
def test_verifiable_credential_fail():
|
|
key = generate_keys()
|
|
did = generate_did(key)
|
|
signing_key = get_signing_key(key)
|
|
|
|
credential = {
|
|
"@context": "https://www.w3.org/2018/credentials/v1",
|
|
"id": "http://example.org/credentials/3731",
|
|
"type": ["VerifiableCredential"],
|
|
"credentialSubject": {
|
|
"id": "did:key:z6MkgGXSJoacuuNdwU1rGfPpFH72GACnzykKTxzCCTZs6Z2M",
|
|
},
|
|
"issuer": did,
|
|
"issuanceDate": now()
|
|
}
|
|
|
|
cred = json.dumps(credential)
|
|
|
|
vc = sign(cred, signing_key, did)
|
|
vc["id"] = "bar"
|
|
verified = verify_vc(json.dumps(vc))
|
|
assert not verified
|
|
|
|
|
|
def test_verifiable_presentation():
|
|
key = generate_keys()
|
|
did = generate_did(key)
|
|
signing_key = get_signing_key(key)
|
|
|
|
credential = {
|
|
"@context": "https://www.w3.org/2018/credentials/v1",
|
|
"id": "http://example.org/credentials/3731",
|
|
"type": ["VerifiableCredential"],
|
|
"credentialSubject": {
|
|
"id": "did:key:z6MkgGXSJoacuuNdwU1rGfPpFH72GACnzykKTxzCCTZs6Z2M",
|
|
},
|
|
"issuer": did,
|
|
"issuanceDate": now()
|
|
}
|
|
|
|
cred = json.dumps(credential)
|
|
|
|
vc = sign(cred, signing_key, did)
|
|
vc_json = json.dumps(vc)
|
|
|
|
holder_key = generate_keys()
|
|
holder_did = generate_did(holder_key)
|
|
holder_signing_key = get_signing_key(holder_key)
|
|
vp = sign_vp(holder_signing_key, holder_did, vc_json)
|
|
verified = verify_vp(json.dumps(vp))
|
|
assert verified
|
|
|
|
|
|
def test_verifiable_presentation_fail1():
|
|
key = generate_keys()
|
|
did = generate_did(key)
|
|
signing_key = get_signing_key(key)
|
|
|
|
credential = {
|
|
"@context": "https://www.w3.org/2018/credentials/v1",
|
|
"id": "http://example.org/credentials/3731",
|
|
"type": ["VerifiableCredential"],
|
|
"credentialSubject": {
|
|
"id": "did:key:z6MkgGXSJoacuuNdwU1rGfPpFH72GACnzykKTxzCCTZs6Z2M",
|
|
},
|
|
"issuer": did,
|
|
"issuanceDate": now()
|
|
}
|
|
|
|
cred = json.dumps(credential)
|
|
|
|
vc = sign(cred, signing_key, did)
|
|
vc_json = json.dumps(vc)
|
|
|
|
holder_key = generate_keys()
|
|
holder_did = generate_did(holder_key)
|
|
holder_signing_key = get_signing_key(holder_key)
|
|
vp = sign_vp(holder_signing_key, holder_did, vc_json)
|
|
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)
|
|
signing_key = get_signing_key(key)
|
|
|
|
credential = {
|
|
"@context": "https://www.w3.org/2018/credentials/v1",
|
|
"id": "http://example.org/credentials/3731",
|
|
"type": ["VerifiableCredential"],
|
|
"credentialSubject": {
|
|
"id": "did:key:z6MkgGXSJoacuuNdwU1rGfPpFH72GACnzykKTxzCCTZs6Z2M",
|
|
},
|
|
"issuer": did,
|
|
"issuanceDate": now()
|
|
}
|
|
|
|
cred = json.dumps(credential)
|
|
|
|
vc = sign(cred, signing_key, did)
|
|
vc_json = json.dumps(vc)
|
|
|
|
holder_key = generate_keys()
|
|
holder_did = generate_did(holder_key)
|
|
holder_signing_key = get_signing_key(holder_key)
|
|
vp = sign_vp(holder_signing_key, holder_did, vc_json)
|
|
vp["id"] = "http://example.org/presentations/3732"
|
|
verified = verify_vp(json.dumps(vp))
|
|
assert not verified
|
|
|