File: //lib/python3/dist-packages/twisted/application/twist/test/__pycache__/test_options.cpython-38.pyc
U
W[�* � @ s� d Z ddlmZmZ ddlmZ ddlmZ ddlm Z ddl
m Z m Z m
Z
ddlmZ dd lmZ dd
lmZ dd lmZ dd lmZ dd
lmZ ddlmZ ddlZ G dd� de j j j �Z dS )z2
Tests for L{twisted.application.twist._options}.
� )�stdout�stderr)�reactor)�version)�
UsageError)�LogLevel�textFileLogObserver�jsonFileLogObserver)�
MemoryReactor� )�
NoSuchReactor)� ServiceMaker)�
ExitStatus)� DummyExit)�_options� )� TwistOptionsNc @ s� e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:S );� OptionsTestsz$
Tests for L{TwistOptions}.
c C s t � | _| �td| j� dS )zX
Patch L{_twist.exit} so we can capture usage and prevent actual exits.
�exitN)r r �patchr ��self� r �M/usr/lib/python3/dist-packages/twisted/application/twist/test/test_options.py� patchExit s zOptionsTests.patchExitc s& g � _ d� fdd� }� �td|� dS )zZ
Patch L{_options.open} so we can capture usage and prevent actual opens.
Nc s( | dkrt d d | ��� j�| |f� tS )N� nocanopen)�IOError�opened�append�NotImplemented)�name�moder r r �fakeOpen- s z(OptionsTests.patchOpen.<locals>.fakeOpenZopenFile)N)r r r )r r" r r r � patchOpen' s zOptionsTests.patchOpenc s$ i � _ � fdd�}� �td|� dS )zo
Patch C{_options.installReactor} so we can capture usage and prevent
actual installs.
c s" | dkrt � �t� }|� j| < |S )N�fusion)r r
�installedReactors)r r r r r �installReactor> s
z8OptionsTests.patchInstallReactor.<locals>.installReactorr& N)r% r r )r r& r r r �patchInstallReactor7 s z OptionsTests.patchInstallReactorc C s t � }| �|�� �d�� dS )z@
L{TwistOptions.getSynopsis} appends arguments.
z plugin [plugin_options]N)r �
assertTrueZ getSynopsis�endswith�r �optionsr r r �
test_synopsisI s �zOptionsTests.test_synopsisc C s< | � � t� }|�� | �| jjtj� | �| jjt � dS )zl
L{TwistOptions.opt_version} exits with L{ExitStatus.EX_OK} and prints
the version.
N)
r r Z opt_version� assertEqualsr �statusr �EX_OK�messager r* r r r � test_versionT s
zOptionsTests.test_versionc C sD | � � t� }|�d� | �t| j�tdg�� | �|d d� dS )zo
L{TwistOptions.installReactor} installs the chosen reactor and sets
the reactor name.
r$ Z reactorNameN)r' r � opt_reactor� assertEqual�setr% r- r* r r r � test_reactorb s
zOptionsTests.test_reactorc C s< | � � t� }d|_|�dg� | �t| j�tdg�� dS )z�
L{TwistOptions.installReactor} installs the chosen reactor after the
command line options have been parsed.
ztest-subcommandz--reactor=fusionr$ N)r' r Z
subCommandZ parseOptionsr3 r4 r% r* r r r �test_installCorrectReactorp s
z'OptionsTests.test_installCorrectReactorc C s" | � � t� }| �t|jd� dS )zn
L{TwistOptions.installReactor} raises UsageError if an unknown reactor
is specified.
ZcoalN)r' r � assertRaisesr r2 r* r r r �test_installReactorBogus~ s z%OptionsTests.test_installReactorBogusc C s t � }| �t|�d�� dS )z�
L{TwistOptions.installReactor} returns the currently installed reactor
when the default reactor name is specified.
�defaultN)r �assertIdenticalr r& r* r r r �test_installReactorDefault� s z'OptionsTests.test_installReactorDefaultc C s& t � }|�d� | �|d tj� dS )zQ
L{TwistOptions.opt_log_level} sets the corresponding log level.
�warnZlogLevelN)r �
opt_log_levelr: r r< r* r r r �test_logLevelValid� s
z OptionsTests.test_logLevelValidc C s t � }| �t|jd� dS )zi
L{TwistOptions.opt_log_level} with an invalid log level name raises
UsageError.
ZcheeseN)r r7 r r= r* r r r �test_logLevelInvalid� s z!OptionsTests.test_logLevelInvalidc C s$ t � }|�|� | �|d |� dS )z�
Set log file name and check the selected output stream.
@param name: The name of the file.
@param expectedStream: The expected stream.
ZlogFileN)r � opt_log_filer: )r r ZexpectedStreamr+ r r r � _testLogFile� s
zOptionsTests._testLogFilec C s | � dt� dS )zW
L{TwistOptions.opt_log_file} given C{"-"} as a file name uses stdout.
�-N)rA r r r r r �test_logFileStdout� s z OptionsTests.test_logFileStdoutc C s | � dt� dS )zW
L{TwistOptions.opt_log_file} given C{"+"} as a file name uses stderr.
�+N)rA r r r r r �test_logFileStderr� s z OptionsTests.test_logFileStderrc C s, | � � t� }|�d� | �dg| j� dS )zX
L{TwistOptions.opt_log_file} opens the given file name in append mode.
�mylog)rF �aN)r# r r@ r3 r r* r r r �test_logFileNamed� s
z OptionsTests.test_logFileNamedc C sJ | � � | �� t� }|�d� | �| jjtj� | � | jj
� d�� dS )z�
L{TwistOptions.opt_log_file} exits with L{ExitStatus.EX_IOERR} if
unable to open the log file due to an L{EnvironmentError}.
r z%Unable to open log file 'nocanopen': N) r r# r r@ r- r r. r �EX_IOERRr( r0 �
startswithr* r r r �test_logFileCantOpen� s
��z!OptionsTests.test_logFileCantOpenc C s4 t � }|�|� | �|d |� | �|d |� dS )z�
Set log file format and check the selected observer.
@param format: The format of the file.
@param expectedObserver: The expected observer.
�fileLogObserverFactory� logFormatN)r �opt_log_formatr: r3 )r �formatZexpectedObserverr+ r r r �_testLogFormat� s
�zOptionsTests._testLogFormatc C s | � dt� dS )�g
L{TwistOptions.opt_log_format} given C{"text"} uses a
L{textFileLogObserver}.
�textN)rP r r r r r �test_logFormatText� s z OptionsTests.test_logFormatTextc C s | � dt� dS )rQ �jsonN)rP r r r r r �test_logFormatJSON� s z OptionsTests.test_logFormatJSONc C s t � }| �t|jd� dS )zk
L{TwistOptions.opt_log_format} given an invalid format name raises
L{UsageError}.
ZfrommageN)r r7 r rN r* r r r �test_logFormatInvalid s z"OptionsTests.test_logFormatInvalidc C sN | � � t� }|�d� |�d� |�� | �|d t� | �|d d� dS )�r
L{TwistOptions.selectDefaultLogObserver} will not override an already
selected observer.
rR �quesorL rM N)r# r rN r@ �selectDefaultLogObserverr: r r3 r* r r r �'test_selectDefaultLogObserverNoOverride
s
�z4OptionsTests.test_selectDefaultLogObserverNoOverridec C s\ G dd� dt �}| �td|� � t� }|�d� |�� | �|d t� | �|d d� dS ) rW c @ s e Zd Zdd� ZdS )zIOptionsTests.test_selectDefaultLogObserverDefaultWithTTY.<locals>.TTYFilec S s dS )NTr r r r r �isatty$ s zPOptionsTests.test_selectDefaultLogObserverDefaultWithTTY.<locals>.TTYFile.isattyN)�__name__�
__module__� __qualname__r[ r r r r �TTYFile# s r_ r rB rL rM rR N) �objectr r r r@ rY r: r r3 )r r_ r+ r r r �+test_selectDefaultLogObserverDefaultWithTTY s
�z8OptionsTests.test_selectDefaultLogObserverDefaultWithTTYc C sD | � � t� }|�d� |�� | �|d t� | �|d d� dS )rW rX rL rM rT N)r# r r@ rY r: r r3 r* r r r �.test_selectDefaultLogObserverDefaultWithoutTTY4 s
�z;OptionsTests.test_selectDefaultLogObserverDefaultWithoutTTYc C s6 t � }|j}|D ] }| �|t� | �|| t� qdS )zM
L{TwistOptions.plugins} is a mapping of available plug-ins.
N)r �plugins�assertIsInstance�strr
)r r+ rc r r r r �test_pluginsTypeE s
z OptionsTests.test_pluginsTypec C s t � }| �d|j� dS )z�
L{TwistOptions.plugins} includes a C{"web"} plug-in.
This is an attempt to verify that something we expect to be in the list
is in there without enumerating all of the built-in plug-ins.
ZwebN)r ZassertInrc r* r r r �test_pluginsIncludeWebQ s z#OptionsTests.test_pluginsIncludeWebc C sP t � }|jD ]>\}}}}| �|t� | �|d� | �t|�� | �|t� q dS )z~
L{TwistOptions.subCommands} is an iterable of tuples as expected by
L{twisted.python.usage.Options}.
N)r � subCommandsrd re r: r( �callable)r r+ r �shortcut�parser�docr r r �test_subCommandsType\ s z!OptionsTests.test_subCommandsTypec C s4 t � }t|j�}tdd� |jD ��}| �||� dS )zW
L{TwistOptions.subCommands} includes a sub-command for every plug-in.
c s s | ]\}}}}|V qd S )Nr )�.0r rj rk rl r r r � <genexpr>q s
z:OptionsTests.test_subCommandsIncludeWeb.<locals>.<genexpr>N)r r4 rc rh r3 )r r+ rc rh r r r �test_subCommandsIncludeWebj s
�z'OptionsTests.test_subCommandsIncludeWebc C s | � � t� }| �t|j� dS )zd
L{TwistOptions.postOptions} raises L{UsageError} is it has no
sub-command.
N)r' r r7 r Z postOptionsr* r r r � test_postOptionsNoSubCommandx s z)OptionsTests.test_postOptionsNoSubCommandN) r\ r] r^ �__doc__r r# r' r, r1 r5 r6 r8 r; r> r? rA rC rE rH rK rP rS rU rV rZ ra rb rf rg rm rp rq r r r r r s:
r )!rr �sysr r Ztwisted.internetr Ztwisted.copyrightr Ztwisted.python.usager Ztwisted.loggerr r r Ztwisted.test.proto_helpersr
Zreactorsr �servicer
Z runner._exitr Zrunner.test.test_runnerr Ztwistr r Ztwisted.trial.unittestZtwistedZtrialZunittestZTestCaser r r r r �<module> s