Your IP : 216.73.216.213
a
XC?h< � @ sZ d dl mZ d dlZd dlZd dlZd dlZd dlZd dlmZ d dl Z d dl
Z
d dl Z d dl m
Z
d dlmZ e
�d�Zdd� ZdKd d
�ZdLd d �Zejd
d� �Zdd� Zi fdd�ZG dd� dedd��Zejdd� �ZG dd� dedd��ZG dd � d ed d ��Z d d �d d!�Z d"d#� Z d$d%� Z d&d'� Z G d(d)� d)e j!�Z"G d*d+� d+e j!�Z#G d,d-� d-e j!�Z$G d.d/� d/e"�Z%G d0d1� d1e"�Z&G d2d3� d3e"�Z'G d4d5� d5e"�Z(G d6d7� d7e"�Z)G d8d9� d9e"�Z*G d:d;� d;e"�Z+G d<d=� d=e"�Z,G d>d?� d?e"�Z-G d@dA� dAe"�Z.G dBdC� dCe"�Z/G dDdE� dEedEdF��Z0e �1dG�G dHdI� dIe"��Z2e3dJk�rVe �4� dS )M� )�
namedtupleN)�dedent)�support)�
script_helperZ_xxsubinterpretersc C s: t �� \}}| �dd�}td|� d|� d��}|t|�fS )N�
z
z-
import contextlib
with open(zU, 'w') as spipe:
with contextlib.redirect_stdout(spipe):
z
)�os�pipe�replacer �open)�script�r�wZindented�wrapped� r �A/opt/alt/python39/lib64/python3.9/test/test__xxsubinterpreters.py�_captured_script s ��
r c C sJ t |�\}}|�& t�| ||� |�� W d � S 1 s<0 Y d S �N)r � interpreters�
run_string�read)�interpZrequest�sharedr Zrpiper r r � _run_output" s r c C sL |d u rt j}t�� }|| }t�| �sHt�� |kr<td��t�d� q d S )Nzinterp is not runningg{�G�z�?)r Z
SHORT_TIMEOUT�time� monotonicr �
is_running� RuntimeError�sleep)r ZtimeoutZ
start_timeZdeadliner r r �_wait_for_interp_to_run) s
r c # s| t �� \�}� �fdd�}tj|d�}|�� t� � d V t|d��}|�d� W d � n1 sf0 Y |�� d S )Nc s t �� td�� d��� d S )Nz7
# wait for "signal"
with open(z5) as rpipe:
rpipe.read()
)r r r r �r r r r �run: s
�z_running.<locals>.run��targetr
Zdone) r r � threading�Thread�startr r
�write�join)r r
r �tZspiper r r �_running7 s (r) c K s t | ||� d S r )� _run_interp)�id�sourcer r r r �
run_interpV s r- c C sF t |�}t�� }|| kr4t�� |kr(t�t||� nt�| ||� d S r )r r �get_main� get_currentr �execr )r+ r, r Z_mainns�mainr r r r* Z s r* c s* e Zd Zedd� �Zd� fdd� Z� ZS )� Interpreterc C s( t || �r|S t |t�r | |�S t�d S r )�
isinstance�str�NotImplementedError)�cls�rawr r r �from_rawh s
zInterpreter.from_rawNc s� t �� }||kr6|sd}n|dkr0td�|���|}nX|d urp|sHd}n|dkrXtd��t|t j�s�t �|�}n |r||dkr�d}|}nt �� }t� �| ||�}|S )Nr1 z)name mismatch (expected "main", got "{}")r z!name mismatch (unexpected "main")) r r. �
ValueError�formatr3 �
InterpreterID�create�super�__new__)r6 �namer+ r1 �self�� __class__r r r> q s, � zInterpreter.__new__)NN)�__name__�
__module__� __qualname__� classmethodr8 r> �
__classcell__r r rA r r2 f s
r2 zname idc c s$ z
d V W n t jy Y n0 d S r )r �ChannelClosedErrorr r r r �expect_channel_closed� s
rI c s6 e Zd Zd
� fdd� Zdd� Zdd� Zdd � Z� ZS ) �
ChannelActionNc s& |sd}|sd}t � �| |||�}|S )N�bothr1 �r= r> )r6 �action�endr r@ rA r r r> � s zChannelAction.__new__c O sb | j dkr | jdvrJt| j��n*| j dv r@| jdvrJt| j��n
t| j ��| jdvr^t| j��d S )N�use)�same�opposite�send�recv��close� force-close)rK rP rQ rR rS )r1 rP �other�extra)rM rN r9 r )r@ �args�kwargsr r r �__init__� s
zChannelAction.__init__c C s2 | j dkr|S | j dkr(|dkr$dS dS | j S d S )NrP rQ rR rS �rN �r@ rN r r r � resolve_end� s
zChannelAction.resolve_endc C st | j dkr|S | j dkr(|d u r$t�|S | j dkrB|d u r>t�|S | j dkrp|jdkrZ|S |rl|jdkrl|S t�d S )NrP rW rX r1 )r r r? )r@ r rW rX r r r �resolve_interp� s
z ChannelAction.resolve_interp)NN)rC rD rE r> r[ r^ r_ rG r r rA r rJ � s rJ zaction end interpc sB e Zd Zddd�� fdd� Zdd� Zdd � Zd
d �d d
�Z� ZS )� ChannelStater F��closedc s t � �| ||�}|S r rL )r6 �pendingrb r@ rA r r r> � s zChannelState.__new__c C s t | �| jd | jd�S �N� ra ��typerc rb �r@ r r r �incr� s zChannelState.incrc C s t | �| jd | jd�S rd rf rh r r r �decr� s zChannelState.decrT��forcec C s2 | j r|r| jdkr| S t| �|r&dn| jdd�S )Nr Tra )rb rc rg )r@ rl r r r rU � s zChannelState.close)r )rC rD rE r> ri rj rU rG r r rA r r` � s r` zpending closedT��
hideclosedc C sr |j r(|dkr"|dkr"|jr"d}q,d}nd}zt| |||�}W n& tjyd |sX|sX� |�� }Y n
0 |rnd�|S )NrO rS FT.)rb rc � _run_actionr rH rU )�cidrM rN �statern Z
expectfail�resultr r r �
run_action� s rs c C s |dkr�|dkr$t �| d� |�� S |dkrx|jsdzt �| � W n t jyX | Y S 0 td��q�t �| � |�� S nt|��n~|dkr�i }|dv r�d||< t j | fi |�� |�
� S |d k� r�d
di}|dv r�d||< t j | fi |�� |j
dd �S t|��d S ) NrO rR � spamrS zexpected ChannelEmptyErrorrU �rS rR TrV rl rk ) r � channel_sendri rc � channel_recv�ChannelEmptyError� Exceptionrj r9 �
channel_closerU )rp rM rN rq rZ r r r ro � s8
� ro c C s> t �� D ]0} | dkrqzt �| � W q ty6 Y q0 qd S )Nr )r �list_all�destroyr �r+ r r r �clean_up_interpreters s r~ c C s6 t �� D ](} zt �| � W q t jy. Y q0 qd S r )r �channel_list_all�channel_destroy�ChannelNotFoundError�rp r r r �clean_up_channels s
r� c @ s e Zd Zdd� ZdS )�TestBasec C s t � t� d S r �r~ r� rh r r r �tearDown) s zTestBase.tearDownN)rC rD rE r� r r r r r� ' s r� c @ s e Zd Zdd� Zdd� ZdS )�IsShareableTestsc C sP g d�}|D ]>}| � |�� | �t�|�� W d � q 1 s@0 Y q d S )N)Nrt �spam�
i����)�subTest�
assertTruer � is_shareable)r@ Z
shareables�objr r r �test_default_shareables3 s �z(IsShareableTests.test_default_shareablesc
C s� G dd� d�}G dd� dt �}ddtdttt� t� d||d �|d
�g }|D ]B}| �t|��� | �t� |�� W d � qJ1 s�0 Y qJd S ) Nc @ s e Zd Zdd� Zdd� ZdS )z3IsShareableTests.test_not_shareable.<locals>.Cheesec S s
|| _ d S r �r? )r@ r? r r r r[ D s z<IsShareableTests.test_not_shareable.<locals>.Cheese.__init__c S s | j S r r� rh r r r �__str__F s z;IsShareableTests.test_not_shareable.<locals>.Cheese.__str__N)rC rD rE r[ r� r r r r �CheeseC s r� c @ s e Zd ZdZdS )z5IsShareableTests.test_not_shareable.<locals>.SubBytesz A subclass of a shareable type.N)rC rD rE �__doc__r r r r �SubBytesI s r� TF.g Y@Z Wensleydalert )
�bytes�NotImplementedrg �objectry r� �repr� assertFalser r� )r@ r� r� Znot_shareablesr� r r r �test_not_shareableB s( ��z#IsShareableTests.test_not_shareableN)rC rD rE r� r� r r r r r� 1 s r� c s` e Zd Z� fdd�Z� fdd�Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z � Z S )�ShareableTypeTestsc s t � �� t�� | _d S r )r= �setUpr �channel_createrp rh rA r r r� e s
zShareableTypeTests.setUpc s t �| j� t� �� d S r )r r� rp r= r� rh rA r r r� i s zShareableTypeTests.tearDownc C sr |D ]h}| � |��J t�| j|� t�| j�}| �||� | �t|�t|�� W d � q1 sb0 Y qd S r )r� r rv rp rw � assertEqual�assertIsrg )r@ �valuesr� �gotr r r �_assert_valuesm s z!ShareableTypeTests._assert_valuesc C s^ dD ]T}| � |��6 t�| j|� t�| j�}| �||� W d � q1 sN0 Y qd S )Nr )r� r rv rp rw r� )r@ r� r� r r r �test_singletonsx s
z"ShareableTypeTests.test_singletonsc C s | � dd| jg� d S )Nrt i' )r� rp rh r r r �
test_types� s
�z ShareableTypeTests.test_typesc C s | � dd� tdd�D �� d S )Nc s s | ]}|j d ddd�V qdS )� �littleT)ZsignedN)�to_bytes)�.0�ir r r � <genexpr>� s �z0ShareableTypeTests.test_bytes.<locals>.<genexpr>���� )r� �rangerh r r r �
test_bytes� s
�z ShareableTypeTests.test_bytesc C s | � g d�� d S )N)z hello worldu 你好世界� )r� rh r r r � test_strs� s z ShareableTypeTests.test_strsc C s* | � t�tdd�tjtj d g�� d S )Nr� r� re )r� � itertools�chainr� �sys�maxsizerh r r r �test_int� s �zShareableTypeTests.test_intc
C s� t jd t j d dd g}|D ]f}| �|��H | �t�� t�| j|� W d � n1 s^0 Y W d � q 1 s|0 Y q d S )Nre r� �� )r� r� r� � assertRaises�
OverflowErrorr rv rp )r@ Zintsr� r r r �test_non_shareable_int� s
� z)ShareableTypeTests.test_non_shareable_int)
rC rD rE r� r� r� r� r� r� r� r� r� rG r r rA r r� c s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� ListAllTestsc C s" t �� }t �� }| �||g� d S r )r r. r{ r� )r@ r1 �idsr r r � test_initial� s zListAllTests.test_initialc C s6 t �� }t �� }t �� }t �� }| �||||g� d S r )r r. r< r{ r� �r@ r1 �first�secondr� r r r �test_after_creating� s
z ListAllTests.test_after_creatingc C s> t �� }t �� }t �� }t �|� t �� }| �|||g� d S r )r r. r< r| r{ r� r� r r r �test_after_destroying� s
z"ListAllTests.test_after_destroyingN)rC rD rE r� r� r� r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )�GetCurrentTestsc C s. t �� }t �� }| �||� | �|t j� d S r )r r. r/ r� �assertIsInstancer; )r@ r1 �curr r r � test_main� s zGetCurrentTests.test_mainc C sR t �� }t �� }t|td��}t|�� �}t �� \}}| �||� | � ||� d S )Nz�
import _xxsubinterpreters as _interpreters
cur = _interpreters.get_current()
print(cur)
assert isinstance(cur, _interpreters.InterpreterID)
)
r r. r< r r �int�stripr{ r� ZassertNotEqual)r@ r1 r �outr� �_�expectedr r r �test_subinterpreter� s z#GetCurrentTests.test_subinterpreterN)rC rD rE r� r� r r r r r� � s r� c @ s e Zd Zdd� Zdd� ZdS )� GetMainTestsc C s0 t �� \}t �� }| �||� | �|t j� d S r )r r{ r. r� r� r; )r@ r� r1 r r r �test_from_main� s
zGetMainTests.test_from_mainc C s< t �� \}t �� }t|td��}t|�� �}| �||� d S )Nz�
import _xxsubinterpreters as _interpreters
main = _interpreters.get_main()
print(main)
assert isinstance(main, _interpreters.InterpreterID)
)r r{ r< r r r� r� r� )r@ r� r r� r1 r r r �test_from_subinterpreter� s
z%GetMainTests.test_from_subinterpreterN)rC rD rE r� r� r r r r r� � s r� c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )�IsRunningTestsc C s t �� }| �t �|�� d S r )r r. r� r )r@ r1 r r r r� � s zIsRunningTests.test_mainc C sd t �� }| �t �|�� t|�� | �t �|�� W d � n1 sF0 Y | �t �|�� d S r )r r< r� r r) r� �r@ r r r r r� � s
.z"IsRunningTests.test_subinterpreterc C s2 t �� }t|td|� d���}| �|�� d� d S )Nz`
import _xxsubinterpreters as _interpreters
if _interpreters.is_running(zZ):
print(True)
else:
print(False)
�True)r r< r r r� r� )r@ r r� r r r r� � s
� z'IsRunningTests.test_from_subinterpreterc C sJ t �� }t �|� | �t�� t �|� W d � n1 s<0 Y d S r )r r< r| r� r r r� r r r �test_already_destroyed� s
z%IsRunningTests.test_already_destroyedc C s8 | � t�� t�d� W d � n1 s*0 Y d S �Ni@B )r� r r r rh r r r �test_does_not_exist s z"IsRunningTests.test_does_not_existc C s8 | � t�� t�d� W d � n1 s*0 Y d S �Nr� )r� r9 r r rh r r r � test_bad_id s zIsRunningTests.test_bad_idN) rC rD rE r� r� r� r� r� r� r r r r r� � s r� c @ sD e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dS )�InterpreterIDTestsc C s" t jddd�}| �t|�d� d S )Nr� Trk )r r; r� r� �r@ r+ r r r �
test_with_int s z InterpreterIDTests.test_with_intc C s4 G dd� dt �}tj|� dd�}| �t|�d� d S )Nc @ s e Zd Zdd� ZdS )z.InterpreterIDTests.test_coerce_id.<locals>.Intc S s dS �Nr� r rh r r r � __index__ s z8InterpreterIDTests.test_coerce_id.<locals>.Int.__index__N�rC rD rE r� r r r r �Int s r� Trk r� )r4 r r; r� r� )r@ r� r+ r r r �test_coerce_id s z!InterpreterIDTests.test_coerce_idc C sf | � ttjt� � | � ttjd� | � ttjd� | � ttjd� | � ttjd� | � ttjd� d S �Ng $@�10s 10r� l )r� � TypeErrorr r; r� r9 r� rh r r r r� s z InterpreterIDTests.test_bad_idc C sH t �� }| �t��" t �t|�d � W d � n1 s:0 Y d S �Nre )r r� r� r r; r� r� r r r r� $ s z&InterpreterIDTests.test_does_not_existc C s" t jddd�}| �t|�d� d S �Nr� Trk r� )r r; r� r4 r� r r r �test_str) s zInterpreterIDTests.test_strc C s" t jddd�}| �t|�d� d S )Nr� Trk zInterpreterID(10))r r; r� r� r� r r r � test_repr- s z InterpreterIDTests.test_reprc C s( t �� }t �t|��}t �� }| �||k� | �||k� | �|t|�k� | �t|�|k� | �|tt|��k� | �tt|��|k� | �|tt|��d k� | �|tt|��k� | �|dd k� | �|td�k� | �|dk� | �||k� | �||k� | �||k� | �||k� d S �N皙�����?r� r� �infr� )r r< r; r� r� �floatr� r4 �r@ �id1�id2�id3r r r �
test_equality1 s$ z InterpreterIDTests.test_equalityN)
rC rD rE r� r� r� r� r� r� r� r r r r r�
s r� c @ sN e Zd Zdd� Ze�d�dd� �Zdd� Zdd � Zd
d � Z d d
� Z
dd� Z dS )� CreateTestsc C s* t �� }| �|t j� | �|t �� � d S r )r r< r� r; �assertInr{ r� r r r � test_in_mainJ s zCreateTests.test_in_mainz*enable this test when working on pystate.cc C sD t � }td�D ] }t�� }t�|� |�|� q| �t|�d� d S )N�d )�setr� r r< r| �addr� �len)r@ �seenr� r+ r r r �test_unique_idP s
zCreateTests.test_unique_idc sn t �� �d � � �fdd�}t j|d�}�� |�� W d � n1 sH0 Y |�� | �� t�� � d S )Nc s t �� � ��� ��� d S r )r r< �acquire�releaser �r+ �lockr r �f] s z%CreateTests.test_in_thread.<locals>.fr! )r# ZLockr$ r% r' r� r r{ �r@ r� r( r r� r �test_in_threadZ s &zCreateTests.test_in_threadc C sJ t �� \}t �� }t|td��}t|�� �}| �tt �� �|||h� d S )Nz�
import _xxsubinterpreters as _interpreters
id = _interpreters.create()
print(id)
assert isinstance(id, _interpreters.InterpreterID)
) r r{ r< r r r� r� r� r� )r@ r1 r� r� r� r r r �test_in_subinterpreteri s
z"CreateTests.test_in_subinterpreterc s^ t �� \}t �� � d �� �fdd�}tj|d�}|�� |�� | �tt �� �|� �h� d S )Nc s t � td��} t| �� ��d S )Nz�
import _xxsubinterpreters as _interpreters
id = _interpreters.create()
print(id)
)r r r� r� )r� �r� r� r r r� z s z6CreateTests.test_in_threaded_subinterpreter.<locals>.fr! ) r r{ r< r# r$ r% r' r� r� )r@ r1 r� r( r r� r � test_in_threaded_subinterpreterv s
z+CreateTests.test_in_threaded_subinterpreterc C sj t t�� �}g }td�D ]}t�� }|�|� q|D ]}t�|� q4t�� }| �t t�� �||hB � d S �N� )r� r r{ r� r< �appendr| r� �r@ �beforer� r� r+ r r r �test_after_destroy_all� s z"CreateTests.test_after_destroy_allc C s` t t�� �}t�� }t�� }t�� }t�|� t�|� t�� }| �t t�� �|||hB � d S r )r� r r{ r< r| r� )r@ r r� r� r� r+ r r r �test_after_destroy_some� s
z#CreateTests.test_after_destroy_someN) rC rD rE r� �unittest�skipr� r� r� r r r r r r r r� H s
r� c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )� DestroyTestsc C sf t �� }t �� }t �� }| �|t �� � t �|� | �|t �� � | �|t �� � | �|t �� � d S r )r r< r� r{ r| � assertNotInr� r r r �test_one� s
zDestroyTests.test_onec C sv t t�� �}t � }td�D ]}t�� }|�|� q| �t t�� �||B � |D ]}t�|� qN| �t t�� �|� d S r )r� r r{ r� r< r� r� r| r r r r �test_all� s zDestroyTests.test_allc sl t �� \� ��t�� t �� � W d � n1 s40 Y � �fdd�}tj|d�}|�� |�� d S )Nc s8 �� t�� t�� � W d � n1 s*0 Y d S r �r� r r r| r �r1 r@ r r r� � s z!DestroyTests.test_main.<locals>.fr! ) r r{ r� r r| r# r$ r% r' r� r r r r� � s
( zDestroyTests.test_mainc C sJ t �� }t �|� | �t�� t �|� W d � n1 s<0 Y d S r )r r< r| r� r r� r r r r� � s
z#DestroyTests.test_already_destroyedc C s8 | � t�� t�d� W d � n1 s*0 Y d S r� r rh r r r r� � s z DestroyTests.test_does_not_existc C s8 | � t�� t�d� W d � n1 s*0 Y d S r� )r� r9 r r| rh r r r r� � s zDestroyTests.test_bad_idc C sJ t �� \}t �� }td|� d��}t �||� | �tt �� �||h� d S )Nzo
import _xxsubinterpreters as _interpreters
try:
_interpreters.destroy(zD)
except RuntimeError:
pass
�r r{ r< r r r� r� )r@ r1 r+ r r r r �test_from_current� s
�
z DestroyTests.test_from_currentc C sR t �� \}t �� }t �� }td|� d��}t �||� | �tt �� �||h� d S )NzZ
import _xxsubinterpreters as _interpreters
_interpreters.destroy(�)
r )r@ r1 r� r� r r r r �test_from_sibling� s
�
z DestroyTests.test_from_siblingc s4 t �� � � fdd�}tj|d�}|�� |�� d S )Nc s t �� � d S r )r r| r r} r r r� � s z.DestroyTests.test_from_other_thread.<locals>.fr! )r r< r# r$ r% r' r� r r} r �test_from_other_thread� s
z#DestroyTests.test_from_other_threadc C s� t �� \}t �� }t|��| | jt �|�d|� d�d� | jtd|� d�d�� t �|� W d � n1 sn0 Y | �t �|�� W d � n1 s�0 Y d S )NzInterp z& should be running before destruction.)�msgz%Should not be able to destroy interp z while it's still running.) r r{ r< r) r� r r� r r| )r@ r1 r r r r �test_still_running� s
�
�(z DestroyTests.test_still_runningN)
rC rD rE r r
r� r� r� r� r r r r r r r r r
� s
r
c s e Zd Zed�ZdZ� fdd�Z� fdd�Zdd� Zd d
� Z d d � Z
e � e
ed
�d�dd� �Zdd� Zdd� Zdd� Zdd� Zdd� Zdd � Zejd5d d ��Zd d!� Zd"d#� Zd$d%� Zd&d'� Z d(d)� Z d*d+� Z d,d-� Z d.d/� Z d0d1� Z!e �"d2�d3d4� �Z#� Z$S )6�RunStringTestszJ
with open('{}', 'w') as out:
out.write('{}')
r� c s t � �� t�� | _d | _d S r )r= r� r r< r+ �_fsrh rA r r r� s
zRunStringTests.setUpc s" | j d ur| j �� t� �� d S r )r rU r= r� rh rA r r r� s
zRunStringTests.tearDownc C sV t d�\}}|�& t�| j|� |�� }W d � n1 s<0 Y | �|d� d S )N�print("it worked!", end="")�
it worked!)r r r r+ r r� )r@ r �filer� r r r � test_success s
&zRunStringTests.test_successc sr t d�\� }|�B � �fdd�}tj|d�}|�� |�� |�� }W d � n1 sX0 Y ��|d� d S )Nr c s t ��j� � d S r )r r r+ r �r r@ r r r� $ s z(RunStringTests.test_in_thread.<locals>.fr! r )r r# r$ r% r' r r� )r@ r r� r( r� r r r r� ! s &z RunStringTests.test_in_threadc C s` t jdd�}td�\}}|�$ t �||� |�� }W d � n1 sF0 Y | �|d� d S )NF)�isolatedz�
import threading
def f():
print('it worked!', end='')
t = threading.Thread(target=f)
t.start()
t.join()
r )r r< r r r r� )r@ Z subinterpr r r� r r r �test_create_thread. s &z!RunStringTests.test_create_thread�forkztest needs os.fork()c C s� dd l }|�d��j}|�d� |�� d}td|j� d|� d��}t�| j|� |� d� |�
� }| � ||� W d � n1 s�0 Y d S )Nr zw+r� zspam spam spam spam spamz�
import os
try:
os.fork()
except RuntimeError:
with open('z3', 'w') as out:
out.write('z')
) �tempfileZNamedTemporaryFiler& �flushr r? r r r+ �seekr r� )r@ r! r r� r Zcontentr r r � test_fork? s
��
zRunStringTests.test_forkc C sf t | j��H | �t�� t�| jd� W d � n1 s:0 Y W d � n1 sX0 Y d S )N�
print("spam"))r) r+ r� r r r rh r r r �test_already_runningU s z#RunStringTests.test_already_runningc C sT d}|t �� v r|d7 }q| �t�� t �|d� W d � n1 sF0 Y d S )Nr re r% )r r{ r� r r r� r r r r� Z s
z"RunStringTests.test_does_not_existc C s: | � t�� t�dd� W d � n1 s,0 Y d S )Nr� r% )r� r9 r r rh r r r �
test_error_ida s z RunStringTests.test_error_idc C s: | � t�� t�dd� W d � n1 s,0 Y d S )Nr� r% )r� r� r r rh r r r r� e s zRunStringTests.test_bad_idc C s< | � t�� t�| jd� W d � n1 s.0 Y d S r� �r� r� r r r+ rh r r r �test_bad_scripti s z RunStringTests.test_bad_scriptc C s< | � t�� t�| jd� W d � n1 s.0 Y d S )Ns
print("spam")r( rh r r r �test_bytes_for_scriptm s z$RunStringTests.test_bytes_for_scriptNc c sz | � tj��}d V W d � n1 s(0 Y |d u r\| �t|j��d�d t|�� n| �t|j�d�||�� d S )N�:r z{}: {})r� r �RunFailedErrorr� r4 � exception�splitr: )r@ �exctyper Zcaughtr r r �assert_run_failedq s $�
�z RunStringTests.assert_run_failedc C s< | � t�� t�| jd� W d � n1 s.0 Y d S )Nz print("spam")r0 � SyntaxErrorr r r+ rh r r r �test_invalid_syntax| s z"RunStringTests.test_invalid_syntaxc C s> | � td�� t�| jd� W d � n1 s00 Y d S )Nr� zraise Exception("spam"))r0 ry r r r+ rh r r r � test_failure� s zRunStringTests.test_failurec C s> | � td�� t�| jd� W d � n1 s00 Y d S )N�42zraise SystemExit(42))r0 �
SystemExitr r r+ rh r r r �test_SystemExit� s z RunStringTests.test_SystemExitc C s~ | � t��" t�| jtd�� W d � n1 s20 Y | � td��" t�| jtd�� W d � n1 sp0 Y d S )NzG
import sys
sys.exit()
r4 zI
import sys
sys.exit(42)
)r0 r5 r r r+ r rh r r r �
test_sys_exit� s 0z RunStringTests.test_sys_exitc C s� t �� \}}ddd d�}td|� d��}t�| j||� t|d��}t�|�}W d � n1 sb0 Y | � |d d� | � |d d
� | � |d d � | �
|d
� d S )N� hams -1)r� �eggs�cheddarz�
eggs = int(eggs)
spam = 42
result = spam + eggs
ns = dict(vars())
del ns['__builtins__']
import pickle
with open(�C, 'wb') as chan:
pickle.dump(ns, chan)
�rbr� �* r9 r� rr �) r: ) r r r r r r+ r
�pickle�loadr� Z assertIsNone�r@ r r
r r �chan�nsr r r �test_with_shared� s ��
(z RunStringTests.test_with_sharedc C s� t �| jtd�� ddi}td�}t �| j||� t�� \}}td|� d��}t �| j|� t|d��}t�|�}W d � n1 s�0 Y | � |d d d � | � |d
d d� | � |d d� d S ) Nzj
spam = 'eggs'
ns1 = dict(vars())
del ns1['__builtins__']
r� r8 zL
ns2 = dict(vars())
del ns2['__builtins__']
zr
ns = dict(vars())
del ns['__builtins__']
import pickle
with open(r; r< �ns1r9 �ns2)
r r r+ r r r r
r? r@ r� )r@ r r r r
rB rC r r r �test_shared_overwrites� s �
(z%RunStringTests.test_shared_overwritesc C s| t �� \}}ddi}td|� d��}t�| j||� t|d��}t�|�}W d � n1 s^0 Y | � |d d� d S )NrC s not __main__z�
spam = 42
ns = dict(vars())
del ns['__builtins__']
import pickle
with open(r; r< )
r r r r r r+ r
r? r@ r� rA r r r �#test_shared_overwrites_default_vars� s �
(z2RunStringTests.test_shared_overwrites_default_varsc C s� t �� \}}t�| jtd|� d��� t|d��}t�|�}W d � n1 sP0 Y t �� \}}t�| jtd|� d��� t|d��}t�|�}W d � n1 s�0 Y | � d|� | �
d|� | � d|� | � d|� d S )Nz�
spam = True
ns = dict(vars())
del ns['__builtins__']
import pickle
with open(zd, 'wb') as chan:
pickle.dump(ns, chan)
del ns, pickle, chan
r< z�
eggs = False
ns = dict(vars())
del ns['__builtins__']
import pickle
with open(r; r� r9 ) r r r r r+ r r
r? r@ r� r )r@ r r
rB rE rF r r r �test_main_reused� s �
( � ( z RunStringTests.test_main_reusedc
C s� t �� \}}td|� d��}t�| j|� t|d��}t�|�}W d � n1 sT0 Y |� d� |� d� | �
|di d d d dd�� d S ) Nz�
spam = 42
ns = dict(vars())
ns['__builtins__'] = str(ns['__builtins__'])
import pickle
with open(r; r< � __builtins__�
__loader__�__main__r= )rC �__annotations__r� � __package__�__spec__r� ) r r r r r r+ r
r? r@ �popr� )r@ r r
r rB rC r r r � test_execution_namespace_is_main s" �
(
�z/RunStringTests.test_execution_namespace_is_mainzblocking foreverc C s� t d�}t�� �P}t�|d|�}t�|��}|�� }W d � n1 sH0 Y W d � n1 sf0 Y | �|d� d S )Na�
from textwrap import dedent
import threading
import _xxsubinterpreters as _interpreters
id = _interpreters.create()
def f():
_interpreters.run_string(id, dedent('''
import time
# Give plenty of time for the main interpreter to finish.
time.sleep(1_000_000)
'''))
t = threading.Thread(target=f)
t.start()
r r )r r Ztemp_dirr Z make_scriptZ spawn_python�waitr� )r@ r �dirname�filename�procZretcoder r r �test_still_running_at_exit s
Dz)RunStringTests.test_still_running_at_exit)N)%rC rD rE r ZSCRIPTZFILENAMEr� r� r r� r r Z
skipUnless�hasattrr r$ r& r� r' r� r) r* �
contextlib�contextmanagerr0 r2 r3 r6 r7 rD rG rH rI rQ r rV rG r r rA r r s8
! r c @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�ChannelIDTestsc C s0 t jddd�}| �t|�d� | �|jd� d S )Nr� Trk rK )r � _channel_idr� r� rN �r@ rp r r r �test_default_kwargs> s z"ChannelIDTests.test_default_kwargsc C s� t jdddd�}| �|jd� t jddddd�}| �|jd� t jdddd�}| �|jd� t jddddd �}| �|jd� t jddddd�}| �|jd
� d S ) Nr� T�rR rl rR F�rR rS rl �rS rl rS )rS rR rl rK )r r[ r� rN r\ r r r �test_with_kwargsD s z ChannelIDTests.test_with_kwargsc C s4 G dd� dt �}tj|� dd�}| �t|�d� d S )Nc @ s e Zd Zdd� ZdS )z*ChannelIDTests.test_coerce_id.<locals>.Intc S s dS r� r rh r r r r� V s z4ChannelIDTests.test_coerce_id.<locals>.Int.__index__Nr� r r r r r� U s r� Trk r� )r4 r r[ r� r� )r@ r� rp r r r r� T s z ChannelIDTests.test_coerce_idc C sf | � ttjt� � | � ttjd� | � ttjd� | � ttjd� | � ttjd� | � ttjd� d S r� )r� r� r r[ r� r9 r� rh r r r r� \ s zChannelIDTests.test_bad_idc C s> | � t�� tjdddd� W d � n1 s00 Y d S )Nr� F�rR rS )r� r9 r r[ rh r r r �test_bad_kwargsd s z ChannelIDTests.test_bad_kwargsc C sJ t �� }| �t j��" t �t|�d � W d � n1 s<0 Y d S r� )r r� r� r� r[ r� r\ r r r r� h s z"ChannelIDTests.test_does_not_existc C s" t jddd�}| �t|�d� d S r� )r r[ r� r4 r\ r r r r� m s zChannelIDTests.test_strc C s� t jddd�}| �t|�d� t jdddd�}| �t|�d� t jdddd�}| �t|�d� t jddddd �}| �t|�d� d S )
Nr� Trk z
ChannelID(10)r^ zChannelID(10, send=True)r` zChannelID(10, recv=True)r_ )r r[ r� r� r\ r r r r� q s zChannelIDTests.test_reprc C s( t �� }t �t|��}t �� }| �||k� | �||k� | �|t|�k� | �t|�|k� | �|tt|��k� | �tt|��|k� | �|tt|��d k� | �|tt|��k� | �|dd k� | �|td�k� | �|dk� | �||k� | �||k� | �||k� | �||k� d S r� )r r� r[ r� r� r� r� r4 )r@ �cid1�cid2Zcid3r r r r� ~ s$ z ChannelIDTests.test_equalityN) rC rD rE r] ra r� r� rc r� r� r� r� r r r r rZ < s
rZ c @ s> e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Ze�d)�d*d+� �Zd,d-� Zd.d/� Zd0d1� Z d2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z!d<d=� Z"d>d?� Z#d@dA� Z$dBdC� Z%dDdE� Z&dFdG� Z'dHdI� Z(dJdK� Z)dLS )M� ChannelTestsc C s t �� }| �|t j� d S r )r r� r� Z ChannelIDr\ r r r �test_create_cid� s z ChannelTests.test_create_cidc C sr t �� }t �� }t �� }t �� }t �� }| �|t|�d � | �|t|�d � | �t|�t|� |||h� d S r� )r r r� r� r� r� )r@ r r� r� r� Zafterr r r �test_sequential_ids� s z ChannelTests.test_sequential_idsc C s\ t �� }t|td��}t|�� �}t �� }t|td��}t|�� �}| �|t|�d � d S )Nz�
import _xxsubinterpreters as _interpreters
cid = _interpreters.channel_create()
print(cid)
re )r r< r r r� r� r� )r@ r� r� rd r� re r r r �test_ids_global� s z ChannelTests.test_ids_globalc C s@ t �� }t j|dd�}t j|dd�}| �|g � | �|g � dS )z=Test listing interpreters for a channel with no associations.T�rR FN)r r� �channel_list_interpretersr� )r@ rp � send_interps� recv_interpsr r r �#test_channel_list_interpreters_none� s
z0ChannelTests.test_channel_list_interpreters_nonec C s� t �� }t �� }t �|d� t j|dd�}t j|dd�}| �||g� | �|g � t �� }t|td|� d��� t j|dd�}t j|dd�}| �||g� | �||g� dS )z(Test basic listing channel interpreters.rR Trj F�e
import _xxsubinterpreters as _interpreters
obj = _interpreters.channel_recv(r N) r r. r� rv rk r� r< r r )r@ �interp0rp rl rm �interp1r r r �$test_channel_list_interpreters_basic� s � z1ChannelTests.test_channel_list_interpreters_basicc C s� t �� }t �� }t �� }t �� }t �� }t �|d� t|td|� d��� t|td|� d��� t|td|� d��� t j|dd�}t j|dd�}| �t |�||h� | �t |�||h� d S )
z?Test listing interpreters for a channel with many associations.rR �_
import _xxsubinterpreters as _interpreters
_interpreters.channel_send(z, "send")
ro r Trj FN)
r r. r< r� rv r r rk r� r� )r@ rp rq �interp2�interp3rp rl rm r r r �'test_channel_list_interpreters_multiple� s& � � � z4ChannelTests.test_channel_list_interpreters_multiplec C s� t �� }t �� }t �� }t �|d� t|td|� d��� t j|dd�}t j|dd�}| �||g� | �||g� t � |� t j|dd�}t j|dd�}| �||g� | �|g � dS )z?Test listing channel interpreters with a destroyed interpreter.rR ro r Trj FN)
r r. r< r� rv r r rk r� r| �r@ rp rq rp rl rm r r r �(test_channel_list_interpreters_destroyed� s �
z5ChannelTests.test_channel_list_interpreters_destroyedc C s> t �� }t �� }t �� }t �� }t �|d� t|td|� d��� t �|d� t|td|� d��� t j|dd�}t j|dd�}| �t |�d� | �t |�d� t j
|dd� t j|dd�}t j|dd�}| �t |�d � | �t |�d� t|td
|� d��� t j|dd�}t j|dd�}| �t |�d � | �||g� d S ) z:Test listing channel interpreters with a released channel.�dataro r Trj Fre r� r �b
import _xxsubinterpreters as _interpreters
_interpreters.channel_release(N) r r. r< r� rv r r rk r� r� �channel_release)r@ rp rq rt rp rl rm r r r �'test_channel_list_interpreters_released s8 � � � z4ChannelTests.test_channel_list_interpreters_releasedc C s� t �� }t �� }t �� }t �|d� t j|dd�}t j|dd�}| �t|�d� | �t|�d� t j|dd� | � t j
�� t j|dd� W d � n1 s�0 Y | � t j
�� t j|dd� W d � n1 s�0 Y dS ) z8Test listing channel interpreters with a closed channel.rR Trj Fre r rk N) r r. r< r� rv rk r� r� rz r� rH rw r r r �%test_channel_list_interpreters_closed4 s ,z2ChannelTests.test_channel_list_interpreters_closedc C sX t �� }t �� }t �� }t �|d� t j|dd�}t j|dd�}| �t|�d� | �t|�d� t j|dd� | � t j
�� t j|dd� W d � n1 s�0 Y t j|dd�}| �t|�d� t |t d|� d ��� | � t j
�� t j|dd� W d � n1 �s0 Y | � t j
�� t j|dd� W d � n1 �sJ0 Y dS )
zCTest listing channel interpreters with a channel's send end closed.rR Trj Fre r N�`
import _xxsubinterpreters as _interpreters
_interpreters.channel_close(�, force=True)
)
r r. r< r� rv rk r� r� rz r� rH r r rw r r r �.test_channel_list_interpreters_closed_send_endJ s( ,� .z;ChannelTests.test_channel_list_interpreters_closed_send_endc C s> t �� }d}t �||� t �|�}| �||� | �||� d S �Nrt )r r� rv rw r� Z assertIsNot)r@ rp Zorigr� r r r �test_send_recv_mainn s
z ChannelTests.test_send_recv_mainc C s t �� }t|td��}d S )Na6
import _xxsubinterpreters as _interpreters
cid = _interpreters.channel_create()
orig = b'spam'
_interpreters.channel_send(cid, orig)
obj = _interpreters.channel_recv(cid)
assert obj is not orig
assert obj == orig
)r r< r r )r@ r� r� r r r � test_send_recv_same_interpreterw s z,ChannelTests.test_send_recv_same_interpreterc C s@ t �� }t �� }t|td|� d���}t �|�}| �|d� d S )Nrs �, b'spam')
rt )r r� r< r r rw r� )r@ rp r� r� r� r r r �%test_send_recv_different_interpreters� s �
z2ChannelTests.test_send_recv_different_interpretersc sV t �� � � fdd�}tj|d�}|�� t �� d� |�� t �� �}| �|d� d S )Nc sD zt �� �} W q4W q t jy0 t�d� Y q 0 q t �� | � d S )Nr� )r rw rx r r rv )r� r� r r r� � s
z8ChannelTests.test_send_recv_different_threads.<locals>.fr! rt ) r r� r# r$ r% rv r' rw r� �r@ r� r( r� r r� r � test_send_recv_different_threads� s
z-ChannelTests.test_send_recv_different_threadsc sf t �� � t �� �d �� ��fdd�}tj|d�}|�� t �� d� |�� t �� �}| � |d� d S )Nc s t �td� � d� � d����d S )Nz�
import time
import _xxsubinterpreters as _interpreters
while True:
try:
obj = _interpreters.channel_recv(z�)
break
except _interpreters.ChannelEmptyError:
time.sleep(0.1)
assert(obj == b'spam')
_interpreters.channel_send(z, b'eggs')
)r r r �rp r� r� r r r� � s
�
�zIChannelTests.test_send_recv_different_interpreters_and_threads.<locals>.fr! rt � eggs)
r r� r< r# r$ r% rv r' rw r� r� r r� r �1test_send_recv_different_interpreters_and_threads� s
z>ChannelTests.test_send_recv_different_interpreters_and_threadsc C s< | � tj�� t�dd� W d � n1 s.0 Y d S )Nr� rt )r� r r� rv rh r r r �test_send_not_found� s z ChannelTests.test_send_not_foundc C s: | � tj�� t�d� W d � n1 s,0 Y d S r� )r� r r� rw rh r r r �test_recv_not_found� s z ChannelTests.test_recv_not_foundc C sB t �� }| �t j�� t �|� W d � n1 s40 Y d S r )r r� r� rx rw r\ r r r �test_recv_empty� s z ChannelTests.test_recv_emptyc C s� t � }t�� }t�||�}t�|d � t�|d� t�|d� t�|d� t�||�}t�||�}t�|�}t�||�}t�||�}| �||� | �|d � | �|d� | �|d� | �|d� | �||� d S )Nre rt r� )r� r r� rw rv r� r� ) r@ �defaultrp Zobj1Zobj2Zobj3Zobj4Zobj5Zobj6r r r �test_recv_default� s$
z ChannelTests.test_recv_defaultc C sR t �� }t �� }t|td�t|jd��}t �|�}| �|d� | �|� � d� d S )Nz�
import _xxsubinterpreters as _interpreters
print(cid.end)
_interpreters.channel_send(cid, b'spam')
r� rt rR )
r r� r< r r �dictrR rw r� r� �r@ rp r r� r� r r r � test_run_string_arg_unresolved� s
�
z+ChannelTests.test_run_string_arg_unresolvedz(disabled until high-level channels existc C s` t �� }t j|dd�}t �� }t|td�t|jd��}t �|�}| � |d� | � |�
� d� d S )NT)Z_resolvez�
import _xxsubinterpreters as _interpreters
print(chan.id.end)
_interpreters.channel_send(chan.id, b'spam')
)rB rt rR ) r r� r[ r< r r r� rR rw r� r� r� r r r � test_run_string_arg_resolved� s
�
z)ChannelTests.test_run_string_arg_resolvedc C s� t �� }t �|d� t �|� t �|� | �t j�� t �|d� W d � n1 sV0 Y | �t j�� t �|� W d � n1 s�0 Y d S �Nrt r� �r r� rv rw rz r� rH r\ r r r �test_close_single_user s
*z#ChannelTests.test_close_single_userc C s t �� }t �� }t �� }t �|td|� d��� t �|td|� d��� t �|� | �t j��(}t �|td|� d��� W d � n1 s�0 Y | �dt |j
�� | �t j��(}t �|td|� d��� W d � n1 s�0 Y | �dt |j
�� d S )Nrs r� z_
import _xxsubinterpreters as _interpreters
_interpreters.channel_recv(r z,
_interpreters.channel_send(z, b'spam')
rH ) r r� r< r r rz r� r, r� r4 r- )r@ rp r� r� �cmr r r �test_close_multiple_users s(
�
�
�*
�*z&ChannelTests.test_close_multiple_usersc C sb t �� }t �|d� t �|� t �|� | �t j�� t �|� W d � n1 sT0 Y d S r� r� r\ r r r �test_close_multiple_times( s
z&ChannelTests.test_close_multiple_timesc
C s� g d�}|D ]�\}}| � ||f��� t�� }t�|d� t�|� tj|||d� | �tj�� t�|d� W d � n1 s�0 Y | �tj�� t�|� W d � n1 s�0 Y W d � q 1 s�0 Y q d S )N))FF�TF)FT)TTrt rb r� )r� r r� rv rw rz r� rH )r@ ZtestsrR rS rp r r r �test_close_empty1 s
*z ChannelTests.test_close_emptyc C sp t �� }t �|d� t �|d� | �t j�� t �|� W d � n1 sL0 Y t �|� t �|d� d S )Nrt r8 r� �r r� rv r� ZChannelNotEmptyErrorrz rw r\ r r r �%test_close_defaults_with_unused_itemsD s (
z2ChannelTests.test_close_defaults_with_unused_itemsc C s� t �� }t �|d� t �|d� | �t j�� t j|dd� W d � n1 sP0 Y t �|� t �|d� t �|� t �|� t j|dd� d S )Nrt r8 T�rS r� r� r\ r r r �*test_close_recv_with_unused_items_unforcedN s ,
z7ChannelTests.test_close_recv_with_unused_items_unforcedc C s� t �� }t �|d� t �|d� t j|dd� | �t j�� t �|d� W d � n1 s\0 Y t �|� t �|� | �t j�� t �|� W d � n1 s�0 Y d S )Nrt r8 Trj r� �r r� rv rz r� rH rw r\ r r r �*test_close_send_with_unused_items_unforced[ s *
z7ChannelTests.test_close_send_with_unused_items_unforcedc C s� t �� }t �|d� t �|d� | �t j�� t j|ddd� W d � n1 sR0 Y t �|� t �|d� t �|� t �|� t j|dd� d S )Nrt r8 Tru r� r� r� r\ r r r �*test_close_both_with_unused_items_unforcedh s .
z7ChannelTests.test_close_both_with_unused_items_unforcedc C s� t �� }t �|d� t �|d� t j|ddd� | �t j�� t �|d� W d � n1 s^0 Y | �t j�� t �|� W d � n1 s�0 Y d S )Nrt r8 Tr` r� r� r\ r r r �(test_close_recv_with_unused_items_forcedu s *z5ChannelTests.test_close_recv_with_unused_items_forcedc C s� t �� }t �|d� t �|d� t j|ddd� | �t j�� t �|d� W d � n1 s^0 Y | �t j�� t �|� W d � n1 s�0 Y d S )Nrt r8 Tr^ r� r� r\ r r r �(test_close_send_with_unused_items_forced� s *z5ChannelTests.test_close_send_with_unused_items_forcedc C s� t �� }t �|d� t �|d� t j|dddd� | �t j�� t �|d� W d � n1 s`0 Y | �t j�� t �|� W d � n1 s�0 Y d S )Nrt r8 Tr_ r� r� r\ r r r �(test_close_both_with_unused_items_forced� s *z5ChannelTests.test_close_both_with_unused_items_forcedc C s� t �� }t �|� | �t j�� t �|d� W d � n1 s@0 Y | �t j�� t �|� W d � n1 sv0 Y d S r� )r r� rz r� rH rv rw r\ r r r �test_close_never_used� s
*z"ChannelTests.test_close_never_usedc C s� t �� }t �|d� t �� }t �|td|� d��� | �t j�� t �|� W d � n1 s`0 Y | �t j�� t � |� W d � n1 s�0 Y d S )Nrt r~ r )
r r� rv r< r r r� rH rw rz �r@ rp r r r r �!test_close_by_unassociated_interp� s
� (z.ChannelTests.test_close_by_unassociated_interpc C s� t �� }t �|d� t �|d� t �|d� t �|� t j|dd� | �t j�� t �|d� W d � n1 sr0 Y | �t j�� t �|� W d � n1 s�0 Y d S )Nrt Trk r� r� r\ r r r �-test_close_used_multiple_times_by_single_user� s
*z:ChannelTests.test_close_used_multiple_times_by_single_userc C s� t �� }| �t j�� t jddd� W d � n1 s80 Y t �|� | �t j�� t j|dd� W d � n1 s|0 Y d S )Nr� Trj )r r� r� r� rk rz rH r\ r r r �.test_channel_list_interpreters_invalid_channel� s ,
z;ChannelTests.test_channel_list_interpreters_invalid_channelc C s@ t �� }| �t�� t �|� W d � n1 s20 Y d S r )r r� r� r� rk r\ r r r �+test_channel_list_interpreters_invalid_args� s z8ChannelTests.test_channel_list_interpreters_invalid_argsN)*rC rD rE rg rh ri rn rr rv rx r| r} r� r� r� r� r� r� r� r� r� r� r� r r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r rf � sL +$
rf c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )�ChannelReleaseTestsa�
- main / interp / other
- run in: current thread / new thread / other thread / different threads
- end / opposite
- force / no force
- used / not used (associated / not associated)
- empty / emptied / never emptied / partly emptied
- closed / not closed
- released / not released
- creator (interp) / other
- associated interpreter not running
- associated interpreter destroyed
c C s� t �� }t �|d� t �|� t j|ddd� | �t j�� t �|d� W d � n1 s\0 Y | �t j�� t �|� W d � n1 s�0 Y d S �Nrt Trb r� �r r� rv rw r{ r� rH r\ r r r �test_single_user� s
*z$ChannelReleaseTests.test_single_userc C sx t �� }t �� }t �� }t �|td|� d��� t|td|� d|� d���}t �|td|� d��� | �|�� d� d S ) Nrs r� ro z,)
_interpreters.channel_release(z+)
print(repr(obj))
z+
_interpreters.channel_release(r zb'spam')r r� r< r r r r� r� )r@ rp r� r� r� r r r �test_multiple_users s
� ��
� z'ChannelReleaseTests.test_multiple_usersc C s� t �� }t �|d� t �|� t �|� | �t j�� t �|d� W d � n1 sV0 Y | �t j�� t �|� W d � n1 s�0 Y d S r� r� r\ r r r �test_no_kwargs s
*z"ChannelReleaseTests.test_no_kwargsc C sn t �� }t �|d� t �|� t j|ddd� | �t j�� t j|ddd� W d � n1 s`0 Y d S )Nrt Trb r� r\ r r r �test_multiple_times s
z'ChannelReleaseTests.test_multiple_timesc C sj t �� }t �|d� t �|d� t j|ddd� | �t j�� t �|� W d � n1 s\0 Y d S )Nrt r8 Trb )r r� rv r{ r� rH rw r\ r r r �test_with_unused_items) s z*ChannelReleaseTests.test_with_unused_itemsc C s� t �� }t �|� | �t j�� t �|d� W d � n1 s@0 Y | �t j�� t �|� W d � n1 sv0 Y d S r� )r r� r{ r� rH rv rw r\ r r r �test_never_used2 s
*z#ChannelReleaseTests.test_never_usedc C s� t �� }t �|d� t �� }t �|td|� d��� t �|�}t �|� | �t j �� t �|d� W d � n1 sv0 Y | �
|d� d S )Nrt rz r r� ) r r� rv r< r r rw r{ r� rH r� )r@ rp r r� r r r �test_by_unassociated_interp; s
�
*z/ChannelReleaseTests.test_by_unassociated_interpc C sh t �� }t �� }t �|td|� d|� d��� | �t j�� t �|� W d � n1 sZ0 Y d S )Nze
import _xxsubinterpreters as _interpreters
obj = _interpreters.channel_send(z5, b'spam')
_interpreters.channel_release(r )r r� r< r r r� rH rw r� r r r �test_close_if_unassociatedJ s
�� z.ChannelReleaseTests.test_close_if_unassociatedc C sR t �� }t �|d � t �|� t �|d� t j|dd� t �|�}| �|d� d S )Nrt Trj )r r� rv rw r{ r� )r@ rp r� r r r �test_partiallyW s
z"ChannelReleaseTests.test_partiallyc C s� t �� }t �|d� t �|d� t �|d� t �|� t j|ddd� | �t j�� t �|d� W d � n1 st0 Y | �t j�� t �|� W d � n1 s�0 Y d S r� r� r\ r r r �'test_used_multiple_times_by_single_userb s
*z;ChannelReleaseTests.test_used_multiple_times_by_single_userN)rC rD rE r� r� r� r� r� r� r� r� r� r� r� r r r r r� � s '
r� c sz e Zd ZdZ� fdd�Zedd� �Zedd� �Zdd � Zdd d �Z d
d� Z
dd� Z dd� Z dd� Z
dd� Zdd� Z� ZS )�ChannelCloseFixtureFc sx | j r i }n4t�|�}t�|�}t�|�}|j||j||j|i}|sHd}t� �| |||||�}t� |_t� |_ ||_
|S )NrP ) �QUICKr2 r8 r? r= r> r� �_preppedr` �_state�_known)r6 rN r rW rX �creatorZknownr@ rA r r r> w s
�zChannelCloseFixture.__new__c C s | j S r �r� rh r r r rq � s zChannelCloseFixture.statec C s> z| j W S ty8 | �| j�}| �|�| _ | j Y S 0 d S r )�_cid�AttributeError�_get_interpreterr� � _new_channel)r@ r� r r r rp � s zChannelCloseFixture.cidc C s | � |�}| �|� |S r )r� �_prep_interpreterr� r r r �get_interpreter� s
z#ChannelCloseFixture.get_interpreterNc C s2 |d u r| j }|dkr&| jjdkr&dS t| jj�S )NrS rR F)rN rq rb �boolr] r r r �expect_closed_error� s
z'ChannelCloseFixture.expect_closed_errorc C s | � |� d S r )r� r� r r r �prep_interpreter� s z$ChannelCloseFixture.prep_interpreterc C s
|| _ d S r r� )r@ rM rr r r r �
record_action� s z!ChannelCloseFixture.record_actionc C s t � t� d S r r� rh r r r �clean_up� s z ChannelCloseFixture.clean_upc C s@ |j dkrt�� S t�� }t|jd|� d�� t�|�| _| jS )Nr1 a
import _xxsubinterpreters
cid = _xxsubinterpreters.channel_create()
# We purposefully send back an int to avoid tying the
# channel to the other interpreter.
_xxsubinterpreters.channel_send(zC, int(cid))
del _xxsubinterpreters
)r? r r� r- r+ rw r� )r@ r� Zchr r r r� � s
�
z ChannelCloseFixture._new_channelc C sj |dv r| j S |dkr | jS |dkr*| jS |}z| j| }W n$ ty` t|� }| j|< Y n0 |S d S )N)rP r rW rX )r rW rX r� �KeyErrorr2 )r@ r r? r r r r� � s z$ChannelCloseFixture._get_interpreterc C sF |j | jv rd S | j�|j � |jdkr,d S t|j d| j� d�� d S )Nr1 a
import _xxsubinterpreters as interpreters
import test.test__xxsubinterpreters as helpers
ChannelState = helpers.ChannelState
try:
cid
except NameError:
cid = interpreters._channel_id(r )r+ r� r� r? r- rp r� r r r r� � s
�z%ChannelCloseFixture._prep_interpreter)N)rC rD rE r� r> �propertyrq rp r� r� r� r� r� r� r� r� rG r r rA r r� p s
r� z end interp other extra creatorz%these tests take several hours to runc @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
�dd�Z dd� Z
dd� Z d"dd�Z dd� Z
d#dd�Zd d � Zd d � Zd d!� ZdS )$�ExhaustiveChannelTestsa�
- main / interp / other
- run in: current thread / new thread / other thread / different threads
- end / opposite
- force / no force
- used / not used (associated / not associated)
- empty / emptied / never emptied / partly emptied
- closed / not closed
- released / not released
- creator (interp) / other
- associated interpreter not running
- associated interpreter destroyed
- close after unbound
c c s� g V | � dd�D ]"}|V | �� D ]}|| V q$q| � dd�D ]"}|V | �� D ]}|| V qTqB| �dd�D ]z}|V | � dd�D ]*}|| }|V | �� D ]}|| V q�q�| � dd�D ]*}|| }|V | �� D ]}|| V q�q�qr| �dd�D ]�}|V | � dd�D ].}|| }|V | �� D ]}|| V �q*�q| � dd�D ].}|| }|V | �� D ]}|| V �qf�qLq�d S )NrP rW rX )�_iter_close_action_sets� _iter_post_close_action_sets�_iter_use_action_sets)r@ Z closeactionsZ postactionsZ
useactions�actionsr r r �iter_action_sets sB z'ExhaustiveChannelTests.iter_action_setsc c s ||f}t dd|�gV t dd|�gV t dd|�t dd|�gV t dd|�gV t dd|�gV t dd|�t dd|�gV |D ]<}|D ]2}|D ](}t dd|�t dd|�t dd|�gV q�q�q�|D ]P}|D ]F}|D ]<}|D ]2}t dd|�t dd|�t dd|�t dd|�gV q�q�q�q�d S )NrO rS rR �rJ )r@ rq rt �interpsru Zinterp4r r r r� 9 s>
�
�
�
�
�
�
�
�z,ExhaustiveChannelTests._iter_use_action_setsc
c s� d}||f}dD ]4}|r dnd}|D ] }|D ]}t |||�gV q,q$qdD ]<} dD ]2}
|D ](} |D ] } t | d| �t |
d| �gV qbqZqRqJd S )Nru r� rV rU rT rS rR r� )
r@ rq rt Zendsr� rl �opr rN ZrecvopZsendoprS rR r r r r� j s
�
�z.ExhaustiveChannelTests._iter_close_action_setsc c s. dD ]$}t dd|�gV t dd|�gV qd S )N)rP rX rW rO rS rR r� r� r r r r� } s
�
�z3ExhaustiveChannelTests._iter_post_close_action_setsc C s |D ]}| � ||� qd S r )rs )r@ �fixr� rM r r r � run_actions� s z"ExhaustiveChannelTests.run_actionsTrm c C s� |� |j�}|�|j|j|j�}|�|� |jdkrZt|j |j
||j |d�}|� ||� n�t
�� }t|jd|j � dt|j
�� dt|�� dt|j �� d|� d|� d|� d�� tt�t
�|�d �tt
�|��d
�}|� ||� d S ) Nr1 rm zB
result = helpers.run_action(
z,
z!,
hideclosed=zB,
)
interpreters.channel_send(zR, result.pending.to_bytes(1, 'little'))
interpreters.channel_send(z2, b'X' if result.closed else b'')
r� )rc rb )r^ rN r_ r rW rX r� r? rs rp rM rq r� r r� r- r+ r� r` r� �
from_bytesrw r� )r@ r� rM rn rN r rr r� r r r rs � sB
������� �
�z!ExhaustiveChannelTests.run_actionc c sB g d�}|D ]0\}}}dD ] }dD ]}t |||||�V q"qq d S )N))r1 r rX )r r1 rX )rq rt rX )rq rt r1 )rP rW r� rb )r� )r@ r r rW rX r� rN r r r �
iter_fixtures� s
z$ExhaustiveChannelTests.iter_fixturesc C st |rdnd}t ||jd�}|�� s4| j||dd� n<| �tj�� | j||dd� W d � n1 sf0 Y d S )NrV rU rP Frm )rJ rN r� rs r� r rH )r@ r� rl r� rU r r r �_close� s z ExhaustiveChannelTests._closeNc C s. |d u s|j dkr�| �tj�� t�|j� W d � n1 s@0 Y | �tj�� t�|jd� W d � n1 sz0 Y | �tj�� t�|j� W d � n1 s�0 Y | �tj�� tj|jdd� W d � n1 s�0 Y n0t|j d� t|j d� t|j d� t|j d� d S ) Nr1 rt Trk zz
with helpers.expect_channel_closed():
interpreters.channel_recv(cid)
z�
with helpers.expect_channel_closed():
interpreters.channel_send(cid, b'spam')
z{
with helpers.expect_channel_closed():
interpreters.channel_close(cid)
z�
with helpers.expect_channel_closed():
interpreters.channel_close(cid, force=True)
)
r? r� r rH rw rp rv rz r- r+ )r@ r� r r r r �_assert_closed_in_interp� s *,*0 z/ExhaustiveChannelTests._assert_closed_in_interpc C s| | � |jj� t|jj�D ]}t�|j� q| �|� dD ]&}|� |�}|j
dkrTq:| �||� q:|� d�}| �||� d S )N)rP rW r1 Zfresh) r� rq rb r� rc r rw rp r� r� r? )r@ r� r� r r r r �_assert_closed� s
z%ExhaustiveChannelTests._assert_closedFc c s� d}| � � D ]�}t� | �� D ]�}|d7 }|dkr: d S |rl|d d dkrTt� t||d�t|��� n2|d d dkr�tddd� td dd� tj�� |||fV q |r td
� q t� d S ) Nr re r� � z ({} actions)� r� r\ �.z---)r� �printr� r: r� r� �stdoutr" )r@ �verboser� r� r� r r r �_iter_close_tests� s$
z(ExhaustiveChannelTests._iter_close_testsc C s dt _| �� D ]
\}}}qd S )NT)r� r� r� �r@ r� r� r� r r r �_skim_close_tests s z(ExhaustiveChannelTests._skim_close_testsc C s� | � � D ]v\}}}| �d�|||���@ |�|j� | �||� | j|dd� | �|� W d � n1 sl0 Y |�� qd S )N� {} {} {}Frk � r� r� r: r� r r� r� r� r� r� r r r �
test_close s (z!ExhaustiveChannelTests.test_closec C s� | � � D ]v\}}}| �d�|||���@ |�|j� | �||� | j|dd� | �|� W d � n1 sl0 Y |�� qd S )Nr� Trk r� r� r r r �test_force_close s (z'ExhaustiveChannelTests.test_force_close)N)F)rC rD rE r� r� r� r� r� r� rs r� r� r� r� r� r� r� r� r r r r r� � s (-1
r� rL )N)N)5� collectionsr rX r� r r? r� �textwrapr r# r r �testr Z test.supportr �
import_moduler r r r rY r) r- r* r2 rI rJ r` rs ro r~ r� ZTestCaser� r� r� r� r� r� r� r� r� r
r rZ rf r� r� r r� rC r1 r r r r �<module> st
&
3$
2?(;]b 7Y : &�q @