Your IP : 216.73.217.129
���]� � @ s� d d l Z d d l Z d d l Z d d l Z d d l m Z d d l Z e j d � Z Gd d � d e j � Z
d d d � Z e d k r� e � d S) � N)�support�resourcec @ s e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z e
j e j
d k d � d d � � Z e j d � d d � � Z e
j e e d � d � e j d d d � d d � � � Z e
j e e d � d � e j d d d � d d � � � Z d S) � ResourceTestc C s_ | j t t j � | j t t j d d � | j t t j � | j t t j d d d � d S)N�* )� assertRaises� TypeErrorr � getrlimit� setrlimit)�self� r �2/opt/alt/python35/lib64/python3.5/test_resource.py� test_args s zResourceTest.test_argsc C sa y t j t j � \ } } Wn t k
r0 Yn- X| j t j | � t j 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 se y t j t j � \ } } Wn t k
r0 Yn1Xz� y# t j t j d | f � d } Wn t k
rq d } Yn Xt t j d � } z� | j d d � yI | j d � | j
� x+ t d � D] } t j
d � | j
� q� WWn t k
r� | s� � Yn X| r t j t j | | f � Wd | j � XWd | rPt j t j | | f � t j t j � Xd S) Ni TF�wb� X� Y� g�������?)r r r r r �
ValueError�openr ZTESTFN�write�flush�range�time�sleep�OSError�close�unlink)r
r r Z limit_set�f�ir r r �test_fsize_enforced" s8
z ResourceTest.test_fsize_enforcedc C s� d } y t j t j � \ } } Wn t k
r6 Ynq Xy t j t j | | f � Wn t t f k
rn Yn Xy t j t j | | f � Wn t t f k
r� Yn Xd S)N�
�2 l +�O� �tC3�y+v� )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 t t j � | j t t j d d � t j t j � } t j t j � } y t j t j � } Wn t t f k
r� Yn Xy t j t j � } Wn t t f k
r� Yn Xd 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 sX y t j t j � } Wn t k
r* Yn* XGd d � d � } t j t j | � � d S)Nc @ s( e Z d Z d d � Z d d � Z d S)z9ResourceTest.test_setrusage_refcount.<locals>.BadSequencec S s d S)N� r )r
r r r �__len__y s zAResourceTest.test_setrusage_refcount.<locals>.BadSequence.__len__c S s, | d k r" t t t d � � � St � 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+ r1 r r r r � BadSequencex s r5 )r r Z
RLIMIT_CPUr r )r
�limitsr5 r r r �test_setrusage_refcountr s
z$ResourceTest.test_setrusage_refcountc C s0 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 sX xQ d d d d d g D]: } t j t � �" | j t t d | � t � Wd QRXq Wd S)NZMSGQUEUEZNICEZRTPRIOZRTTIMEZ
SIGPENDING�RLIMIT_)�
contextlib�suppressr r8 �getattrr r9 )r
�attrr r r �test_linux_constants� s z!ResourceTest.test_linux_constants� c
C sR xK d d d g D]: } t j t � �" | j t t d | � t � Wd QRXq Wd S)NZSWAPZSBSIZEZNPTSr< )r= r> r r8 r? r r9 )r
r@ r r r �test_freebsd_contants� s z"ResourceTest.test_freebsd_contants�prlimitz
no prlimitr* � �$ c C s� | j t t j � t j � d k rA | j t t j d t j � | j t t j d t j � t j t j � } | j
t j d t j � | � | j
t j d t j | � | � d S)Nr r, ���) r r r rD �os�geteuid�PermissionError� RLIMIT_AS�ProcessLookupErrorr r )r
�limitr r r � test_prlimit� s
zResourceTest.test_prlimitc sT G� f d d � d � } t j t j � � | j t j d t j | � � � � d S)Nc s. e Z d Z d d � Z � f d d � Z d S)z2ResourceTest.test_prlimit_refcount.<locals>.BadSeqc S s d S)Nr* r )r
r r r r+ � s z:ResourceTest.test_prlimit_refcount.<locals>.BadSeq.__len__c s � | d S)Nr, r )r
r0 )r6 r r r1 � s z>ResourceTest.test_prlimit_refcount.<locals>.BadSeq.__getitem__N)r2 r3 r4 r+ r1 r )r6 r r �BadSeq� s rO r )r r rK r rD )r
rO r )r6 r �test_prlimit_refcount� s z"ResourceTest.test_prlimit_refcountN)r2 r3 r4 r
r r$ r( r) r7 r: �unittestZ
skipUnless�sys�platformrA r Zrequires_freebsd_versionrC �hasattrr Zrequires_linux_versionrN rP r r r r r s / '$r c C s t j t � d S)N)r Z run_unittestr )�verboser r r � test_main� s rV �__main__)
r= rR rH rQ Ztestr r �
import_moduler ZTestCaser rV r2 r r r r �<module> s �