3
�}:aF! � @ s� d Z ddlZddlZddlmZ ddlZddlmZmZmZm Z m
Z
e
j d�Z e j
je
jd��ZG dd� dej�ZG d d
� d
ej�ZG d d � d ej�ZG d
d� dej�ZG dd� dej�Zedkr�ej� dS )zTests for josepy.jws.� N)�mock)�errors� json_util�jwa�jwk� test_utilzcert.pemzrsa512_key.pemc @ s e Zd ZdZdd� Zdd� ZdS )�
MediaTypeTestz Tests for josepy.jws.MediaType.c C sF ddl m} | jd|jd�� | jd|jd�� | jtj|jd� d S )Nr )� MediaTypezapplication/app�appzapp;foo)�
josepy.jwsr � assertEqual�decode� assertRaisesr �DeserializationError)�selfr � r � /usr/lib/python3.6/jws_test.py� test_decode s
zMediaTypeTest.test_decodec C s4 ddl m} | jd|jd�� | jd|jd�� d S )Nr )r r
zapplication/appzapplication/app;foo)r r r �encode)r r r r r � test_encode s zMediaTypeTest.test_encodeN)�__name__�
__module__� __qualname__�__doc__r r r r r r r s r c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )�
HeaderTestz Tests for josepy.jws.Header.c C s< ddl m} |dd�| _|dd�| _|d d�| _|� | _d S )
Nr )�Header�foo)r �bar�a�b)�crit)r r )r r �header1�header2r �empty)r r r r r �setUp" s
zHeaderTest.setUpc C s, ddl m} | j|ddd�| j| j � d S )Nr )r r r r )r r )r r )r r r r! r )r r r r r �test_add_non_empty) s z HeaderTest.test_add_non_emptyc C s0 | j | j| j| j � | j | j| j| j � d S )N)r r! r# )r r r r �test_add_empty. s zHeaderTest.test_add_emptyc C s | j t| jj| j� d S )N)r � TypeErrorr! �__add__r" )r r r r �test_add_overlapping_error2 s z%HeaderTest.test_add_overlapping_errorc C s | j t| jjd� d S )NZxxx)r r' r! r( )r r r r �test_add_wrong_type_error5 s z$HeaderTest.test_add_wrong_type_errorc C s* ddl m} | jtj|jdddgi� d S )Nr )r r r r )r r r r r � from_json)r r r r r � test_crit_decode_always_errors8 s z)HeaderTest.test_crit_decode_always_errorsc C s� ddl m} |ttfd�}|j� }tjjtjjtj�}t j
|�}| j |d||gi� | j ||j |�� t j
d| �|d d< | j
tj|j |� d S )Nr )r )�x5cr- s xxx)r r �CERT�to_partial_json�OpenSSLZcryptoZdump_certificateZ
FILETYPE_ASN1�wrapped�base64Z b64encoder r+ r r r )r r �headerZjobjZ cert_asn1Zcert_b64r r r �test_x5c_decoding= s
z HeaderTest.test_x5c_decodingc C s: | j d| jj� � | j d| jj� � | jtj| jj� d S )Nr r )r r! Zfind_keyr" r r �Errorr )r r r r �
test_find_keyI s zHeaderTest.test_find_keyN) r r r r r$ r% r& r) r* r, r4 r6 r r r r r s r c @ s e Zd ZdZdd� Zdd� ZdS )�
SignatureTestz Tests for josepy.jws.Signature.c C sH ddl m} ddl m} | j|d|tjd�d�|jddd id��� d S )
Nr )r )� Signatures foo)�alg)� signaturer3 ZZm9vr9 �RS256)r r r8 r r r; r+ )r r r8 r r r �test_from_jsonR s z SignatureTest.test_from_jsonc C s& ddl m} | jtj|jddi� d S )Nr )r8 r: r )r r8 r r r r+ )r r8 r r r �test_from_json_no_alg_errorZ s z)SignatureTest.test_from_json_no_alg_errorN)r r r r r< r= r r r r r7 O s r7 c @ sp e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )�JWSTestzTests for josepy.jws.JWS.c C sv t | _| jj� | _ddlm} |jd| jtjd�| _ |jd| jtjt
ddg�d�| _ |jd| jtjt
dg�d�| _ d S )Nr )�JWSs foo)�payload�keyr9 r r9 )r@ rA r9 Zprotect)
�KEYZprivkeyZ
public_key�pubkeyr r? �signr r; � unprotected� frozenset� protected�mixed)r r? r r r r$ c s
z
JWSTest.setUpc C sF | j | jjjj| j� | j | jjjj| j� | j | jjjj| j� d S )N)r rE r: �combinedr rC rG rH )r r r r �test_pubkey_jwkq s zJWSTest.test_pubkey_jwkc C s | j | jj� d� d S )NT)�assertIsrE �verify)r r r r �test_sign_unprotectedv s z JWSTest.test_sign_unprotectedc C s | j | jj� d� d S )NT)rK rG rL )r r r r �test_sign_protectedy s zJWSTest.test_sign_protectedc C s | j | jj� d� d S )NT)rK rH rL )r r r r �test_sign_mixed| s zJWSTest.test_sign_mixedc C sX | j j� }| jd|� ddlm} |j|�}| j| j |� | jdht|jj j
� �� d S )Nsr eyJhbGciOiAiUlMyNTYifQ.Zm9v.OHdxFVj73l5LpxbFp1AmYX4yJM0Pyb_893n1zQjpim_eLS5J1F61lkvrCrCDErTEJnBGOGesJ72M7b6Ve1cAJAr )r? r9 ) rH Z
to_compactr r r? � from_compactZassertNotEqual�setr: rI Z not_omitted)r Zcompactr? rH r r r � test_compact_lost_unprotected s
z%JWSTest.test_compact_lost_unprotectedc C s" ddl m} | jtj|jd� d S )Nr )r? � .)r r? r r r rP )r r? r r r �$test_from_compact_missing_components� s z,JWSTest.test_from_compact_missing_componentsc C s| | j jdd�}| jjdd�}| jd|� | jd|� |d j� |d< ddlm} | j|j|�| j � | j|j|�| j� d S )NT)�flatrG r3 r )r? ) rG r/ rE Z assertNotIn�to_jsonr r? r r+ )r Zprotected_jobjZunprotected_jobjr? r r r �test_json_omitempty� s zJWSTest.test_json_omitemptyc C s� t j| jjj�t jd�| jjjt j| jjjjd��d�}|j� }|d j� |d< | j | jj
dd�|� ddl m } | j | j|j
|�� d S ) Ns foozutf-8)r: r@ r3 rG r3 T)rU r )r? )r �encode_b64joserH r: r3 rG r �copyrV r r/ r r? r+ )r �jobj_to� jobj_fromr? r r r �test_json_flat� s
zJWSTest.test_json_flatc C sn | j jftjd�d�}|j� }|d d j� g|d< | j| j jdd�|� ddlm } | j| j |j
|�� d S )Ns foo)�
signaturesr@ r] r F)rU )r? ) rH r: r rX rY rV r r/ r r? r+ )r rZ r[ r? r r r �test_json_not_flat� s zJWSTest.test_json_not_flatc C s( ddl m} | jtj|jf dd�� d S )Nr )r? r )r] r: )r r? r r r r+ )r r? r r r �test_from_json_mixed_flat� s z!JWSTest.test_from_json_mixed_flatc C s$ ddl m} t|j| jj� �� d S )Nr )r? )r r? �hashr+ rH rV )r r? r r r �test_from_json_hashable� s z JWSTest.test_from_json_hashableN)r r r r r$ rJ rM rN rO rR rT rW r\ r^ r_ ra r r r r r> ` s
r> c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�CLITestc C s t jd�| _d S )Nzrsa512_key.pem)r Z vector_path�key_path)r r r r r$ � s z
CLITest.setUpc C sX ddl m} tjd��8}d|j_tjd�� | jd|jdg�� W d Q R X W d Q R X d S ) Nr )�CLIz sys.stdinz&{"payload": "foo", "signature": "xxx"}z
sys.stdout� rL ���)r rd r �patch�read� return_valuer �run)r rd �sinr r r �test_unverified� s
zCLITest.test_unverifiedc C s� ddl m} tjd��j}d|j_tjd��L}|jdd| jdd d
d g� |jj d d d |j_| j
d|jd
g�� W d Q R X W d Q R X d S )Nr )rd z sys.stdinr z
sys.stdoutrD z-kz-ar; z-pr re rL ) r rd r rg rh ri rj rc �write�
mock_callsr )r rd rk �soutr r r � test_json� s
zCLITest.test_jsonc C s� ddl m} tjd��p}d|j_tjd��R}|jddd| jg� |jj d d d |j_| j
d|jdd
d d d| jg�� W d Q R X W d Q R X d S )
Nr )rd z sys.stdinr z
sys.stdoutz --compactrD z-kre rL z--ktyZRSA) r rd r rg rh ri rj rc rm rn r )r rd rk ro r r r � test_compact� s
zCLITest.test_compactN)r r r r$ rl rp rq r r r r rb � s rb �__main__)r r2 Zunittestr r0 Zjosepyr r r r r Zload_comparable_certr. ZJWKRSA�loadZ load_vectorrB ZTestCaser r r7 r> rb r �mainr r r r �<module> s
0c$