File: //opt/cppython/lib/python3.8/test/__pycache__/audit-tests.cpython-38.opt-2.pyc
U
-D�gT$ � @ s� d dl Z d dlZG dd� d�Zdd� Zdd� Zdd � Zd
d � Ze jd d
� �Zdd� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zed*kr�d d+lmZ e� ejd, Ze � e � dS )-� Nc @ sF e Zd Zdefdd�Zdd� Zdd� Zdd � Zed
d � �Z d d
� Z
dS )�TestHookNc C s |pd| _ || _g | _d| _d S )N� F)�raise_on_events�exc_type�seen�closed)�selfr r r r �//opt/cppython/lib/python3.8/test/audit-tests.py�__init__ s
zTestHook.__init__c G s t �| � | S �N)�sys� addaudithook�r �ar r r � __enter__ s
zTestHook.__enter__c G s | � � d S r )�closer r r r �__exit__ s zTestHook.__exit__c C s
d| _ d S )NT)r �r r r r r s zTestHook.closec C s dd� | j D �S )Nc S s g | ] }|d �qS )r r ��.0�ir r r �
<listcomp>% s z(TestHook.seen_events.<locals>.<listcomp>)r r r r r � seen_events# s zTestHook.seen_eventsc C s6 | j r
d S | j�||f� || jkr2| �d| ��d S )Nz
saw event )r r �appendr r )r �event�argsr r r �__call__' s
zTestHook.__call__) �__name__�
__module__� __qualname__� RuntimeErrorr
r r r �propertyr r r r r r r s
r c C s | |krt | �d|����d S )N� should equal ��AssertionError��x�yr r r � assertEqual0 s r( c C s | |krt | �d|����d S )Nz should be in r# ��elZseriesr r r �assertIn5 s r+ c C s | |krt | �d|����d S )Nz should not be in r# r) r r r � assertNotIn: s r, c C sP t | �t |�kr"t| �d|����tdd� t| |�D ��rLt| �d|����d S )Nr" c s s | ]\}}||kV qd S r r )r ZixZiyr r r � <genexpr>B s z&assertSequenceEqual.<locals>.<genexpr>)�lenr$ �any�zipr% r r r �assertSequenceEqual? s r1 c
c s> z
d V W n. t k
r8 } zt|t�r(� W 5 d }~X Y nX d S r )�
BaseException�
isinstancer$ )Zex_type�exr r r � assertRaisesF s
r5 c C sN t � �>} t�dddd� t| jd d d� t| jd d d� W 5 Q R X d S )N�
test_event� � � r )r7 r8 r9 )r r �auditr( r ��hookr r r �
test_basicQ s r= c
C sN t dd��:} t � �(}t�d� td| j� td|j� W 5 Q R X W 5 Q R X d S )N�sys.addaudithook�r r6 )r r r: r+ r r, �Zhook1Zhook2r r r �test_block_add_hookX s
rA c C sB t t��0 tdtd��} t� �}W 5 Q R X W 5 Q R X W 5 Q R X d S )Nr> )r r )r5 r2 r r@ r r r �!test_block_add_hook_baseexceptionb s
�rB c
C s~ dd l } G dd� d�}| �|� �}| �d�}td| �|�� tdd��.}tt�� | �|� W 5 Q R X | �|� W 5 Q R X d S )Nr c @ s e Zd Zdd� ZdS )z test_pickle.<locals>.PicklePrintc S s t dfS )N)�Pwned!)�str)r �pr r r �
__reduce_ex__q s z.test_pickle.<locals>.PicklePrint.__reduce_ex__N)r r r rF r r r r � PicklePrintp s rG )r �b�cr7 r8 r9 rC zpickle.find_classr? )�pickle�dumpsr( �loadsr r5 r )rJ rG Z payload_1Z payload_2r< r r r � test_picklem s
rM c C s� G dd� d�} G dd� d�}G dd� d| �}| � }t � �<}d|_|f|_tjd �||f� |j|_d |_||_W 5 Q R X d
d � |j D �}t
|d f|df|df|d
fg|� d S )Nc @ s e Zd ZdS )ztest_monkeypatch.<locals>.AN�r r r r r r r �A� s rO c @ s e Zd ZdS )ztest_monkeypatch.<locals>.BNrN r r r r �B� s rP c @ s e Zd ZdS )ztest_monkeypatch.<locals>.CNrN r r r r �C� s rQ �X� __bases__�{ c S s( g | ] \}}|d kr|d |d f�qS )zobject.__setattr__r r7 r �r �er r r r r � s z$test_monkeypatch.<locals>.<listcomp>r � __class__) r r rS �type�__dict__�__set__r
Znew_attrrW r r1 )rO rP rQ r r< �actualr r r �test_monkeypatch� s �r\ c C s< zdd l } | �� j}W n tk
r. d }Y nX tdhd���}ttjd dfttjdftddfttjd d d
d d d d d d
� f |tjd ffD ]*^}}|s�q�t t
�� ||� W 5 Q R X q�W 5 Q R X dd� |j D �}dd� |j D �}t dd� tjd dftjdfdtjd d f|�r tjd dfnd fD �|� t g |� d S )Nr �openr? r8 �r�rbr9 �wb�w���Fc W s dS )Nr7 r �r r r r �<lambda>� � ztest_open.<locals>.<lambda>c S s0 g | ](\}}|d kr|d r|d |d f�qS )r] r7 r r rU r r r r � s z test_open.<locals>.<listcomp>c S s0 g | ](\}}|d kr|d s|d |d f�qS )r] r7 r r8 r rU r r r r � s c S s g | ]}|d k r|�qS r r r r r r r � s �)r9 ra )
�sslZcreate_default_context�load_dh_params� ImportErrorr r] r �argv�
executabler5 r r r1 )rf rg r< �fnr Z actual_modeZ actual_flagr r r � test_open� s<
�
�� �rl c s� g � � fdd�} t �| �}zNt� �>}td� d|_td� d|_td� d|_td � d
|_W 5 Q R X W 5 t �|� X td gd � � d S )
Nc s | j tjjkr� �|� d S r )�f_coder r �__code__r )�framer r �Ztracedr r �trace� s z test_cantrace.<locals>.trace�1F�2T�3r7 �4r �call� )r �settracer �evalZ __cantrace__r1 )rq �oldr< r rp r �
test_cantrace� s
r{ c C sF dd l } t� �.}| � dd� t|jd d d d� d� W 5 Q R X d S )Nr rb � r7 r8 )rb r| )�mmapr r( r )r} r< r r r � test_mmap� s r~ c s0 dd� � � fdd�} t �| � � t _td��d S )Nc S s | t k rt�| ||� d S r )r r �__excepthook__)r � exc_value�exc_tbr r r �
excepthook� s z#test_excepthook.<locals>.excepthookc sr | dkrnt |d |d �s8td|d �d|d �d���|d � kr\td|d � d � � ���t| t|d �� d S )
Nzsys.excepthookr8 r7 zExpected isinstance(z, �)r � Expected � == )r3 � TypeError�
ValueError�print�repr�r r �r� r r r< � s z test_excepthook.<locals>.hookz fatal-error)r r
r� r r; r r� r �test_excepthook� s
r� c sD ddl m} dd� � � fdd�}t�|� � t_| td�dd � d S ) Nr )�write_unraisable_excc S s d S r r )Zhookargsr r r �unraisablehook s z+test_unraisablehook.<locals>.unraisablehookc sL | dkrH|d � kr,t d|d � d� � ���t| t|d j�|d j� d S )Nzsys.unraisablehookr r� r� r7 )r� r� r� r� �err_msgr� �r� r r r< s z!test_unraisablehook.<locals>.hookznonfatal-errorzfor audit hook test)Z _testcapir� r r
r� r )r� r< r r� r �test_unraisablehook s
r� c C s~ ddl m} m}m}m} dd� }t�|� | |d�}||d� z||d� W n tk
r` Y n
X td��|� � }||� d S )Nr )�OpenKey�EnumKey�CloseKey�HKEY_LOCAL_MACHINEc S s | � d�sd S t| f|�� d S )Nzwinreg.��
startswithr� r� r r r r< s
ztest_winreg.<locals>.hookZSoftwarei' z%Expected EnumKey(HKLM, 10000) to fail)
�winregr� r� r� r� r r
�OSErrorr �Detach)r� r� r� r� r< �kZkvr r r � test_winreg s
r� c C sj dd l } dd� }t�|� | �� | � | j| j�}z(z|�d� W n tk
rV Y nX W 5 |�� X d S )Nr c S s | � d�rt| f|�� d S )Nzsocket.r� r� r r r r< 3 s
ztest_socket.<locals>.hook)z 127.0.0.1i� ) �socketr r
� gethostname�AF_INET� SOCK_STREAMr �bind� Exception)r� r< �sockr r r � test_socket0 s
r� c C sJ dd l } dd� }t�|� | jdd� t� }|g}| �|� | �|� d S )Nr c S s | � d�rt| f|�� d S )Nzgc.r� r� r r r r< I s
ztest_gc.<locals>.hookr7 )Z
generation)�gcr r
� get_objects�objectZ
get_referrersZ
get_referents)r� r< r& r' r r r �test_gcF s
r� c C s6 dd l } dd� }t�|� | �� D ]}t|t�r"q"d S )Nr c W s d S r r rc r r r rd [ re z test_not_in_gc.<locals>.<lambda>)r� r r
r� r3 �list)r� r< �or r r �test_not_in_gcX s
r� �__main__)�suppress_msvcrt_assertsr7 ) �
contextlibr r r( r+ r, r1 �contextmanagerr5 r= rA rB rM r\ rl r{ r~ r� r� r� r� r� r� r Z test.supportr� ri �test�globalsr r r r �<module> s6 $
+!