File: //usr/local/lib/python3.7/test/__pycache__/test_resource.cpython-37.opt-2.pyc
B
��g/ � @ sl d dl Z d dlZd dlZd dlZd dlmZ d dlZe�d�ZG dd� dej �Z
d dd�Z e dkrhe � dS )
� N)�support�resourcec @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z e
� e j
dkd�dd� �Zdd� Ze
� eed�d�e�ddd�dd� ��Ze
� eed�d�e�ddd�d d � ��Zd S ) � ResourceTestc C sF | � ttj� | � ttjdd� | � ttj� | � ttjddd� d S )N�* )� assertRaises� TypeErrorr � getrlimit� setrlimit)�self� r �./usr/local/lib/python3.7/test/test_resource.py� test_args s zResourceTest.test_argsc C sN yt �t j�\}}W n tk
r( Y n"X | �t j|� t �t j||f� d S )N)r r � RLIMIT_FSIZE�AttributeError� assertEqualZ
RLIM_INFINITYr )r
�cur�maxr r r �test_fsize_ismax s z ResourceTest.test_fsize_ismaxc C s yt �t j�\}}W n tk
r( Y n�X z�yt �t jd|f� d}W n tk
r^ d}Y nX ttjd�}zz|� d� y:|� d� |�
� x"t d�D ]}t �
d� |�
� q�W W n tk
r� |sȂ Y nX |r�t �t j||f� W d |�� X W d |�rt �t j||f� t�tj� X d S ) Ni TF�wbs XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX� Y� g�������?)r r r r r �
ValueError�openr ZTESTFN�write�flush�range�timeZsleep�OSError�close�unlink)r
r r Z limit_set�f�ir r r �test_fsize_enforced" s8
z ResourceTest.test_fsize_enforcedc C s� dd }yt �t j�\}}W n tk
r0 Y nbX yt �t j||f� W n ttfk
r` Y nX yt �t j||f� W n ttfk
r� Y nX d S )N�
�2 )r r r r r �
OverflowErrorr )r
Ztoo_bigr r r r r �test_fsize_toobigQ s z ResourceTest.test_fsize_toobigc C s� | � ttj� | � ttjdd� t�tj�}t�tj�}yt�tj�}W n ttfk
r` Y nX yt�tj �}W n ttfk
r� Y nX d S )Nr )
r r r Z getrusageZ RUSAGE_SELFZRUSAGE_CHILDRENZ RUSAGE_BOTHr r Z
RUSAGE_THREAD)r
Z usageselfZ
usagechildrenZ usagebothZ usage_threadr r r �test_getrusageb s zResourceTest.test_getrusagec C sH yt �t j�}W n tk
r$ Y n X G dd� d�}t �t j|� � d S )Nc @ s e Zd Zdd� Zdd� ZdS )z9ResourceTest.test_setrusage_refcount.<locals>.BadSequencec S s dS )N� r )r
r r r �__len__y s zAResourceTest.test_setrusage_refcount.<locals>.BadSequence.__len__c S s |dkrt ttd���S t�d S )N)r � i@B )�len�tupler �
IndexError)r
�keyr r r � __getitem__{ s zEResourceTest.test_setrusage_refcount.<locals>.BadSequence.__getitem__N)�__name__�
__module__� __qualname__r) r/ r r r r � BadSequencex s r3 )r r Z
RLIMIT_CPUr r )r
�limitsr3 r r r �test_setrusage_refcountr s z$ResourceTest.test_setrusage_refcountc C s$ t �� }| �|t� | �|d� d S )Nr )r Z getpagesize�assertIsInstance�intZassertGreaterEqual)r
Zpagesizer r r �
test_pagesize� s zResourceTest.test_pagesizeZlinuxztest requires Linuxc
C s> x8dD ]0}t �t�� | �ttd| �t� W d Q R X qW d S )N)ZMSGQUEUEZNICEZRTPRIOZRTTIMEZ
SIGPENDING�RLIMIT_)�
contextlib�suppressr r6 �getattrr r7 )r
�attrr r r �test_linux_constants� s
z!ResourceTest.test_linux_constantsc
C s> x8dD ]0}t �t�� | �ttd| �t� W d Q R X qW d S )N)ZSWAPZSBSIZEZNPTSr9 )r: r; r r6 r< r r7 )r
r= r r r �test_freebsd_contants� s
z"ResourceTest.test_freebsd_contants�prlimitz
no prlimitr( � �$ c C s` | � ttj� | � ttjdtj� t�tj�}| �t�dtj�|� | �t�dtj|�|� d S )N���r )r r r r@ �ProcessLookupError� RLIMIT_ASr r )r
�limitr r r � test_prlimit� s
zResourceTest.test_prlimitc s< G � fdd�d�}t �t j�� | �t �dt j|� �� � d S )Nc s e Zd Zdd� Z� fdd�ZdS )z2ResourceTest.test_prlimit_refcount.<locals>.BadSeqc S s dS )Nr( r )r
r r r r) � s z:ResourceTest.test_prlimit_refcount.<locals>.BadSeq.__len__c s � | d S )Nr* r )r
r. )r4 r r r/ � s z>ResourceTest.test_prlimit_refcount.<locals>.BadSeq.__getitem__N)r0 r1 r2 r) r/ r )r4 r r �BadSeq� s rH r )r r rE r r@ )r
rH r )r4 r �test_prlimit_refcount� s z"ResourceTest.test_prlimit_refcountN)r0 r1 r2 r
r r" r& r' r5 r8 �unittestZ
skipUnless�sys�platformr> r? �hasattrr r Zrequires_linux_versionrG rI r r r r r s / r c C s t �t� d S )N)r Z run_unittestr )�verboser r r � test_main� s rO �__main__)N)
r: rK �osrJ Ztestr r �
import_moduler ZTestCaser rO r0 r r r r �<module> s
!