a
    cl_M                     @   s`   d Z ddlZddlZddlZddlmZ dZdZej	edZ
ej	edZG dd	 d	eZdS )
z"Blackbox tests for traffic_summary    N)BlackboxTestCasezscript/traffic_summary.plz$python/samba/tests/blackbox/testdataztraffic_summary.pdmlztraffic_summary.expectedc                   @   s   e Zd Zdd Zdd ZdS )TrafficSummaryTestsc                 C   s"   d}t j|dd}| }|dkS )zwCheck that perl XML::Twig module is installed.
        Traffic summary depends on this module being installed.
        zperl -MXML::Twig -e 1T)shellr   )
subprocessPopenwait)selflinepZretcode r   F/usr/lib/python3/dist-packages/samba/tests/blackbox/traffic_summary.py
check_twig"   s    zTrafficSummaryTests.check_twigc                 C   s   |   s| d |  T}dtt|f }t| | | tt	 }t|	 }| 
|| W d    n1 st0    Y  d S )Nz&Perl module XML::Twig is not installedz	%s %s >%s)r   ZskipTestZmktempSCRIPTINPUTprintZ	check_runopenEXPECTED_FN	readlinesZassertEqual)r   outputZcommandZexpectedZactualr   r   r   test_traffic_summary+   s    


z(TrafficSummaryTests.test_traffic_summaryN)__name__
__module____qualname__r   r   r   r   r   r   r       s   	r   )__doc__osr   ZtempfileZsamba.testsr   r   ZDATA_DIRpathjoinr   r   r   r   r   r   r   <module>   s   