File: //lib/python3.6/site-packages/supervisor/tests/__pycache__/test_socket_manager.cpython-36.pyc
3
�k�`�! � @ s� d Z ddlZddlZddlZddlZddlZddlZy ddlZW n ek
rX dZY nX ddl m
Z
ddl m Z ddl m
Z
ddl mZ G dd� d�ZG d d
� d
ej�ZG d d � d ej�ZG d
d� dej�Zdd� Zdd� Zedkr�ejdd� dS )z(Test suite for supervisor.socket_manager� N)�DummySocketConfig)� DummyLogger)�UnixStreamSocketConfig)�InetStreamSocketConfigc @ s$ e Zd Zdd� Zdd� Zdd� ZdS )�Subjectc C s
d| _ d S )N� )�value)�self� r
�)/usr/lib/python3.6/test_socket_manager.py�__init__ s zSubject.__init__c C s | j S )N)r )r r
r
r �getValue s zSubject.getValuec C s
|| _ d S )N)r )r �valr
r
r �setValue s zSubject.setValueN)�__name__�
__module__� __qualname__r r
r r
r
r
r r s r c @ s< e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
S )� ProxyTestc C s
d| _ d S )NF)�on_deleteCalled)r r
r
r �setUp! s zProxyTest.setUpc C s ddl m} |S )Nr )�Proxy)�supervisor.socket_managerr )r r r
r
r �_getTargetClass$ s zProxyTest._getTargetClassc O s | j � ||�S )N)r )r �args�kwr
r
r �_makeOne( s zProxyTest._makeOnec C s
d| _ d S )NT)r )r r
r
r �setOnDeleteCalled+ s zProxyTest.setOnDeleteCalledc C s | j t� �}| jd|j� � d S )Nr )r r � assertEqualr
)r �proxyr
r
r �test_proxy_getattr. s z ProxyTest.test_proxy_getattrc C s< | j t� | jd�}| jd|j� � d }t� | j| j� d S )N)Z on_deleter )r r r r r
�
gc_collect�
assertTruer )r r r
r
r �test_on_delete2 s
zProxyTest.test_on_deleteN) r r r 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 )�ReferenceCounterTestc C s
d| _ d S )NF)�running)r r
r
r r ; s zReferenceCounterTest.setUpc C s
d| _ d S )NT)r$ )r r
r
r �start> s zReferenceCounterTest.startc C s
d| _ d S )NF)r$ )r r
r
r �stopA s zReferenceCounterTest.stopc C s ddl m} |S )Nr )�ReferenceCounter)r r' )r r' r
r
r r D s z$ReferenceCounterTest._getTargetClassc O s | j � ||�S )N)r )r r r r
r
r r H s z ReferenceCounterTest._makeOnec C s� | j | j| jd�}| j| j� |j� | j| j� | jd|j� � |j� | j| j� | jd|j� � |j � | j| j� | jd|j� � |j � | j| j� | jd|j� � d S )N)�on_zero� on_non_zero� � r )
r r& r% � assertFalser$ Z incrementr! r Z get_count� decrement)r �ctrr
r
r �test_incr_and_decrK s z'ReferenceCounterTest.test_incr_and_decrc C s$ | j | j| jd�}| jt|j� d S )N)r( r) )r r& r% � assertRaises� Exceptionr- )r r. r
r
r � test_decr_at_zero_raises_error[ s z3ReferenceCounterTest.test_decr_at_zero_raises_errorN)
r r r r r% r& r r r/ r2 r
r
r
r r# 9 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!S )"�SocketManagerTestc C s
t � d S )N)r )r r
r
r �tearDowna s zSocketManagerTest.tearDownc C s ddl m} |S )Nr )�
SocketManager)r r5 )r r5 r
r
r r d s z!SocketManagerTest._getTargetClassc O s | j � ||�S )N)r )r r r r
r
r r h s zSocketManagerTest._makeOnec C s< t d�}| j|�}d|jt|�|jf }| jt|�|� d S )Nr+ z<%s at %s for %s>)r r � __class__�idZurlr �repr)r �conf� sock_managerZexpectedr
r
r � test_reprk s
zSocketManagerTest.test_reprc C s& t d�}| j|�}| j||j� � d S )Nr+ )r r r �config)r r9 r: r
r
r �test_get_configr s
z!SocketManagerTest.test_get_configc C s> t dd�}| j|�}| j|j|� |j� }| j|j� d� d S )NZ localhost�a� � 127.0.0.1)r? r> )r r r �
socket_config�
get_socket� getsockname)r r9 r: �sockr
r
r �test_tcp_w_hostnamew s
z%SocketManagerTest.test_tcp_w_hostnamec C s> t dd�}| j|�}| j|j|� |j� }| j|j� d� d S )N� 127.0.0.1�a� )rE rF )r r r r@ rA rB )r r9 r: rC r
r
r �
test_tcp_w_ip~ s
z SocketManagerTest.test_tcp_w_ipc C sV t j� \}}t|�}| j|�}| j|j|� |j� }| j|j� |� d }tj |� d S )N)
�tempfileZmkstempr r r r@ rA rB �os�close)r Ztf_fdZtf_namer9 r: rC r
r
r � test_unix� s
zSocketManagerTest.test_unixc C s< t d�}| j|�}| j|j� � |j� }| j|j� � t|j� �}|j� }t|j� �}| j||� | j ||� | j d|j
� � | j|j� � | j|j j � d }t
� | j|j� � | j|j j � d }t
� | j|j� � | j|j j � |j� }| j|j� � t|j� �}| j||� ~t
� | j|j� � | j|j j � d S )Nr+ )r r r, � is_preparedrA r! r7 Z_getZassertNotEqualr Zget_socket_ref_count�socket� close_calledr ) r r9 r: rC Zsock_idZsock2Zsock2_idZsock3Zsock3_idr
r
r �test_socket_lifecycle� s:
z'SocketManagerTest.test_socket_lifecyclec C s� t d�}t� }| j||d�}|j� }| jt|j�d� | jdt|� |jd � ~t� | jt|j�d� | jdt|� |jd � d S )Nr* )�loggerzCreating socket %sr r+ zClosing socket %s) r r r rA r �len�datar8 r )r r9 rP r: rC r
r
r � test_logging� s z SocketManagerTest.test_loggingc C sP t d�}| j|�}|j� }| j|j� � | j|j� | j|j� | j|j� d S )Nr* ) r r rA r! rL r, Z bind_calledZ
listen_calledrN )r r9 r: rC r
r
r �test_prepare_socket� s
z%SocketManagerTest.test_prepare_socketc C sB t ddd�}| j|�}|j� }| j|j� � | j|j|j� � d S )Nr* �* )�backlog)r r rA r! rL r �listen_backlogZ get_backlog)r r9 r: rC r
r
r �+test_prepare_socket_uses_configured_backlog� s
z=SocketManagerTest.test_prepare_socket_uses_configured_backlogc C s@ t dd d�}| j|�}|j� }| j|j� � | j|jtj� d S )Nr* )rV ) r r rA r! rL r rW rM Z SOMAXCONN)r r9 r: rC r
r
r �;test_prepare_socket_uses_somaxconn_if_no_backlog_configured� s
zMSocketManagerTest.test_prepare_socket_uses_somaxconn_if_no_backlog_configuredc C s< t dd�}| j|�}|j� }| j|�}| jtj|j� ~d S )Nz 127.0.0.1ia� )r r rA r0 rM �error)r r9 r: rC Z
sock_manager2r
r
r � test_tcp_socket_already_taken� s
z/SocketManagerTest.test_tcp_socket_already_takenc C s& t d�}| j|�}| jtj|j� d S )Nz/notthere/foo.sock)r r r0 rM rZ rA )r r9 r: r
r
r �test_unix_bad_sock� s
z$SocketManagerTest.test_unix_bad_sockc C sp t dd�}| j|�}| j|j� � y|j� | j� W n4 tk
rj } z| j|jd d� W Y d d }~X nX d S )Nz 127.0.0.1ia� r z Socket has not been prepared) r r r, rL Z_closeZfailr1 r r )r r9 r: �er
r
r �#test_close_requires_prepared_socket� s
z5SocketManagerTest.test_close_requires_prepared_socketN)r r r r4 r r r; r= rD rG rK rO rS rT rX rY r[ r\ r^ r
r
r
r r3 _ s
+ r3 c C s$ t d k r tj� tj� tj� d S )N)�__pypy__�gcZcollectr
r
r
r r � s r c C s t jtjt �S )N)�unittestZ
findTestCases�sys�modulesr r
r
r
r �
test_suite� s rd �__main__)Z defaultTest)�__doc__r` rb rI ra rM rH r_ � ImportErrorZsupervisor.tests.baser r Zsupervisor.datatypesr r r ZTestCaser r# r3 r rd r �mainr
r
r
r �<module> s.
&