3
}:a2 @ s d Z ddlZddlZddlmZmZmZmZ ejdZ ej
dZ ej
dZ ej
dZej
dZej
d ZG d
d d ejZG d d
d
ZG dd dejeZG dd dejeZG dd dejeZedkrej dS )zTests for josepy.jwk. N)errors json_util test_utilutilzdsa512_key.pemzrsa256_key.pemzrsa512_key.pemzec_p256_key.pemzec_p384_key.pemzec_p521_key.pemc @ s e Zd ZdZdd Zdd ZdS )JWKTestzTests for josepy.jwk.JWK.c C s" ddl m} | jtj|jt d S )Nr )JWK)
josepy.jwkr assertRaisesr ErrorloadDSA_PEM)selfr r /usr/lib/python3.6/jwk_test.py test_load s zJWKTest.test_loadc C s" ddl m} | jtj|jt d S )Nr )JWKRSA)r r r r r
r r )r
r r r r test_load_subclass_wrong_type s z%JWKTest.test_load_subclass_wrong_typeN)__name__
__module__ __qualname____doc__r r r r r r r s r c @ s* e Zd ZU dZeZedd Zdd ZdS )JWKTestBaseMixinz"Mixin test for JWK subclass tests.c C s | j | j| jj d S )N) assertEqual
thumbprintjwk)r
r r r test_thumbprint_private s z(JWKTestBaseMixin.test_thumbprint_privatec C s | j | j| jj j d S )N)r r r
public_key)r
r r r test_thumbprint_public# s z'JWKTestBaseMixin.test_thumbprint_publicN) r r r r NotImplementedr bytesr r r r r r r s
r c @ sD e Zd ZdZdZdd Zdd Zdd Zd d
Zd d Z d
d Z
dS )
JWKOctTestz Tests for josepy.jwk.JWKOct.s 羨`p}>I͒'o1(i c C s. ddl m} |dd| _dtjdd| _d S )Nr )JWKOcts foo)keyoct)ktyk)r r! r r Zencode_b64josejobj)r
r! r r r setUp. s zJWKOctTest.setUpc C s | j | jj | j d S )N)r r to_partial_jsonr& )r
r r r test_to_partial_json3 s z JWKOctTest.test_to_partial_jsonc C s& ddl m} | j| j|j| j d S )Nr )r! )r r! r r from_jsonr& )r
r! r r r test_from_json6 s zJWKOctTest.test_from_jsonc C s ddl m} t|j| j d S )Nr )r! )r r! hashr* r& )r
r! r r r test_from_json_hashable: s z"JWKOctTest.test_from_json_hashablec C s$ ddl m} | j| j|jd d S )Nr )r! s foo)r r! r r r )r
r! r r r r > s zJWKOctTest.test_loadc C s | j | jj | j d S )N)ZassertIsr r )r
r r r test_public_keyB s zJWKOctTest.test_public_keyN) r r r r r r' r) r+ r- r r. r r r r r ' s r c @ s e Zd ZdZdZdd Zdd Zdd Zd d
Zd d Z d
d Z
dd Z dd Z dd Z
dd Zdd Zdd Zdd Zd d Zd S )
JWKRSATestz Tests for josepy.jwk.JWKRSA.s K#3ʘˀ< H*l$&|O Yc C s ddl m} |tj d| _dddd| _|tj jd| _|tj d| _ dddd| _
|td| _ | jj | _
d | j
d
< | jj | _| jjd d d d
ddd | j | _d S )Nr )r )r" RSAAQABz+m2Fylv-Uz7trgTW8EBHP3FQSMeZs2GNQ6VRo1sIVJEk)r$ enzVrHVztFHtH92ucFJD_N_HW9AsdRsUuHUBBBDlHwNlRd3fp580rv2-6QWE30cWgdmJS86ObRz6lUTor4R0T-3C5Qz+lPQED_EPTV0UIBfNI3KP2d9Jlrc2mrMllmf946bu-CEdZzUVNZn4lLLBD1R6NE8TKNQZwcfKfc7kl5jfqXArCRSURQZCWJFq43QvT5Bm5iN8n1okQZbHh2u7etM8LKKCF2pY2UdQZoi45cEkbVoJjAbnQpFY87Q)r4 pqZdpZdqZqi)r r
RSA256_KEYr jwk256
jwk256json_wrappedjwk256_not_comparable
RSA512_KEYjwk512
jwk512jsonprivatecopyprivate_json_small private_jsonupdater )r
r r r r r' M s.
zJWKRSATest.setUpc C s& | j | jjtj | j| j| j d S )N)assertIsInstancer; r" r ZComparableRSAKeyr r8 )r
r r r test_init_auto_comparablen s z$JWKRSATest.test_init_auto_comparablec C s" ddl m} | jd|jd d S )Nr )r AA)r r r
_encode_param)r
r r r r test_encode_param_zeror s z!JWKRSATest.test_encode_param_zeroc C s$ | j | j| j | j | j| j d S )N)r r8 r= )r
r r r test_equalsx s zJWKRSATest.test_equalsc C s$ | j | j| j | j | j| j d S )N)assertNotEqualr8 r= )r
r r r test_not_equals| s zJWKRSATest.test_not_equalsc C s* ddl m} | j| j|jtjd d S )Nr )r zrsa256_key.pem)r r r r? r r load_vector)r
r r r r r s zJWKRSATest.test_loadc C s | j | j| jj d S )N)r r8 r? r )r
r r r r. s zJWKRSATest.test_public_keyc C s@ | j | jj | j | j | jj | j | j | jj | j d S )N)r r8 r( r9 r= r> r? rB )r
r r r r) s z JWKRSATest.test_to_partial_jsonc C sR ddl m} | j| j|j| j | j| j|j| j | j| j|j| j d S )Nr )r )
r r r r8 r* r9 r= r> r? rB )r
r r r r r+ s zJWKRSATest.test_from_jsonc C s& ddl m} | j| j|j| j d S )Nr )r )r r r r? r* rA )r
r r r r test_from_json_private_small s z'JWKRSATest.test_from_json_private_smallc C s, ddl m} | jd= | jtj|j| j d S )Nr )r r6 )r r rB r r r
r* )r
r r r r %test_from_json_missing_one_additional s z0JWKRSATest.test_from_json_missing_one_additionalc C s ddl m} t|j| j d S )Nr )r )r r r, r* r9 )r
r r r r r- s z"JWKRSATest.test_from_json_hashablec C sD ddl m} | jtj|jdddd | jtj|jdddd d S )Nr )r r0 r1 )r$ r2 r3 1)r r r r DeserializationErrorr* )r
r r r r test_from_json_non_schema_errors s
z+JWKRSATest.test_from_json_non_schema_errorsc C s0 ddl m} |jd}| jtj|j d d S )Nr )r a {
"kty": "RSA",
"kid": "bilbo.baggins@hobbiton.example",
"use": "sig",
"n": "n4EPtAOCc9AlkeQHPzHStgAbgs7bTZLwUBZdR8_KuKPEHLd4rHVTeT-O-XV2jRojdNhxJWTDvNd7nqQ0VEiZQHz_AJmSCpMaJMRBSFKrKb2wqVwGU_NsYOYL-QtiWN2lbzcEe6XC0dApr5ydQLrHqkHHig3RBordaZ6Aj-oBHqFEHYpPe7Tpe-OfVfHd1E6cS6M1FZcD1NNLYD5lFHpPI9bTwJlsde3uhGqC0ZCuEHg8lhzwOHrtIQbS0FVbb9k3-tVTU4fg_3L_vniUFAKwuCLqKnS2BYwdq_mzSnbLY7h_qixoR7jig3__kRhuaxwUkRz5iaiQkqgc5gHdrNP5zw",
"e": "AQAB"
}s@ f63838e96077ad1fc01c3f8405774dedc0641f558ebb4b40dccf5f9b6d66a932)r r Z
json_loadsr binasciiZhexlifyr )r
r r" r r r test_thumbprint_go_jose s z"JWKRSATest.test_thumbprint_go_joseN)r r r r r r' rE rH rI rK r r. r) r+ rM rN r- rR rT r r r r r/ F s !r/ c @ s e Zd ZdZdZdd Zdd Zdd Zd d
Zd d Z d
d Z
dd Z dd Z dd Z
dd Zdd Zdd Zdd Zd d Zd d Zd!S )" JWKECTestzTests for josepy.jwk.JWKEC.s L lFJ$ jO dRc C s ddl m} |tj d| _|tj d| _|tj d| _|tj j d| _
ddddd| _ dd d
d d| _ dd d
dd| _
|td| _dddddd| _| j| _d S )Nr )JWKEC)r" ECzP-256z+jjQtV-fA7J_tK8dPzYq7jRPNjF8r5p6LW2R25S2Gw5Uz+EPAw8_8z7PYKsHH6hlGSlsWxFoFl7-0vM0QRGbmnvCc)r$ crvxyzP-384Z@tIhpNtEXkadUbrY84rYGgApFM1X_3l3EWQRuOP1IWtxlTftrZQwneJZF0k0eRn00z@KW2Gp-TThDXmZ-9MJPnD8hv-X130SVvfZRl1a04HPVwIbvLe87mvA_iuOa-myUyvzP-521ZXAFkdl6cKzBmP18U8fffpP4IZN2eED45hDcwRPl5ZeClwHcLtnMBMuWYFFO_Nzm6DL2MhpN0zI2bcMLJd95aY2tPszXAYvZq3wByjt7nQd8nYMqhFNCL3j_-U6GPWZet1hYBY_XZHrC4yIV0R4JnssRAY9eqc1EElpCc4hziis1jiV1iR4WZ+xReNQBKqqTthG8oTmBdhp4EQYImSK1dVqfa2yyMn2rc)r4 rY rZ rX r$ )r rV EC_P256_KEYr r8 EC_P384_KEYjwk384 EC_P521_KEYjwk521r: r; r9
jwk384json
jwk521jsonr? rB r )r
rV r r r r' s0
zJWKECTest.setUpc C s& | j | jjtj | j| j| j d S )N)rD r; r" r ZComparableECKeyr r8 )r
r r r rE s z#JWKECTest.test_init_auto_comparablec C s$ ddl m} | jd|jdd d S )Nr )rV rF )r rV r rG )r
rV r r r rH s z JWKECTest.test_encode_param_zeroc C s4 | j | j| j | j | j| j | j | j| j d S )N)r r8 r] r_ )r
r r r rI s zJWKECTest.test_equalsc C sd | j | j| j | j | j| j | j | j| j | j | j| j | j | j| j | j | j| j d S )N)rJ r8 r] r_ )r
r r r rK s zJWKECTest.test_not_equalsc C s* ddl m} | j| j|jtjd d S )Nr )rV zec_p256_key.pem)r rV r r? r r rL )r
rV r r r r s zJWKECTest.test_loadc C s | j | j| jj d S )N)r r8 r? r )r
r r r r. s zJWKECTest.test_public_keyc C sT | j | jj | j | j | jj | j | j | jj | j | j | jj | j d S )N)
r r8 r( r9 r] r` r_ ra r? rB )r
r r r r) s z JWKECTest.test_to_partial_jsonc C sh ddl m} | j| j|j| j | j| j|j| j | j| j|j| j | j| j
|j| j d S )Nr )r ) r r r r8 r* r9 r] r` r_ ra r? rB )r
r r r r r+ s zJWKECTest.test_from_jsonc C s* ddl m} | jd= | jt|j| j d S )Nr )r rY )r r rB r KeyErrorr* )r
r r r r #test_from_json_missing_x_coordinate s z-JWKECTest.test_from_json_missing_x_coordinatec C s* ddl m} | jd= | jt|j| j d S )Nr )r rZ )r r rB r rc r* )r
r r r r #test_from_json_missing_y_coordinate s z-JWKECTest.test_from_json_missing_y_coordinatec C s ddl m} t|j| j d S )Nr )r )r r r, r* r9 )r
r r r r r- s z!JWKECTest.test_from_json_hashablec C sH ddl m} | jtj|jddddd | jtj|jdddd d d S )
Nr )r rW zP-256r1 z+m2Fylv-Uz7trgTW8EBHP3FQSMeZs2GNQ6VRo1sIVJEk)r$ rX rY rZ z+jjQtV-fA7J_tK8dPzYq7jRPNjF8r5p6LW2R25S2Gw5UrP )r r r r rQ r* )r
r r r r rR " s
z*JWKECTest.test_from_json_non_schema_errorsc C s, ddl m} | jtj|jddddd d S )Nr )r rW zP-255z+jjQtV-fA7J_tK8dPzYq7jRPNjF8r5p6LW2R25S2Gw5Uz+EPAw8_8z7PYKsHH6hlGSlsWxFoFl7-0vM0QRGbmnvCc)r$ rX rY rZ )r r r r rQ r* )r
r r r r test_unknown_crv_name+ s z JWKECTest.test_unknown_crv_namec C sl ddl m}m} dd l}d}|j|}|j }|jj|d }| j|d d | jt |d |j
| d S )Nr )rV r s -----BEGIN EC PRIVATE KEY-----
MHcCAQEEICZ7LCI99Na2KZ/Fq8JmJROakGJ5+J7rHiGSPoO36kOAoAoGCCqGSM49
AwEHoUQDQgAEGS5RvStca15z2FEanCM3juoX7tE/LB7iD44GWawGE40APAl/iZuH
31wQfst4glTZpxkpEI/MzNZHjiYnqrGeSw==
-----END EC PRIVATE KEY-----rZ ) r rV r josepyr r( r Zdecode_b64joser lenr* )r
rV r rh datar" rZ r r r test_encode_y_leading_zero_p2563 s
z)JWKECTest.test_encode_y_leading_zero_p256N)r r r r r r' rE rH rI rK r r. r) r+ rd re r- rR rf rk r r r r rU s" ! rU __main__)r rS Zunittestrh r r r r rL r Zload_rsa_private_keyr7 r< Zload_ec_private_keyr[ r\ r^ ZTestCaser r r r/ rU r mainr r r r s"
u