← Back
Editing: enhancements_test.cpython-38.pyc
U �]� � @ sl d Z ddlZddlZddlZddlmZ ddlmZ G dd� dej �Z G dd� dej �Z e d krhe� � dS ) zModule for enhancement UI.� N)�errors)�utilc @ sF e Zd ZdZdd� Zdd� Zedd� �Ze� d�d d � �Z d d � Z d S )�AskTestzTest the ask method.c C s t �t j� d S �N)�logging�disableZCRITICAL��self� r �I/usr/lib/python3/dist-packages/certbot/tests/display/enhancements_test.py�setUp s z AskTest.setUpc C s t �t j� d S r )r r ZNOTSETr r r r �tearDown s zAskTest.tearDownc C s ddl m} ||�S )Nr )�ask)� certbot.display.enhancementsr )�clsZ enhancementr r r r �_call s z AskTest._call�!certbot.display.enhancements.utilc C s$ t jdf|� j_| �| �d�� d S )N� Zredirect�� display_util�OK�menu� return_valueZ assertTruer �r Z mock_utilr r r � test_redirect s zAskTest.test_redirectc C s | � tj| jd� d S )NZunknown_enhancement)Z assertRaisesr �Errorr r r r r �test_key_error s zAskTest.test_key_errorN) �__name__� __module__� __qualname__�__doc__r r � classmethodr �mock�patchr r r r r r r s r c @ sR e Zd ZdZedd� �Ze�d�dd� �Ze�d�dd� �Z e�d�d d � �Z d S ) � RedirectTestz$Test the redirect_by_default method.c C s ddl m} |� S )Nr )�redirect_by_default)r r% )r r% r r r r # s zRedirectTest._callr c C s" t jdf|� j_| �| �� � d S �Nr r r r r r � test_secure( s zRedirectTest.test_securec C s" t jdf|� j_| �| �� � d S r& )r ZCANCELr r � assertFalser r r r r � test_cancel- s zRedirectTest.test_cancelc C s" t jdf|� j_| �| �� � d S )Nr )r r r r r( r r r r r � test_easy2 s zRedirectTest.test_easyN) r r r r r! r r" r# r' r) r* r r r r r$ ! s r$ �__main__)r r Zunittestr" Zcertbotr Zcertbot.displayr r ZTestCaser r$ r �mainr r r r �<module> s
Save File
Cancel