3
}:a7# @ s d Z ddlZddlmZ ddlmZmZ ejdZejdZejdZ ej
dZ ej
d Z ej
d
Z
G d d d ejZG d
d dejZG dd dejZG dd dejZedkrej dS )zTests for josepy.jwa. N)mock)errors test_utilzrsa256_key.pemzrsa512_key.pemzrsa1024_key.pemzec_p256_key.pemzec_p384_key.pemzec_p521_key.pemc @ sH e Zd ZdZdd Zdd Zdd Zdd Zd
d Zd d
Z dd Z
dS )JWASignatureTestz"Tests for josepy.jwa.JWASignature.c C s4 ddl m} G dd d|}|d| _|d| _d S )Nr ) JWASignaturec @ s e Zd Zdd Zdd ZdS )z'JWASignatureTest.setUp..MockSigc S s
t d S )N)NotImplementedError)selfkeymsg r /usr/lib/python3.6/jwa_test.pysign s z,JWASignatureTest.setUp..MockSig.signc S s
t d S )N)r )r r r
sigr r r verify s z.JWASignatureTest.setUp..MockSig.verifyN)__name__
__module__ __qualname__r
r r r r r MockSig s r Sig1Sig2)
josepy.jwar r r )r r r r r r setUp s
zJWASignatureTest.setUpc C s | j | j| j d S )N) assertEqualr )r r r r test_eq" s zJWASignatureTest.test_eqc C s | j | j| j d S )N)assertNotEqualr r )r r r r test_ne% s zJWASignatureTest.test_nec C s | j | jd d S )N )r r )r r r r test_ne_other_type( s z#JWASignatureTest.test_ne_other_typec C s( | j dt| j | j dt| j d S )Nr r )r reprr r )r r r r test_repr+ s zJWASignatureTest.test_reprc C s( | j | jj d | j | jj d d S )Nr r )r r Zto_partial_jsonr )r r r r test_to_partial_json/ s z%JWASignatureTest.test_to_partial_jsonc C s. ddl m} ddl m} | j|jd| d S )Nr )r )RS256r! )r r r! assertIs from_json)r r r! r r r test_from_json3 s z JWASignatureTest.test_from_jsonN) r r r __doc__r r r r r r r$ r r r r r s r c @ s e Zd Zdd ZdS ) JWAHSTestc C sX ddl m} d}| j|jdd| | j|jdd|d | j|jdd|d d d S ) Nr )HS256s R͔A.:'i
+UZԳzOs some keys fooT !F)r r' r r
r" r )r r' r r r r test_it; s
zJWAHSTest.test_itN)r r r r) r r r r r& 9 s r& c @ sL e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dS ) JWARSTestc C s( ddl m} | jtj|jtj d d S )Nr )r! s foo)r r! assertRaisesr Errorr
RSA512_KEY
public_key)r r! r r r test_sign_no_private_partH s z#JWARSTest.test_sign_no_private_partc C sD ddl m} ddl m} | jtj|jtd | jtj|jtd d S )Nr )r! )PS256s foo)r r! r0 r+ r r, r
RSA256_KEY)r r! r0 r r r test_sign_key_too_smallL s z!JWARSTest.test_sign_key_too_smallc C s` ddl m} d}| j|jtd| | j|jtj d|d | j|jtj d|d d d S )Nr )r! s@ |Ʋ(*:N&}Ot"YY) &ǙS OVҹ.>}s fooTr( F)r r! r r
r- r" r r. )r r! r r r r test_rsR s zJWARSTest.test_rsc C sT ddl m} |jtd}| j|jtj d|d | j|jtj d|d d d S )Nr )r0 s fooTr( F)r r0 r
RSA1024_KEYr" r r. )r r0 r r r r test_ps_ s
zJWARSTest.test_psc C s4 ddl m} tj }|j|d | j|jjd d S )Nr )r! messageT)r r! r MagicMockr
r" called)r r! r r r r test_sign_new_apif s zJWARSTest.test_sign_new_apic C sj ddl m} tjdgd}tj }||j_|j|d | j|jjd | j|j jd | j|j
jd d S )Nr )r! signer)specr6 T) r r! r r7 r: return_valuer
r" r8 updatefinalize)r r! r r: r r r test_sign_old_apil s zJWARSTest.test_sign_old_apic C s6 ddl m} tj }|j|dd | j|jjd d S )Nr )r! r6 signatureT)r r! r r7 r r" r8 )r r! r r r r test_verify_new_apiv s z JWARSTest.test_verify_new_apic C sl ddl m} tjdgd}tj }||j_|j|dd | j|jjd | j|j jd | j|jjd d S )Nr )r! verifier)r; r6 r@ T)
r r! r r7 rB r< r r" r8 r= )r r! r rB r r r test_verify_old_api| s z JWARSTest.test_verify_old_apiN) r r r r/ r2 r3 r5 r9 r? rA rC r r r r r* F s
r* c @ s\ e Zd Zdd Zdd Zdd Zdd Zd d
Zd d Zd
d Z dd Z
dd Z dd Z dS ) JWAECTestc C s( ddl m} | jtj|jtj d d S )Nr )ES256s foo)r rE r+ r r, r
EC_P256_KEYr. )r rE r r r r/ s z#JWAECTest.test_sign_no_private_partc C s: ddl m} d}|jt|}| j|jtj ||d d S )Nr )rE s fooT)r rE r
rF r" r r. )r rE r6 r@ r r r test_es256_sign_and_verify s z$JWAECTest.test_es256_sign_and_verifyc C s: ddl m} d}|jt|}| j|jtj ||d d S )Nr )ES384s fooT)r rH r
EC_P384_KEYr" r r. )r rH r6 r@ r r r test_es384_sign_and_verify s z$JWAECTest.test_es384_sign_and_verifyc C s> ddl m}m} d}|jt|}| j|jtj ||d d S )Nr )rE rH s fooF) r rE rH r
rF r" r rI r. )r rE rH r6 r@ r r r test_verify_with_wrong_jwa s z$JWAECTest.test_verify_with_wrong_jwac C sb ddl m} ddlm} ddlm} d}|jt|}|j|j | }| j
|j |j ||d d S )Nr )rE )ec)default_backends fooF)
r rE Z)cryptography.hazmat.primitives.asymmetricrL Z cryptography.hazmat.backendsrM r
rF Zgenerate_private_key SECP256R1r" r r. )r rE rL rM r6 r@ Z
different_keyr r r test_verify_with_different_key s z(JWAECTest.test_verify_with_different_keyc
C sb ddl m} ddlm} tj| d}tjd}d|_|j|d W d Q R X | j |jj
d d S ) Nr )rE )rN )curvez josepy.jwa.decode_dss_signaturer6 T)r r ) r rE ,cryptography.hazmat.primitives.asymmetric.ecrN r r7 patchr< r
r" r8 )r rE rN r decode_patchr r r r9 s zJWAECTest.test_sign_new_apic
C s ddl m} ddlm} tjdg| d}tj }||j_tjd}d |_|j |d W d Q R X | j
|jj d | j
|j j d | j
|j
j d d S )
Nr )rE )rN r: )r; rP z josepy.jwa.decode_dss_signaturer6 T)r r )r rE rQ rN r r7 r: r< rR r
r" r8 r= r> )r rE rN r r: rS r r r r? s zJWAECTest.test_sign_old_apic C sf dd l }ddlm} ddlm} tjd| d}|j|dd|j|j d d | j
|jj d
d S ) Nr )rE )rN )key_sizerP r6 T) mathr rE rQ rN r r7 r ceilrU r" r8 )r rY rE rN r r r r rA s "z JWAECTest.test_verify_new_apic C s dd l }ddlm} ddlm} tjdgd| d}tj }||j_|j |dd|j
|j d d
| j |jj
d | j |jj
d | j |j j
d d S ) Nr )rE ) SECP521R1rB i )r; rU rP r6 rV rW rX T)rY r rE rQ r[ r r7 rB r< r rZ rU r" r8 r= )r rY rE r[ r rB r r r rC s "z JWAECTest.test_verify_old_apic C sl ddl m} ddlm} |jdddddd }tjd
*}d|_|j|j d }| j
t |d W d Q R X d S )Nr )ES512)JWKzXAf9KP6DqLRbtit6NS_LRIaCP_-NdC5l5R2ugbILdfpv6dS9R4wUPNxiGw-vVWumA56Yo1oBnEm8ZdR4W-u1lPHw5zXAD4i4STyJ07iZJkHkpKEOuICpn6IHknzwAlrf-1w1a5dqOsRe30EECSN4vFxaeAmtdBSCKBwCq7h1q4bPgMrMUvFZXAHAlXxrabjcx_yBxGObnm_DkEQMJK1E69OHY3x3VxF5VXoKc93CG4GLoaPvphZQvZnt5EfExQoPktwOMIVhBHaFRzP-521ZEC)dxyZcrvZktyz josepy.jwa.decode_dss_signatures testrX B )r r ) r r\ Z
josepy.jwkr] r# r rR r< r
r r len)r r\ r] r rS r r r r test_signature_size s
z JWAECTest.test_signature_sizeN)
r r r r/ rG rJ rK rO r9 r? rA rC rd r r r r rD s
rD __main__)r% Zunittestr Zjosepyr r Zload_rsa_private_keyr1 r- r4 Zload_ec_private_keyrF rI Z EC_P521_KEYZTestCaser r& r* rD r mainr r r r s
*
A^