← Back
Editing: test_usage.cpython-38.pyc
U ��W[]\ � @ st d Z ddlmZmZ ddlmZ ddlmZ G dd� dej�Z G dd� dej �Z G d d � d ej�Z G d d � d ej �Z G d d� dej�ZG dd� dej�ZG dd� dej �ZG dd� dej �ZG dd� dej�ZG dd� dej�ZG dd� dej�ZG dd � d ej �ZG d d � d ej �ZG d d � d ej �ZG d!d"� d"ej �ZG d#d$� d$ej �ZG d%d&� d&ej �ZG d'd(� d(ej �Zd)S )*zK Tests for L{twisted.python.usage}, a command line option parsing library. � )�division�absolute_import)�unittest)�usagec @ sV e Zd Zddddgdddgdd d gd d d d ggZdddgddggZdd� Zdd� Zd S )� WellBehaved�long�w�defaultzand a docstring�another�nz no docstring�longonlyN�noshort� shortless�exceptzthis one got docstring�aflag�fzO flagallicious docstringness for this here �flout�oc C s d| j d< d S )N�PONY!�myflag��opts��self� r �9/usr/lib/python3/dist-packages/twisted/test/test_usage.py� opt_myflag s zWellBehaved.opt_myflagc C s d|f | j d< d S )Nz%s WITH A PONY!�myparamr �r �valuer r r � opt_myparam" s zWellBehaved.opt_myparam)�__name__� __module__� __qualname__� optParameters�optFlagsr r r r r r r s ���� r c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) �ParseCorrectnessTestszZ Test L{usage.Options.parseOptions} for correct values under good conditions. c C s$ d� � | _t� | _| j�| j� dS )zL Instantiate and parseOptions a well-behaved Options class. �A--long Alpha -n Beta --shortless Gamma -f --myflag --myparam TofuN)�split�niceArgVr �nice� parseOptionsr r r r �setUp, s zParseCorrectnessTests.setUpc C sT | � | jjd d� | � | jjd d� | � | jjd d� | � | jjd d� d S ) z1 Parameters have correct values. r ZAlphar ZBetar r r ZGammaN�� assertEqualr* r r r r r �test_checkParameters9 s z*ParseCorrectnessTests.test_checkParametersc C s, | � | jjd d� | � | jjd d� dS )z, Flags have correct values. r � r r Nr- r r r r �test_checkFlagsB s z%ParseCorrectnessTests.test_checkFlagsc C s, | � | jjd d� | � | jjd d� dS )zB Custom flags and parameters have correct values. r r r zTofu WITH A PONY!Nr- r r r r �test_checkCustomsI s z'ParseCorrectnessTests.test_checkCustomsN)r! r" r# �__doc__r, r/ r1 r2 r r r r r& ' s r& c @ sL e Zd Zddddegddddegdddd egd ddd eggZd d � ZeZdS )� TypedOptions�foointN� zFoo int�foofloat��Q� ��@z Foo float�eggintzEgg int without default�eggfloatzEgg float without defaultc C s || _ dS )zg This option has an underscore in its name to exercise the _ to - translation. N)�underscoreValuer r r r �opt_under_scoreZ s z TypedOptions.opt_under_score)r! r" r# �int�floatr$ r< Zopt_ur r r r r4 R s �r4 c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dS )� TypedTestszK Test L{usage.Options.parseOptions} for options with forced types. c C s t � | _d S �N)r4 r r r r r r, h s zTypedTests.setUpc C s� g }| j �|� | �| j jd d� | �| j jd t� | �| j jd d� | �| j jd t� | �| j jd � | �| j jd � dS )z, Default values are parsed. r5 r6 r7 r8 r9 r: N)r r+ r. r �assertIsInstancer= r> � assertIsNone�r �argVr r r �test_defaultValuesk s z TypedTests.test_defaultValuesc C s� d� � }| j�|� | �| jjd d� | �| jjd t� | �| jjd d� | �| jjd t� | �| jjd d� | �| jjd t� | �| jjd d � | �| jjd t� d S ) z2 int and float values are parsed. z8--fooint 912 --foofloat -823.1 --eggint 32 --eggfloat 21r5 i� r7 g����̸��r9 � r: g 5@N)r( r r+ r. r rA r= r> rC r r r �test_parsingValuesy s z TypedTests.test_parsingValuesc C s$ | j �ddg� | �| j jd� dS )zw A dash in an option name is translated to an underscore before being dispatched to a handler. z --under-score�fooN�r r+ r. r; r r r r �test_underscoreOption� s z TypedTests.test_underscoreOptionc C s$ | j �ddg� | �| j jd� dS )zE An option name with a dash in it can have an alias. z-u�barNrI r r r r �test_underscoreOptionAlias� s z%TypedTests.test_underscoreOptionAliasc C s d� � }| �tj| jj|� dS )z7 Passing wrong values raises an error. z --fooint eggN)r( � assertRaisesr � UsageErrorr+ rC r r r �test_invalidValues� s z TypedTests.test_invalidValuesN) r! r" r# r3 r, rE rG rJ rL rO r r r r r? d s r? c @ s e Zd ZdddddggZdS )�WrongTypedOptions�barwrongNzBar with wrong coerceZhe)r! r" r# r$ r r r r rP � s �rP c @ s8 e Zd Zdd� Zdd� Zddddegdddd eggZdS ) �WeirdCallableOptionsc C s t d��d S )NZOuch)� RuntimeError�r r r r �_bar� s zWeirdCallableOptions._barc C s t d��d S )NZYay)� ValueErrorrT r r r �_foo� s zWeirdCallableOptions._foorQ NzBar with strange callableZfoowrongzFoo with strange callable)r! r" r# rU rW r$ r r r r rR � s �rR c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �WrongTypedTestszF Test L{usage.Options.parseOptions} for wrong coerce options. c C s" t � }d�� }| �t|j|� dS )z2 Using a non-callable type fails. z--barwrong eggN)rP r( rM � TypeErrorr+ �r �usrD r r r �test_nonCallable� s z WrongTypedTests.test_nonCallablec C s t � }g }|�|� dS )zP The coerce functions are not called if no values are provided. N)rR r+ rZ r r r �test_notCalledInDefault� s z'WrongTypedTests.test_notCalledInDefaultc C sR t � }d�� }| �tj|j|�}| �t|�d� t � }d�� }| �t|j|� dS )zI Errors raised by coerce functions are handled properly. z--foowrong blahz&Parameter type enforcement failed: Yayz--barwrong blahN) rR r( rM r rN r+ r. �strrS )r r[ rD �er r r �test_weirdCallable� s z"WrongTypedTests.test_weirdCallableN)r! r" r# r3 r\ r] r` r r r r rX � s rX c @ s e Zd Zdd� ZdS )� OutputTestsc C s. t � }| �tj|jdg�}| �t|�d� dS )zF Error output case adjustment does not mangle options z-Zzoption -Z not recognizedN)r rM r rN r+ r. r^ )r Zoptr_ r r r �test_uppercasing� s �z OutputTests.test_uppercasingN)r! r" r# rb r r r r ra � s ra c @ s e Zd ZdgZdgZdS )�InquisitionOptions)�expectr_ )�torture-device�t� comfy-chairz set preferred torture deviceN)r! r" r# r% r$ r r r r rc � s ��rc c @ s e Zd ZddgZdS )�HolyQuestOptions)� horseback�hz use a horse)� for-grail�gN)r! r" r# r% r r r r rh � s �rh c @ s* e Zd ZdgZddedfddedfgZdS ) �SubCommandOptions)�europian-swallowNz$set default swallow type to Europian� inquisition�inquestzPerform an inquisition� holyquestZquestzEmbark upon a holy questN)r! r" r# r% rc rh � subCommandsr r r r rm � s ���rm c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dS )�SubCommandTestszJ Test L{usage.Options.parseOptions} for options with subcommands. c C sd t � }|�ddg� | �|d � | �|jd� | �|jt� | �|jd � | �|jd d� dS )z- A subcommand is recognized. �--europian-swallowro rn rd re rg N) rm r+ � assertTruer. � subCommandrA � subOptionsrc � assertFalse�r r r r r �test_simpleSubcommand s z%SubCommandTests.test_simpleSubcommandc C sf t � }|�dddg� | �|d � | �|jd� | �|jt� | �|jd � | �|jd d� dS ) zA Flags and options of a subcommand are assigned. ro �--expect�--torture-device=featherrn rd re �featherN� rm r+ rx r. rv rA rw rc ru ry r r r �"test_subcommandWithFlagsAndOptions s z2SubCommandTests.test_subcommandWithFlagsAndOptionsc C sf t � }|�dddg� | �|d � | �|jd� | �|jt� | �|jd � | �|jd d� d S ) zG Flags and options of a subcommand alias are assigned. rp r{ r| rn ro rd re r} Nr~ ry r r r �'test_subcommandAliasWithFlagsAndOptions s z7SubCommandTests.test_subcommandAliasWithFlagsAndOptionsc C sb t � }|�ddg� | �|d � | �|jd� | �|jt� | �|jd � | �|jd � dS )zG Flags and options of another subcommand are assigned. rq z --for-grailrn ri rk N) rm r+ rx r. rv rA rw rh ru ry r r r �)test_anotherSubcommandWithFlagsAndOptions* s z9SubCommandTests.test_anotherSubcommandWithFlagsAndOptionsc C s@ t � }|�dg� | �|d � | �|j� | �t|d�� dS )z� If no subcommand is specified and no default subcommand is assigned, a subcommand will not be implied. rt rn rw N)rm r+ ru rB rv rx �hasattrry r r r �test_noSubcommand6 s z!SubCommandTests.test_noSubcommandc C sh t � }d|_|�dg� | �|d � | �|jd� | �|jt� | � |jd � | �|jd d� dS ) zK Flags and options in the default subcommand are assigned. rp rt rn ro rd re rg N) rm ZdefaultSubCommandr+ ru r. rv rA rw rc rx ry r r r �test_defaultSubcommandA s z&SubCommandTests.test_defaultSubcommandc s` G dd� dt j�� G � fdd�dt j�}|� }|�dg� | �t|jd�� | �|jj|� dS )zw The parseOptions method from the Options object specified for the given subcommand is called. c @ s e Zd Zdd� ZdS )zDSubCommandTests.test_subCommandParseOptionsHasParent.<locals>.SubOptc _ s | j | _tjj| f|�|� d S r@ )�parent� sawParentr �Optionsr+ )r �a�kwr r r r+ T s zQSubCommandTests.test_subCommandParseOptionsHasParent.<locals>.SubOpt.parseOptionsN)r! r" r# r+ r r r r �SubOptS s r� c s e Zd Zdd� dfgZdS )zASubCommandTests.test_subCommandParseOptionsHasParent.<locals>.OptrH r rK N�r! r" r# rr r �r� r r �OptW s �r� rH r� N)r r� r+ ru r� rw r. r� )r r� r r r� r �$test_subCommandParseOptionsHasParentN s z4SubCommandTests.test_subCommandParseOptionsHasParentc s� G dd� dt j�� G � fdd�dt j�}G � fdd�dt j�}|� }|�dg� |� }|�dg� | �t|jd �� | �t|jd �� | �|jj|� | �|jj|� d S ) zh The .parent pointer is correct even when the same Options class is used twice. c @ s e Zd ZdS )z:SubCommandTests.test_subCommandInTwoPlaces.<locals>.SubOptN)r! r" r# r r r r r� e s r� c s e Zd Zdd� dfgZdS )z:SubCommandTests.test_subCommandInTwoPlaces.<locals>.OptFoorH r �quuxNr� r r� r r �OptFoog s �r� c s e Zd Zdd� dfgZdS )z:SubCommandTests.test_subCommandInTwoPlaces.<locals>.OptBarrK �br� Nr� r r� r r �OptBark s �r� rH rK r� N)r r� r+ ru r� rw ZfailUnlessIdenticalr� )r r� r� ZoFooZoBarr r� r �test_subCommandInTwoPlaces` s z*SubCommandTests.test_subCommandInTwoPlacesN) r! r" r# r3 rz r r� r� r� r� r� r� r r r r rs s rs c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) �HelpStringTestsz& Test generated help strings. c C s d� � | _t� | _dS )z; Instantiate a well-behaved Options class. r' N)r( r) r r* r r r r r, } s zHelpStringTests.setUpc C s@ z| j �� W n, tk r: } z| �|� W 5 d}~X Y nX dS )z, __str__ shouldn't go boom. N)r* �__str__� ExceptionZfail)r r_ r r r � test_noGoBoom� s z HelpStringTests.test_noGoBoomc C sF dd� t | j��� D �}| �t|�dk� | �|d �d�dk� dS )zK Extra whitespace in flag and parameters docs is stripped. c S s g | ]}|� d �dkr|�qS )r r )�find)�.0�sr r r � <listcomp>� s zJHelpStringTests.test_whitespaceStripFlagsAndParameters.<locals>.<listcomp>r Z flagalliciousN)r^ r* � splitlinesru �lenr� )r �linesr r r �&test_whitespaceStripFlagsAndParameters� s z6HelpStringTests.test_whitespaceStripFlagsAndParametersN)r! r" r# r3 r, r� r� r r r r r� y s r� c @ s e Zd ZdZdd� Zdd� ZdS )�PortCoerceTestsz3 Test the behavior of L{usage.portCoerce}. c C s: | � dt�d�� | � dt�d�� | � dt�d�� dS )z4 Test the answers with valid input. r �0i� Z3210i�� Z65535N)r. r � portCoercer r r r �test_validCoerce� s z PortCoerceTests.test_validCoercec C sD | � ttjd� | � ttjd� | � ttjd� | � ttjd� dS )z" Test error path. � z-21Z212189rH N)rM rV r r� r r r r �test_errorCoerce� s z PortCoerceTests.test_errorCoerceN)r! r" r# r3 r� r� r r r r r� � s r� c @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dS )�ZshCompleterTestsz� Test the behavior of the various L{twisted.usage.Completer} classes for producing output usable by zsh tab-completion system. c C sN t �� }|�dt j�}| �|d� t jddd�}|�dt j�}| �|d� dS )zX Completer produces zsh shell-code that produces no completion matches. � some-optionz :some-option:� some actionT�Zdescr�repeat�*:some action:N)r � Completer� _shellCode�_ZSHr. �r �c�gotr r r �test_completer� s z ZshCompleterTests.test_completerc C st t �� }|�dt j�}| �|d� t �d�}|�dt j�}| �|d� t jdddd�}|�dt j�}| �|d� d S ) zn CompleteFiles produces zsh shell-code that completes file names according to a glob. r� z :some-option (*):_files -g "*"z*.pyz$:some-option (*.py):_files -g "*.py"r� Tr� z%*:some action (*.py):_files -g "*.py"N)r � CompleteFilesr� r� r. r� r r r � test_files� s z ZshCompleterTests.test_filesc C sN t �� }|�dt j�}| �|d� t jddd�}|�dt j�}| �|d� dS )zV CompleteDirs produces zsh shell-code that completes directory names. r� z:some-option:_directoriesr� Tr� z*:some action:_directoriesN)r � CompleteDirsr� r� r. r� r r r � test_dirs� s zZshCompleterTests.test_dirsc C s� t �d�}|�dt j�}| �|d� t �dddg�}|�dt j�}| �|d� t jdddgdd d �}|�dt j�}| �|d � d S ) zw CompleteList produces zsh shell-code that completes words from a fixed list of possibilities. �ABCr� z:some-option:(A B C)�1�2�3z:some-option:(1 2 3)r� Tr� z*:some action:(1 2 3)N)r � CompleteListr� r� r. r� r r r � test_list� s �zZshCompleterTests.test_listc C s� t �d�}|�dt j�}| �|d� t �dddg�}|�dt j�}| �|d� t jdddgdd d �}|�dt j�}d }| �||� d S ) z� CompleteMultiList produces zsh shell-code that completes multiple comma-separated words from a fixed list of possibilities. r� r� z-:some-option:_values -s , 'some-option' A B Cr� r� r� z-:some-option:_values -s , 'some-option' 1 2 3r� Tr� z.*:some action:_values -s , 'some action' 1 2 3N)r �CompleteMultiListr� r� r. )r r� r� Zexpectedr r r �test_multiList� s �z ZshCompleterTests.test_multiListc C sN t �� }|�dt j�}| �|d� t jddd�}|�dt j�}| �|d� dS )zd CompleteUsernames produces zsh shell-code that completes system usernames. r� z:some-option:_usersr� Tr� z*:some action:_usersN)r �CompleteUsernamesr� r� r. �r r� �outr r r �test_usernames s z ZshCompleterTests.test_usernamesc C sN t �� }|�dt j�}| �|d� t jddd�}|�dt j�}| �|d� dS )zc CompleteGroups produces zsh shell-code that completes system group names. r� z:group:_groupsr� Tr� z*:some action:_groupsN)r �CompleteGroupsr� r� r. r� r r r � test_groups s z ZshCompleterTests.test_groupsc C sN t �� }|�dt j�}| �|d� t jddd�}|�dt j�}| �|d� dS )zU CompleteHostnames produces zsh shell-code that completes hostnames. r� z:some-option:_hostsr� Tr� z*:some action:_hostsN)r �CompleteHostnamesr� r� r. r� r r r �test_hostnames' s z ZshCompleterTests.test_hostnamesc C sV t �� }|�dt j�}| �|�d�� t jddd�}|�dt j�}| �|�d�� dS )z� CompleteUserAtHost produces zsh shell-code that completes hostnames or a word of the form <username>@<hostname>. r� z:host | user@host:r� Tr� r� N)r �CompleteUserAtHostr� r� ru � startswithr� r r r �test_userAtHost4 s z!ZshCompleterTests.test_userAtHostc C sN t �� }|�dt j�}| �|d� t jddd�}|�dt j�}| �|d� dS )zv CompleteNetInterfaces produces zsh shell-code that completes system network interface names. r� z :some-option:_net_interfacesr� Tr� z *:some action:_net_interfacesN)r �CompleteNetInterfacesr� r� r. r� r r r �test_netInterfacesB s z$ZshCompleterTests.test_netInterfacesN)r! r" r# r3 r� r� r� r� r� r� r� r� r� r� r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )� CompleterNotImplementedTestszs Using an unknown shell constant with the various Completer() classes should raise NotImplementedError c C sl t jt jt jt jt jt jt jt jt j t j g }|D ]6}z |� }W n |d�}Y nX | � t |j dd� q0dS )zM Using an unknown shellType should raise NotImplementedError NZbad_shell_type)r r� r� r� r� r� r� r� r� r� r� rM �NotImplementedErrorr� )r �classes�cls�actionr r r �test_unknownShellV s& � �z.CompleterNotImplementedTests.test_unknownShellN)r! r" r# r3 r� r r r r r� Q s r� c @ s@ e Zd ZdZG dd� de�Zdd� Zdd� Zdd � Zd d � Z d S ) �FlagFunctionTestsz* Tests for L{usage.flagFunction}. c @ s( e Zd ZdZdd� Zdd� Zdd� ZdS ) zFlagFunctionTests.SomeClassz> Dummy class for L{usage.flagFunction} tests. c C s dS )z� A one argument method to be tested by L{usage.flagFunction}. @param a: a useless argument to satisfy the function's signature. Nr )r r� r r r �oneArgs s z"FlagFunctionTests.SomeClass.oneArgc C s dS )zU A no argument method to be tested by L{usage.flagFunction}. Nr r r r r �noArgz s z!FlagFunctionTests.SomeClass.noArgc C s dS )aG A multiple arguments method to be tested by L{usage.flagFunction}. @param a: a useless argument to satisfy the function's signature. @param b: a useless argument to satisfy the function's signature. @param c: a useless argument to satisfy the function's signature. Nr )r r� r� r� r r r �manyArgs s z$FlagFunctionTests.SomeClass.manyArgsN)r! r" r# r3 r� r� r� r r r r � SomeClasso s r� c C s | � dt�| �� j�� dS )zs L{usage.flagFunction} returns C{False} if the method checked allows exactly one argument. FN)�assertIsr � flagFunctionr� r� r r r r � test_hasArg� s z FlagFunctionTests.test_hasArgc C s | � dt�| �� j�� dS )zq L{usage.flagFunction} returns C{True} if the method checked allows exactly no argument. TN)r� r r� r� r� r r r r � test_noArg� s z FlagFunctionTests.test_noArgc C s, | � tjtj| �� j�}| �dt|�� dS )z L{usage.flagFunction} raises L{usage.UsageError} if the method checked allows more than one argument. z$Invalid Option function for manyArgsN�rM r rN r� r� r� r. r^ �r �excr r r �test_tooManyArguments� s �z'FlagFunctionTests.test_tooManyArgumentsc C s. | � tjtj| �� jd�}| �dt|�� dS )z� L{usage.flagFunction} uses the given method name in the error message raised when the method allows too many arguments. Zflubudufz$Invalid Option function for flubudufNr� r� r r r �,test_tooManyArgumentsAndSpecificErrorMessage� s �z>FlagFunctionTests.test_tooManyArgumentsAndSpecificErrorMessageN) r! r" r# r3 �objectr� r� r� r� r� r r r r r� j s r� c @ s e Zd ZdZdd� ZdS )�OptionsInternalTestsz6 Tests internal behavior of C{usage.Options}. c C sP G dd� dt j�}|� }| �t�dddgd�|jd |jd |jd d�� dS )zs Options which are synonyms to another option are aliases towards the longest option name. c @ s e Zd Zdd� ZeZeZdS )z;OptionsInternalTests.test_optionsAliasesOrder.<locals>.Optsc S s dS )z~ This is an option method with a very long name, that is going to be aliased. Nr r r r r �opt_very_very_long� s zNOptionsInternalTests.test_optionsAliasesOrder.<locals>.Opts.opt_very_very_longN)r! r" r# r� Z opt_shortZopt_sr r r r �Opts� s r� r� �short�very-very-long)r� r� r� N)r r� r. �dict�fromkeysZsynonyms)r r� r r r r �test_optionsAliasesOrder� s ���z-OptionsInternalTests.test_optionsAliasesOrderN)r! r" r# r3 r� r r r r r� � s r� N) r3 Z __future__r r Z twisted.trialr Ztwisted.pythonr r� r ZTestCaser& r4 r? rP rR rX ra rc rh rm rs r� r� r� r� r� r� r r r r �<module> s, +@ $ w# E
Save File
Cancel