File: //usr/local/lib/python3.10/test/__pycache__/test__xxsubinterpreters.cpython-310.opt-1.pyc
o
�i�; � @ sj 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 d dlmZ e�d�Zdd � ZdLd
d �ZdLd d
�Zejdd� �Zdd� Zi fdd�ZG dd� dedd��Zejdd� �ZG dd� dedd��Z G d d � 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/d0� d0e#�Z&G d1d2� d2e#�Z'G d3d4� d4e#�Z(G d5d6� d6e#�Z)G d7d8� d8e#�Z*G d9d:� d:e#�Z+G d;d<� d<e#�Z,G d=d>� d>e#�Z-G d?d@� d@e#�Z.G dAdB� dBe#�Z/G dCdD� dDe#�Z0G dEdF� dFedFdG��Z1e �2dH�G dIdJ� dJe#��Z3e4dKk�r3e �5� dS dS )M� )�
namedtupleN)�dedent)�support)�
import_helper)�
script_helperZ_xxsubinterpretersc C s> t �� \}}| �dd�}td|� d|� d��}|t|dd�fS )N�
z
z-
import contextlib
with open(zg, 'w', encoding="utf-8") as spipe:
with contextlib.redirect_stdout(spipe):
z
�utf-8��encoding)�os�pipe�replacer �open)�script�r�wZindented�wrapped� r �9/usr/local/lib/python3.10/test/test__xxsubinterpreters.py�_captured_script s �
�r c C sJ t |�\}}|� t�| ||� |�� W d � S 1 s w Y d S �N)r � interpreters�
run_string�read)�interpZrequest�sharedr Zrpiper r r � _run_output# s
$�r c C sX |d u rt j}t�� }|| }t�| �s*t�� |kr td��t�d� t�| �rd S d S )Nzinterp is not runningg{�G�z�?)r Z
SHORT_TIMEOUT�time� monotonicr �
is_running� RuntimeError�sleep)r �timeoutZ
start_time�deadliner r r �_wait_for_interp_to_run* s
�r$ c # s� � t �� \�}� �fdd�}tj|d�}|�� t� � d V t|ddd��
}|�d� W d � n1 s6w Y |�� d S )Nc s t �� td�� d��� d S )Nz7
# wait for "signal"
with open(zG, encoding="utf-8") as rpipe:
rpipe.read()
)r r r r �r r r r �run; s
�z_running.<locals>.run��targetr r r �done) r r � threading�Thread�startr$ r �write�join)r r r& �tZspiper r% r �_running8 s � � r0 c K s t | ||� d S r )� _run_interp)�id�sourcer r r r �
run_interpW s r4 c C sH t |�}t�� }|| krt�� |krt�t||� d S t�| ||� d S r )r r �get_main� get_currentr �execr )r2 r3 r Z_mainns�mainr r r r1 [ s r1 c s* e Zd Zedd� �Zd� fdd� Z� ZS )� Interpreterc C s$ t || �r|S t |t�r| |�S t�r )�
isinstance�str�NotImplementedError)�cls�rawr r r �from_rawi s
zInterpreter.from_rawNc s� t �� }||kr|s
d}n |dkrtd�|���|}n,|d ur8|s$d}n|dkr,td��t|t j�s7t �|�}n|r>|dkrCd}|}nt �� }t� �| ||�}|S )Nr8 z)name mismatch (expected "main", got "{}")r z!name mismatch (unexpected "main")) r r5 �
ValueError�formatr: �
InterpreterID�create�super�__new__)r= �namer2 r8 �self�� __class__r r rE r s. �
� zInterpreter.__new__�NN)�__name__�
__module__� __qualname__� classmethodr? rE �
__classcell__r r rH r r9 g s
r9 zname idc c s& � zd V W d S t jy Y d S w r )r �ChannelClosedErrorr r r r �expect_channel_closed� s � �rQ c s6 e Zd Zd
� fdd� Zdd� Zdd� Zdd � Z� ZS ) �
ChannelActionNc s& |sd}|sd}t � �| |||�}|S )N�bothr8 �rD rE )r= �action�endr rG rH r r rE � s zChannelAction.__new__c O sb | j dkr| jdvrt| j��n| j dv r | jdvr t| j��nt| j ��| jdvr/t| j��d S )N�use)�same�opposite�send�recv��close� force-close)rS rX rY rZ r[ )r8 rX �other�extra)rU rV r@ r )rG �args�kwargsr r r �__init__� s
�
�
�zChannelAction.__init__c C s. | j dkr|S | j dkr|dkrdS dS | j S )NrX rY rZ r[ �rV �rG rV r r r � resolve_end� s
zChannelAction.resolve_endc C st | j dkr|S | j dkr|d u rt�|S | j dkr!|d u r t�|S | j dkr8|jdkr-|S |r6|jdkr6|S t�d S )NrX r_ r` r8 )r r rF )rG r r_ r` r r r �resolve_interp� s"
�z ChannelAction.resolve_interprJ )rK rL rM rE rc rf rg rO r r rH r rR � s
rR zaction end interpc sB e Zd Zddd�� fdd� Zdd� Zdd � Zd
d �d d
�Z� ZS )� ChannelStater F��closedc s t � �| ||�}|S r rT )r= �pendingrj rG rH r r rE � s zChannelState.__new__c C s t | �| jd | jd�S �N� ri ��typerk rj �rG r r r �incr� � zChannelState.incrc C s t | �| jd | jd�S rl rn rp r r r �decr� rr zChannelState.decrT��forcec C s8 | j r |r
| jdkr | S t| �|rddd�S | jdd�S )Nr Tri )rj rk ro )rG ru r r r r] � s zChannelState.close)r )rK rL rM rE rq rs r] rO r r rH r rh � s
rh zpending closedT��
hideclosedc C st |j r|dkr|dkr|jrd}nd}nd}z t| |||�}W n tjy3 |s,|s,� |�� }Y |S w |r8d�|S )NrW r[ FT.)rj rk � _run_actionr rP r] )�cidrU rV �staterw Z
expectfail�resultr r r �
run_action� s
�r| c C s� |dkr=|dkrt �| d� |�� S |dkr9|js0z
t �| � W td�� t jy/ | Y S w t �| � |�� S t|��|dkrXi }|dv rKd||< t j | fi |�� |�
� S |d krwd
di}|dv rhd||< t j | fi |�� |j
dd �S t|��) NrW rZ � spamr[ zexpected ChannelEmptyErrorr] �r[ rZ Tr^ ru rt ) r � channel_sendrq rk � channel_recv�ChannelEmptyError� Exceptionrs r@ �
channel_closer] )ry rU rV rz rb r r r rx � s: ��
� rx c C s< t �� D ]} | dkr qzt �| � W q ty Y qw d S )Nr )r �list_all�destroyr �r2 r r r �clean_up_interpreters s ��r� c C s4 t �� D ]} zt �| � W q t jy Y qw d S r )r �channel_list_all�channel_destroy�ChannelNotFoundError�ry r r r �clean_up_channels s ��r� c @ � e Zd Zdd� ZdS )�TestBasec C � t � t� d S r �r� r� rp r r r �tearDown* �
zTestBase.tearDownN)rK rL rM r� r r r r r� ( s r� c @ � e Zd Zdd� Zdd� ZdS )�IsShareableTestsc C sP g d�}|D ] }| � |�� | �t�|�� W d � n1 s w Y qd S )N)Nr} �spam�
i����)�subTest�
assertTruer � is_shareable)rG Z
shareables�objr r r �test_default_shareables4 s � ���z(IsShareableTests.test_default_shareablesc
C s� G dd� d�}G dd� dt �}ddtdttt� t� d||d �|d
�g }|D ]!}| �t|��� | �t� |�� W d � n1 sAw Y q%d S ) Nc @ r� )z3IsShareableTests.test_not_shareable.<locals>.Cheesec S s
|| _ d S r �rF )rG rF r r r rc E �
z<IsShareableTests.test_not_shareable.<locals>.Cheese.__init__c S � | j S r r� rp r r r �__str__G s z;IsShareableTests.test_not_shareable.<locals>.Cheese.__str__N)rK rL rM rc r� r r r r �CheeseD s r� c @ s e Zd ZdZdS )z5IsShareableTests.test_not_shareable.<locals>.SubBytesz A subclass of a shareable type.N)rK rL rM �__doc__r r r r �SubBytesJ s r� TF.g Y@Z Wensleydaler} )
�bytes�NotImplementedro �objectr� r� �repr� assertFalser r� )rG r� r� Znot_shareablesr� r r r �test_not_shareableC s. �� ���z#IsShareableTests.test_not_shareableN)rK rL rM r� r� r r r r r� 2 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 � t � �� t�� | _d S r )rD �setUpr �channel_createry rp rH r r r� f �
zShareableTypeTests.setUpc s t �| j� t� �� d S r )r r� ry rD r� rp rH r r r� j s zShareableTypeTests.tearDownc C sr |D ]4}| � |��% t�| j|� t�| j�}| �||� | �t|�t|�� W d � n1 s1w Y qd S r )r� r r ry r� � assertEqual�assertIsro )rG �valuesr� �gotr r r �_assert_valuesn s ���z!ShareableTypeTests._assert_valuesc C s^ dD ]*}| � |�� t�| j|� t�| j�}| �||� W d � n1 s'w Y qd S )Nr )r� r r ry r� r� )rG r� r� r r r �test_singletonsy s ���z"ShareableTypeTests.test_singletonsc C s | � dd| jg� d S )Nr} i' )r� ry rp 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)�signedN)�to_bytes)�.0�ir r r � <genexpr>� s � �z0ShareableTypeTests.test_bytes.<locals>.<genexpr>���� )r� �rangerp r r r �
test_bytes� s
�z ShareableTypeTests.test_bytesc C s | � g d�� d S )N)z hello worldu 你好世界� )r� rp r r r � test_strs� � z ShareableTypeTests.test_strsc C s* | � t�tdd�tjtj d g�� d S )Nr� r� rm )r� � itertools�chainr� �sys�maxsizerp r r r �test_int� s
�zShareableTypeTests.test_intc
C s� t jd t j d dd g}|D ]3}| �|��$ | �t�� t�| j|� W d � n1 s/w Y W d � n1 s>w Y qd S )Nrm r� �� )r� r� r� � assertRaises�
OverflowErrorr r ry )rG Zintsr� r r r �test_non_shareable_int� s
� �� ���z)ShareableTypeTests.test_non_shareable_int)
rK rL rM r� r� r� r� r� r� r� r� r� rO r r rH r r� d s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� ListAllTestsc C s" t �� }t �� }| �||g� d S r )r r5 r� r� )rG r8 �idsr r r � test_initial� s zListAllTests.test_initialc C s6 t �� }t �� }t �� }t �� }| �||||g� d S r )r r5 rC r� r� �rG r8 �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 r5 rC r� r� r� r� r r r �test_after_destroying� s
z"ListAllTests.test_after_destroyingN)rK rL rM r� r� r� r r r r r� � s r� c @ r� )�GetCurrentTestsc C s. t �� }t �� }| �||� | �|t j� d S r )r r5 r6 r� �assertIsInstancerB )rG r8 �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 r5 rC r r �int�stripr� r� ZassertNotEqual)rG r8 r �outr� �_�expectedr r r �test_subinterpreter� s z#GetCurrentTests.test_subinterpreterN)rK rL rM r� r� r r r r r� � � r� c @ r� )� GetMainTestsc C s0 t �� \}t �� }| �||� | �|t j� d S r )r r� r5 r� r� rB )rG r� r8 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� rC r r r� r� r� )rG r� r r� r8 r r r �test_from_subinterpreter� s
z%GetMainTests.test_from_subinterpreterN)rK rL rM r� r� r r r r r� � r� r� c @ sF e Zd Zdd� Ze�d�dd� �Zdd� Zdd � Zd
d � Z d d
� Z
dS )�IsRunningTestsc C s t �� }| �t �|�� d S r )r r5 r� r )rG r8 r r r r� � s zIsRunningTests.test_mainzFails on FreeBSDc C sd t �� }| �t �|�� t|�� | �t �|�� W d � n1 s#w Y | �t �|�� d S r )r rC r� r r0 r� �rG 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 rC r r r� r� )rG r r� r r r r� � s
�z'IsRunningTests.test_from_subinterpreterc C sL t �� }t �|� | �t�� t �|� W d � d S 1 s w Y d S r )r rC r� r� r r r� r r r �test_already_destroyed �
"�z%IsRunningTests.test_already_destroyedc C �: | � t�� t�d� W d � d S 1 sw Y d S �Ni@B )r� r r r rp r r r �test_does_not_exist � "�z"IsRunningTests.test_does_not_existc C r� �Nr� )r� r@ r r rp r r r � test_bad_id
r� zIsRunningTests.test_bad_idN) rK rL rM r� �unittest�skipr� 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� Trt )r rB r� r� �rG r2 r r r �
test_with_int s z InterpreterIDTests.test_with_intc C �4 G dd� dt �}tj|� dd�}| �t|�d� d S )Nc @ r� )z.InterpreterIDTests.test_coerce_id.<locals>.Intc S � dS �Nr� r rp r r r � __index__ � z8InterpreterIDTests.test_coerce_id.<locals>.Int.__index__N�rK rL rM r� r r r r �Int � r� Trt r� )r; r rB r� r� )rG r� r2 r r r �test_coerce_id � z!InterpreterIDTests.test_coerce_idc C �f | � ttjt� � | � ttjd� | � ttjd� | � ttjd� | � ttjd� | � ttjd� d S �Ng $@�10s 10r� l )r� � TypeErrorr rB r� r@ r� rp r r r r� � z InterpreterIDTests.test_bad_idc C sJ t �� }| �t�� t �t|�d � W d � d S 1 s w Y d S �Nrm )r r� r� r rB r� r� r r r r� & s "�z&InterpreterIDTests.test_does_not_existc C �" t jddd�}| �t|�d� d S �Nr� Trt r� )r rB r� r; r� r r r �test_str+ � zInterpreterIDTests.test_strc C r )Nr� Trt zInterpreterID(10))r rB r� r� r� r r r � test_repr/ r z InterpreterIDTests.test_reprc C �( 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 rC rB r� r� �floatr� r; �rG �id1�id2�id3r r r �
test_equality3 �$ z InterpreterIDTests.test_equalityN)
rK rL rM 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 rC r� rB �assertInr� r� r r r � test_in_mainL 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 rC r� �addr� �len)rG �seenr� r2 r r r �test_unique_idR s
zCreateTests.test_unique_idc sn t �� �d � � �fdd�}t j|d�}�� |�� W d � n1 s$w Y |�� | �� t�� � d S )Nc s t �� � ��� ��� d S r )r rC �acquire�releaser �r2 �lockr r �f_ s z%CreateTests.test_in_thread.<locals>.fr' )r* �Lockr+ r, r. r r r� �rG r r/ r r r �test_in_thread\ 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� rC r r r� r� r� r )rG r8 r
r� r r r r �test_in_subinterpreterk 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 | s z6CreateTests.test_in_threaded_subinterpreter.<locals>.fr' ) r r� rC r* r+ r, r. r� r )rG r8 r r/ r r$ r � test_in_threaded_subinterpreterx s
z+CreateTests.test_in_threaded_subinterpreterc C sj t t�� �}g }td�D ] }t�� }|�|� q |D ]}t�|� qt�� }| �t t�� �||hB � d S �N� )r r r� r� rC �appendr� r� �rG �beforer� r� r2 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� rC r� r� )rG r* r
r r r2 r r r �test_after_destroy_some� s
z#CreateTests.test_after_destroy_someN) rK rL rM r r� r� r r" r# r% r+ r, r r r r r J 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 rC 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�|� q'| �t t�� �|� d S r& )r r r� r� rC r r� r� r) r r r �test_all� s zDestroyTests.test_allc sl t �� \� ��t��
t �� � W d � n1 sw Y � �fdd�}tj|d�}|�� |�� d S )Nc s: �� t�� t�� � W d � d S 1 sw Y d S r �r� r r r� r �r8 rG r r r � r� z!DestroyTests.test_main.<locals>.fr' ) r r� r� r r� r* r+ r, r. r! r r2 r r� � s
� zDestroyTests.test_mainc C sL t �� }t �|� | �t�� t �|� W d � d S 1 s w Y d S r )r rC r� r� r r� r r r r� � r� z#DestroyTests.test_already_destroyedc C r� r� r1 rp r r r r� � r� z DestroyTests.test_does_not_existc C r� r� )r� r@ r r� rp r r r r� � r� 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� rC r r r� r )rG r8 r2 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(�)
r3 )rG r8 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 � � z.DestroyTests.test_from_other_thread.<locals>.fr' )r rC 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 s7w Y | �t �|�� W d � d S 1 sOw 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� rC r0 r� r r� r r� )rG r8 r r r r �test_still_running� s
�
� �"�z DestroyTests.test_still_runningN)
rK rL rM r/ r0 r� r� r� r� r4 r6 r8 r: r r r r r- � s
r- c s� e Zd Z� fdd�Zdd� Zdd� Zdd� Ze�e e
d �d
�d d � �Z d
d� Z dd� Z
dd� Zdd� Zdd� Zdd� Zejd1dd��Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Z d,d-� Z e� d.�d/d0� �Z � Z S )2�RunStringTestsc r� r )rD r� r rC r2 rp rH r r r� r� zRunStringTests.setUpc C sV t d�\}}|� t�| j|� |�� }W d � n1 s w Y | �|d� d S )N�print("it worked!", end="")�
it worked!)r r r r2 r r� )rG r �filer� r r r � test_success s
�zRunStringTests.test_successc sr t d�\� }|�! � �fdd�}tj|d�}|�� |�� |�� }W d � n1 s,w Y ��|d� d S )Nr< c s t ��j� � d S r )r r r2 r �r rG r r r r� z(RunStringTests.test_in_thread.<locals>.fr' r= )r r* r+ r, r. r r� )rG r> r r/ r� r r@ r r" s
� z RunStringTests.test_in_threadc C s` t jdd�}td�\}}|� t �||� |�� }W d � n1 s#w 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 rC r r r r� )rG 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 }|jddd��6}|�d� |�� d}td|j� d|� d ��}t�| j|� |� d� |�
� }| � ||� W d � d S 1 sDw Y d S )
Nr zw+r r r� zspam spam spam spam spamz�
import os
try:
os.fork()
except RuntimeError:
with open('zE', 'w', encoding='utf-8') as out:
out.write('z')
) �tempfile�NamedTemporaryFiler- �flushr rF r r r2 �seekr r� )rG rD r> r� r Zcontentr r r � test_fork5 s
�
�
"�zRunStringTests.test_forkc C sx t | j��- | �t�� t�| jd� W d � n1 s w Y W d � d S W d � d S 1 s5w Y d S )N�
print("spam"))r0 r2 r� r r r rp r r r �test_already_runningK s ��"�z#RunStringTests.test_already_runningc C s` d}|t �� v r|d7 }|t �� v s| �t�� t �|d� W d � d S 1 s)w Y d S )Nr rm rI )r r� r� r r r� r r r r� P s � "�z"RunStringTests.test_does_not_existc C �< | � t�� t�dd� W d � d S 1 sw Y d S )Nr� rI )r� r@ r r rp r r r �
test_error_idW � "�z RunStringTests.test_error_idc C rK )Nr� rI )r� r� r r rp r r r r� [ rM zRunStringTests.test_bad_idc C �> | � t�� t�| jd� W d � d S 1 sw Y d S r� �r� r� r r r2 rp r r r �test_bad_script_ � "�z RunStringTests.test_bad_scriptc C rN )Ns
print("spam")rO rp r r r �test_bytes_for_scriptc rQ z$RunStringTests.test_bytes_for_scriptNc c s~ � | � tj�� }d V W d � n1 sw Y |d u r0| �t|j��d�d t|�� d S | �t|j�d�||�� d S )N�:r z{}: {})r� r �RunFailedErrorr� r; � exception�splitrA )rG �exctyper9 Zcaughtr r r �assert_run_failedg s � ��
�z RunStringTests.assert_run_failedc C rN )Nz print("spam")rX � SyntaxErrorr r r2 rp r r r �test_invalid_syntaxr s "�z"RunStringTests.test_invalid_syntaxc C �@ | � td�� t�| jd� W d � d S 1 sw Y d S )Nr� zraise Exception("spam"))rX r� r r r2 rp r r r � test_failurew � "�zRunStringTests.test_failurec C r[ )N�42zraise SystemExit(42))rX �
SystemExitr r r2 rp r r r �test_SystemExit{ r] z RunStringTests.test_SystemExitc C s� | � t�� t�| jtd�� W d � n1 sw Y | � td�� t�| jtd�� W d � d S 1 s9w Y d S )NzG
import sys
sys.exit()
r^ zI
import sys
sys.exit(42)
)rX r_ r r r2 r rp r r r �
test_sys_exit s �"�z RunStringTests.test_sys_exitc C s� t �� \}}ddd d�}td|� d��}t�| j||� t|d��
}t�|�}W d � n1 s1w 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� �* rc r� r{ �) rd ) r r r r r r2 r �pickle�loadr� Z assertIsNone�rG 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 sCw Y | � |d d d � | � |d
d d� | � |d d� d S ) Nzj
spam = 'eggs'
ns1 = dict(vars())
del ns1['__builtins__']
r� rb zL
ns2 = dict(vars())
del ns2['__builtins__']
zr
ns = dict(vars())
del ns['__builtins__']
import pickle
with open(re rf �ns1rc �ns2)
r r r2 r r r r ri rj r� )rG r r r r rl rm 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/w Y | � |d d� d S )NrK s not __main__z�
spam = 42
ns = dict(vars())
del ns['__builtins__']
import pickle
with open(re rf )
r r r r r r2 r ri rj r� rk 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 s(w Y t �� \}}t�| jtd|� d��� t|d��
}t�|�}W d � n1 sUw 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
rf z�
eggs = False
ns = dict(vars())
del ns['__builtins__']
import pickle
with open(re r� rc ) r r r r r2 r r ri rj r r. )rG r r rl ro rp 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 s*w 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(re rf � __builtins__�
__loader__�__main__rg )rK �__annotations__r� � __package__�__spec__r� ) r r r r r r2 r ri rj �popr� )rG r r r rl rm r r r � test_execution_namespace_is_main� s$
� �
�z/RunStringTests.test_execution_namespace_is_mainzblocking foreverc C s� t d�}t�� �(}t�|d|�}t�|�� }|�� }W d � n1 s$w Y W d � n1 s3w 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� )rG r �dirname�filename�proc�retcoder r r �test_still_running_at_exit s
�� �z)RunStringTests.test_still_running_at_exitr )!rK rL rM r� r? r" rB r� Z
skipUnless�hasattrr rH rJ r� rL r� rP rR �
contextlib�contextmanagerrX rZ r\ r` ra rn rq rr rs r{ r� r� rO r r rH r r; s4
! 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� Trt rS )r � _channel_idr� r� rV �rG ry r r r �test_default_kwargs4 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�rZ ru rZ F�rZ r[ ru �r[ ru r[ )r[ rZ ru rS )r r� r� rV r� r r r �test_with_kwargs: s z ChannelIDTests.test_with_kwargsc C r� )Nc @ r� )z*ChannelIDTests.test_coerce_id.<locals>.Intc S r� r� r rp r r r r� L r� z4ChannelIDTests.test_coerce_id.<locals>.Int.__index__Nr� r r r r r� K r� r� Trt r� )r; r r� r� r� )rG r� ry r r r r� J r� z ChannelIDTests.test_coerce_idc C r� r� )r� r� r r� r� r@ r� rp r r r r� R r zChannelIDTests.test_bad_idc C s@ | � t�� tjdddd� W d � d S 1 sw Y d S )Nr� F�rZ r[ )r� r@ r r� rp r r r �test_bad_kwargsZ s "�z ChannelIDTests.test_bad_kwargsc C sL t �� }| �t j�� t �t|�d � W d � d S 1 s w Y d S r )r r� r� r� r� r� r� r r r r� ^ s "�z"ChannelIDTests.test_does_not_existc C r r )r r� r� r; r� r r r r c r zChannelIDTests.test_strc C s� t jddd�}| �t|�d� t jdddd�}| �t|�d� t jdddd�}| �t|�d� t jddddd �}| �t|�d� d S )
Nr� Trt z
ChannelID(10)r� zChannelID(10, send=True)r� zChannelID(10, recv=True)r� )r r� r� r� r� r r r r g s zChannelIDTests.test_reprc C r r )r r� r� r� r� r r� r; )rG �cid1�cid2Zcid3r r r r t r z ChannelIDTests.test_equalityN) rK rL rM r� r� r� r� r� r� r r r r r r r r� 2 s
r� 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� � 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 )rG 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)
rm )r rC r r r� r� r� )rG r
r� r� r r� 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�rZ FN)r r� �channel_list_interpretersr� )rG ry � 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.rZ Tr� F�e
import _xxsubinterpreters as _interpreters
obj = _interpreters.channel_recv(r5 N) r r5 r� r r� r� rC r r )rG �interp0ry r� r� �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.rZ �_
import _xxsubinterpreters as _interpreters
_interpreters.channel_send(z, "send")
r� r5 Tr� FN)
r r5 rC r� r r r r� r� r )rG r� r� �interp2�interp3ry r� r� 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.rZ r� r5 Tr� FN)
r r5 rC r� r r r r� r� r� �rG r� r� ry r� r� 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.�datar� r5 Tr� Frm r� r �b
import _xxsubinterpreters as _interpreters
_interpreters.channel_release(N) r r5 rC r� r r r r� r� r �channel_release)rG r� r� r� ry r� r� 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 sOw Y | � t j
�� t j|dd� W d � dS 1 smw Y dS ) z8Test listing channel interpreters with a closed channel.rZ Tr� Frm r rt N) r r5 rC r� r r� r� r r� r� rP r� r r r �%test_channel_list_interpreters_closed* s �"�z2ChannelTests.test_channel_list_interpreters_closedc C sV 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 sOw Y t j|dd�}| �t|�d� t |t d|� d ��� | � t j
�� t j|dd� W d � n1 s�w Y | � t j
�� t j|dd� W d � dS 1 s�w Y dS )
zCTest listing channel interpreters with a channel's send end closed.rZ Tr� Frm r N�`
import _xxsubinterpreters as _interpreters
_interpreters.channel_close(�, force=True)
)
r r5 rC r� r r� r� r r� r� rP r r r� r r r �.test_channel_list_interpreters_closed_send_end@ s. � � �"�z;ChannelTests.test_channel_list_interpreters_closed_send_endc C s> t �� }d}t �||� t �|�}| �||� | �||� d S �Nr} )r r� r r� r� Z assertIsNot)rG ry Zorigr� r r r �test_send_recv_maind 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 rC r r )rG r
r� r r r � test_send_recv_same_interpreterm r� z,ChannelTests.test_send_recv_same_interpreterc C s@ t �� }t �� }t|td|� d���}t �|�}| �|d� d S )Nr� �, b'spam')
r} )r r� rC r r r� r� )rG ry r
r� r� r r r �%test_send_recv_different_interpretersy s �
z2ChannelTests.test_send_recv_different_interpretersc sV t �� � � fdd�}tj|d�}|�� t �� d� |�� t �� �}| �|d� d S )Nc sB zt �� �} W n t jy t�d� Y nw qt �� | � d S )NTr )r r� r� r r! r )r� r� r r r � s
��z8ChannelTests.test_send_recv_different_threads.<locals>.fr' r} ) r r� r* r+ r, r r. r� r� �rG 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 �ry r
r� r r r � s
�
�zIChannelTests.test_send_recv_different_interpreters_and_threads.<locals>.fr' r} � eggs)
r r� rC r* r+ r, r r. r� 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 � d S 1 sw Y d S )Nr� r} )r� r r� r rp r r r �test_send_not_found� s "�z ChannelTests.test_send_not_foundc C s< | � tj�� t�d� W d � d S 1 sw Y d S r� )r� r r� r� rp r r r �test_recv_not_found� s "�z ChannelTests.test_recv_not_foundc C sD t �� }| �t j�� t �|� W d � d S 1 sw Y d S r )r r� r� r� r� 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 )Nrm r} r� )r� r r� r� r r� r� ) rG �defaultry Zobj1�obj2Zobj3Zobj4Zobj5Zobj6r 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� r} rZ )
r r� rC r r �dictrZ r� r� r� �rG ry 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')
)rl r} rZ ) r r� r� rC r r r� rZ r� r� r� r� r r r � test_run_string_arg_resolved� s
�
z)ChannelTests.test_run_string_arg_resolvedc C � t �� }t �|d� t �|� t �|� | �t j�� t �|d� W d � n1 s+w Y | �t j�� t �|� W d � d S 1 sGw Y d S �Nr} r� �r r� r r� r� r� rP r� r r r �test_close_single_user� �
� "�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 sFw Y | �dt |j
�� | �t j��}t �|td|� d��� W d � n1 sqw Y | �dt |j
�� d S )Nr� r� z_
import _xxsubinterpreters as _interpreters
_interpreters.channel_recv(r5 z,
_interpreters.channel_send(z, b'spam')
rP ) r r� rC r r r� r� rT r r; rU )rG ry r
r �cmr r r �test_close_multiple_users s,
�
�
� �
� �z&ChannelTests.test_close_multiple_usersc C sd t �� }t �|d� t �|� t �|� | �t j�� t �|� W d � d S 1 s+w 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 ]i\}}| � ||f��V t�� }t�|d� t�|� tj|||d� | �tj�� t�|d� W d � n1 s@w Y | �tj��
t�|� W d � n1 s[w Y W d � n1 sjw Y qd S )N))FF�TF)FT)TTr} r� r� )r� r r� r r� r� r� rP )rG ZtestsrZ r[ ry r r r �test_close_empty' s"
� �� ���z ChannelTests.test_close_emptyc C sp t �� }t �|d� t �|d� | �t j��
t �|� W d � n1 s&w Y t �|� t �|d� d S )Nr} rb r� �r r� r r� ZChannelNotEmptyErrorr� r� r� r r r �%test_close_defaults_with_unused_items: s �
z2ChannelTests.test_close_defaults_with_unused_itemsc C s� t �� }t �|d� t �|d� | �t j�� t j|dd� W d � n1 s(w Y t �|� t �|d� t �|� t �|� t j|dd� d S )Nr} rb T�r[ r� r� r� r r r �*test_close_recv_with_unused_items_unforcedD 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.w Y t �|� t �|� | �t j�� t �|� W d � d S 1 sTw Y d S )Nr} rb Tr� r� �r r� r r� r� rP r� r� r r r �*test_close_send_with_unused_items_unforcedQ s �
"�z7ChannelTests.test_close_send_with_unused_items_unforcedc C s� t �� }t �|d� t �|d� | �t j�� t j|ddd� W d � n1 s)w Y t �|� t �|d� t �|� t �|� t j|dd� d S )Nr} rb Tr~ r� r� r� r� r r r �*test_close_both_with_unused_items_unforced^ s �
z7ChannelTests.test_close_both_with_unused_items_unforcedc C � t �� }t �|d� t �|d� t j|ddd� | �t j�� t �|d� W d � n1 s/w Y | �t j�� t �|� W d � d S 1 sKw Y d S )Nr} rb Tr� r� r� r� r r r �(test_close_recv_with_unused_items_forcedk � � "�z5ChannelTests.test_close_recv_with_unused_items_forcedc C r� )Nr} rb Tr� r� r� r� r r r �(test_close_send_with_unused_items_forcedv r� 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 s0w Y | �t j�� t �|� W d � d S 1 sLw Y d S )Nr} rb Tr� r� r� r� r r r �(test_close_both_with_unused_items_forced� s � "�z5ChannelTests.test_close_both_with_unused_items_forcedc C � t �� }t �|� | �t j�� t �|d� W d � n1 s w Y | �t j�� t �|� W d � d S 1 s<w Y d S r� )r r� r� r� rP r r� r� r r r �test_close_never_used� �
� "�z"ChannelTests.test_close_never_usedc C s� t �� }t �|d� t �� }t �|td|� d��� | �t j��
t �|� W d � n1 s0w Y | �t j�� t � |� W d � d S 1 sLw Y d S )Nr} r� r� )
r r� r rC r r r� rP r� r� �rG ry 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 s9w Y | �t j�� t �|� W d � d S 1 sUw Y d S )Nr} Trt 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 s w Y t �|� | �t j�� t j|dd� W d � d S 1 s?w Y d S )Nr� Tr� )r r� r� r� r� r� rP r� r r r �.test_channel_list_interpreters_invalid_channel� s �
"�z;ChannelTests.test_channel_list_interpreters_invalid_channelc C sB t �� }| �t�� t �|� W d � d S 1 sw Y d S r )r r� r� r� r� r� r r r �+test_channel_list_interpreters_invalid_args� s "�z8ChannelTests.test_channel_list_interpreters_invalid_argsN)*rK rL rM 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� r� r� r� r r r r r� � sN +$
r� c @ sd e Zd ZdZ dd� 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.w Y | �t j�� t �|� W d � d S 1 sJw Y d S �Nr} Tr� r� �r r� r r� r� r� rP 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 ) Nr� r� r� z,)
_interpreters.channel_release(z+)
print(repr(obj))
z+
_interpreters.channel_release(r5 zb'spam')r r� rC r r r r� r� )rG ry r
r r� r r r �test_multiple_users� s
�� �
�z'ChannelReleaseTests.test_multiple_usersc C r� r� r� r� r r r �test_no_kwargs r� z"ChannelReleaseTests.test_no_kwargsc C sp t �� }t �|d� t �|� t j|ddd� | �t j�� t j|ddd� W d � d S 1 s1w Y d S )Nr} Tr� r� r� r r r �test_multiple_times s
"�z'ChannelReleaseTests.test_multiple_timesc C sl t �� }t �|d� t �|d� t j|ddd� | �t j�� t �|� W d � d S 1 s/w Y d S )Nr} rb Tr� )r r� r r� r� rP r� r� r r r �test_with_unused_items s "�z*ChannelReleaseTests.test_with_unused_itemsc C r� r� )r r� r� r� rP r r� r� r r r �test_never_used( r� z#ChannelReleaseTests.test_never_usedc C s� t �� }t �|d� t �� }t �|td|� d��� t �|�}t �|� | �t j �� t �|d� W d � n1 s;w Y | �
|d� d S )Nr} r� r5 r� ) r r� r rC r r r� r� r� rP r� )rG ry r r� r r r �test_by_unassociated_interp1 s
�
�z/ChannelReleaseTests.test_by_unassociated_interpc C sj t �� }t �� }t �|td|� d|� d��� | �t j�� t �|� W d � d S 1 s.w Y d S )Nze
import _xxsubinterpreters as _interpreters
obj = _interpreters.channel_send(z5, b'spam')
_interpreters.channel_release(r5 )r r� rC r r r� rP r� r� r r r �test_close_if_unassociated@ s
� � "�z.ChannelReleaseTests.test_close_if_unassociatedc C sR t �� }t �|d � t �|� t �|d� t j|dd� t �|�}| �|d� d S )Nr} Tr� )r r� r r� r� r� )rG ry r� r r r �test_partiallyM 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 s:w Y | �t j�� t �|� W d � d S 1 sVw Y d S r� r� r� r r r �'test_used_multiple_times_by_single_userX s
� "�z;ChannelReleaseTests.test_used_multiple_times_by_single_userN)rK rL rM 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 ri }nt�|�}t�|�}t�|�}|j||j||j|i}|s$d}t� �| |||||�}t� |_t� |_ ||_
|S )NrX ) �QUICKr9 r? rF rD rE r �_preppedrh �_state�_known)r= rV r r_ r` �creatorZknownrG rH r r rE m s
�zChannelCloseFixture.__new__c C r� r �r� rp r r r rz � s zChannelCloseFixture.statec C s: z| j W S ty | �| j�}| �|�| _ | j Y S w r )�_cid�AttributeError�_get_interpreterr� � _new_channel)rG r� r r r ry � 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| jjdkrdS t| jj�S )Nr[ rZ F)rV rz rj �boolre r r r �expect_closed_error� s
z'ChannelCloseFixture.expect_closed_errorc C s | � |� d S r )r� r� r r r �prep_interpreter� r7 z$ChannelCloseFixture.prep_interpreterc C s
|| _ d S r r� )rG rU r{ r r r �
record_action� r� z!ChannelCloseFixture.record_actionc C r� r r� rp r r r �clean_up� r� z ChannelCloseFixture.clean_upc C s@ |j dkr t�� S t�� }t|jd|� d�� t�|�| _| jS )Nr8 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
)rF r r� r4 r2 r� r� )rG r� Zchr r r r� � s
� z ChannelCloseFixture._new_channelc C sf |dv r| j S |dkr| jS |dkr| jS |}z| j| }W |S ty2 t|� }| j|< Y |S w )N)rX r r_ r` )r r_ r` r� �KeyErrorr9 )rG r rF r r r r� � s ��z$ChannelCloseFixture._get_interpreterc C sF |j | jv rd S | j�|j � |jdkrd S t|j d| j� d�� d S )Nr8 a
import _xxsubinterpreters as interpreters
import test.test__xxsubinterpreters as helpers
ChannelState = helpers.ChannelState
try:
cid
except NameError:
cid = interpreters._channel_id(r5 )r2 r� r rF r4 ry r� r r r r� � s
�z%ChannelCloseFixture._prep_interpreterr )rK rL rM r� rE �propertyrz ry r� r� r� r� r� r� r� r� rO r r rH r r� f s
r� z end interp other extra creatorz%these tests take several hours to runc @ s� e Zd ZdZ dd� 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 qq
| � dd�D ]}|V | �� D ]}|| V q+q"| �dd�D ]=}|V | � dd�D ]}|| }|V | �� D ]}|| V qRqE| � dd�D ]}|| }|V | �� D ]}|| V qnqaq:| �dd�D ]=}|V | � dd�D ]}|| }|V | �� D ]}|| V q�q�| � dd�D ]}|| }|V | �� D ]}|| V q�q�q~d S )NrX r_ r` )�_iter_close_action_sets� _iter_post_close_action_sets�_iter_use_action_sets)rG Z closeactionsZ postactionsZ
useactions�actionsr r r �iter_action_sets sV � � � � �� � ���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 ]}|D ]}t dd|�t dd|�t dd|�gV qIqEqA|D ](}|D ]#}|D ] }|D ]}t dd|�t dd|�t dd|�t dd|�gV qnqjqfqbd S )NrW r[ rZ �rR )rG r� r� �interpsr� Zinterp4r r r r / sL �
�
�
�
�
�
�
���
�����z,ExhaustiveChannelTests._iter_use_action_setsc
c s� � d}||f}dD ]}|rdnd}|D ]}|D ]
}t |||�gV qqq dD ] } dD ]}
|D ]} |D ]} t | d| �t |
d| �gV q2q.q*q&d S )Nr~ r� r^ r] r\ r[ rZ r )
rG r� r� Zendsr ru �opr rV ZrecvopZsendopr[ rZ r r r r ` s, �
���
�����z.ExhaustiveChannelTests._iter_close_action_setsc c s0 � dD ]}t dd|�gV t dd|�gV qd S )N)rX r` r_ rW r[ rZ r r� r r r r s s �
�
��z3ExhaustiveChannelTests._iter_post_close_action_setsc C s |D ]}| � ||� qd S r )r| )rG �fixr rU r r r � run_actions| s �z"ExhaustiveChannelTests.run_actionsTrv c C s� |� |j�}|�|j|j|j�}|�|� |jdkr.t|j |j
||j |d�}|� ||� d S t
�� }t|jd|j � dt|j
�� dt|�� dt|j �� d|� d|� d|� d�� tt�t
�|�d �tt
�|��d
�}|� ||� d S ) Nr8 rv 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� )rk rj )rf rV rg r r_ r` r� rF r| ry rU rz r� r r� r4 r2 r� rh r� �
from_bytesr� r� )rG r rU rw rV r r{ r� r r r r| � sB
�������
� �z!ExhaustiveChannelTests.run_actionc c sD � g d�}|D ]\}}}dD ]}dD ] }t |||||�V qqqd S )N))r8 r r` )r r8 r` )r� r� r` )r� r� r8 )rX r_ r� r� )r� )rG r r r_ r` r� rV r r r �
iter_fixtures� s ����z$ExhaustiveChannelTests.iter_fixturesc C sx |rdnd}t ||jd�}|�� s| j||dd� d S | �tj�� | j||dd� W d � d S 1 s5w Y d S )Nr^ r] rX Frv )rR rV r� r| r� r rP )rG r ru r r] r r r �_close� s "�z ExhaustiveChannelTests._closeNc C s2 |d u s |j dkr| �tj�� t�|j� W d � n1 s w Y | �tj�� t�|jd� W d � n1 s=w Y | �tj�� t�|j� W d � n1 sYw Y | �tj�� tj|jdd� W d � d S 1 sxw Y d S t|j d� t|j d� t|j d� t|j d� d S ) Nr8 r} Trt 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)
)
rF r� r rP r� ry r r� r4 r2 )rG r r r r r �_assert_closed_in_interp� s" � � �"� z/ExhaustiveChannelTests._assert_closed_in_interpc C s| | � |jj� t|jj�D ]}t�|j� q
| �|� dD ]}|� |�}|j
dkr*q | �||� q |� d�}| �||� d S )N)rX r_ r8 Zfresh) r� rz rj r� rk r r� ry r r� rF )rG r r� r r r r �_assert_closed� s
z%ExhaustiveChannelTests._assert_closedFc c s� � d}| � � D ]V}t� | �� D ]F}|d7 }|dkr d S |r7|d d dkr+t� t||d�t|��� n|d d dkrEtddd� td dd� tj�� |||fV q|r]td
� qt� d S ) Nr rm r� � z ({} actions)� r� rd �.z---)r �printr rA r r� �stdoutrF )rG �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 �rG r� r r r r r �_skim_close_tests� s �z(ExhaustiveChannelTests._skim_close_testsc C � | � � D ];\}}}| �d�|||��� |�|j� | �||� | j|dd� | �|� W d � n1 s6w Y |�� qd S )N� {} {} {}Frt � r r� rA r� r r
r
r r� r r r r �
test_close� � �
�z!ExhaustiveChannelTests.test_closec C r )Nr Trt r r r r r �test_force_close r z'ExhaustiveChannelTests.test_force_closer )F)rK rL rM r� r r r r r
r| r r
r r r r r r r r r r r � s$ -1
r rv r )6� collectionsr r� r� r ri r� �textwrapr r* r r� �testr Z test.supportr r �
import_moduler r r r$ r� r0 r4 r1 r9 rQ rR rh r| rx r� r� ZTestCaser� r� r� r� r� r� r� r� r r- r; r� r� r� r� r� r rK r8 r r r r �<module> sz
&
3$
2?);]b +Y : &�q
@ �