Your IP : 216.73.216.213
U
i�fT$ � @ 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 �5/opt/alt/python38/lib64/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# �ZelZseriesr 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 r0 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_typeZexr r r � assertRaisesF s
r3 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 )r5 r6 r7 )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 r4 )r r r8 r* r r+ �Zhook1Zhook2r r r �test_block_add_hookX s
r? 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 )r3 r1 r r> r r r �!test_block_add_hook_baseexceptionb s
�r@ 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 rD r r r r � PicklePrintp s rE )r �b�cr5 r6 r7 rA zpickle.find_classr= )�pickle�dumpsr( �loadsr r3 r )rH rE Z payload_1Z payload_2r: r r r � test_picklem s
rK 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 rM c @ s e Zd ZdS )ztest_monkeypatch.<locals>.BNrL r r r r �B� s rN c @ s e Zd ZdS )ztest_monkeypatch.<locals>.CNrL r r r r �C� s rO �X� __bases__�{ c S s( g | ] \}}|d kr|d |d f�qS )zobject.__setattr__r r5 r �r �er r r r r � s z$test_monkeypatch.<locals>.<listcomp>r � __class__) r r rQ �type�__dict__�__set__r
Znew_attrrU r r0 )rM rN rO r r: Zactualr r r �test_monkeypatch� s �rY 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= r6 �r�rbr7 �wb�w���Fc W s dS )Nr5 r �r r r r �<lambda>� � ztest_open.<locals>.<lambda>c S s0 g | ](\}}|d kr|d r|d |d f�qS )rZ r5 r r rS r r r r � s z test_open.<locals>.<listcomp>c S s0 g | ](\}}|d kr|d s|d |d f�qS )rZ r5 r r6 r rS r r r r � s c S s g | ]}|d k r|�qS r r r r r r r � s �)r7 r^ )
�sslZcreate_default_context�load_dh_params� ImportErrorr rZ r �argv�
executabler3 r r r0 )rc rd r: �fnr Z actual_modeZ actual_flagr r r � test_open� s<
�
�� �ri 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�3r5 �4r Zcall� )r �settracer �evalZ __cantrace__r0 )rn �oldr: r rm r �
test_cantrace� s
rw c C sF dd l } t� �.}| � dd� t|jd d d d� d� W 5 Q R X d S )Nr r_ � r5 r6 )r_ rx )�mmapr r( r )ry r: r r r � test_mmap� s rz c s0 dd� � � fdd�} t �| � � t _td��d S )Nc S s | t k rt�| ||� d S r )r r �__excepthook__)r � exc_valueZexc_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.excepthookr6 r5 zExpected isinstance(z, �)r � Expected � == )r2 � TypeError�
ValueError�print�repr�r r �r} r r r: � s z test_excepthook.<locals>.hookz fatal-error)r r
r} r r9 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� r5 )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 ZDetach)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
Z gethostnameZAF_INETZ SOCK_STREAMr Zbind� Exception)r� r: Zsockr 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>.hookr5 )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 r` r r r ra [ rb z test_not_in_gc.<locals>.<lambda>)r� r r
r� r2 �list)r� r: �or r r �test_not_in_gcX s
r� �__main__)�suppress_msvcrt_assertsr5 ) �
contextlibr r r( r* r+ r0 �contextmanagerr3 r; r? r@ rK rY ri rw rz r� r� r� r� r� r� r Z test.supportr� rf Ztest�globalsr r r r �<module> s6 $
+!