File: //lib/python3/dist-packages/josepy/__pycache__/jws_test.cpython-38.pyc
U
��]H! � @ s� d Z ddlZddlZddlZddlZddlmZmZmZm Z m
Z
e
� d�Z e j
�e
�d��ZG dd� dej�ZG dd � d ej�ZG d
d � d ej�ZG d d
� d
ej�ZG dd� dej�Zedkr�e�� dS )zTests for josepy.jws.� N)�errors� json_util�jwa�jwk� test_utilzcert.pem�rsa512_key.pemc @ s e Zd ZdZdd� Zdd� ZdS )�
MediaTypeTestz Tests for josepy.jws.MediaType.c C sF ddl m} | �d|�d�� | �d|�d�� | �tj|jd� d S )Nr �� MediaType�application/app�appzapp;foo)�
josepy.jwsr
� assertEqual�decode� assertRaisesr �DeserializationError��selfr
� r �1/usr/lib/python3/dist-packages/josepy/jws_test.py� test_decode s �zMediaTypeTest.test_decodec C s4 ddl m} | �d|�d�� | �d|�d�� d S )Nr r r r zapplication/app;foo)r
r
r �encoder 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�| _|dd�| _|� | _d S )Nr ��Header�foo)r �bar��a�b)�crit)r
r �header1�header2r% �empty�r r r r r �setUp" s
zHeaderTest.setUpc C s, ddl m} | �|ddd�| j| j � d S )Nr 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 � d S �N)r r& r( �r r r r �test_add_empty. s zHeaderTest.test_add_emptyc C s | � t| jj| j� d S r, )r � TypeErrorr& �__add__r' r- r r r �test_add_overlapping_error2 s z%HeaderTest.test_add_overlapping_errorc C s | � t| jjd� d S )NZxxx)r r/ r& r0 r- r r r �test_add_wrong_type_error5 s z$HeaderTest.test_add_wrong_type_errorc C s* ddl m} | �tj|jdddgi� d S )Nr r r% r# r$ )r
r r r r � from_jsonr) r r r � test_crit_decode_always_errors8 s
�z)HeaderTest.test_crit_decode_always_errorsc C s� ddl m} |ttfd�}|�� }tj�tjjtj�}t �
|�}| � |d||gi� | � ||� |�� t �
d| �|d d< | �
tj|j |� d S )Nr r )�x5cr5 s xxx)r
r �CERT�to_partial_json�OpenSSLZcryptoZdump_certificateZ
FILETYPE_ASN1�wrapped�base64Z b64encoder r3 r r r )r r �headerZjobjZ cert_asn1Zcert_b64r r r �test_x5c_decoding= s �
z HeaderTest.test_x5c_decodingc C s: | � d| j�� � | � d| j�� � | �tj| 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. r1 r2 r4 r< r> 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} | �|d|tjd�d�|�ddd id��� d S )
Nr r �� Signature� foo)�alg)� signaturer; ZZm9vrC �RS256)r
r rA r r rE r3 )r r rA r r r �test_from_jsonR s ��z SignatureTest.test_from_jsonc C s& ddl m} | �tj|jddi� d S )Nr r@ rD r )r
rA r r r r3 )r rA r r r �test_from_json_no_alg_errorZ s
�z)SignatureTest.test_from_json_no_alg_errorN)r r r r rF rG r r r r r? O s r? 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 | _| j�� | _ddlm} |jd| jtjd�| _ |jd| jtjt
ddg�d�| _ |jd| jtjt
dg�d�| _ d S )Nr ��JWSrB )�payload�keyrC r rC )rK rL rC Zprotect)
�KEYZprivkeyZ
public_key�pubkeyr
rJ �signr rE � unprotected� frozenset� protected�mixed�r rJ r r r r* c s( �
� �z
JWSTest.setUpc C sF | � | jjjj| j� | � | jjjj| j� | � | jjjj| j� d S r, )r rP rD �combinedr rN rR rS r- r r r �test_pubkey_jwkq s zJWSTest.test_pubkey_jwkc C s | � | j�� � d S r, )�
assertTruerP �verifyr- r r r �test_sign_unprotectedv s z JWSTest.test_sign_unprotectedc C s | � | j�� � d S r, )rW rR rX r- r r r �test_sign_protectedy s zJWSTest.test_sign_protectedc C s | � | j�� � d S r, )rW rS rX r- r r r �test_sign_mixed| s zJWSTest.test_sign_mixedc C s\ | j �� }| �d|� ddlm} |�|�}| �| j |� | �tdg�t|jj �
� �� d S )Nsr eyJhbGciOiAiUlMyNTYifQ.Zm9v.OHdxFVj73l5LpxbFp1AmYX4yJM0Pyb_893n1zQjpim_eLS5J1F61lkvrCrCDErTEJnBGOGesJ72M7b6Ve1cAJAr rI rC ) rS Z
to_compactr r
rJ � from_compactZassertNotEqual�setrD rU Z not_omitted)r ZcompactrJ rS r r r � test_compact_lost_unprotected s
�
�z%JWSTest.test_compact_lost_unprotectedc C s" ddl m} | �tj|jd� d S )Nr rI � .)r
rJ r r r r\ rT r r r �$test_from_compact_missing_components� s z,JWSTest.test_from_compact_missing_componentsc C s� | j jdd�}| jjdd�}| �d|k� | �d|k� |d �� |d< ddlm} | �|�|�| j � | �|�|�| j� d S )NT�ZflatrR r; r rI ) rR r7 rP rW �to_jsonr
rJ r r3 )r Zprotected_jobjZunprotected_jobjrJ r r r �test_json_omitempty� s zJWSTest.test_json_omitemptyc C s� t �| jjj�t �d�| jjjt �| jjj�d��d�}|�� }|d �� |d< | � | jj
dd�|� ddl m } | � | j|�
|�� d S ) NrB zutf-8)rD rK r; rR r; Tra r rI )r �encode_b64joserS rD r; rR r �copyrb r r7 r
rJ r3 �r Zjobj_toZ jobj_fromrJ r r r �test_json_flat� s ��� zJWSTest.test_json_flatc C sn | j jft�d�d�}|�� }|d d �� g|d< | �| j jdd�|� ddlm } | �| j |�
|�� d S )NrB )�
signaturesrK rh r Fra rI ) rS rD r rd re rb r r7 r
rJ r3 rf r r r �test_json_not_flat� s � zJWSTest.test_json_not_flatc C s( ddl m} | �tj|jddd�� d S )Nr rI r r )rh rD )r
rJ r r r r3 rT r r r �test_from_json_mixed_flat� s �z!JWSTest.test_from_json_mixed_flatc C s$ ddl m} t|�| j�� �� d S )Nr rI )r
rJ �hashr3 rS rb rT r r r �test_from_json_hashable� s z JWSTest.test_from_json_hashableN)r r r r r* rV rY rZ r[ r^ r` rc rg ri rj rl r r r r rH ` s
rH c @ s, e Zd Zdd� Zdd� Zdd� Zdd� Zd S )
�CLITestc C s t �d�| _d S )Nr )r Z vector_path�key_pathr- r r r r* � s z
CLITest.setUpc
C sX ddl m} t�d��8}d|j_t�d�� | �d|�dg�� W 5 Q R X W 5 Q R X d S )Nr ��CLI� sys.stdinz&{"payload": "foo", "signature": "xxx"}�
sys.stdout���rX )r
rp �mock�patch�read� return_valuer �run)r rp �sinr r r �test_unverified� s
zCLITest.test_unverifiedc C s� ddl m} t�d��j}d|j_t�d��L}|�dd| jdd d
d g� |jj d d d |j_| �
d|�d
g�� W 5 Q R X W 5 Q R X d S )Nr ro rq r rr rO �-kz-arE z-pr � rX � r
rp rt ru rv rw rx rn �writeZ
mock_callsr �r rp ry Zsoutr r r � test_json� s �zCLITest.test_jsonc
C s� ddl m} t�d��p}d|j_t�d��R}|�ddd| jg� |jj d d d |j_| �
d|�dd
d d d| jg�� W 5 Q R X W 5 Q R X d S )
Nr ro rq r rr z --compactrO r{ r| rX z--ktyZRSAr} r r r r � test_compact� s
�zCLITest.test_compactN)r r r r* rz r� r� r r r r rm � s rm �__main__)r r: Zunittestrt r8 Zjosepyr r r r r Zload_comparable_certr6 ZJWKRSA�loadZ load_vectorrM ZTestCaser r r? rH rm r �mainr r r r �<module> s
0d$