142 lines
4.1 KiB
Python
142 lines
4.1 KiB
Python
import json
|
|
import multicodec
|
|
import multiformats
|
|
import nacl.encoding
|
|
|
|
from did import generate_keys, generate_did, get_signing_key
|
|
from sign_vc import sign
|
|
from sign_vp import sign_vp
|
|
from verify import verify_vc
|
|
from verify_vp import verify_vp
|
|
from utils import now
|
|
|
|
|
|
def test_generated_did_key():
|
|
key = generate_keys()
|
|
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.get('x', '')
|
|
missing_padding = len(k_x) % 4
|
|
if missing_padding:
|
|
k_x += '=' * (4 - missing_padding)
|
|
|
|
assert key.get('kty') == 'OKP'
|
|
assert key.get('crv') == 'Ed25519'
|
|
assert key.get('kid') == 'Generated'
|
|
assert k_x == x
|
|
assert key.get('d') is not None
|
|
|
|
|
|
def test_credential():
|
|
# import pdb; pdb.set_trace()
|
|
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_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_vc(json.dumps(vp))
|
|
assert verified
|
|
|