File: //usr/local/lib/python3.7/unittest/test/testmock/__pycache__/testsealable.cpython-37.opt-2.pyc
B
"��g� � @ sH d dl Z d dl mZ G dd� d�ZG dd� de j�ZedkrDe �� dS )� N)�mockc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� SampleObjectc C s d| _ d| _d S )N� )� attr_sample1� attr_sample2)�self� r �?/usr/local/lib/python3.7/unittest/test/testmock/testsealable.py�__init__ s zSampleObject.__init__c C s d S )Nr )r r r r �method_sample1
s zSampleObject.method_sample1c C s d S )Nr )r r r r �method_sample2
s zSampleObject.method_sample2N)�__name__�
__module__� __qualname__r
r r r r r r r s r 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 d � Zd d � Zd!d"� Zd#d$� Zd%S )&� TestSealablec C sD t �� }| �|jt j� | �|�� t j� | �|�� �� t j� d S )N)r �MockZassertIsInstance�test�test2)r �mr r r �,test_attributes_return_more_mocks_by_default s z9TestSealable.test_attributes_return_more_mocks_by_defaultc C sN t �� }t �|� | �t�� |j W d Q R X | �t�� |� W d Q R X d S )N)r r �seal� assertRaises�AttributeErrorr )r r r r r �.test_new_attributes_cannot_be_accessed_on_seal s
z;TestSealable.test_new_attributes_cannot_be_accessed_on_sealc C s2 t �� }t �|� | �t�� d|_W d Q R X d S )Nr )r r r r r r )r r r r r �)test_new_attributes_cannot_be_set_on_seal# s
z6TestSealable.test_new_attributes_cannot_be_set_on_sealc C s6 t �� }d|j_t �|� d|j_| �|jjd� d S )Nr � )r r r r r � assertEqual)r r r r r �+test_existing_attributes_can_be_set_on_seal* s
z8TestSealable.test_existing_attributes_can_be_set_on_sealc C s< t �� }d|j_t �|� | �t�� d|j_W d Q R X d S )Nr )r r r r r r r �test3)r r r r r �2test_new_attributes_cannot_be_set_on_child_of_seal2 s
z?TestSealable.test_new_attributes_cannot_be_set_on_child_of_sealc C s. t �� }d|j_t �|� | �|�� d� d S )N� )r r r � return_valuer r )r r r r r �+test_existing_attributes_allowed_after_seal: s
z8TestSealable.test_existing_attributes_allowed_after_sealc C s( t jdd�}t �|� | �|jd� d S )Nr )�
test_value)r r r r r# )r r r r r �.test_initialized_attributes_allowed_after_sealB s
z;TestSealable.test_initialized_attributes_allowed_after_sealc C s2 t �� }t �|� | �t�� |� W d Q R X d S )N)r r r r r )r r r r r � test_call_on_sealed_mock_failsH s
z+TestSealable.test_call_on_sealed_mock_failsc C s( t jdd�}t �|� | �|� d� d S )N� )r! )r r r r )r r r r r �)test_call_on_defined_sealed_mock_succeedsO s
z6TestSealable.test_call_on_defined_sealed_mock_succeedsc C sv t �� }d|j�� _t �|� | �|j�� jd� | �t�� |j�� j W d Q R X | �t�� |jj W d Q R X d S )N� )
r r �test1r r r r r r �test4)r r r r r �&test_seals_recurse_on_added_attributesU s
z3TestSealable.test_seals_recurse_on_added_attributesc C s� t �� }d|jjd _d|jjdd� _t �|� | �|jjd jd� | �|jjdd� jd� | �t�� |jjd j W d Q R X | �t�� |jjdd� j W d Q R X d S )Nr( �ar r& )
r Z MagicMockr) r r r r r r r* )r r r r r �#test_seals_recurse_on_magic_methodsa s
z0TestSealable.test_seals_recurse_on_magic_methodsc C sZ t jdd�}t jdd�|j_d|jj_t �|� | �|jjjd� |jjj d|jj_d S )NZ root_mock)�nameZ
not_sealedr( r )r r r) r r r r r* )r r r r r �,test_seals_dont_recurse_on_manual_attributeso s
z9TestSealable.test_seals_dont_recurse_on_manual_attributesc C s\ t �t�}d|_d|_t �|� | �|jd� | �|jd� | �t�� |j W d Q R X d S )Nr r )
r r r r Z attr_sample3r r r r r )r r r r r �)test_integration_with_spec_att_definitionz s
z6TestSealable.test_integration_with_spec_att_definitionc C sN t �t�}d|j_t �|� | �|�� d� | �t�� |� � W d Q R X d S )Nr )
r r r r r! r r r r r )r r r r r �,test_integration_with_spec_method_definition� s
z9TestSealable.test_integration_with_spec_method_definitionc C s, t �t�}| �t�� d|j_W d Q R X d S )Nr )r r r r r Zmethod_sample3r! )r r r r r �:test_integration_with_spec_method_definition_respects_spec� s
zGTestSealable.test_integration_with_spec_method_definition_respects_specc C sD t �� }t �|� | �t�� }|j W d Q R X | �dt|j�� d S )N� SECRETE_name) r r r r r r3 �assertIn�str� exception)r r �cmr r r �(test_sealed_exception_has_attribute_name� s
z5TestSealable.test_sealed_exception_has_attribute_namec C s\ t jdd�}|jjjj t �|� | �t��}|jjjjj W d Q R X | �
dt |j �� d S )NZ mock_name)r. z&mock_name.test1.test2.test3.test4.boom)
r r r) r r r* r r r Zboomr4 r5 r6 )r r r7 r r r �"test_attribute_chain_is_maintained� s
z/TestSealable.test_attribute_chain_is_maintainedc C s` t �� }|�� j�� j t �|� | �t��}|�� j�� �� W d Q R X | � dt
|j �� d S )Nz mock.test1().test2.test3().test4) r r r) r r r* r r r r4 r5 r6 )r r r7 r r r � test_call_chain_is_maintained� s
z*TestSealable.test_call_chain_is_maintainedN)r
r r r r r r r r" r$ r% r' r+ r- r/ r0 r1 r2 r8 r9 r: r r r r r s$
r �__main__)Zunittestr r ZTestCaser r
�mainr r r r �<module> s $