File: //usr/local/lib/python3.10/test/__pycache__/test_resource.cpython-310.opt-2.pyc
o
�i* � @ sx d dl Z d dlZd dlZd dlmZ d dlmZ d dlmZ d dlZe� d�Z
G dd� dej �Z e
dkr:e�� dS dS ) � N)�support)�
import_helper)� os_helper�resourcec @ s� e Zd Zdd� Ze�ejdkd�dd� �Zdd� Z d d
� Z
d d � Z e�ejdkd
�dd� �Z dd� Z
e�ejdkd�dd� �Zdd� Ze�eed�d�e�ddd �d d � ��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.10/test/test_resource.py� test_args s zResourceTest.test_args�vxworksz0setting RLIMIT_FSIZE is not supported on VxWorksc C sN z
t �t j�\}}W n
ty Y d S w | �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> z
t �t j�\}}W n
ty Y d S w zwz
t �t jd|f� d}W n ty. d}Y nw ttjd�}z>|� d� z|� d� |�
� t d�D ] }t �
d� |�
� qIW n tyb |s`� Y nw |rnt �t j||f� W |�� n|�� w W |r�t �t j||f� t�tj� d S |r�t �t j||f� t�tj� w ) Ni TF�wbs XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX� 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% sJ �
�
� ����z ResourceTest.test_fsize_enforcedc C s� dd }z
t �t j�\}}W n
ty Y d S w z t �t j||f� W n ttfy/ Y nw z t �t j||f� W d S ttfyH Y d S w )N�
�2 )r r r r r �
OverflowErrorr )r Ztoo_bigr r r r r �test_fsize_toobigT s ���z ResourceTest.test_fsize_toobigc C s� | � ttj� | � ttjdd� t�tj�}t�tj�}zt�tj�}W n ttfy/ Y nw z t�tj �}W d S ttfyE Y d S w r )
r r
r Z getrusageZ RUSAGE_SELFZRUSAGE_CHILDRENZ RUSAGE_BOTHr r Z
RUSAGE_THREAD)r Z usageselfZ
usagechildrenZ usagebothZ usage_threadr r r �test_getrusagee s ��zResourceTest.test_getrusagez.setting RLIMIT_CPU is not supported on VxWorksc C sH zt �t j�}W n
ty Y d S w G dd� d�}t �t j|� � d S )Nc @ s e Zd Zdd� Zdd� ZdS )z9ResourceTest.test_setrusage_refcount.<locals>.BadSequencec S � dS �N� r r
r r r �__len__~ � zAResourceTest.test_setrusage_refcount.<locals>.BadSequence.__len__c S s |dv r t ttd���S t�)N)r � i@B )�len�tupler �
IndexError�r �keyr r r � __getitem__� s zEResourceTest.test_setrusage_refcount.<locals>.BadSequence.__getitem__N��__name__�
__module__� __qualname__r1 r9 r r r r � BadSequence} s r> )r r Z
RLIMIT_CPUr r )r �limitsr> r r r �test_setrusage_refcountu 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_pagesize�linuxztest requires Linuxc C �N dD ]"}t �t�� | �ttd| �t� W d � n1 s w Y qd S )N)ZMSGQUEUEZNICEZRTPRIOZRTTIMEZ
SIGPENDING�RLIMIT_��
contextlib�suppressr rA �getattrr rB �r �attrr r r �test_linux_constants� s ���z!ResourceTest.test_linux_constantsc C rE )N)ZSWAPZSBSIZEZNPTSrF rG rK r r r �test_freebsd_contants� s ���z"ResourceTest.test_freebsd_contants�prlimitz
no prlimitr0 � �$ c C s` | � ttj� | � ttjdtj� t�tj�}| �t�dtj�|� | �t�dtj|�|� d S )N���r )r r
r rO �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 r. r/ r r
r r r r1 � r2 z:ResourceTest.test_prlimit_refcount.<locals>.BadSeq.__len__c s � | d S )Nr3 r r7 �r? r r r9 � s z>ResourceTest.test_prlimit_refcount.<locals>.BadSeq.__getitem__Nr: r rW r r �BadSeq� s rX r )r r rT r rO )r rX r rW r �test_prlimit_refcount� s
�z"ResourceTest.test_prlimit_refcountN)r; r< r= r �unittestZskipIf�sys�platformr r( r, r- r@ rC Z
skipUnlessrM rN �hasattrr r Zrequires_linux_versionrV rY r r r r r
s. �
/ �
r �__main__)rH r[ rZ �testr Z test.supportr r r! �
import_moduler ZTestCaser r; �mainr r r r �<module> s
% �