File: //lib/python3/dist-packages/josepy/__pycache__/jwa_test.cpython-38.pyc
U
��]- � @ s� d Z ddlZddlZddlmZmZ e�d�Ze�d�Ze�d�Z G dd� dej
�Z G d d
� d
ej
�Z G d d � d ej
�Z
ed
kr�e�� dS )zTests for josepy.jwa.� N)�errors� test_utilzrsa256_key.pemzrsa512_key.pemzrsa1024_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
�1/usr/lib/python3/dist-packages/josepy/jwa_test.py�sign s z,JWASignatureTest.setUp.<locals>.MockSig.signc S s
t � �d S r 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� d S r )� assertEqualr �r
r
r
r �test_eq s zJWASignatureTest.test_eqc C s | � | j| j� d S r )�assertNotEqualr r r r
r
r �test_ne# s zJWASignatureTest.test_nec C s | � | jd� d S )N� )r r r r
r
r �test_ne_other_type& s z#JWASignatureTest.test_ne_other_typec C s( | � dt| j�� | � dt| j�� d S �Nr r )r �reprr r r r
r
r � test_repr) s zJWASignatureTest.test_reprc C s( | � | j�� d� | � | j�� d� d S r! )r r Zto_partial_jsonr r r
r
r �test_to_partial_json- s z%JWASignatureTest.test_to_partial_jsonc C s0 ddl m} ddl m} | �|�d�|k� d S )Nr r ��RS256r& )r r r& �
assertTrueZ from_json)r
r r& r
r
r �test_from_json1 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 s\ ddl m} d}| �|�dd�|� | �|�dd|�dk� | �|�dd|d �dk� d S ) Nr )�HS256s �R�͔���A.:'i��
�+��UZԳ�zOs some key� fooT� !F)r r+ r r r' r )r
r+ r r
r
r �test_it9 s �zJWAHSTest.test_itN)r r r r. r
r
r
r r* 7 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} | �tj|jt�� d� d S )Nr r% r, )r r& � assertRaisesr �Errorr �
RSA512_KEY�
public_key)r
r& r
r
r �test_sign_no_private_partF s �z#JWARSTest.test_sign_no_private_partc C sD ddl m} ddl m} | �tj|jtd� | �tj|jtd� d S )Nr r% ��PS256r, )r r& r6 r0 r r1 r �
RSA256_KEY)r
r& r6 r
r
r �test_sign_key_too_smallK s z!JWARSTest.test_sign_key_too_smallc C s\ ddl m} d}| �|�td�|� | �|�t�� d|�� | �|�t�� d|d �� d S )Nr r% s@ |Ʋ��(���*:�N&}��O�t��"Y�Y��)� ��� &�ǙS��� OV��ҹ.>}�r, r- ) r r& r r r2 r' r r3 � assertFalse)r
r& r r
r
r �test_rsQ s � �zJWARSTest.test_rsc C sP ddl m} |�td�}| �|�t�� d|�� | �|�t�� d|d �� d S )Nr r5 r, r- )r r6 r � RSA1024_KEYr' r r3 r9 )r
r6 r r
r
r �test_ps^ s �zJWARSTest.test_psc C s2 ddl m} t�� }|�|d� | �|jj� d S )Nr r% �message)r r& �mock� MagicMockr r' �called�r
r& r r
r
r �test_sign_new_apie s zJWARSTest.test_sign_new_apic C sZ ddl m} tjdgd�}t�� }||j_|�|d� | �t|jj |j
j |j j g�� d S )Nr r% �signer��specr= ) r r& r> r? rC � return_valuer r' �allr@ �update�finalize)r
r& r rC r
r
r �test_sign_old_apik s �zJWARSTest.test_sign_old_apic C s4 ddl m} t�� }|�|dd� | �|jj� d S )Nr r% r= � signature)r r& r> r? r r' r@ rA r
r
r �test_verify_new_apiv s z JWARSTest.test_verify_new_apic C s\ ddl m} tjdgd�}t�� }||j_|�|dd� | �t|jj |j
j |jj g�� d S )Nr r% �verifierrD r= rK ) r r& r> r? rM rF r r' rG r@ rH )r
r& r rM r
r
r �test_verify_old_api| s �z JWARSTest.test_verify_old_apiN) r r r r4 r8 r: r< rB rJ rL rN r
r
r
r r/ D s
r/ �__main__)r) Zunittestr> Zjosepyr r Zload_rsa_private_keyr7 r2 r; ZTestCaser r* r/ r �mainr
r
r
r �<module> s
*
D