← Back
Editing: 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$
Save File
Cancel