3
�}:aV7 � @ s� d Z ddlZddlZddlmZ ddlmZmZmZmZ ej d�Z
ej d�Z G dd� dej
�ZG d d
� d
ej
�ZG d d � d ej
�ZG d
d� dej
�ZG dd� dej
�Zedkr�ej� dS )zTests for josepy.json_util.� N)�mock)�errors�
interfaces� test_util�utilzcert.pemzcsr.pemc @ sH e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dS )� FieldTestz!Tests for josepy.json_util.Field.c C sV ddl m} xDtjddgddgddg�D ]&\}}}| j|d||d�j|�d� q(W d S )Nr )�FieldTF�foo)�default� omitempty)�josepy.json_utilr � itertools�product�assertIsZomit)�selfr r
r �value� r �$/usr/lib/python3.6/json_util_test.py�test_no_omit_boolean s
z FieldTest.test_no_omit_booleanc C sz t j� }dd� }dd� }ddlm} |d�}|j|�}| jd|j|�� |j|�}| jd|j|�� | jd |j|�� d S )
Nc S s dS )N�dr )� unused_valuer r r �decoder s z+FieldTest.test_descriptors.<locals>.decoderc S s dS )N�er )r r r r �encoder s z+FieldTest.test_descriptors.<locals>.encoderr )r r r r ) r � MagicMockr r r � assertEqual�encoder �decode)r �
mock_valuer r r �fieldr r r �test_descriptors s
zFieldTest.test_descriptorsc C sL G dd� dt j�}|� }ddlm} | j|j|�|� | jd|j|�� d S )Nc @ s e Zd Zdd� Zedd� �ZdS )z<FieldTest.test_default_encoder_is_partial.<locals>.MockFieldc S s dS )Nr r )r r r r �to_partial_json. s zLFieldTest.test_default_encoder_is_partial.<locals>.MockField.to_partial_jsonc S s d S )Nr )�cls�jobjr r r � from_json1 s zFFieldTest.test_default_encoder_is_partial.<locals>.MockField.from_jsonN)�__name__�
__module__� __qualname__r! � classmethodr$ r r r r � MockField, s r) r )r r )r ZJSONDeSerializabler r r �default_encoderZassertNotEqual)r r) Z
mock_fieldr r r r � test_default_encoder_is_partial+ s
z)FieldTest.test_default_encoder_is_partialc C s* t j� }ddlm} | j|j|�|� d S )Nr )r )r r r r r r* )r r r r r r � test_default_encoder_passthrough; s z*FieldTest.test_default_encoder_passthroughc C s( ddl m} | jd|jdddg�� d S )Nr )r � � � )r- r. r/ )r r r �default_decoder)r r r r r �"test_default_decoder_list_to_tuple@ s z,FieldTest.test_default_decoder_list_to_tuplec C s@ ddl m} |jddi�}| j|tj� | j|tjdd�� d S )Nr )r �xr. )r2 )r r r0 �assertIsInstancer Z
frozendictr )r r �objr r r �'test_default_decoder_dict_to_frozendictD s z1FieldTest.test_default_decoder_dict_to_frozendictc C s* t j� }ddlm} | j|j|�|� d S )Nr )r )r r r r r r0 )r r r r r r � test_default_decoder_passthroughJ s z*FieldTest.test_default_decoder_passthroughN) r% r& r' �__doc__r r r+ r, r1 r5 r6 r r r r r s r c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� JSONObjectWithFieldsMetaTestz4Tests for josepy.json_util.JSONObjectWithFieldsMeta.c s~ ddl m} ddl m} |d�� _|d�� _G � fdd�dt|d�}G d d
� d
|�}G � fd d �d |�}|� _|� _|� _d S )
Nr )r )�JSONObjectWithFieldsMetaZBazZBaz2c s e Zd ZdZ� jZdS )z-JSONObjectWithFieldsMetaTest.setUp.<locals>.A�barN)r: )r% r& r' � __slots__r �bazr )r r r �A[ s r= )� metaclassc @ s e Zd ZdS )z-JSONObjectWithFieldsMetaTest.setUp.<locals>.BN)r% r& r' r r r r �B_ s r? c s e Zd Z� jZdS )z-JSONObjectWithFieldsMetaTest.setUp.<locals>.CN)r% r& r' �field2r< r )r r r �Cb s rA ) r r r9 r r@ �object�a_cls�b_cls�c_cls)r r r9 r= r? rA r )r r �setUpS s
z"JSONObjectWithFieldsMetaTest.setUpc C s0 | j d| ji| jj� | j d| ji| jj� d S )Nr< )r r rC �_fieldsrD )r r r r � test_fieldsi s z(JSONObjectWithFieldsMetaTest.test_fieldsc C s | j d| ji| jj� d S )Nr< )r r@ rE rG )r r r r �test_fields_inheritancen s z4JSONObjectWithFieldsMetaTest.test_fields_inheritancec C s$ | j d| jj� | j d| jj� d S )Nr: r< )r: r< )r< )r rC r; rD )r r r r �
test_slotsr s z'JSONObjectWithFieldsMetaTest.test_slotsc C s$ | j d| jj� | j f | jj� d S )Nr: )r: )r rC Z _orig_slotsrD )r r r r �test_orig_slotsv s z,JSONObjectWithFieldsMetaTest.test_orig_slotsN) r% r& r' r7 rF rH rI rJ rK r r r r r8 P s r8 c @ sp e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )�JSONObjectWithFieldsTestz0Tests for josepy.json_util.JSONObjectWithFields.c sF ddl m} ddl m� G � fdd�d|�}|| _|d ddd�| _d S ) Nr )�JSONObjectWithFields)r c sP e Zd Z� dddd� dd� d�Z� d�Z� d�Zejd d
� �Zejd d
� �Zd S )
z@JSONObjectWithFieldsTest.setUp.<locals>.MockJSONObjectWithFieldsr2 Tc S s | d S )Nr. r )r2 r r r �<lambda>� s zIJSONObjectWithFieldsTest.setUp.<locals>.MockJSONObjectWithFields.<lambda>c S s | d S )Nr. r )r2 r r r rN � s )r r r �y�Zc S s | dkrt j� �| S )Ni� )r �SerializationError)r r r r rO � s zBJSONObjectWithFieldsTest.setUp.<locals>.MockJSONObjectWithFields.yc S s | dkrt j� �| S )Ni� )r �DeserializationError)r r r r rO � s N)r% r& r' r2 rO �zr r r )r r r �MockJSONObjectWithFields� s rT r. r/ )r2 rO rS )r rM r rT r )r rM rT r )r r rF � s
z JSONObjectWithFieldsTest.setUpc C s | j | j| jddd�� d S )Nr. r/ )rO rS )r r rT )r r r r �test_init_defaults� s z+JSONObjectWithFieldsTest.test_init_defaultsc C s" | j d| jdddd�jd�� d S )N�
� r )r2 rO rS r2 )r rT r )r r r r � test_encode� s
z$JSONObjectWithFieldsTest.test_encodec C s | j tj| jjd� d S )Nr )� assertRaisesr �Errorr r )r r r r �test_encode_wrong_field� s z0JSONObjectWithFieldsTest.test_encode_wrong_fieldc C s | j tj| jdd d�jd� d S )Ni� )rO rS rO )rY r rQ rT r )r r r r �+test_encode_serialization_error_passthrough� s zDJSONObjectWithFieldsTest.test_encode_serialization_error_passthroughc C s | j | jj� ddd�� d S )Nr. r/ )rO rP )r r �fields_to_partial_json)r r r r �'test_fields_to_partial_json_omits_empty� s z@JSONObjectWithFieldsTest.test_fields_to_partial_json_omits_emptyc C s& | j d ddd�| jjddd��� d S )Nr. r/ )r2 rO rS )rO rP )r rT �fields_from_json)r r r r �-test_fields_from_json_fills_default_for_empty� s
zFJSONObjectWithFieldsTest.test_fields_from_json_fills_default_for_emptyc C sh | j tj| jjddi� | j tj| jjddi� | j tj| jjddd�� | j tj| jjddd�� d S )NrO r rP )r2 rO )r2 rP )rY r rR rT r_ )r r r r �&test_fields_from_json_fails_on_missing� s z?JSONObjectWithFieldsTest.test_fields_from_json_fails_on_missingc C s( | j | jdddd�j� dddd�� d S )Nr- r. r/ )r2 rO rS )r2 rO rP )r rT r! )r r r r �#test_fields_to_partial_json_encoder� s z<JSONObjectWithFieldsTest.test_fields_to_partial_json_encoderc C s( | j dddd�| jjdddd��� d S )Nr. r/ )r2 rO rS � )r2 rO rP )r rT r_ )r r r r � test_fields_from_json_decoder� s
z6JSONObjectWithFieldsTest.test_fields_from_json_decoderc C s | j tj| jdddd�j� d S )Nr- i� r/ )r2 rO rS )rY r rQ rT r! )r r r r �-test_fields_to_partial_json_error_passthrough� s zFJSONObjectWithFieldsTest.test_fields_to_partial_json_error_passthroughc C s | j tj| jjdddd�� d S )Nrc i� r/ )r2 rO rP )rY r rR rT r$ )r r r r �'test_fields_from_json_error_passthrough� s z@JSONObjectWithFieldsTest.test_fields_from_json_error_passthroughN)r% r& r' r7 rF rU rX r[ r\ r^ r` ra rb rd re rf r r r r rL | s rL c @ s| e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd S ) �DeEncodersTestc C s d| _ d| _d S )Na� MIIB3jCCAYigAwIBAgICBTkwDQYJKoZIhvcNAQELBQAwdzELMAkGA1UEBhMCVVMxETAPBgNVBAgMCE1pY2hpZ2FuMRIwEAYDVQQHDAlBbm4gQXJib3IxKzApBgNVBAoMIlVuaXZlcnNpdHkgb2YgTWljaGlnYW4gYW5kIHRoZSBFRkYxFDASBgNVBAMMC2V4YW1wbGUuY29tMB4XDTE0MTIxMTIyMzQ0NVoXDTE0MTIxODIyMzQ0NVowdzELMAkGA1UEBhMCVVMxETAPBgNVBAgMCE1pY2hpZ2FuMRIwEAYDVQQHDAlBbm4gQXJib3IxKzApBgNVBAoMIlVuaXZlcnNpdHkgb2YgTWljaGlnYW4gYW5kIHRoZSBFRkYxFDASBgNVBAMMC2V4YW1wbGUuY29tMFwwDQYJKoZIhvcNAQEBBQADSwAwSAJBAKx1c7RR7R_drnBSQ_zfx1vQLHUbFLh1AQQQ5R8DZUXd36efNK79vukFhN9HFoHZiUvOjm0c-pVE6K-EdE_twuUCAwEAATANBgkqhkiG9w0BAQsFAANBAC24z0IdwIVKSlntksllvr6zJepBH5fMndfk3XJp10jT6VE-14KNtjh02a56GoraAvJAT5_H67E8GvJ_ocNnB_oa� MIIBXTCCAQcCAQAweTELMAkGA1UEBhMCVVMxETAPBgNVBAgMCE1pY2hpZ2FuMRIwEAYDVQQHDAlBbm4gQXJib3IxDDAKBgNVBAoMA0VGRjEfMB0GA1UECwwWVW5pdmVyc2l0eSBvZiBNaWNoaWdhbjEUMBIGA1UEAwwLZXhhbXBsZS5jb20wXDANBgkqhkiG9w0BAQEFAANLADBIAkEArHVztFHtH92ucFJD_N_HW9AsdRsUuHUBBBDlHwNlRd3fp580rv2-6QWE30cWgdmJS86ObRz6lUTor4R0T-3C5QIDAQABoCkwJwYJKoZIhvcNAQkOMRowGDAWBgNVHREEDzANggtleGFtcGxlLmNvbTANBgkqhkiG9w0BAQsFAANBAHJH_O6BtC9aGzEVCMGOZ7z9iIRHWSzr9x_bOzn7hLwsbXPAgO1QxEwL-X-4g20Gn9XBE1N9W6HCIEut2d8wACg)�b64_cert�b64_csr)r r r r rF � s
zDeEncodersTest.setUpc C s0 ddl m} |d�}| j|t� | jd|� d S )Nr )�encode_b64jose� x�eA)r rj r3 �strr )r rj �encodedr r r �test_encode_b64jose� s z"DeEncodersTest.test_encode_b64josec C s0 ddl m} |d�}| j|t� | jd|� d S )Nr )�decode_b64joserl rk )r rp r3 �bytesr )r rp �decodedr r r �test_decode_b64jose� s z"DeEncodersTest.test_decode_b64josec C s ddl m} | jtj|d� d S )Nr )rp r2 )r rp rY r rR )r rp r r r �!test_decode_b64jose_padding_error� s z0DeEncodersTest.test_decode_b64jose_padding_errorc C sL ddl m} | jd|ddd�� | jtj|ddd� | jtj|ddd� d S ) Nr )rp s foo�Zm9vr/ )�sizer. rc )r rp r rY r rR )r rp r r r �test_decode_b64jose_size s z'DeEncodersTest.test_decode_b64jose_sizec C sR ddl m} | jd|dddd�� | jd|dddd�� | jtj|dd dd� d S )
Nr )rp s fooru r/ T)rv �minimumr. rc )r rp r rY r rR )r rp r r r � test_decode_b64jose_minimum_size s
z/DeEncodersTest.test_decode_b64jose_minimum_sizec C s0 ddl m} |d�}| jd|� | j|t� d S )Nr )� encode_hex16s foo�666f6f)r rz r r3 rm )r rz rn r r r �test_encode_hex16 s z DeEncodersTest.test_encode_hex16c C s0 ddl m} |d�}| jd|� | j|t� d S )Nr )� decode_hex16r{ s foo)r r} r r3 rq )r r} rr r r r �test_decode_hex16 s z DeEncodersTest.test_decode_hex16c C sR ddl m} | jd|dddd�� | jd|dddd�� | jtj|dd dd� d S )
Nr )r} s foor{ r/ T)rv rx r. rc )r r} r rY r rR )r r} r r r � test_decode_hex16_minimum_size s
z-DeEncodersTest.test_decode_hex16_minimum_sizec C s ddl m} | jtj|d� d S )Nr )r} r2 )r r} rY r rR )r r} r r r � test_decode_hex16_odd_length% s z+DeEncodersTest.test_decode_hex16_odd_lengthc C s" ddl m} | j| j|t�� d S )Nr )� encode_cert)r r� r rh �CERT)r r� r r r �test_encode_cert) s z DeEncodersTest.test_encode_certc C sD ddl m} || j�}| j|tj� | j|t� | jt j
|d� d S )Nr )� decode_cert� ) r r� rh r3 r �ComparableX509r r� rY r rR )r r� Zcertr r r �test_decode_cert- s
z DeEncodersTest.test_decode_certc C s" ddl m} | j| j|t�� d S )Nr )�
encode_csr)r r� r ri �CSR)r r� r r r �test_encode_csr4 s z DeEncodersTest.test_encode_csrc C sD ddl m} || j�}| j|tj� | j|t� | jt j
|d� d S )Nr )�
decode_csrr� ) r r� ri r3 r r� r r� rY r rR )r r� Zcsrr r r �test_decode_csr8 s
z DeEncodersTest.test_decode_csrN)r% r& r' rF ro rs rt rw ry r| r~ r r� r� r� r� r� r r r r rg � s rg c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )� TypedJSONObjectWithFieldsTestc C sH ddl m} G dd� d|�}|jG dd� d|��}|| _|dd�| _d S ) Nr )�TypedJSONObjectWithFieldsc @ s e Zd Zi ZdZdS )zPTypedJSONObjectWithFieldsTest.setUp.<locals>.MockParentTypedJSONObjectWithFields�typeN)r% r& r' ZTYPESZtype_field_namer r r r �#MockParentTypedJSONObjectWithFieldsH s r� c @ s( e Zd ZdZdZedd� �Zdd� ZdS ) zJTypedJSONObjectWithFieldsTest.setUp.<locals>.MockTypedJSONObjectWithFields�testr c S s d|d iS )Nr r )r" r# r r r r_ R s z[TypedJSONObjectWithFieldsTest.setUp.<locals>.MockTypedJSONObjectWithFields.fields_from_jsonc S s
d| j iS )Nr )r )r r r r r] V s zaTypedJSONObjectWithFieldsTest.setUp.<locals>.MockTypedJSONObjectWithFields.fields_to_partial_jsonN)r )r% r& r' �typr; r( r_ r] r r r r � MockTypedJSONObjectWithFieldsL s r� r: )r )r r� �register�
parent_cls�msg)r r� r� r� r r r rF B s z#TypedJSONObjectWithFieldsTest.setUpc C s | j | jj� ddd�� d S )Nr� r: )r� r )r r� r! )r r r r �test_to_partial_json\ s z2TypedJSONObjectWithFieldsTest.test_to_partial_jsonc C s. x(g f ddgD ]}| j tj| jj|� qW d S )NrW Zasd)rY r rR r� r$ )r r r r r � test_from_json_non_dict_failsb s z;TypedJSONObjectWithFieldsTest.test_from_json_non_dict_failsc C s | j tj| jji � d S )N)rY r rR r� r$ )r r r r �!test_from_json_dict_no_type_failsg s z?TypedJSONObjectWithFieldsTest.test_from_json_dict_no_type_failsc C s | j tj| jjddi� d S )Nr� r: )rY r ZUnrecognizedTypeErrorr� r$ )r r r r �!test_from_json_unknown_type_failsk s z?TypedJSONObjectWithFieldsTest.test_from_json_unknown_type_failsc C s" | j ddi| jjddd��� d S )Nr r: r� )r� r )r r� r$ )r r r r �test_from_json_returns_objo s z8TypedJSONObjectWithFieldsTest.test_from_json_returns_objN) r% r& r' rF r� r� r� r� r� r r r r r� @ s r� �__main__)r7 r
Zunittestr Zjosepyr r r r Zload_comparable_certr� Zload_comparable_csrr� ZTestCaser r8 rL rg r� r% �mainr r r r �<module> s
D,]g4