← Back
Editing: 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
Save File
Cancel