Edit File: 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. &
Back to File Manager