File: //usr/lib/python3/dist-packages/certbot/tests/__pycache__/reporter_test.cpython-38.pyc
U
�]�
� @ sJ d Z ddlZddlZddlZddlZG dd� dej�ZedkrFe�� dS )zTests for certbot.reporter.� Nc @ sp e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� ZdS )� ReporterTestz$Tests for certbot.reporter.Reporter.c C s6 ddl m} |�tjdd��| _tj| _t� � t_d S )Nr )�reporterF)�quiet)
Zcertbotr ZReporter�mockZ MagicMock�sys�stdout�
old_stdout�six�StringIO)�selfr � r �=/usr/lib/python3/dist-packages/certbot/tests/reporter_test.py�setUp s zReporterTest.setUpc C s | j t_d S �N)r r r �r r r r
�tearDown s zReporterTest.tearDownc C sF | j �d| j j� | j �� tj�� }| �d|k� | �d|k� d S )Nz
Line 1
Line 2zLine 1
zLine 2)r � add_message� LOW_PRIORITY�print_messagesr r �getvalue�
assertTrue�r �outputr r r
�test_multiline_message s
z#ReporterTest.test_multiline_messagec C s dd� t j_| �� d S )Nc S s dS �NTr r r r r
�<lambda> � z3ReporterTest.test_tty_print_empty.<locals>.<lambda>)r r �isatty�test_no_tty_print_emptyr r r r
�test_tty_print_empty s z!ReporterTest.test_tty_print_emptyc C sZ | j �� | �tj�� d� zt�W n tk
rB | j �� Y nX | �tj�� d� d S )N� )r r Z assertEqualr r r �
ValueErrorr r r r
r s
z$ReporterTest.test_no_tty_print_emptyc C s dd� t j_| �� d S )Nc S s dS r r r r r r
r * r z7ReporterTest.test_tty_successful_exit.<locals>.<lambda>)r r r �_successful_exit_commonr r r r
�test_tty_successful_exit) s z%ReporterTest.test_tty_successful_exitc C s | � � d S r )r" r r r r
�test_no_tty_successful_exit- s z(ReporterTest.test_no_tty_successful_exitc C s dd� t j_| �� d S )Nc S s dS r r r r r r
r 1 r z9ReporterTest.test_tty_unsuccessful_exit.<locals>.<lambda>)r r r �_unsuccessful_exit_commonr r r r
�test_tty_unsuccessful_exit0 s z'ReporterTest.test_tty_unsuccessful_exitc C s | � � d S r )r% r r r r
� test_no_tty_unsuccessful_exit4 s z*ReporterTest.test_no_tty_unsuccessful_exitc C sX | � � | j�� tj�� }| �d|k� | �d|k� | �d|k� | �d|k� d S �NzIMPORTANT NOTES:�High�Med�Low)�
_add_messagesr r r r r r r r r r
r" 7 s
z$ReporterTest._successful_exit_commonc C sv | � � zt�W n tk
r. | j�� Y nX tj�� }| �d|k� | �d|k� | �d|k� | �d|k� d S r( )r, r! r r r r r r r r r r
r% @ s
z&ReporterTest._unsuccessful_exit_commonc C sB | j �d| j j� | j jd| j jdd� | j jd| j jdd� d S )Nr) r* F)Zon_crashr+ )r r Z
HIGH_PRIORITYZMEDIUM_PRIORITYr r r r r
r, L s � �zReporterTest._add_messagesN)�__name__�
__module__� __qualname__�__doc__r r r r r r# r$ r& r' r" r% r, r r r r
r s r �__main__) r0 r Zunittestr r ZTestCaser r- �mainr r r r
�<module> s K