Your IP : 216.73.216.213
U
i�f� � @ s` 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� Zd d
� ZdMd d �Zejd
d� �Zdd� Zi fdd�Zdd� 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 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� dEe#�Z0G dFdG� dGedGdH��Z1e �2dI�G dJdK� dKe#��Z3e4dLk�r\e �5� dS )N� )�
namedtupleN)�dedent)�support)�
script_helperZ_xxsubinterpretersc s& t j�� fdd�tt� �d �D ��S )Nc 3 s | ]}t � |�V qd S �N)� combinations)�.0�r�Zsets� �A/opt/alt/python38/lib64/python3.8/test/test__xxsubinterpreters.py� <genexpr> s �zpowerset.<locals>.<genexpr>� )� itertools�chain�
from_iterable�range�lenr
r r
r �powerset s �r c 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)�scriptr �wZindented�wrappedr r r �_captured_script s ��
r c
C s> t |�\}}|�$ t�| ||� |�� W 5 Q R � S Q R X d S r )r � interpreters�
run_string�read)�interpZrequest�sharedr Zrpiper r r � _run_output( s r# c # s` t �� \�}� �fdd�}tj|d�}|�� d V t|d��}|�d� W 5 Q R X |�� 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 �run2 s
�z_running.<locals>.run��targetr Zdone)r r � threading�Thread�startr �write�join)r! r r% �tZspiper r$ r �_running/ s r. c K s t | ||� d S r )� _run_interp)�id�sourcer" r r r �
run_interpM s r2 c C sF t |�}t�� }|| kr4t�� |kr(t�t||� nt�| ||� d S r )r r �get_main� get_current� RuntimeError�execr )r0 r1 r" Z_mainns�mainr r r r/ Q s r/ c s0 � ��fdd�}t j|d�}|�� |�� d S )Nc s t � ��� d S r )Z_runr �r0 r" r1 r r r% ^ s z run_interp_threaded.<locals>.runr& )r( r) r* r, )r0 r1 r" r% r- r r8 r �run_interp_threaded] s r9 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_rawg s
zInterpreter.from_rawNc s� t �� }||kr6|sd}n|dkr0td�|���|}nX|d k rp|sHd}n|dkrXtd��t|t j�s�t �|�}n |r||dkr�d}|}nt �� }t� �| ||�}|S )Nr7 z)name mismatch (expected "main", got "{}")r! z!name mismatch (unexpected "main")) r r3 �
ValueError�formatr; �
InterpreterID�create�super�__new__)r> �namer0 r7 �self�� __class__r r rF p s, � zInterpreter.__new__)NN)�__name__�
__module__� __qualname__� classmethodr@ rF �
__classcell__r r rI r r: e s
r: zname idc c s& z
d V W n t jk
r Y nX d S 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�bothr7 �rE rF )r> �action�endr! rH rI r r rF � s zChannelAction.__new__c O sb | j dkr | jdkrJt| j��n*| j dkr@| jdkrJt| j��n
t| j ��| jdkr^t| j��d S )N�use)�same�opposite�send�recv��close� force-close)rS rX rY rZ r[ )r7 rX �other�extra)rU rV rA r! )rH �args�kwargsr r r �__init__� s
zChannelAction.__init__c C s2 | j dkr|S | j dkr(|dkr$dS dS | j S d S )NrX rY rZ r[ �rV �rH rV r r r � resolve_end� s
zChannelAction.resolve_endc C st | j dkr|S | j dkr(|d kr$t�|S | j dkrB|d kr>t�|S | j dkrp|jdkrZ|S |rl|jdkrl|S t�d S )NrX r_ r` r7 )r! r5 rG )rH r! r_ r` r r r �resolve_interp� s
z ChannelAction.resolve_interp)NN)rK rL rM rF rc rf rg rO r r rI 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 rH rI r r rF � s zChannelState.__new__c C s t | �| jd | jd�S �Nr ri ��typerk rj �rH r r r �incr� s zChannelState.incrc C s t | �| jd | jd�S rl rm ro r r r �decr� s zChannelState.decrT��forcec C s2 | j r|r| jdkr| S t| �|r&dn| jdd�S )Nr Tri )rj rk rn )rH rs r r r r] � s zChannelState.close)r )rK rL rM rF rp rq r] rO r r rI r rh � s rh zpending closedT��
hideclosedc C st |j r(|dkr"|dkr"|jr"d}q,d}nd}zt| |||�}W n( tjk
rf |sZ|sZ� |�� }Y n
X |rpd�|S )NrW r[ FT.)rj rk � _run_actionr rP r] )�cidrU rV �stateru Z
expectfail�resultr r r �
run_action� s rz c C s� |dkr�|dkr$t �| d� |�� S |dkrz|jsfzt �| � W n t jk
rZ | Y S X td��q�t �| � |�� S nt|��nt|dkr�i }|dkr�d||< t j | f|� |�
� S |d kr�d
di}|dkr�d||< t j | f|� |j
dd �S t|��d S ) NrW rZ � spamr[ zexpected ChannelEmptyErrorr] �r[ rZ Tr^ rs rr ) r � channel_sendrp rk � channel_recv�ChannelEmptyError� Exceptionrq rA �
channel_closer] )rw rU rV rx rb r r r rv � s:
� rv c C s@ t �� D ]2} | dkrqzt �| � W q tk
r8 Y qX qd S )Nr )r �list_all�destroyr5 �r0 r r r �clean_up_interpreters s r� c C s8 t �� D ]*} zt �| � W q t jk
r0 Y qX qd S r )r �channel_list_all�channel_destroy�ChannelNotFoundError�rw r r r �clean_up_channels s
r� c @ s e Zd Zdd� ZdS )�TestBasec C s t � t� d S r �r� r� ro r r r �tearDown( s zTestBase.tearDownN)rK rL rM r� r r r r r� & s r� c @ s e Zd Zdd� Zdd� ZdS )�IsShareableTestsc
C sB d ddddg}|D ]*}| � |�� | �t�|�� W 5 Q R X qd S )Nr{ �spam�
i����)�subTest�
assertTruer � is_shareable)rH Z
shareables�objr r r �test_default_shareables2 s � �z(IsShareableTests.test_default_shareablesc
C s~ G dd� d�}G dd� dt �}ddtdttt� t� d||d �|d
�g }|D ].}| �t|��� | �t� |�� W 5 Q R X qJd S ) Nc @ s e Zd Zdd� Zdd� ZdS )z3IsShareableTests.test_not_shareable.<locals>.Cheesec S s
|| _ d S r �rG )rH rG r r r rc C s z<IsShareableTests.test_not_shareable.<locals>.Cheese.__init__c S s | j S r r� ro r r r �__str__E s z;IsShareableTests.test_not_shareable.<locals>.Cheese.__str__N)rK rL rM rc r� r r r r �CheeseB s r� c @ s e Zd ZdS )z5IsShareableTests.test_not_shareable.<locals>.SubBytesN)rK rL rM r r r r �SubBytesH s r� TF.g Y@Z Wensleydaler{ )
�bytes�NotImplementedrn �objectr� r� �repr� assertFalser r� )rH r� r� Znot_shareablesr� r r r �test_not_shareableA s( ��z#IsShareableTests.test_not_shareableN)rK rL rM r� r� r r r r r� 0 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 )rE �setUpr �channel_createrw ro rI r r r� d s
zShareableTypeTests.setUpc s t �| j� t� �� d S r )r r� rw rE r� ro rI r r r� h s zShareableTypeTests.tearDownc
C s^ |D ]T}| � |��@ t�| j|� t�| j�}| �||� | �t|�t|�� W 5 Q R X qd S r )r� r r} rw r~ � assertEqual�assertIsrn )rH �valuesr� �gotr r r �_assert_valuesl s z!ShareableTypeTests._assert_valuesc
C sJ dD ]@}| � |��, t�| j|� t�| j�}| �||� W 5 Q R X qd S )Nr )r� r r} rw r~ r� )rH r� r� r r r �test_singletonsw s
z"ShareableTypeTests.test_singletonsc C s | � dd| jg� d S )Nr{ i' )r� rw ro 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)r �ir r r r
� s �z0ShareableTypeTests.test_bytes.<locals>.<genexpr>���� )r� r ro r r r �
test_bytes� s
�z ShareableTypeTests.test_bytesc C s | � dddg� d S )Nz hello worldu 你好世界� )r� ro r r r � test_strs� s z ShareableTypeTests.test_strsc C s* | � t�tdd�tjtj d g�� d S )Nr� r� r )r� r r r �sys�maxsizero r r r �test_int� s �zShareableTypeTests.test_intc C sd t jd t j d dd g}|D ]>}| �|��* | �t�� t�| j|� W 5 Q R X W 5 Q R X q d S )Nr r� �� )r� r� r� � assertRaises�
OverflowErrorr r} rw )rH 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 rI r r� b s r� c @ s$ e Zd Zdd� Zdd� Zdd� ZdS )� ListAllTestsc C s" t �� }t �� }| �||g� d S r )r r3 r� r� )rH r7 �idsr r r � test_initial� s zListAllTests.test_initialc C s6 t �� }t �� }t �� }t �� }| �||||g� d S r )r r3 rD r� r� �rH r7 �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 r3 rD 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 @ s e Zd Zdd� Zdd� ZdS )�GetCurrentTestsc C s. t �� }t �� }| �||� | �|t j� d S r )r r3 r4 r� �assertIsInstancerC )rH r7 �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 r3 rD r# r �int�stripr� r� ZassertNotEqual)rH r7 r! �outr� �_�expectedr r r �test_subinterpreter� s z#GetCurrentTests.test_subinterpreterN)rK rL rM 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� r3 r� r� rC )rH r� r7 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� rD r# r r� r� r� )rH r� r! r� r7 r r r �test_from_subinterpreter� s
z%GetMainTests.test_from_subinterpreterN)rK rL rM 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 r3 r� �
is_running)rH r7 r r r r� � s zIsRunningTests.test_mainc C sP t �� }| �t �|�� t|�� | �t �|�� W 5 Q R X | �t �|�� d S r )r rD r� r� r. r� �rH 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 rD r# r r� r� )rH r! r� r r r r� � s
� z'IsRunningTests.test_from_subinterpreterc C s6 t �� }t �|� | �t�� t �|� W 5 Q R X d S r )r rD r� r� r5 r� r� r r r �test_already_destroyed� s
z%IsRunningTests.test_already_destroyedc C s$ | � t�� t�d� W 5 Q R X d S �Ni@B )r� r5 r r� ro r r r �test_does_not_exist s z"IsRunningTests.test_does_not_existc C s$ | � t�� t�d� W 5 Q R X d S �Nr� )r� rA r r� ro r r r � test_bad_id s zIsRunningTests.test_bad_idN) rK rL rM 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� Trr )r rC r� r� �rH r0 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 ro r r r � __index__ s z8InterpreterIDTests.test_coerce_id.<locals>.Int.__index__N�rK rL rM r� r r r r �Int s r� Trr r� )r< r rC r� r� )rH r� r0 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 rC r� rA r� ro r r r r� s z InterpreterIDTests.test_bad_idc C s4 t �� }| �t�� t �t|�d � W 5 Q R X d S �Nr )r r� r� r5 rC r� r� r r r r� # s z&InterpreterIDTests.test_does_not_existc C s" t jddd�}| �t|�d� d S �Nr� Trr r� )r rC r� r< r� r r r �test_str( s zInterpreterIDTests.test_strc C s" t jddd�}| �t|�d� d S )Nr� Trr zInterpreterID(10))r rC 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 rD rC r� r� �floatr� r< �rH �id1�id2�id3r r r �
test_equality0 s$ 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 rD r� rC �assertInr� r� r r r � test_in_mainI 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 rD r� �addr� r )rH �seenr� r0 r r r �test_unique_idO s
zCreateTests.test_unique_idc sZ t �� �d � � �fdd�}t j|d�}�� |�� W 5 Q R X |�� | �� t�� � d S )Nc s t �� � ��� ��� d S r )r rD �acquire�releaser �r0 �lockr r �f\ s z%CreateTests.test_in_thread.<locals>.fr& )r( ZLockr) r* r, r� r r� �rH r� r- r r� r �test_in_threadY 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� rD r# r r� r� r� r� )rH r7 r� r� r� r r r �test_in_subinterpreterh 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� y s z6CreateTests.test_in_threaded_subinterpreter.<locals>.fr& ) r r� rD r( r) r* r, r� r� )rH r7 r� r- r r r � test_in_threaded_subinterpreteru 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 rD �appendr� r� �rH �beforer� r� r0 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� rD r� r� )rH r r� r� r� r0 r r r �test_after_destroy_some� s
z#CreateTests.test_after_destroy_someN) rK rL rM r� �unittest�skipr� r� r� r r r r r r r r� G 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 rD 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 rD r� r� r� r r r r �test_all� s zDestroyTests.test_allc sX t �� \� ��t�� t �� � W 5 Q R X � �fdd�}tj|d�}|�� |�� d S )Nc s$ �� t�� t�� � W 5 Q R X d S r �r� r5 r r� r �r7 rH r r r� � s z!DestroyTests.test_main.<locals>.fr& ) r r� r� r5 r� r( r) r* r, r� r r r r� � s
zDestroyTests.test_mainc C s6 t �� }t �|� | �t�� t �|� W 5 Q R X d S r )r rD r� r� r5 r� r r r r� � s
z#DestroyTests.test_already_destroyedc C s$ | � t�� t�d� W 5 Q R X d S r� r ro r r r r� � s z DestroyTests.test_does_not_existc C s$ | � t�� t�d� W 5 Q R X d S r� )r� rA r r� ro 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� rD r r r� r� )rH r7 r0 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 )rH r7 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 rD 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 5 Q R X | �t �|�� W 5 Q R X 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� rD r. r� r� r� r5 r� )rH r7 r! r r r �test_still_running� s
�
�z DestroyTests.test_still_runningN)
rK rL rM 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�Zedd� �Z d d
� Z
d d � Z d
d� Z e
�eed�d�dd� �Zdd� Zdd� Zdd� Zdd� Zdd � Zd d � Zejd7d d!��Zd"d#� Zd$d%� Z d&d'� Z d(d)� Z d*d+� Z d,d-� Z d.d/� Z!d0d1� Z"d2d3� Z#e
�$d4�d5d6� �Z%� Z&S )8�RunStringTestszJ
with open('{}', 'w') as out:
out.write('{}')
r� c s t � �� t�� | _d | _d S r )rE r� r rD r0 �_fsro rI r r r� s
zRunStringTests.setUpc s" | j d k r| j �� t� �� d S r )r r] rE r� ro rI r r r� s
zRunStringTests.tearDownc C s | j d krt| �| _ | j S r )r Z FSFixturero r r r �fs s
zRunStringTests.fsc C sB t d�\}}|� t�| j|� |�� }W 5 Q R X | �|d� d S )N�print("it worked!", end="")�
it worked!�r r r r0 r r� �rH r �filer� r r r � test_success s
zRunStringTests.test_successc s^ t d�\� }|�8 � �fdd�}tj|d�}|�� |�� |�� }W 5 Q R X ��|d� d S )Nr c s t ��j� � d S r )r r r0 r �r rH r r r� ) s z(RunStringTests.test_in_thread.<locals>.fr& r )r r( r) r* r, r r� )rH r r� r- r� r r! r r� & s z RunStringTests.test_in_threadc C sB t d�\}}|� t�| j|� |�� }W 5 Q R X | �|d� d S )Nz�
import threading
def f():
print('it worked!', end='')
t = threading.Thread(target=f)
t.start()
t.join()
r r r r r r �test_create_thread3 s
z!RunStringTests.test_create_thread�forkztest needs os.fork()c C s| dd l }|�d��`}|�d� |�� d}td|j� d|� d��}t�| j|� |� d� |�
� }| � ||� W 5 Q R X 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 rG r r r0 �seekr r� )rH r$ r r� r Zcontentr r r � test_forkC s
��
zRunStringTests.test_forkc
C s> t | j��* | �t�� t�| jd� W 5 Q R X W 5 Q R X d S )N�
print("spam"))r. r0 r� r5 r r ro r r r �test_already_runningY s z#RunStringTests.test_already_runningc C s@ d}|t �� kr|d7 }q| �t�� t �|d� W 5 Q R X d S )Nr r r( )r r� r� r5 r r� r r r r� ^ s
z"RunStringTests.test_does_not_existc C s& | � t�� t�dd� W 5 Q R X d S )Nr� r( )r� rA r r ro r r r �
test_error_ide s z RunStringTests.test_error_idc C s& | � t�� t�dd� W 5 Q R X d S )Nr� r( )r� r� r r ro r r r r� i s zRunStringTests.test_bad_idc C s( | � t�� t�| jd� W 5 Q R X d S r� �r� r� r r r0 ro r r r �test_bad_scriptm s z RunStringTests.test_bad_scriptc C s( | � t�� t�| jd� W 5 Q R X d S )Ns
print("spam")r+ ro r r r �test_bytes_for_scriptq s z$RunStringTests.test_bytes_for_scriptNc c sf | � tj�� }d V W 5 Q R X |d krH| �t|j��d�d t|�� n| �t|j�d�||�� d S )N�:r z{}: {})r� r �RunFailedErrorr� r< � exception�splitrB )rH �exctyper Zcaughtr r r �assert_run_failedu s �
�z RunStringTests.assert_run_failedc C s( | � t�� t�| jd� W 5 Q R X d S )Nz print("spam")r3 � SyntaxErrorr r r0 ro r r r �test_invalid_syntax� s z"RunStringTests.test_invalid_syntaxc C s* | � td�� t�| jd� W 5 Q R X d S )Nr� zraise Exception("spam"))r3 r� r r r0 ro r r r � test_failure� s zRunStringTests.test_failurec C s* | � td�� t�| jd� W 5 Q R X d S )N�42zraise SystemExit(42))r3 �
SystemExitr r r0 ro r r r �test_SystemExit� s z RunStringTests.test_SystemExitc C sV | � t�� t�| jtd�� W 5 Q R X | � td�� t�| jtd�� W 5 Q R X d S )NzG
import sys
sys.exit()
r7 zI
import sys
sys.exit(42)
)r3 r8 r r r0 r ro 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 5 Q R X | � |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� �* r< r� ry �) r= ) r r r r r r0 r �pickle�loadr� Z assertIsNone�rH 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 5 Q R X | � |d d d � | � |d
d d� | � |d d� d S ) Nzj
spam = 'eggs'
ns1 = dict(vars())
del ns1['__builtins__']
r� r; zL
ns2 = dict(vars())
del ns2['__builtins__']
zr
ns = dict(vars())
del ns['__builtins__']
import pickle
with open(r> r? �ns1r< �ns2)
r r r0 r r r r rB rC r� )rH r" r r r rE rF r r r �test_shared_overwrites� s �
z%RunStringTests.test_shared_overwritesc C sh t �� \}}ddi}td|� d��}t�| j||� t|d��}t�|�}W 5 Q R X | � |d d� d S )NrK s not __main__z�
spam = 42
ns = dict(vars())
del ns['__builtins__']
import pickle
with open(r> r? )
r r r r r r0 r rB rC r� rD 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 5 Q R X t �� \}}t�| jtd|� d��� t|d��}t�|�}W 5 Q R X | � 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� r< ) r r r r r0 r r rB rC r� r )rH r r rE rH rI r r r �test_main_reused� s �
� z RunStringTests.test_main_reusedc
C s| t �� \}}td|� d��}t�| j|� t|d��}t�|�}W 5 Q R X |� 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@ )rK �__annotations__�__doc__� __package__�__spec__r� ) r r r r r r0 r rB rC �popr� )rH r r r rE rF r r r � test_execution_namespace_is_main s" �
�z/RunStringTests.test_execution_namespace_is_mainzblocking foreverc
C sX t d�}t�� �2}t�|d|�}t�|��}|�� }W 5 Q R X W 5 Q R X | �|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� )rH r �dirname�filename�procZretcoder r r �test_still_running_at_exit$ s
z)RunStringTests.test_still_running_at_exit)N)'rK rL rM r ZSCRIPTZFILENAMEr� r� �propertyr r r� r" r Z
skipUnless�hasattrr r' r) r� r* r� r, r- �
contextlib�contextmanagerr3 r5 r6 r9 r: rG rJ rK rL rU r
rZ rO r r rI r r s<
! 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� Trr rS )r � _channel_idr� r� rV �rH rw r r r �test_default_kwargsB 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 rs rZ F�rZ r[ rs �r[ rs r[ )r[ rZ rs rS )r r` r� rV ra r r r �test_with_kwargsH 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 ro r r r r� Z s z4ChannelIDTests.test_coerce_id.<locals>.Int.__index__Nr� r r r r r� Y s r� Trr r� )r< r r` r� r� )rH r� rw r r r r� X s z ChannelIDTests.test_coerce_idc C sf | � ttjt� � | � ttjd� | � ttjd� | � ttjd� | � ttjd� | � ttjd� d S r� )r� r� r r` r� rA r� ro r r r r� ` s zChannelIDTests.test_bad_idc C s* | � t�� tjdddd� W 5 Q R X d S )Nr� F�rZ r[ )r� rA r r` ro r r r �test_bad_kwargsh s z ChannelIDTests.test_bad_kwargsc C s6 t �� }| �t j�� t �t|�d � W 5 Q R X d S r� )r r� r� r� r` r� ra r r r r� l s z"ChannelIDTests.test_does_not_existc C s" t jddd�}| �t|�d� d S r� )r r` r� r< ra r r r r� q 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� Trr z
ChannelID(10)rc zChannelID(10, send=True)re zChannelID(10, recv=True)rd )r r` r� r� ra r r r r� u 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� r< )rH �cid1�cid2Zcid3r r r r� � s$ z ChannelIDTests.test_equalityN) rK rL rM rb rf r� r� rh r� r� r� r� r r r r r_ @ 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� Ze�d�dd� �Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Z d2d3� Z d4d5� Z d6d7� Z d8S )9� ChannelTestsc C s t �� }| �|t j� d S r )r r� r� Z ChannelIDra 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� )rH 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)
r )r rD r# r r� r� r� )rH r� r� ri r� rj r r r �test_ids_global� s z ChannelTests.test_ids_globalc C s> t �� }d}t �||� t �|�}| �||� | �||� d S �Nr{ )r r� r} r~ r� Z assertIsNot)rH rw Zorigr� r r r �test_send_recv_main� 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 rD r# r )rH r� r� r r r � test_send_recv_same_interpreter� s z,ChannelTests.test_send_recv_same_interpreterc C s@ t �� }t �� }t|td|� d���}t �|�}| �|d� d S )N�_
import _xxsubinterpreters as _interpreters
_interpreters.channel_send(�, b'spam')
r{ )r r� rD r# r r~ r� )rH rw 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 sF zt �� �} W q6W q t jk
r2 t�d� Y q X q t �� | � d S )Nr� )r r~ r �time�sleepr} )r� r� r r r� � s
z8ChannelTests.test_send_recv_different_threads.<locals>.fr& r{ ) r r� r( r) r* r} r, r~ r� �rH 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 �rw r� r� r r r� � s
��zIChannelTests.test_send_recv_different_interpreters_and_threads.<locals>.fr& r{ � eggs)
r r� rD r( r) r* r} r, r~ r� rw r ry r �1test_send_recv_different_interpreters_and_threads� s
z>ChannelTests.test_send_recv_different_interpreters_and_threadsc C s( | � tj�� t�dd� W 5 Q R X d S )Nr� r{ )r� r r� r} ro r r r �test_send_not_found s z ChannelTests.test_send_not_foundc C s& | � tj�� t�d� W 5 Q R X d S r� )r� r r� r~ ro r r r �test_recv_not_found s z ChannelTests.test_recv_not_foundc C s. t �� }| �t j�� t �|� W 5 Q R X d S r )r r� r� r r~ ra r r r �test_recv_empty s z ChannelTests.test_recv_emptyc 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� rD r# r �dictrZ r~ r� r� �rH rw 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')
)rE r{ rZ ) r r� r` rD 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 sr t �� }t �|d� t �|� t �|� | �t j�� t �|d� W 5 Q R X | �t j�� t �|� W 5 Q R X d S �Nr{ rz �r r� r} r~ r� r� rP ra r r r �test_close_single_userA 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 5 Q R X | �dt |j
�� | �t j�� }t �|td|� d��� W 5 Q R X | �dt |j
�� d S )Nrr rs z_
import _xxsubinterpreters as _interpreters
_interpreters.channel_recv(r z,
_interpreters.channel_send(z, b'spam')
rP ) r r� rD r r r� r� r/ r� r< r0 )rH rw r� r� �cmr r r �test_close_multiple_usersL s(
�
�
�
�z&ChannelTests.test_close_multiple_usersc C sN t �� }t �|d� t �|� t �|� | �t j�� t �|� W 5 Q R X d S ro r� ra r r r �test_close_multiple_timesd s
z&ChannelTests.test_close_multiple_timesc C s� ddddg}|D ]�\}}| � ||f��z t�� }t�|d� t�|� tj|||d� | �tj�� t�|d� W 5 Q R X | �tj�� t�|� W 5 Q R X W 5 Q R X qd S )N)FF�TF)FT)TTr{ rg rz )r� r r� r} r~ r� r� rP )rH ZtestsrZ r[ rw r r r �test_close_emptym s �
z ChannelTests.test_close_emptyc C s\ t �� }t �|d� t �|d� | �t j�� t �|� W 5 Q R X t �|� t �|d� d S )Nr{ r; rz �r r� r} r� ZChannelNotEmptyErrorr� r~ ra 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 5 Q R X t �|� t �|d� t �|� t �|� t j|dd� d S )Nr{ r; T�r[ rz r� ra r r r �*test_close_recv_with_unused_items_unforced� s
z7ChannelTests.test_close_recv_with_unused_items_unforcedc C s� t �� }t �|d� t �|d� t j|dd� | �t j�� t �|d� W 5 Q R X t �|� t �|� | �t j�� t �|� W 5 Q R X d S )Nr{ r; T�rZ rz �r r� r} r� r� rP r~ ra 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 5 Q R X t �|� t �|d� t �|� t �|� t j|dd� d S )Nr{ r; Tr| rz r� r� ra r r r �*test_close_both_with_unused_items_unforced� s
z7ChannelTests.test_close_both_with_unused_items_unforcedc C sz t �� }t �|d� t �|d� t j|ddd� | �t j�� t �|d� W 5 Q R X | �t j�� t �|� W 5 Q R X d S )Nr{ r; Tre rz r� ra r r r �(test_close_recv_with_unused_items_forced� s z5ChannelTests.test_close_recv_with_unused_items_forcedc C sz t �� }t �|d� t �|d� t j|ddd� | �t j�� t �|d� W 5 Q R X | �t j�� t �|� W 5 Q R X d S )Nr{ r; Trc rz r� ra 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 5 Q R X | �t j�� t �|� W 5 Q R X d S )Nr{ r; Trd rz r� ra 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 5 Q R X | �t j�� t �|� W 5 Q R X d S ro )r r� r� r� rP r} r~ ra 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 5 Q R X | �t j�� t � |� W 5 Q R X d S )Nr{ z`
import _xxsubinterpreters as _interpreters
_interpreters.channel_close(z, force=True)
)
r r� r} rD r r r� rP r~ r� �rH rw 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 5 Q R X | �t j�� t �|� W 5 Q R X d S )Nr{ Trr rz r� ra r r r �-test_close_used_multiple_times_by_single_user� s
z:ChannelTests.test_close_used_multiple_times_by_single_userN) rK rL rM rl rm rn rp rq rt 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 rk � s8
rk c @ s\ e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dd� Z dS )�ChannelReleaseTestsc C sx t �� }t �|d� t �|� t j|ddd� | �t j�� t �|d� W 5 Q R X | �t j�� t �|� W 5 Q R X d S �Nr{ Trg rz �r r� r} r~ �channel_releaser� rP ra 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 ) Nrr rs ze
import _xxsubinterpreters as _interpreters
obj = _interpreters.channel_recv(z,)
_interpreters.channel_release(z+)
print(repr(obj))
z+
_interpreters.channel_release(r zb'spam')r r� rD r r r# r� r� )rH rw r� r� r� r r r �test_multiple_users, s
� ��
� z'ChannelReleaseTests.test_multiple_usersc C sr t �� }t �|d� t �|� t �|� | �t j�� t �|d� W 5 Q R X | �t j�� t �|� W 5 Q R X d S r� r� ra r r r �test_no_kwargs@ s
z"ChannelReleaseTests.test_no_kwargsc C sZ t �� }t �|d� t �|� t j|ddd� | �t j�� t j|ddd� W 5 Q R X d S )Nr{ Trg r� ra r r r �test_multiple_timesK s
z'ChannelReleaseTests.test_multiple_timesc C sV t �� }t �|d� t �|d� t j|ddd� | �t j�� t �|� W 5 Q R X d S )Nr{ r; Trg )r r� r} r� r� rP r~ ra r r r �test_with_unused_itemsT s z*ChannelReleaseTests.test_with_unused_itemsc C s\ t �� }t �|� | �t j�� t �|d� W 5 Q R X | �t j�� t �|� W 5 Q R X d S ro )r r� r� r� rP r} r~ ra r r r �test_never_used] s
z#ChannelReleaseTests.test_never_usedc C s| t �� }t �|d� t �� }t �|td|� d��� t �|�}t �|� | �t j �� t �|d� W 5 Q R X | �
|d� d S )Nr{ zb
import _xxsubinterpreters as _interpreters
_interpreters.channel_release(r rz ) r r� r} rD r r r~ r� r� rP r� )rH rw r! r� r r r �test_by_unassociated_interpf s
�
z/ChannelReleaseTests.test_by_unassociated_interpc C sT t �� }t �� }t �|td|� d|� d��� | �t j�� t �|� W 5 Q R X d S )Nze
import _xxsubinterpreters as _interpreters
obj = _interpreters.channel_send(z5, b'spam')
_interpreters.channel_release(r )r r� rD r r r� rP r~ r� r r r �test_close_if_unassociatedu 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� )rH rw r� r r r �test_partially� s
z"ChannelReleaseTests.test_partiallyc C s� t �� }t �|d� t �|d� t �|d� t �|� t j|ddd� | �t j�� t �|d� W 5 Q R X | �t j�� t �|� W 5 Q R X d S r� r� ra r r r �'test_used_multiple_times_by_single_user� 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� � 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 )NrX ) �QUICKr: r@ rG rE rF r� �_preppedrh �_state�_known)r> rV r! r_ r` �creatorZknownrH rI r r rF � s&
�zChannelCloseFixture.__new__c C s | j S r �r� ro r r r rx � s zChannelCloseFixture.statec C s@ z| j W S tk
r: | �| j�}| �|�| _ | j Y S X d S r )�_cid�AttributeError�_get_interpreterr� � _new_channel)rH r� r r r rw � s zChannelCloseFixture.cidc C s | � |�}| �|� |S r )r� �_prep_interpreterr� r r r �get_interpreter� s
z#ChannelCloseFixture.get_interpreterNc C s2 |d kr| j }|dkr&| jjdkr&dS t| jj�S )Nr[ rZ F)rV rx 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� s z$ChannelCloseFixture.prep_interpreterc C s
|| _ d S r r� )rH rU ry r r r �
record_action� s z!ChannelCloseFixture.record_actionc C s t � t� d S r r� ro r r r �clean_up� s z ChannelCloseFixture.clean_upc C s@ |j dkrt�� S t�� }t|jd|� d�� t�|�| _| jS )Nr7 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
)rG r r� r2 r0 r~ r� )rH r� Zchr r r r� � s
�
z ChannelCloseFixture._new_channelc C sl |dkr| j S |dkr | jS |dkr*| jS |}z| j| }W n& tk
rb t|� }| j|< Y nX |S d S )N)rX r! r_ r` )r! r_ r` r� �KeyErrorr: )rH r! rG r r r r� � s z$ChannelCloseFixture._get_interpreterc C sF |j | jkrd S | j�|j � |jdkr,d S t|j d| j� d�� d S )Nr7 a
import _xxsubinterpreters as interpreters
import test.test__xxsubinterpreters as helpers
ChannelState = helpers.ChannelState
try:
cid
except NameError:
cid = interpreters._channel_id(r )r0 r� r� rG r2 rw r� r r r r� � s
�z%ChannelCloseFixture._prep_interpreter)N)rK rL rM r� rF r[ rx rw r� r� r� r� r� r� r� r� rO r r rI r r� � s
r� z end interp other extra creatorz%these tests take several hours to runc @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d �d
d�Zdd� Z dd� Z
d!dd�Z dd� Z d"dd�Z
dd � Zd d � Zd d � ZdS )#�ExhaustiveChannelTestsc 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 )NrX r_ r` )�_iter_close_action_sets� _iter_post_close_action_sets�_iter_use_action_sets)rH Z closeactionsZ postactionsZ
useactions�actionsr r r �iter_action_sets7 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 )NrW r[ rZ �rR )rH �interp1�interp2�interpsZinterp3Zinterp4r r r r� d 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 )Nr| r� r^ r] r\ r[ rZ r� )
rH r� r� Zendsr� rs �opr! rV ZrecvopZsendopr[ rZ r r r r� � s
�
�z.ExhaustiveChannelTests._iter_close_action_setsc c s. dD ]$}t dd|�gV t dd|�gV qd S )N)rX r` r_ rW r[ rZ r� r� r r r r� � s
�
�z3ExhaustiveChannelTests._iter_post_close_action_setsc C s |D ]}| � ||� qd S r )rz )rH �fixr� rU r r r � run_actions� s z"ExhaustiveChannelTests.run_actionsTrt 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 ) Nr7 rt 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� rG rz rw rU rx r� r r� r2 r0 r� rh r� �
from_bytesr~ r� )rH r� rU ru rV r! ry r� r r r rz � sB
��������
�z!ExhaustiveChannelTests.run_actionc c sF ddddg}|D ]0\}}}dD ] }dD ]}t |||||�V q&q qd S )N)r7 r! r` )r! r7 r` )r� r� r` )r� r� r7 )rX r_ r� rg )r� )rH r r! r_ r` r� rV r r r �
iter_fixtures� s �z$ExhaustiveChannelTests.iter_fixturesc C s` |rdnd}t ||jd�}|�� s4| j||dd� n(| �tj�� | j||dd� W 5 Q R X d S )Nr^ r] rX Frt )rR rV r� rz r� r rP )rH r� rs r� r] r r r �_close� s z ExhaustiveChannelTests._closeNc C s� |d ks|j dkr�| �tj�� t�|j� W 5 Q R X | �tj�� t�|jd� W 5 Q R X | �tj�� t�|j� W 5 Q R X | �tj�� tj|jdd� W 5 Q R X n0t|j d� t|j d� t|j d� t|j d� d S ) Nr7 r{ Trr 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)
)
rG r� r rP r~ rw r} r� r2 r0 )rH 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
dkrTq:| �||� q:|� d�}| �||� d S )N)rX r_ r7 Zfresh) r� rx rj r rk r r~ rw r� r� rG )rH 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 r r� � z ({} actions)� r� rd �.z---)r� �printr� rB r r� �stdoutr% )rH �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� �rH r� r� r� r r r �_skim_close_tests- s z(ExhaustiveChannelTests._skim_close_testsc
C sp | � � D ]b\}}}| �d�|||���6 |�|j� | �||� | j|dd� | �|� W 5 Q R X |�� qd S )N� {} {} {}Frr � r� r� rB r� r! r� r� r� r� r� r r r �
test_close2 s z!ExhaustiveChannelTests.test_closec
C sp | � � D ]b\}}}| �d�|||���6 |�|j� | �||� | j|dd� | �|� W 5 Q R X |�� qd S )Nr� Trr r� r� r r r �test_force_close> s z'ExhaustiveChannelTests.test_force_close)N)F)rK rL rM r� r� r� r� r� rz r� r� r� r� r� r� r� r� r r r r r� s +-1
r� rO )N)6� collectionsr r] r r rB r� �textwrapr r( ru r Ztestr Z test.supportr �
import_moduler r r r# r^ r. r2 r/ r9 r: rQ rR rh rz rv r� r� ZTestCaser� r� r� r� r� r� r� r� r� r r r_ rk r� r� r
r� rK r7 r r r r �<module> sr
&
3$
2?(;]b <Y _ &�q @