3
�}:a� � @ s� d Z ddlZddlZddlmZ G dd� dej�ZG dd� dej�ZG dd � d ej�ZG d
d � d ej�Z G d d
� d
ej�Z
e dkr�ej � dS )zTests for josepy.util.� N)� test_utilc @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )�ComparableX509Testz%Tests for josepy.util.ComparableX509.c C sL t jd�| _t jd�| _t jd�| _t jd�| _t jd�| _t jd�| _d S )Nzcsr.pemz csr-san.pemzcert.pemz cert-san.pem) r Zload_comparable_csr�req1�req2� req_otherZload_comparable_cert�cert1�cert2�
cert_other)�self� r � /usr/lib/python3.6/util_test.py�setUp s zComparableX509Test.setUpc C s | j | jj� d� d S )NT)�assertIsr Z has_expired)r
r r r �test_getattr_proxy s z%ComparableX509Test.test_getattr_proxyc C s$ | j | j| j� | j | j| j� d S )N)� assertEqualr r r r )r
r r r �test_eq s zComparableX509Test.test_eqc C s$ | j | j| j� | j | j| j� d S )N)�assertNotEqualr r r r )r
r r r �test_ne s zComparableX509Test.test_nec C s | j | jd� | j | jd� d S )N� )r r r )r
r r r �test_ne_wrong_types! s z&ComparableX509Test.test_ne_wrong_typesc C sd | j t| j�t| j�� | jt| j�t| j�� | j t| j�t| j�� | jt| j�t| j�� d S )N) r �hashr r r r r r r )r
r r r � test_hash% s z ComparableX509Test.test_hashc C s2 x,| j | jfD ] }| jt|�dj|j�� qW d S )Nz<ComparableX509({0!r})>)r r r �repr�format�wrapped)r
Zx509r r r � test_repr, s
z ComparableX509Test.test_reprN) �__name__�
__module__� __qualname__�__doc__r
r r r r r r r r r r r s
r c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )�ComparableRSAKeyTestz'Tests for josepy.util.ComparableRSAKey.c C s( t jd�| _t jd�| _t jd�| _d S )Nzrsa256_key.pemzrsa512_key.pem)r Zload_rsa_private_key�key�key_same�key2)r
r r r r
5 s zComparableRSAKeyTest.setUpc C s | j d| jj� d S )N� )r r! �key_size)r
r r r r ; s z'ComparableRSAKeyTest.test_getattr_proxyc C s | j | j| j� d S )N)r r! r" )r
r r r r > s z ComparableRSAKeyTest.test_eqc C s | j | j| j� d S )N)r r! r# )r
r r r r A s z ComparableRSAKeyTest.test_nec C s | j | jd� d S )Nr )r r! )r
r r r �test_ne_different_typesD s z,ComparableRSAKeyTest.test_ne_different_typesc C s | j | j| jj� d S )N)r r! r" �_wrapped)r
r r r �test_ne_not_wrappedG s z(ComparableRSAKeyTest.test_ne_not_wrappedc C s$ ddl m} | j|d�|d�� d S )Nr )�ComparableRSAKeyr )� josepy.utilr) r )r
r) r r r �test_ne_no_serializationK s z-ComparableRSAKeyTest.test_ne_no_serializationc C sF | j t| j�t� | jt| j�t| j�� | jt| j�t| j�� d S )N)�assertIsInstancer r! �intr r" r r# )r
r r r r O s z ComparableRSAKeyTest.test_hashc C s | j t| j�jd�d� d S )Nz'<ComparableRSAKey(<cryptography.hazmat.T)r r r! �
startswith)r
r r r r T s z ComparableRSAKeyTest.test_reprc C s" ddl m} | j| jj� |� d S )Nr )r) )r* r) r, r! �
public_key)r
r) r r r �test_public_keyX s z$ComparableRSAKeyTest.test_public_keyN)r r r r r
r r r r& r( r+ r r r0 r r r r r 2 s r c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )�ComparableECKeyTestz&Tests for josepy.util.ComparableECKey.c C s4 t jd�| _t jd�| _t jd�| _t jd�| _d S )Nzec_p256_key.pemzec_p384_key.pemzec_p521_key.pem)r Zload_ec_private_key�p256_key�
p256_key_same�p384_key�p521_key)r
r r r r
` s zComparableECKeyTest.setUpc C s | j d| jj� d S )Nr$ )r r2 r% )r
r r r r g s z&ComparableECKeyTest.test_getattr_proxyc C s | j | j| j� d S )N)r r2 r3 )r
r r r r j s zComparableECKeyTest.test_eqc C s$ | j | j| j� | j | j| j� d S )N)r r2 r4 r5 )r
r r r r m s zComparableECKeyTest.test_nec C s | j | jd� d S )Nr )r r2 )r
r r r r& q s z+ComparableECKeyTest.test_ne_different_typesc C s | j | j| jj� d S )N)r r2 r3 r' )r
r r r r( t s z'ComparableECKeyTest.test_ne_not_wrappedc C s$ ddl m} | j|d�|d�� d S )Nr )�ComparableECKeyr )r* r6 r )r
r6 r r r r+ x s z,ComparableECKeyTest.test_ne_no_serializationc C s^ | j t| j�t� | jt| j�t| j�� | jt| j�t| j�� | jt| j�t| j�� d S )N) r, r r2 r- r r3 r r4 r5 )r
r r r r | s z ComparableECKeyTest.test_hashc C s | j t| j�jd�d� d S )Nz&<ComparableECKey(<cryptography.hazmat.T)r r r2 r. )r
r r r r � s z ComparableECKeyTest.test_reprc C s" ddl m} | j| jj� |� d S )Nr )r6 )r* r6 r, r2 r/ )r
r6 r r r r0 � s z#ComparableECKeyTest.test_public_keyN)r r r r r
r r r r& r( r+ r r r0 r r r r r1 ] s r1 c @ sp e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )�ImmutableMapTestz#Tests for josepy.util.ImmutableMap.c C s| ddl m} G dd� d|�}G dd� d|�}|| _|| _| jddd �| _| jddd
�| _| jd d d �| _| jddd �| _d S )
Nr )� ImmutableMapc @ s e Zd ZdZdS )z!ImmutableMapTest.setUp.<locals>.A�x�yN)r9 r: )r r r � __slots__r r r r �A� s r< c @ s e Zd ZdZdS )z!ImmutableMapTest.setUp.<locals>.Br9 r: N)r9 r: )r r r r; r r r r �B� s r= � � )r9 r: )r: r9 � � )r* r8 r< r= �a1�a1_swap�a2�b)r
r8 r< r= r r r r
� s zImmutableMapTest.setUpc C s> | j | jddd�| jjdd�� | j | j| jjddd�� d S )Nr? )r9 r: )r9 r@ rA )r r< rB �updaterD )r
r r r � test_update� s z ImmutableMapTest.test_updatec C s | j t| jjd� d S )N�z)� assertRaises�KeyErrorrB � __getitem__)r
r r r �&test_get_missing_item_raises_key_error� s z7ImmutableMapTest.test_get_missing_item_raises_key_errorc C s | j | j| j� d S )N)r rB rC )r
r r r �"test_order_of_args_does_not_matter� s z3ImmutableMapTest.test_order_of_args_does_not_matterc C s( | j t| jdd� | j t| jdd� d S )Nr> )r9 r? )r: )rI � TypeErrorr< )r
r r r �test_type_error_on_missing� s z+ImmutableMapTest.test_type_error_on_missingc C s. | j t| jddd� | j t| jdddd� d S )Nr> r? )r9 rH r@ )r9 r: rH )rI rN r< )r
r r r � test_type_error_on_unrecognized� s z0ImmutableMapTest.test_type_error_on_unrecognizedc C sD | j d| jj� | j d| jj� | j d| jj� | j d| jj� d S )Nr> r? )r rB r9 r: rC )r
r r r �
test_get_attr� s z ImmutableMapTest.test_get_attrc C s | j ttj| jjd�d� d S )Nr9 �
)rI �AttributeError� functools�partialrB � __setattr__)r
r r r �$test_set_attr_raises_attribute_error� s z5ImmutableMapTest.test_set_attr_raises_attribute_errorc C s4 | j | j| j� | j | j| j� | j| j| j� d S )N)r rB rD r )r
r r r �
test_equal� s zImmutableMapTest.test_equalc C s | j td�t| j�� d S )Nr> r? )r> r? )r r rB )r
r r r r � s zImmutableMapTest.test_hashc C s | j t| jdi d�j� d S )Nr> )r9 r: )rI rN r< �__hash__)r
r r r �test_unhashable� s z ImmutableMapTest.test_unhashablec C sT | j dt| j�� | j dt| j�� | j dt| j�� | j dt| jddd��� d S )Nz A(x=1, y=2)z B(x=1, y=2)zB(x='foo', y='bar')ZfooZbar)r9 r: )r r rB rC rE r= )r
r r r r � s zImmutableMapTest.test_reprN)r r r r r
rG rL rM rO rP rQ rW rX r rZ r r r r r r7 � s r7 c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dS )�frozendictTestz!Tests for josepy.util.frozendict.c C s ddl m} |ddd�| _d S )Nr )�
frozendictr> �2)r9 r: )r* r\ �fdict)r
r\ r r r r
� s zfrozendictTest.setUpc C s( ddl m} | j| j|ddd��� d S )Nr )r\ r> r] )r9 r: )r* r\ r r^ )r
r\ r r r �test_init_dict� s z frozendictTest.test_init_dictc C s& ddl m} | jt|ddij� � d S )Nr )r\ �arE )r* r\ rI rN �items)r
r\ r r r �!test_init_other_raises_type_error� s z0frozendictTest.test_init_other_raises_type_errorc C s | j dt| j�� d S )Nr? )r �lenr^ )r
r r r �test_len� s zfrozendictTest.test_lenc C s | j t| j�t� d S )N)r, r r^ r- )r
r r r r � s zfrozendictTest.test_hashc C s$ | j d| jj� | j d| jj� d S )Nr> r] )r r^ r9 r: )r
r r r r � s z!frozendictTest.test_getattr_proxyc C s | j t| jjd� d S )NrH )rI rS r^ � __getattr__)r
r r r �#test_getattr_raises_attribute_error� s z2frozendictTest.test_getattr_raises_attribute_errorc C s | j t| jjdd� d S )NrH r@ )rI rS r^ rV )r
r r r �test_setattr_immutable� s z%frozendictTest.test_setattr_immutablec C s | j dt| j�� d S )Nzfrozendict(x=1, y='2'))r r r^ )r
r r r r � s zfrozendictTest.test_reprN)
r r r r r
r_ rb rd r r rf rg r r r r r r[ � s r[ �__main__)
r rT ZunittestZjosepyr ZTestCaser r r1 r7 r[ r �mainr r r r �<module> s )+.D$