3
�}:a7# � @ s� d Z ddlZddlmZ ddlmZmZ ejd�Zejd�Zejd�Z ej
d�Z ej
d �Z ej
d
�Z
G d d � d ej�ZG d
d� dej�ZG dd� dej�ZG dd� dej�Zedkr�ej� 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.<locals>.MockSigc S s
t � �d S )N)�NotImplementedError)�self�key�msg� r � /usr/lib/python3.6/jwa_test.py�sign s z,JWASignatureTest.setUp.<locals>.MockSig.signc S s
t � �d S )N)r )r r r
�sigr r r �verify s z.JWASignatureTest.setUp.<locals>.MockSig.verifyN)�__name__�
__module__� __qualname__r
r r r r r �MockSig s r �Sig1�Sig2)�
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&}��O�t��"Y�Y��)� ��� &�Ǚ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 �update�finalize)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)�d�x�yZcrvZktyz 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 �<module> s
*
A^