File: //usr/local/lib/python3.10/unittest/test/testmock/__pycache__/testsealable.cpython-310.opt-1.pyc
o
�i� � @ sL d dl Z d dl mZ G dd� d�ZG dd� de j�Zedkr$e �� dS dS )� N)�mockc @ s e Zd Zdd� Zdd� ZdS )� SampleObjectc C � d S �N� ��selfr r �@/usr/local/lib/python3.10/unittest/test/testmock/testsealable.py�method_sample1 � zSampleObject.method_sample1c C r r r r r r r �method_sample2 r zSampleObject.method_sample2N)�__name__�
__module__� __qualname__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%d&� Zd'S )(� TestSealablec C sD t �� }| �|jt j� | �|�� t j� | �|�� �� t j� d S r )r �Mock�assertIsInstance�test�test2�r �mr r r �,test_attributes_return_more_mocks_by_default s z9TestSealable.test_attributes_return_more_mocks_by_defaultc C sx t �� }t �|� | �t�� |j W d � n1 s w Y | �t�� |� W d � d S 1 s5w Y d S r �r r �seal� assertRaises�AttributeErrorr r r r r �.test_new_attributes_cannot_be_accessed_on_seal s
� "�z;TestSealable.test_new_attributes_cannot_be_accessed_on_sealc C sH t �� }t �|� | �t�� d|_W d � d S 1 s w Y d S �N� r r r r r �)test_new_attributes_cannot_be_set_on_seal �
"�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 � assertEqualr r r r �+test_existing_attributes_can_be_set_on_seal% s
z8TestSealable.test_existing_attributes_can_be_set_on_sealc C sR t �� }d|j_t �|� | �t��
d|j_W d � d S 1 s"w Y d S r )r r r r r r r �test3r r r r �2test_new_attributes_cannot_be_set_on_child_of_seal- 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 �+test_existing_attributes_allowed_after_seal5 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 �.test_initialized_attributes_allowed_after_seal= �
z;TestSealable.test_initialized_attributes_allowed_after_sealc C sH t �� }t �|� | �t�� |� W d � d S 1 s w Y d S r )r r r r r r r r r � test_call_on_sealed_mock_failsC r 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 �)test_call_on_defined_sealed_mock_succeedsJ r+ z6TestSealable.test_call_on_defined_sealed_mock_succeedsc C s� t �� }d|j�� _t �|� | �|j�� jd� | �t�� |j�� j W d � n1 s/w Y | �t��
|jj W d � d S 1 sIw Y d S �N� )
r r �test1r r$ r r" r r �test4r r r r �&test_seals_recurse_on_added_attributesP 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 � n1 sHw Y | �t�� |jjdd� j W d � d S 1 sgw Y d S )Nr0 �ar! r- )
r � MagicMockr1 r r$ r r" r r r2 r r r r �#test_seals_recurse_on_magic_methods\ 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_sealedr0 r )r r r1 r r$ r r" r2 r r r r �,test_seals_dont_recurse_on_manual_attributesj s
z9TestSealable.test_seals_dont_recurse_on_manual_attributesc C sr t �t�}d|_d|_t �|� | �|jd� | �|jd� | �t�� |j W d � dS 1 s2w Y dS )z0You are not restricted when using mock with specr r& N)
r r r Z attr_sample1Z attr_sample3r r" r r Z attr_sample2r r r r �)test_integration_with_spec_att_definitionu s
"�z6TestSealable.test_integration_with_spec_att_definitionc C sd t �t�}d|j_t �|� | �|�� d� | �t��
|� � W d � dS 1 s+w Y dS )z<You need to define the methods, even if they are in the specr N)
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 sB t �t�}| �t��
d|j_W d � dS 1 sw Y dS )z)You cannot define methods out of the specr& N)r r r r r Zmethod_sample3r' r r r r �:test_integration_with_spec_method_definition_respects_spec� s
"�zGTestSealable.test_integration_with_spec_method_definition_respects_specc C sX t �� }t �|� | �t�� }|j W d � n1 s w Y | �dt|j�� d S )N� SECRETE_name) r r r r r r= �assertIn�str� exception�r r �cmr r r �(test_sealed_exception_has_attribute_name� s
�z5TestSealable.test_sealed_exception_has_attribute_namec C sp t jdd�}|jjjj t �|� | �t��}|jjjjj W d � n1 s(w Y | �
dt |j �� d S )NZ mock_namer7 z&mock_name.test1.test2.test3.test4.boom)
r r r1 r r$ r2 r r r Zboomr> r? r@ rA r r r �"test_attribute_chain_is_maintained� s
�z/TestSealable.test_attribute_chain_is_maintainedc C st t �� }|�� j�� j t �|� | �t��}|�� j�� �� W d � n1 s*w Y | � dt
|j �� d S )Nz mock.test1().test2.test3().test4) r r r1 r r$ r2 r r r r> r? r@ rA r r r � test_call_chain_is_maintained� s
�z*TestSealable.test_call_chain_is_maintainedc
C sH G dd� d�}dD �]�}| j |d���� tj||d�}d|j_d|jj_t�|� | �|j tj
� | �|jtj � | �|j tj � | �|jtj � | �|jj
tj
� | �|jjtj � | �|�� d� d|j_| �|�� d� | �|j�� d� d|jj_| �|j�� d� | �t�� |� � W d � n1 s�w Y | �t�� d |_W d � n1 s�w Y | �t�� |� � W d � n1 s�w Y d
|j _| �|� � d
� | �t�� |j W d � n1 s�w Y | �t�� d |_W d � n 1 �sw Y | �t�� |�� W d � n 1 �s"w Y | �t��
|j�
� W d � n 1 �s=w Y | �t�� |jj W d � n 1 �sWw Y | �t�� d |j_W d � n 1 �sqw Y | �t��
|j�� W d � n 1 �s�w Y W d � n 1 �s�w Y q d S ) Nc @ s. e Zd ZdZdd� Zdd� ZG dd� d�ZdS ) z1TestSealable.test_seal_with_autospec.<locals>.Foor c S � dS r r r r r r �bar1� � z6TestSealable.test_seal_with_autospec.<locals>.Foo.bar1c S rF )Nr! r r r r r �bar2� rH z6TestSealable.test_seal_with_autospec.<locals>.Foo.bar2c @ s e Zd ZdZdd� ZdS )z5TestSealable.test_seal_with_autospec.<locals>.Foo.Bazr& c S rF r/ r r r r r �ban� rH z9TestSealable.test_seal_with_autospec.<locals>.Foo.Baz.banN)r
r r �bazrJ r r r r �Baz� s rL N)r
r r �foorG rI rL r r r r �Foo� s
rN )TF)�spec_setr4 �bZnew_aZnew_br rI )ZsubTestr Zcreate_autospecrG r' rL rJ r r rM ZNonCallableMagicMockr5 rI rK r" r � TypeErrorr ZbarZ missing_attrZmissing_method)r rN rO rM r r r �test_seal_with_autospec� sn
� �
� � �
� �
�
� �� ���z$TestSealable.test_seal_with_autospecN)r
r r r r r r# r% r( r* r, r. r3 r6 r9 r: r; r< rC rD rE rR r r r r r s(
r �__main__)Zunittestr r ZTestCaser r
�mainr r r r �<module> s a �