o

�/aM�@s`dZd
dl
Z
d
dlZd
dlZd
dlmZ
dZdZe
j�	ed�Z
e
j�	ed�ZGdd	�d	e�ZdS)
z"Blackbox tests for traffic_summary�N)
�BlackboxTestCasezscript/traffic_summary.plz$python/samba/tests/blackbox/testdataztraffic_summary.pdmlztraffic_summary.expectedc@seZ
dZd
d�Zdd�ZdS)�TrafficSummaryTestsc
Cs"d
}
tj
|
dd�}|��}|dkS)zwCheck that perl XML::Twig module is installed.
        Traffic summary depends on this module being installed.
        zperl -MXML::Twig -e 1T)
�shellr
)�
subprocess�Popen�wait)�self�line�
p�retcode�r�F/usr/lib/python3/dist-packages/samba/tests/blackbox/traffic_summary.py�
check_twig"s


zTrafficSummaryTests.check_twigc
Cs�|��s	|�
d
�

|���+}
dtt|
f}t|�

|�|�

tt�
�	�}t|
�
�	�}|�
||�
Wd�
dS1s;w



Y
dS)Nz&Perl module XML::Twig is not installedz	%s %s >%s)r�skipTest�mktemp�SCRIPT�INPUT�print�	check_run�open�EXPECTED_FN�	readlines�assertEqual)r�output�command�expected�actualrrr
�test_traffic_summary+s










"�z(TrafficSummaryTests.test_traffic_summaryN)�__name__�
__module__�__qualname__rrrrrr
r s	r)
�__doc__�osr�tempfile�samba.testsrr�DATA_DIR�path�joinrrrrrrr
�<module>
s