Edit File: test_daemon.cpython-36.pyc
3 ��X( � @ sZ d dl Z d dlZd dlZd dlZd dlZd dlZd dlmZmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZ d dlZejdd� �Zejdd� �Zdd� Zd d � Zd d � Z d d� Z dd� Z dd� Z dd� Z dd� Z!dd� Z"dd� Z#dd � Z$d d � Z%d d � Z&d!d"� Z'd#d$� Z(d%d&� Z)d'd(� Z*d)d*� Z+d+d,� Z,e j-d2k�rBe.Z/ne0Z/d.d/� Z1d0d1� Z2dS )3� N)�booted�is_fifo�_is_fifo� is_socket� _is_socket�is_socket_inet�_is_socket_inet�is_socket_unix�_is_socket_unix�is_socket_sockaddr�_is_socket_sockaddr�is_mq�_is_mq� listen_fds�notifyc c sH y d V W n8 t k rB } z | jtjkr0tj� � W Y d d } ~ X nX d S )N)�OSError�errnoZENOSYS�pytest�skip)�e� r �#/usr/lib64/python3.6/test_daemon.py� skip_enosys s r c c s4 t j| �}z |V W d |d j� |d j� X d S )Nr � )�socketZ socketpair�close)Zfamily�pairr r r �closing_socketpair s r c C s* t jjd�rt� s&t�nt� dks&t�d S )Nz/run/systemd/systemFT> FT)�os�path�existsr �AssertionErrorr r r r � test_booted& s r" c C sJ | j d�j}tj|� tj|tjtjB �}t|d �s8t �t||�sFt �d S )Nz test.fifo) �join�strpath�posix�mkfifor �open�O_RDONLY� O_NONBLOCKr r! )�tmpdirr �fdr r r � test__is_fifo. s r, c C sR | j d�}|jd� |j}tj|tjtjB �}t|d � s>t�t||� sNt�d S )Nz test.fifo�boo) r# �writer$ r r' r( r) r r! )r* �filer r+ r r r �test__is_fifo_file6 s r0 c C s\ | j d�j}tjt�� tdd � s(t�W d Q R X tjt�� td|� sNt�W d Q R X d S )Nz test.fifor ���r1 )r# r$ r �raisesr r r! )r* r r r r �test__is_fifo_bad_fd? s r3 c C sr | j d�j}tj|� tj|tjtjB �}tj|d�}t |d �sDt �t ||�sRt �t |d �s`t �t ||�snt �d S )Nz test.fifo�r) r# r$ r% r& r r' r( r) �fdopenr r! )r* r r+ r/ r r r � test_is_fifoH s r6 c C s~ | j d�}|jd� |j}tj|tjtjB �}tj|d�}t|d � sJt �t||� sZt �t|d � sjt �t||� szt �d S )Nz test.fifor- r4 ) r# r. r$ r r' r( r) r5 r r! )r* r/ r r+ r r r �test_is_fifo_fileS s r7 c C s\ | j d�j}tjt�� tdd � s(t�W d Q R X tjt�� td|� sNt�W d Q R X d S )Nz test.fifor r1 r1 )r# r$ r r2 r r r! )r* r r r r �test_is_fifo_bad_fd_ s r8 c C s> yt | �S tk r8 } z|jtjks*t�dS d }~X nX d S )NF)r r r �EBADFr! )�arg�errorr r r � is_mq_wrapperh s r< c C s> yt | �S tk r8 } z|jtjks*t�dS d }~X nX d S )NF)r r r r9 r! )r: r; r r r �_is_mq_wrapperp s r= c C s t tj��� } x�| D ]�}t|� s&t�t|� s4t�t|� sBt�t� � t|d� sZt�W d Q R X |j � }t|� szt�t|� s�t�t|� s�t�t� � t|d� s�t�W d Q R X t |� s�t�t |� s�t�t |� s�t�t� � t |d� s�t�W d Q R X qW W d Q R X d S )Nz127.0.0.1:2000)r r �AF_UNIXr r! r< r r r �filenor r= r r )r �sockr+ r r r �test_no_mismatchx s$ rA c C sL t tj���6} �x,| D �]"}x�||j� fD ]�}t|�s:t�t|tj�sJt�t|tj� s\t�t|tjtj�spt�t|tjtj� s�t�t � � t |dtjdd� s�t�W d Q R X q*W t |�s�t�t |tj�s�t�t |tj� s�t�t |tjtj�s�t�t |tjtj� �st�t � � t |dtjdd� �s0t�W d Q R X qW W d Q R X d S )Nz 8.8.8.8:2000r ) r r r> r? r r! �AF_INET� SOCK_STREAM� SOCK_DGRAMr r r r )r r@ r: r r r �test_is_socket� s" & rE c E C s, t jtjtj����} | jd � | j� \}}dj|�}�x�d D �]�}�xT| | j� fD �]B}t� � t |dtj �stt �W d Q R X t� � t |d| tj �s�t �W d Q R X t� � t |d| |d�s�t �W d Q R X t� � t |d| d d�s�t �W d Q R X t� �" t |d| | d� �st �W d Q R X t j t�� t |ddd� W d Q R X t� � t |dtj � �sft �W d Q R X t� � t |dtj� �s�t �W d Q R X qTW t j t�� t|ddd� W d Q R X t� � t|dtj � �s�t �W d Q R X t� � t|dtj� �st �W d Q R X | jd � q>W W d Q R X d S ) N� 127.0.0.1r z:{}r )� listeningi@� )Zflowinfoz129.168.11.11:23� )rF r )r r r1 )� contextlib�closingr rB �bindZ getsockname�formatr? r r rC r! r r2 � ValueErrorrD r Zlisten)r@ ZaddrZportrG r: r r r �test_is_socket_sockaddr� s: & $ rN c C s� t tj���} x�| D ]�}|j� }t|�s*t�t|tj�s:t�t|tj� sLt�t|tjtj�s`t�t|tjtj� svt�t|�s�t�t|tj�s�t�t|tj� s�t�t|tjtj�s�t�t|tjtj� st�qW W d Q R X d S )N) r r r> r? r r! rB rC rD )r r@ r+ r r r �test__is_socket� s rO c C s~ t tj��j} xb| D ]Z}xT||j� fD ]D}t|�s4t�t|dd� sFt�t|tj�sVt�t|tj� s$t�q$W qW W d Q R X d S )Nz /no/such/path)r )r r r> r? r r! rC rD )r r@ r: r r r �test_is_socket_unix� s rP c C sr t tj��^} xV| D ]N}|j� }t|�s*t�t|ddd� s>t�t|tj�sNt�t|tj� st�qW W d Q R X d S )Nr r z /no/such/pathr1 )r r r> r? r r! rC rD )r r@ r+ r r r �test__is_socket_unix� s rQ c C sF t jd� t jd� t� g ks"t�td�g ks2t�td�g ksBt�d S )N� LISTEN_FDS� LISTEN_PIDTF)r �unsetenvr r! r r r r �test_listen_fds_no_fds� s rU c C sZ dt jd< tt j� �t jd< td�dddgks2t�td�dddgksHt�t� g ksVt�d S ) N�3rR rS F� � � T)r �environ�str�getpidr r! r r r r �test_listen_fds� s r] c C sP dt jd< tt j� �t jd< td�dgks.t�t� dgks>t�t� g ksLt�d S )N�1rR rS FrW )r rZ r[ r\ r r! r r r r � test_listen_fds_default_unset� s r_ c C s� t d�dkst�t� � t dg d�dks,t�W d Q R X t dddgd�dksNt�t dtj� d�dksft�t dtj� d d�dks�t�d S ) NzREADY=1Fz FDSTORE=1)�fdsr � )�pid)rb r` )r )r r! r r r\ r r r r �test_notify_no_socket s rc rW c / C s� dt jd< tjt�� td� W d Q R X tjt��$ t� � tdg d� W d Q R X W d Q R X tjt�� tdddgd� W d Q R X tjt�� tdt j� d� W d Q R X tjt�� tdt j� d d � W d Q R X d S ) Nz /dev/null� NOTIFY_SOCKETzREADY=1z FDSTORE=1)r` r ra )rb )rb r` )r )r rZ r r2 �connection_errorr r r\ r r r r �test_notify_bad_socket s rf c C s� | j d�j}tjtjtj�}y|j|� W n2 tjk r\ } ztjd| � W Y d d }~X nX t tdd�}|j tj |d� |t j d< td�s�t�t� � tdg d �s�t�W d Q R X tddd gd �s�t�tdt j� d �s�t�tdt j� d d �s�t�d S )Nr zfailed to bind socket (%s)� SO_PASSCRED� r rd zREADY=1z FDSTORE=1)r` ra )rb )rb r` )r )r# r$ r r> rD rK r; r Zxfail�getattrZ setsockoptZ SOL_SOCKETr rZ r r! r r\ )r* r r@ r rg r r r �test_notify_with_socket s rj )rW )3�sysr r% r rI r Zsystemd.daemonr r r r r r r r r r r r r r r r �contextmanagerr r r" r, r0 r3 r6 r7 r8 r< r= rA rE rN rO rP rQ rU r] r_ rc � version_info�ConnectionRefusedErrorre r rf rj r r r r �<module> sB D &