← Back
Editing: util_test.cpython-38.pyc
U ��]� � @ s� d Z ddlZddlZddlZddlmZ G dd� dej�ZG dd� dej�ZG dd � d ej�Z G d d � d ej�Z e d kr�e� � 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 �d�| _t �d�| _t �d�| _t �d�| _t �d�| _t �d�| _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 �2/usr/lib/python3/dist-packages/josepy/util_test.py�setUp s zComparableX509Test.setUpc C s | � | j�� � d S �N)� assertTruer Z has_expiredr r r r �test_getattr_proxy s z%ComparableX509Test.test_getattr_proxyc C s$ | � | j| j� | � | j| j� d S r )� assertEqualr r r r r r r r �test_eq s zComparableX509Test.test_eqc C s$ | � | j| j� | � | j| j� d S r )�assertNotEqualr r r r r r r r �test_ne s zComparableX509Test.test_nec C s | � | jd� | � | jd� d S �N� )r r r r r r r �test_ne_wrong_types" s z&ComparableX509Test.test_ne_wrong_typesc C sd | � t| j�t| j�� | �t| j�t| j�� | � t| j�t| j�� | �t| j�t| j�� d S r ) r �hashr r r r r r r r r r r � test_hash& s z ComparableX509Test.test_hashc C s. | j | jfD ] }| �t|�d�|j�� q 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 �d�| _t �d�| _t �d�| _d S )Nzrsa256_key.pemzrsa512_key.pem)r Zload_rsa_private_key�key�key_same�key2r r r r r 6 s zComparableRSAKeyTest.setUpc C s | � d| jj� d S )N� )r r$ Zkey_sizer r r r r < s z'ComparableRSAKeyTest.test_getattr_proxyc C s | � | j| j� d S r )r r$ r% r r r r r ? s z ComparableRSAKeyTest.test_eqc C s | � | j| j� d S r )r r$ r& r r r r r B s z ComparableRSAKeyTest.test_nec C s | � | jd� d S r )r r$ r r r r �test_ne_different_typesE s z,ComparableRSAKeyTest.test_ne_different_typesc C s | � | j| jj� d S r )r r$ r% Z_wrappedr r r r �test_ne_not_wrappedH s z(ComparableRSAKeyTest.test_ne_not_wrappedc C s$ ddl m} | �|d�|d�� d S )Nr ��ComparableRSAKeyr )� josepy.utilr+ r �r r+ r r r �test_ne_no_serializationL s z-ComparableRSAKeyTest.test_ne_no_serializationc C sJ | � tt| j�t�� | �t| j�t| j�� | �t| j�t| j�� d S r ) r � isinstancer r$ �intr r% r r& r r r r r P s z ComparableRSAKeyTest.test_hashc C s | � t| j��d�� d S )Nz'<ComparableRSAKey(<cryptography.hazmat.)r r r$ � startswithr r r r r U s �z ComparableRSAKeyTest.test_reprc C s& ddl m} | �t| j�� |�� d S )Nr r* )r, r+ r r/ r$ Z public_keyr- r r r �test_public_keyY s z$ComparableRSAKeyTest.test_public_keyN)r r r! r" r r r r r( r) r. r r r2 r r r r r# 3 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 )�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�r r r! � __slots__r r r r �Af s r: c @ s e Zd ZdZdS )z!ImmutableMapTest.setUp.<locals>.Br5 Nr8 r r r r �Bi s r; � � r5 )r7 r6 � � )r, r4 r: r; �a1�a1_swap�a2�b)r r4 r: r; r r r r a s zImmutableMapTest.setUpc C s> | � | jddd�| jjdd�� | � | j| jjddd�� d S )Nr= r5 �r6 r> r? )r r: r@ �updaterB r r r r � test_updatet s z ImmutableMapTest.test_updatec C s | � t| jjd� d S �N�z)� assertRaises�KeyErrorr@ � __getitem__r r r r �&test_get_missing_item_raises_key_errorx s z7ImmutableMapTest.test_get_missing_item_raises_key_errorc C s | � | j| j� d S r )r r@ rA 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< rD r= )r7 �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= )r6 rH r> )r6 r7 rH rN r r r r � test_type_error_on_unrecognized� s z0ImmutableMapTest.test_type_error_on_unrecognizedc C sD | � d| jj� | � d| jj� | � d| jj� | � d| jj� d S )Nr< r= )r r@ r6 r7 rA r r r r � test_get_attr� s z ImmutableMapTest.test_get_attrc C s | � tt�| jjd�d� d S )Nr6 � )rI �AttributeError� functools�partialr@ � __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� d S r )r r@ rB r r r r r � test_equal� s zImmutableMapTest.test_equalc C s | � td�t| j�� d S )N)r< r= )r r r@ r r r r r � s zImmutableMapTest.test_hashc C s | � t| jdi d�j� d S )Nr< r5 )rI rO r: �__hash__r r r r �test_unhashable� s z ImmutableMapTest.test_unhashablec C sT | � dt| j�� | � dt| j�� | � dt| j�� | � dt| jddd��� d S )Nz A(x=1, y=2)z B(x=1, y=2)zB(x='foo', y='bar')ZfooZbarr5 )r r r@ rA rC r; r r r r r � s zImmutableMapTest.test_reprN)r r r! r" r rF rL rM rP rQ rR rX rY r r[ r r r r r r3 ^ s r3 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< �2r5 )r, r_ �fdict�r r_ r r r r � s zfrozendictTest.setUpc C s( ddl m} | �| j|ddd��� d S r] )r, r_ r ra rb r r r �test_init_dict� s z frozendictTest.test_init_dictc C s( ddl m} | �t|t�ddi�� d S )Nr r^ �arC )r, r_ rI rO �sixZ iteritemsrb r r r �!test_init_other_raises_type_error� s z0frozendictTest.test_init_other_raises_type_errorc C s | � dt| j�� d S )Nr= )r �lenra r r r r �test_len� s zfrozendictTest.test_lenc C s | � tt| j�t�� d S r )r r/ r ra r0 r r r r r � s zfrozendictTest.test_hashc C s$ | � d| jj� | � d| jj� d S )Nr< r` )r ra r6 r7 r r r r r � s z!frozendictTest.test_getattr_proxyc C s | � t| jjd� d S rG )rI rT ra � __getattr__r r r r �#test_getattr_raises_attribute_error� s z2frozendictTest.test_getattr_raises_attribute_errorc C s | � t| jjdd� d S )NrH r> )rI rT ra rW r r r r �test_setattr_immutable� s z%frozendictTest.test_setattr_immutablec C s | � dt| j�� d S )Nzfrozendict(x=1, y='2'))r r ra r r r r r � s zfrozendictTest.test_reprN) r r r! r" r rc rf rh r r rj rk r r r r r r\ � s r\ �__main__) r" rU Zunittestre Zjosepyr ZTestCaser r# r3 r\ r �mainr r r r �<module> s )+D$
Save File
Cancel