File: //lib64/python3.6/test/__pycache__/test_resource.cpython-36.pyc
3
\X � @ sl d dl Z d dlZd dlZd dlZd dlmZ d dlZejd�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
j e j
dkd�dd� �Zejd�dd� �Ze
j eed�d�ejddd�dd � ��Ze
j eed�d�ejddd�d d � ��Zd S ) � ResourceTestc C sF | j ttj� | j ttjdd� | j ttj� | j ttjddd� d S )N�* )� assertRaises� TypeErrorr � getrlimit� setrlimit)�self� r �%/usr/lib64/python3.6/test_resource.py� test_args s zResourceTest.test_argsc C sN yt jt j�\}}W n tk
r( Y n"X | jt j|� t jt 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 jt j�\}}W n tk
r( Y n�X z�yt jt jd|f� d}W n tk
r^ d}Y nX ttjd�}z~|j dd � y:|j d� |j
� x"t d�D ]}t j
d� |j
� q�W W n tk
r� |ŝ Y nX |r�t jt j||f� W d |j� X W d |�rt jt j||f� tjtj� X d S ) Ni TF�wb� X� 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 jt j�\}}W n tk
r0 Y nbX yt jt j||f� W n ttfk
r` Y nX yt jt 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� | j ttj� | j ttjdd� tjtj�}tjtj�}ytjtj�}W n ttfk
r` Y nX ytjtj �}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 jt j�}W n tk
r$ Y n X G dd� d�}t jt 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 )Nr � i@B )r r+ )�len�tupler �
IndexError)r
�keyr r r � __getitem__{ s zEResourceTest.test_setrusage_refcount.<locals>.BadSequence.__getitem__N)�__name__�
__module__� __qualname__r* r0 r r r r � BadSequencex s r4 )r r Z
RLIMIT_CPUr r )r
�limitsr4 r r r �test_setrusage_refcountr s z$ResourceTest.test_setrusage_refcountc C s$ t j� }| j|t� | j|d� d S )Nr )r Z getpagesize�assertIsInstance�intZassertGreaterEqual)r
Zpagesizer r r �
test_pagesize� s zResourceTest.test_pagesize�linuxztest requires Linuxc
C s> x8dD ]0}t jt�� | jttd| �t� W d Q R X qW d S )N�MSGQUEUE�NICE�RTPRIO�RTTIME�
SIGPENDING�RLIMIT_)r; r< r= r> r? )�
contextlib�suppressr r7 �getattrr r8 )r
�attrr r r �test_linux_constants� s
z!ResourceTest.test_linux_constants� c
C s> x8dD ]0}t jt�� | jttd| �t� W d Q R X qW d S )N�SWAP�SBSIZE�NPTSr@ )rG rH rI )rA rB r r7 rC r r8 )r
rD r r r �test_freebsd_contants� s
z"ResourceTest.test_freebsd_contants�prlimitz
no prlimitr) � �$ c C s` | j ttj� | j ttjdtj� tjtj�}| jtjdtj�|� | jtjdtj|�|� d S )Nr+ r ���)r r r rK �ProcessLookupError� RLIMIT_ASr r )r
�limitr r r � test_prlimit� s
zResourceTest.test_prlimitc s< G � fdd�d�}t jt j�� | jt jdt 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/ )r5 r r r0 � s z>ResourceTest.test_prlimit_refcount.<locals>.BadSeq.__getitem__N)r1 r2 r3 r* r0 r )r5 r r �BadSeq� s rS r )r r rP r rK )r
rS r )r5 r �test_prlimit_refcount� s z"ResourceTest.test_prlimit_refcountN)r1 r2 r3 r
r r# r' r( r6 r9 �unittestZ
skipUnless�sys�platformrE r Zrequires_freebsd_versionrJ �hasattrr Zrequires_linux_versionrR rT r r r r r s / r c C s t jt� d S )N)r Z run_unittestr )�verboser r r � test_main� s rZ �__main__)N)
rA rV �osrU Ztestr r �
import_moduler ZTestCaser rZ r1 r r r r �<module> s
"