File: //lib/python3/dist-packages/twisted/application/twist/test/__pycache__/test_options.cpython-310.pyc
o
�b�, � @ s� d Z ddlmZmZ ddlmZmZmZmZm Z m
Z
ddl Z ddl
mZ ddlmZ ddlmZ ddlmZmZmZmZ dd lmZ dd
lmZ d d l m Z d d
l m Z d dl m!Z! d dl"m#Z# d dl$m%Z% ddl%m&Z& G dd� de j'j(j)�Z*dS )z2
Tests for L{twisted.application.twist._options}.
� )�stderr�stdout)�Callable�Dict�List�Optional�TextIO�TupleN)�version)�reactor)� IReactorCore)�FileLogObserver�LogLevel�jsonFileLogObserver�textFileLogObserver)�
UsageError)�
MemoryReactor� )�
NoSuchReactor)�
ExitStatus)� DummyExit)� ServiceMaker)�_options� )� TwistOptionsc @ sJ e Zd ZdZd@dd�Zd@dd�Zd@dd �Zd@d
d �Zd@d d
�Zd@dd�Z d@dd�Z
d@dd�Z d@dd�Z d@dd�Z
d@dd�Zdededdfd d �Zd@d d �Zd@d d!�Zd@d"d#�Zd@d$d%�Zd&ed'eegef ddfd(d)�Zd@d*d+�Zd@d,d-�Zd@d.d/�Zd@d0d1�Z d@d2d3�Z d@d4d5�Z d@d6d7�Z d@d8d9�Z d@d:d;�Z!d@d<d=�Z"d@d>d?�Z#dS )A� OptionsTestsz$
Tests for L{TwistOptions}.
�returnNc 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 s8 g � _ ddtdtt dtf� fdd�
}� �td|� dS ) zZ
Patch L{_options.open} so we can capture usage and prevent actual opens.
N�name�moder c s( | dkr
t d d | ��� j�| |f� tS )N� nocanopen)�OSError�opened�append�NotImplemented)r$ r% r r! r" �fakeOpen1 s z(OptionsTests.patchOpen.<locals>.fakeOpen�openFile)N)r( �strr r r r )r r+ r! r r" � patchOpen+ s zOptionsTests.patchOpenc s. i � _ dtdtf� fdd� }� �td|� dS )zo
Patch C{_options.installReactor} so we can capture usage and prevent
actual installs.
r$ r c s" | dkrt � �t� }|� j| < |S )N�fusion)r r �installedReactors)r$ r r r! r" �installReactorA s
z8OptionsTests.patchInstallReactor.<locals>.installReactorr1 N)r0 r- r r r )r r1 r! r r" �patchInstallReactor: s z OptionsTests.patchInstallReactorc C s t � }| �|�� �d�� dS )z@
L{TwistOptions.getSynopsis} appends arguments.
z plugin [plugin_options]N)r �
assertTrue� getSynopsis�endswith�r �optionsr! r! r" �
test_synopsisK 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 � opt_version� assertEqualsr �statusr �EX_OK�messager
r6 r! r! r" � test_versionS s
zOptionsTests.test_versionc C s@ | � � t� }|�d� | �t| j�dh� | �|d d� dS )zo
L{TwistOptions.installReactor} installs the chosen reactor and sets
the reactor name.
r/ � reactorNameN)r2 r � opt_reactor� assertEqual�setr0 r: r6 r! r! r" � test_reactor` s
zOptionsTests.test_reactorc C s8 | � � t� }d|_|�dg� | �t| j�dh� dS )z�
L{TwistOptions.installReactor} installs the chosen reactor after the
command line options have been parsed.
ztest-subcommandz--reactor=fusionr/ N)r2 r �
subCommand� parseOptionsrA rB r0 r6 r! r! r" �test_installCorrectReactorm s
z'OptionsTests.test_installCorrectReactorc C s" | � � t� }| �t|jd� dS )zn
L{TwistOptions.installReactor} raises UsageError if an unknown reactor
is specified.
�coalN)r2 r � assertRaisesr r@ r6 r! r! r" �test_installReactorBogusz 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 r1 r6 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.
�warn�logLevelN)r �
opt_log_levelrK r rM r6 r! r! r" �test_logLevelValid� s
z OptionsTests.test_logLevelValidc C � t � }| �t|jd� dS )zi
L{TwistOptions.opt_log_level} with an invalid log level name raises
UsageError.
�cheeseN)r rH r rO r6 r! r! r" �test_logLevelInvalid� � z!OptionsTests.test_logLevelInvalidr$ �expectedStreamc 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.
�logFileN)r � opt_log_filerK )r r$ rU r7 r! r! r" � _testLogFile� s
zOptionsTests._testLogFilec C � | � dt� dS )zW
L{TwistOptions.opt_log_file} given C{"-"} as a file name uses stdout.
�-N)rX r r r! r! r" �test_logFileStdout� � z OptionsTests.test_logFileStdoutc C rY )zW
L{TwistOptions.opt_log_file} given C{"+"} as a file name uses stderr.
�+N)rX r r r! r! r" �test_logFileStderr� r\ 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)r_ �aN)r. r rW rA r( r6 r! r! r" �test_logFileNamed� s
z OptionsTests.test_logFileNamedc C sX | � � | �� t� }|�d� | �| jjtj� | � | jj
� | � | 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 rW r: r r; r �EX_IOERR�assertIsNotNoner= r3 �
startswithr6 r! r! r" �test_logFileCantOpen� s
��z!OptionsTests.test_logFileCantOpen�format�expectedObserverFactoryc C s4 t � }|�|� | �|d |� | �|d |� dS )z�
Set log file format and check the selected observer factory.
@param format: The format of the file.
@param expectedObserverFactory: The expected observer factory.
�fileLogObserverFactory� logFormatN)r �opt_log_formatrK rA )r rf rg r7 r! r! r" �_testLogFormat� s
zOptionsTests._testLogFormatc C rY )�g
L{TwistOptions.opt_log_format} given C{"text"} uses a
L{textFileLogObserver}.
�textN)rk r r r! r! r" �test_logFormatText� � z OptionsTests.test_logFormatTextc C rY )rl �jsonN)rk r r r! r! r" �test_logFormatJSON� ro z OptionsTests.test_logFormatJSONc C rQ )zk
L{TwistOptions.opt_log_format} given an invalid format name raises
L{UsageError}.
�frommageN)r rH r rj r6 r! r! r" �test_logFormatInvalid� rT 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.
rm �quesorh ri N)r. r rj rW �selectDefaultLogObserverrK r rA r6 r! r! r" �'test_selectDefaultLogObserverNoOverride� s
z4OptionsTests.test_selectDefaultLogObserverNoOverridec C sZ G dd� d�}| � td|� � t� }|�d� |�� | �|d t� | �|d d� dS ) rt c @ s e Zd Zdefdd�ZdS )zIOptionsTests.test_selectDefaultLogObserverDefaultWithTTY.<locals>.TTYFiler c S s dS )NTr! r r! r! r" �isatty s zPOptionsTests.test_selectDefaultLogObserverDefaultWithTTY.<locals>.TTYFile.isattyN)�__name__�
__module__� __qualname__�boolrx r! r! r! r" �TTYFile s r} r rZ rh ri rm N)r r r rW rv rK r rA )r r} r7 r! r! r" �+test_selectDefaultLogObserverDefaultWithTTY s
z8OptionsTests.test_selectDefaultLogObserverDefaultWithTTYc C sD | � � t� }|�d� |�� | �|d t� | �|d d� dS )rt ru rh ri rp N)r. r rW rv rK r rA r6 r! r! r" �.test_selectDefaultLogObserverDefaultWithoutTTY 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�assertIsInstancer- r )r r7 r� r$ r! r! r" �test_pluginsType- 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.
�webN)r �assertInr� r6 r! r! r" �test_pluginsIncludeWeb8 s z#OptionsTests.test_pluginsIncludeWebc C sP t � }|jD ] \}}}}| �|t� | �|d� | �t|�� | �|t� qdS )z~
L{TwistOptions.subCommands} is an iterable of tuples as expected by
L{twisted.python.usage.Options}.
N)r � subCommandsr� r- rK r3 �callable)r r7 r$ �shortcut�parser�docr! r! r" �test_subCommandsTypeB s �z!OptionsTests.test_subCommandsTypec C s0 t � }t|j�}dd� |jD �}| �||� dS )zW
L{TwistOptions.subCommands} includes a sub-command for every plug-in.
c S s h | ]\}}}}|�qS r! r! )�.0r$ r� r� r� r! r! r" � <setcomp>V s z:OptionsTests.test_subCommandsIncludeWeb.<locals>.<setcomp>N)r rB r� r� rA )r r7 r� r� r! r! r" �test_subCommandsIncludeWebO 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)r2 r rH r � postOptionsr6 r! r! r" � test_postOptionsNoSubCommandZ s z)OptionsTests.test_postOptionsNoSubCommand)r N)$ry rz r{ �__doc__r# r. r2 r8 r> rC rF rI rL rP rS r- r rX r[ r^ ra re r r
rk rn rq rs rw r~ r r� r� r� r� r� r! r! r! r" r sH
� �
�
r )+r� �sysr r �typingr r r r r r �twisted.trial.unittest�twisted�twisted.copyrightr
�twisted.internetr �twisted.internet.interfacesr �twisted.loggerr
r r r �twisted.python.usager �twisted.test.proto_helpersr �reactorsr � runner._exitr �runner.test.test_runnerr �servicer �twistr r �trial�unittest�TestCaser r! r! r! r" �<module> s"