File: //lib/python3/dist-packages/twisted/words/test/__pycache__/test_irc.cpython-310.pyc
o
�bċ � @ sF d Z ddlZddlZddlZddlmZmZ ddlmZ ddl m
Z
m Z ddl m
Z
ddlmZ ddlmZmZ G d d
� d
e
�ZG d d � d e�ZG d
d� de�ZG dd� de�ZG dd� de�Zddjejd�djejej d�dej ej d� gZ G dd� de�Z G d d � d ej!�Z"G d d � d e�Z#G d d!� d!e�Z$G d"d#� d#ej�Z%G d$d%� d%e�Z&G d&d'� d'e%�Z'd(d)� Z(G d*d+� d+e�Z)G d,d-� d-e�Z*G d.d/� d/ej�Z+G d0d1� d1e�Z,G d2d3� d3e�Z-G d4d5� d5e�Z.G d6d7� d7ej�Z/G d8d9� d9e�Z0G d:d;� d;e�Z1G d<d=� d=e�Z2G d>d?� d?e�Z3G d@dA� dAe�Z4G dBdC� dCe�Z5dS )Dz+
Tests for L{twisted.words.protocols.irc}.
� N)�protocol�task)�FilePath)�StringIOWithoutClosing�StringTransport)�TestCase)�irc)� IRCClient�
attributesc @ s e Zd Zdd� ZdS )� IRCTestCasec C sN |}t |t�r |�d�}t |t�r t |d t�r dd� |D �}| �||� dS )a$
A buffer is always bytes, but sometimes
we need to compare it to a utf-8 unicode string
@param buf: the buffer
@type buf: L{bytes} or L{unicode} or L{list}
@param val: the value to compare
@type val: L{bytes} or L{unicode} or L{list}
�utf-8r c S s g | ]}|� d ��qS )�utf8)�decode)�.0�b� r �=/usr/lib/python3/dist-packages/twisted/words/test/test_irc.py�
<listcomp>% s z6IRCTestCase.assertEqualBufferValue.<locals>.<listcomp>N)�
isinstance�strr �list� assertEqual)�self�buf�val� bufferValuer r r �assertEqualBufferValue s
z"IRCTestCase.assertEqualBufferValueN)�__name__�
__module__� __qualname__r r r r r r s r c @ sd e Zd ZdZdZdd� Zdd� Zdd� Zd d
� Zd d � Z d
d� Z
dd� Z dd� Z dd� Z
dd� ZdS )�ModeParsingTestsz>
Tests for L{twisted.words.protocols.irc.parseModes}.
)�klbr c C s | � tjtjdg � dS )zI
Parsing an empty mode string raises L{irc.IRCBadModes}.
� N�� assertRaisesr � IRCBadModes�
parseModes�r r r r �test_emptyModes0 � z ModeParsingTests.test_emptyModesc C sT | � tjtjdg � | � tjtjdg � | � tjtjdg � | � tjtjdg � dS )z�
Parsing a mode string that contains an empty sequence (either a C{+} or
C{-} followed directly by another C{+} or C{-}, or not followed by
anything at all) raises L{irc.IRCBadModes}.
z++kz-+k�+�-Nr# r' r r r �test_emptyModeSequence6 s z'ModeParsingTests.test_emptyModeSequencec C �, | � tjtjdg � | � tjtjdg � dS )zp
Parsing a mode string that does not start with C{+} or C{-} raises
L{irc.IRCBadModes}.
�foo�%Nr# r' r r r �test_malformedModesA � z$ModeParsingTests.test_malformedModesc C r- )zk
Parsing a mode string that contains no mode characters raises
L{irc.IRCBadModes}.
r* r+ Nr# r' r r r �test_nullModesI r1 z ModeParsingTests.test_nullModesc C sX t �dg �\}}| �|dg� | �|g � t �dg �\}}| �|g � | �|dg� dS )z�
Parsing a single mode setting with no parameters results in that mode,
with no parameters, in the "added" direction and no modes in the
"removed" direction.
�+s��sN�-sN�r r& r �r �added�removedr r r �test_singleModeQ s z ModeParsingTests.test_singleModec C s\ t �dg �\}}| �|g d�� | �|g � t �dg �\}}| �|g � | �|ddg� dS )z�
Parsing a single-direction mode setting with multiple modes and no
parameters, results in all modes falling into the same direction group.
z+stn)r4 ��tN��nNz-ntr> r< Nr7 r8 r r r �test_singleDirection_ s z%ModeParsingTests.test_singleDirectionc C s2 t �dg �\}}| �|g d�� | �|dg� dS )zL
Parsing a multi-direction mode setting with no parameters.
z+s-n+ti)r4 r< ��iNr> Nr7 r8 r r r �test_multiDirectionl s z$ModeParsingTests.test_multiDirectionc C s0 t �dg �\}}| �|g d�� | �|g � dS )z�
Parsing a multi-direction mode setting containing two consecutive mode
sequences with the same direction results in the same result as if
there were only one mode sequence in the same direction.
z+sn+ti)r4 r> r< rA Nr7 r8 r r r �test_consecutiveDirectiont s z*ModeParsingTests.test_consecutiveDirectionc C s8 | � tjtjdg | j� | � tjtjdg d�| j� dS )z�
If the number of mode parameters does not match the number of modes
expecting parameters, L{irc.IRCBadModes} is raised.
z+kz+kl)r. �10�lulz_extra_paramN)r$ r r% r& �
paramModesr' r r r �test_mismatchedParams~ s �z&ModeParsingTests.test_mismatchedParamsc C s� t �dg d�| j�\}}| �|g d�� | �|g � t �dddg| j�\}}| �|g � | �|g d�� t �dg d �| j�\}}| �|g d
�� | �|g � d S ) z�
Modes which require parameters are parsed and paired with their relevant
parameter, modes which do not require parameters do not consume any of
the parameters.
z+klbb)�somekey�42�nick!user@host� other!*@*)��krI )�lrJ �r rK �r rL z-klbbrK rL ))rN N)rO NrP rQ z+knbb)rI rK rL )rM r> rP rQ N)r r& rG r r8 r r r �test_parameters� s0 �� � � �� z ModeParsingTests.test_parametersN)r r r �__doc__rG r( r, r0 r2 r; r@ rC rD rH rR r r r r r ) s
r c @ � e Zd ZdZdd� ZdS )� MiscTestsz,
Tests for miscellaneous functions.
c
C s^ | � t�tjdg d��d� dd� }| � t�|g dgdgdgd gg�g d gdgdgdg� d
S ) z�
Apply a function of two arguments cumulatively to the items of
a sequence, from right to left, so as to reduce the sequence to
a single value.
r )� � � � ���c S s | � d|� | S )Nr )�insert)rO �xr r r � insertTop� s z'MiscTests.test_foldr.<locals>.insertToprV rW rX rY N)r r �_foldr�operator�sub)r r] r r r �
test_foldr� s
.�zMiscTests.test_foldrN)r r r rS ra r r r r rU � � rU c @ 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&S )'�FormattedTextTestsz>
Tests for parsing and assembling formatted IRC text.
c C s t �|�}| �||� dS )a+
Assert that C{text} is parsed and assembled to the same value as what
C{expectedFormatted} is assembled to. This provides a way to ignore
meaningless differences in the formatting structure that would be
difficult to detect without rendering the structures.
N)r �parseFormattedText�assertAssemblesTo)r �text�expectedFormatted� formattedr r r �assertAssembledEqually� s
z)FormattedTextTests.assertAssembledEquallyc C s8 t �|�}t �|�}| �t �|�|d||||f � dS )zg
Assert that C{formatted} and C{expectedFormatted} assemble to the same
value.
z$%r (%r) is not equivalent to %r (%r)N)r �assembleFormattedTextr )r rh rg rf � expectedTextr r r re � s
��z$FormattedTextTests.assertAssemblesToc C s | � dtj� dS )zO
An empty string parses to a I{normal} attribute with no text.
r" N)ri �A�normalr' r r r �test_parseEmpty� s z"FormattedTextTests.test_parseEmptyc C s0 | � t�tj�d� | � t�tjd �d� dS )z�
An attribute with no text assembles to the empty string. An attribute
whose text is the empty string assembles to two control codes: C{off}
and that of the attribute.
r" zN)r r rj rl rm �boldr' r r r �test_assembleEmpty� s z%FormattedTextTests.test_assembleEmptyc C � | � t�tjd �d� dS )zi
A I{normal} string assembles to a string prefixed with the I{off}
control code.
�hellozhelloN)r r rj rl rm r' r r r �test_assembleNormal� � z&FormattedTextTests.test_assembleNormalc C rq )zt
A I{bold} string assembles to a string prefixed with the I{off} and
I{bold} control codes.
rr zhelloN)r r rj rl ro r' r r r �test_assembleBold rt z$FormattedTextTests.test_assembleBoldc C rq )z
An I{underline} string assembles to a string prefixed with the I{off}
and I{underline} control codes.
rr z helloN)r r rj rl � underliner' r r r �test_assembleUnderline � �z)FormattedTextTests.test_assembleUnderlinec C rq )z�
A I{reverse video} string assembles to a string prefixed with the I{off}
and I{reverse video} control codes.
rr zhelloN)r r rj rl � reverseVideor' r r r �test_assembleReverseVideo rx z,FormattedTextTests.test_assembleReverseVideoc C � | � t�tjjd �d� dS )z�
A I{foreground color} string assembles to a string prefixed with the
I{off} and I{color} (followed by the relevant foreground color code)
control codes.
rr z 02helloN)r r rj rl �fg�bluer' r r r � test_assembleForegroundColor s �z/FormattedTextTests.test_assembleForegroundColorc C r{ )a
A I{background color} string assembles to a string prefixed with the
I{off} and I{color} (followed by a I{,} to indicate the absence of a
foreground color, followed by the relevant background color code)
control codes.
rr z
,02helloN)r r rj rl �bgr} r' r r r � test_assembleBackgroundColor( s �z/FormattedTextTests.test_assembleBackgroundColorc C s& | � t�tjjtjjd �d� dS )z�
A I{foreground} and I{background} color string assembles to a string
prefixed with the I{off} and I{color} (followed by the relevant
foreground color, I{,} and the relevant background color code) control
codes.
rr z 05,02helloN)r r rj rl r| �redr r} r' r r r �test_assembleColor3 s �z%FormattedTextTests.test_assembleColorc C s\ | � t�tjdtjd f �d� | � t�tjtjjtj j
d df tj d f �d� dS )zK
Nested attributes retain the attributes of their parents.
rr � worldzhello worldz yayz 05,03hello05 world yayN) r r rj rl ro rv rm r| r� r �greenry r' r r r �test_assembleNested? s � ���z&FormattedTextTests.test_assembleNestedc C s | � t�d�tjd � dS )zk
Parsing unformatted text results in text with attributes that
constitute a no-op.
rr N)r r rd rl rm r' r r r �test_parseUnformattedTextQ rt z,FormattedTextTests.test_parseUnformattedTextc C sf | � t�d�tjjd � | � t�d�tjjtjjd � | � t�d�tjjdtjjd f � dS )z
Correctly formatted text with colors uses 2 digits to specify
foreground and (optionally) background.
z01yay�yayz
01,02yayz01yay02yipee�yipeeN)r r rd rl r| �blackr r} r' r r r �test_colorFormattingX s ��z'FormattedTextTests.test_colorFormattingc C s� | � dtjjd � | � dtjjd � | � dtjjtjjd � | � dtjjtjjd � | � dtjjtjjd � | � d
tjd � | � d tjd
� | � dtjjd � | � dtj� dS )a�
Formatted text with colors can use 1 digit for both foreground and
background, as long as the text part does not begin with a digit.
Foreground and background colors are only processed to a maximum of 2
digits per component, anything else is treated as text. Color sequences
must begin with a digit, otherwise processing falls back to unformatted
text.
z
1kinda validz kinda validz999,999kinda validz9,999kinda validz1,2kinda validz1,999kinda validz 9kinda validz1,242 is a special numberz2 is a special numberz ,02oopsz,02oopszwrong�wrongz1,hellorr zN) ri rl r| r� r� r r} �yellowrm r' r r r �test_weirdColorFormattingf s$ ����z,FormattedTextTests.test_weirdColorFormattingc C sH | � dtjtjjd df � | � dtjtjjtjjd df � dS )zb
An empty color format specifier clears foreground and background
colors.
z 01yayresetr� �resetz01,02yayresetN)ri rl rm r| r� r r} r' r r r �test_clearColorFormatting� s � �z,FormattedTextTests.test_clearColorFormattingc C sj | � dtjtjtjd df � | � dtjtjjd df � | � dtjtjjtjjd df � dS )zL
A reset format specifier clears all formatting attributes.
z yayresetr� r� z 01yayresetz01,02yayresetN) ri rl rm ro rv r| r� r r} r' r r r �test_resetFormatting� s �� �z'FormattedTextTests.test_resetFormattingc C sB | � t�t�tjtjtjtjj tj
j d df ��d� dS )zZ
Strip formatting codes from formatted text, leaving only the text parts.
rr r� z hello worldN) r r �stripFormattingrj rl ro rv ry r| r� r r� r' r r r �test_stripFormatting� s ���� �z'FormattedTextTests.test_stripFormattingN)r r r rS ri re rn rp rs ru rw rz r~ r� r� r� r� r� r� r� r� r� r r r r rc � s(
rc c @ rT )� FormattingStateAttributeTestszD
Tests for L{twisted.words.protocols.irc._FormattingState}.
c
C s� | � t�� t�� � | � t�� tjdd�� | � tjddddtjd d�tjddddtjd d�� | �tjdd�tjdd�� dS )z�
L{irc._FormattingState}s must have matching character attribute
values (bold, underline, etc) with the same values to be considered
equal.
F)�offTr} )ro rv r� ry �
foreground)ro N)r r �_FormattingState� _IRC_COLORS�assertNotEqualr' r r r �
test_equality� s* ����z+FormattingStateAttributeTests.test_equalityN)r r r rS r� r r r r r� � rb r� z3Hello, this is a nice string with no complications.z!xargs{NUL}might{NUL}like{NUL}this)�NULz"embedded{CR}newline{CR}{NL}FUN{NL})�CR�NLz,escape!%(X)s escape!%(M)s %(X)s%(X)sa %(M)s0)�X�Mc @ s e Zd Zdd� Zdd� ZdS )� QuotingTestsc C �& t D ]}| �|t�t�|��� qdS )z<
Testing client-server level quote/dequote.
N)�stringSubjectsr r �
lowDequote�lowQuote�r r5 r r r �test_lowquoteSanity� � �z QuotingTests.test_lowquoteSanityc C r� )z;
Testing CTCP message level quote/dequote.
N)r� r r � ctcpDequote� ctcpQuoter� r r r �test_ctcpquoteSanity� r� z!QuotingTests.test_ctcpquoteSanityN)r r r r� r� r r r r r� � s r� c @ s$ e Zd ZdZdZdd� Zdd� ZdS )�
DispatcherzS
A dispatcher that exposes one known command and handles unknown commands.
�dispc C s ||fS )z9
A known command that returns its input.
r )r �ar r r r � disp_working� s zDispatcher.disp_workingc C s
|||fS )zM
Handle unknown commands by returning their name and inputs.
r )r �namer� r r r r � disp_unknown� s
zDispatcher.disp_unknownN)r r r rS �prefixr� r� r r r r r� � s
r� c @ �( e Zd ZdZdd� Zdd� Zdd� ZdS ) �DispatcherTestsz3
Tests for L{irc._CommandDispatcherMixin}.
c C s, t � }d}|jdg|�R � }| �||� dS )zD
Dispatching a command invokes the correct handler.
�rV rW �workingN�r� �dispatchr )r r� �args�resr r r �
test_dispatch s z DispatcherTests.test_dispatchc C s6 t � }d}d}|j|g|�R � }| �||f| � dS )zM
Dispatching an unknown command invokes the default handler.
�missingr� Nr� )r r� r� r� r� r r r �test_dispatchUnknown s
z$DispatcherTests.test_dispatchUnknownc C s" t � }d|_| �tj|jd� dS )z�
Dispatching an unknown command, when no default handler is present,
results in an exception being raised.
N�bar)r� r� r$ r �UnhandledCommandr� )r r� r r r �test_dispatchMissingUnknown s z+DispatcherTests.test_dispatchMissingUnknownN)r r r rS r� 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 d9dd�Z d9dd�Z
d9dd�Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Z d5d6� Z d7d8� Z dS ):�ServerSupportedFeatureTestszE
Tests for L{ServerSupportedFeatures} and related functions.
c C sr | � t�d�d� | � t�g �d� | � t�d�d� | � t�dd�d� | � t�d�d� | � t�d�d� dS )zt
L{_intOrDefault} converts values to C{int} if possible, otherwise
returns a default value.
Nr" rr � �123�{ )r r �
_intOrDefaultr' r r r �test_intOrDefault' s z-ServerSupportedFeatureTests.test_intOrDefaultc
C s� dddgffdddgffdddgffddg d�ffddd gffd
dd gffd dd
gffdddgffg}t jj}|D ]\}}||�}| �||� q9| �t|d� | �t|d� | �t|d� | �t|d� dS )a
L{ServerSupportedFeatures._splitParam} splits ISUPPORT parameters
into key and values. Parameters without a separator are split into a
key and a list containing only the empty string. Escaped parameters
are unescaped.
�FOOr" zFOO=zFOO=1�1z FOO=1,2,3)r� �2�3z
FOO=A\x20BzA Bz FOO=\x5Cxz\xzFOO=\�\zFOO=\nz\nzFOO=\xzFOO=\xNNzFOO=\xNz
FOO=\x20\xN)r �ServerSupportedFeatures� _splitParamr r$ �
ValueError)r �paramsr� �param�expectedr� r r r �test_splitParam3 s" � z+ServerSupportedFeatureTests.test_splitParamc C s$ t j�g d��}| �|g d�� dS )z�
L{ServerSupportedFeatures._splitParamArgs} splits ISUPPORT parameter
arguments into key and value. Arguments without a separator are
split into a key and an empty string.
)�A:1�B:2zC:�D))rl r� )�Br� )�Cr" )r� r" N)r r� �_splitParamArgsr �r r� r r r �test_splitParamArgsP s z/ServerSupportedFeatureTests.test_splitParamArgsc C s( t j�g d�t j�}| �|g d�� dS )z�
L{ServerSupportedFeatures._splitParamArgs} uses the argument processor
passed to convert ISUPPORT argument values to some more suitable
form.
)r� r� r� ))rl rV )r� rW )r� NN)r r� r� r� r r� r r r � test_splitParamArgsProcessorY s
�z8ServerSupportedFeatureTests.test_splitParamArgsProcessorc C s@ t jj}| �|d�d� | �t|d� | �|d�ddd�� dS )a
L{ServerSupportedFeatures._parsePrefixParam} parses the ISUPPORT PREFIX
parameter into a mapping from modes to prefix symbols, returns
L{None} if there is no parseable prefix parameter or raises
C{ValueError} if the prefix parameter is malformed.
r" Nrr z(ov)@+��@r )r* rV )�o�v)r r� �_parsePrefixParamr r$ r� )r r� r r r �test_parsePrefixParamd s z1ServerSupportedFeatureTests.test_parsePrefixParamc C sx t jj}| �|g d��ddddd�� | �|g d��ddddd�� | �|g d ��ddddd�� | �t|g d
�� d S ) aO
L{ServerSupportedFeatures._parseChanModesParam} parses the ISUPPORT
CHANMODES parameter into a mapping from mode categories to mode
characters. Passing fewer than 4 parameters results in the empty string
for the relevant categories. Passing more than 4 parameters raises
C{ValueError}.
)r" r" r" r" r" �� addressModesr� �setParam�noParam)r rN rO �imnpstr rN rO r� )r rN rO r" )r� r �c�d�eN)r r� �_parseChanModesParamr r$ r� )r r� r r r �test_parseChanModesParamp s
�
�
�z4ServerSupportedFeatureTests.test_parseChanModesParamc C s� t �� }|�g d�� | �|�d�d� | �|�d�ddg� | �|�d�d� | �|�d �d
� | �|�d �d � | �|�d�� |�d
g� | �|�d�� |�d
g� dS )a
L{ServerSupportedFeatures.parse} changes the internal state of the
instance to reflect the features indicated by the parsed ISUPPORT
parameters, including unknown parameters and unsetting previously set
parameters.
)�MODES=4zCHANLIMIT=#:20,&:10�INVEXz EXCEPTS=Zz
UNKNOWN=A,B,C�MODESrY � CHANLIMIT)�#� )�&�
r� �I�EXCEPTS�Z�UNKNOWN)rl r� r� z-INVEXN)r r� �parser �
getFeature�
assertTrue�
hasFeature� assertFalse)r � supportedr r r �
test_parse� s � z&ServerSupportedFeatureTests.test_parsec C s$ t �� }dd� |D �}|�|� |S )z�
Parse all specified features according to the ISUPPORT specifications.
@type features: C{list} of C{(featureName, value)}
@param features: Feature names and values to parse
@rtype: L{irc.ServerSupportedFeatures}
c S s g | ] \}}d � ||p d��qS )z{}={}r" )�format)r r� �valuer r r r � s z6ServerSupportedFeatureTests._parse.<locals>.<listcomp>)r r� r� )r �featuresr� r r r �_parse� s
z"ServerSupportedFeatureTests._parseNc C s | � ||fg�}|�|�S )z�
Parse a feature, with the given name and value, according to the
ISUPPORT specifications and return the parsed value.
)r r� )r r� r� r� r r r �
_parseFeature� s
z)ServerSupportedFeatureTests._parseFeaturec C s@ | � | �|d�|� | � | �|d�|� | � | �|d�d� dS )zW
Perform some common tests on a feature known to use L{_intOrDefault}.
N�notanintrJ �* �r r )r r� �defaultr r r �_testIntOrDefaultFeature� s z4ServerSupportedFeatureTests._testIntOrDefaultFeaturec C sL t �� j| }|du rdg}| �|�}| �|�|�� | �|�|�|� dS )z�
Features known to have default values are reported as being present by
L{irc.ServerSupportedFeatures.hasFeature}, and their value defaults
correctly, when they don't appear in an ISUPPORT message.
N)�DEFINITELY_NOT� a_feature)r r� � _featuresr r� r� r r� )r r� r� r r� r r r �_testFeatureDefault� s
z/ServerSupportedFeatureTests._testFeatureDefaultc C s� | � d� | � ddg� | � ddg� | �| �dd�ddddd�� | �| �dd�ddddd�� | �| �dd �dd
d d d�� d
S )z�
The CHANMODES ISUPPORT parameter is parsed into a C{dict} giving the
four mode categories, C{'addressModes'}, C{'param'}, C{'setParam'}, and
C{'noParam'}.
� CHANMODES)r zb,,lk,)r z b,,lk,ha,haz,,,r" r� z,A,,rl z
A,Bc,Def,Ghij�Bc�Def�GhijN�r
r r r' r r r �test_support_CHANMODES� s
�
�
�z2ServerSupportedFeatureTests.test_support_CHANMODESc C s | � | �dd�dg� dS )z�
The IDCHAN support parameter is parsed into a sequence of two-tuples
giving channel prefix and ID length pairs.
�IDCHANz!:5)�!�5Nr r' r r r �test_support_IDCHAN� s z/ServerSupportedFeatureTests.test_support_IDCHANc C sL | � | �dd�ddg� | � | �dd�g d�� | � | �dd�g d�� dS ) z~
The MAXLIST support parameter is parsed into a sequence of two-tuples
giving modes and their limits.
�MAXLISTz
b:25,eI:50�r � ��eI�2 zb:25,eI:50,a:3.1415)r r )r� Nzb:25,eI:50,a:notanintNr r' r r r �test_support_MAXLIST� s �
�
�z0ServerSupportedFeatureTests.test_support_MAXLISTc C � | � | �dd�d� dS )zr
The NETWORK support parameter is parsed as the network name, as
specified by the server.
�NETWORK�IRCNetNr r' r r r �test_support_NETWORK � z0ServerSupportedFeatureTests.test_support_NETWORKc C s | � | �d�d� dS )z�
The SAFELIST support parameter is parsed into a boolean indicating
whether the safe "list" command is supported or not.
�SAFELISTTNr r' r r r �test_support_SAFELIST s z1ServerSupportedFeatureTests.test_support_SAFELISTc C r )z�
The STATUSMSG support parameter is parsed into a string of channel
status that support the exclusive channel notice method.
� STATUSMSGz@+Nr r' r r r �test_support_STATUSMSG r z2ServerSupportedFeatureTests.test_support_STATUSMSGc C sV | � | �dd�ddd�� | � | �dd�dddd�� | � | �dd �dddd�� dS )
z�
The TARGMAX support parameter is parsed into a dictionary, mapping
strings to integers, of the maximum number of targets for a particular
command.
�TARGMAXzPRIVMSG:4,NOTICE:3rY rX )�PRIVMSG�NOTICEz PRIVMSG:4,NOTICE:3,KICK:3.1415N)r& r' �KICKz PRIVMSG:4,NOTICE:3,KICK:notanintr r' r r r �test_support_TARGMAX s
�
�
�z0ServerSupportedFeatureTests.test_support_TARGMAXc C � t �� jd }| �d|� dS )a
The NICKLEN support parameter is parsed into an integer value
indicating the maximum length of a nickname the client may use,
otherwise, if the parameter is missing or invalid, the default value
(as specified by RFC 1459) is used.
�NICKLENN�r r� r r �r r r r r �test_support_NICKLEN* � z0ServerSupportedFeatureTests.test_support_NICKLENc C r* )z�
The CHANNELLEN support parameter is parsed into an integer value
indicating the maximum channel name length, otherwise, if the
parameter is missing or invalid, the default value (as specified by
RFC 1459) is used.
�
CHANNELLENNr, r- r r r �test_support_CHANNELLEN4 r/ z3ServerSupportedFeatureTests.test_support_CHANNELLENc C s" | � d� | �| �dd�d� dS )zt
The CHANTYPES support parameter is parsed into a tuple of
valid channel prefix characters.
� CHANTYPESz#&%)r� r� r/ Nr r' r r r �test_support_CHANTYPES> s
z2ServerSupportedFeatureTests.test_support_CHANTYPESc C � | � d� dS )z�
The KICKLEN support parameter is parsed into an integer value
indicating the maximum length of a kick message a client may use.
�KICKLENN�r r' r r r �test_support_KICKLENG � z0ServerSupportedFeatureTests.test_support_KICKLENc C sh | � d� | � ddg� | �| �dd�d� | �| �dd�dddd�� | �| �dd �dddd�� dS )
z�
The PREFIX support parameter is parsed into a dictionary mapping
modes to two-tuples of status symbol and priority.
�PREFIX)r9 rr Nz(ohv)@%+r� )r/ rV )r* rW )r� �hr� z(hov)@%+r r' r r r �test_support_PREFIXN s
�
�z/ServerSupportedFeatureTests.test_support_PREFIXc C r4 )z�
The TOPICLEN support parameter is parsed into an integer value
indicating the maximum length of a topic a client may set.
�TOPICLENNr6 r' r r r �test_support_TOPICLEN` r8 z1ServerSupportedFeatureTests.test_support_TOPICLENc C r4 )a_
The MODES support parameter is parsed into an integer value
indicating the maximum number of "variable" modes (defined as being
modes from C{addressModes}, C{param} or C{setParam} categories for
the C{CHANMODES} ISUPPORT parameter) which may by set on a channel
by a single MODE command from a client.
r� Nr6 r' r r r �test_support_MODESg s z.ServerSupportedFeatureTests.test_support_MODESc C �* | � | �dd�d� | � | �d�d� dS )z�
The EXCEPTS support parameter is parsed into the mode character
to be used for "ban exception" modes. If no parameter is specified
then the character C{e} is assumed.
r� r� r� Nr r' r r r �test_support_EXCEPTSq � z0ServerSupportedFeatureTests.test_support_EXCEPTSc C r? )z�
The INVEX support parameter is parsed into the mode character to be
used for "invite exception" modes. If no parameter is specified then
the character C{I} is assumed.
r� r� r� Nr r' r r r �test_support_INVEXz rA z.ServerSupportedFeatureTests.test_support_INVEX�N) r r r rS r� r� r� r� r� r� r� r r r r
r r r r r" r$ r) r. r1 r3 r7 r; r= r> r@ rB r r r r r� " s:
r� c @ s e Zd ZdZdS )�IRCClientWithoutLoginr N)r r r � performLoginr r r r rD � s rD c @ s@ e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dS )� CTCPTestszK
Tests for L{twisted.words.protocols.irc.IRCClient} CTCP handling.
c C sN t � | _t�| j�| _t� | _| j�| j� | �| jj � | �| jj
d � d S rC ) r �filer � FileWrapper� transportrD �client�makeConnection�
addCleanup�loseConnection�connectionLostr' r r r �setUp� s zCTCPTests.setUpc C sV dt jt jt j d� }dt jt jt j d� }| j�|� | j�� }| �||� dS )z�Testing CTCP query ERRMSG.
Not because this is this is an especially important case in the
field, but it does go through the entire dispatch/decode/encode
process.
z@:nick!guy@over.there PRIVMSG #theChan :%(X)cERRMSG t%(X)c%(EOL)s�r� �EOLz>NOTICE nick :%(X)cERRMSG t :No error has occurred.%(X)c%(EOL)sN) r �X_DELIMr� �LFrJ � dataReceivedrG �getvaluer )r �errQuery�errReply�replyr r r � test_ERRMSG� s ����
zCTCPTests.test_ERRMSGc C sP d| j _| j �ddd� dtjtjtj | j jd� }| j�� }| � ||� dS )z�
If attributes for version information on L{IRCClient} are set to
L{None}, the parts of the CTCP VERSION response they correspond to
are omitted.
�
FrobozzIRC�nick!guy@over.there�#theChanNz1NOTICE nick :%(X)cVERSION %(vname)s::%(X)c%(EOL)s)r� rQ �vname)
rJ � versionName�ctcpQuery_VERSIONr rR r� rS rG rU r �r � versionReplyrX r r r �test_noNumbersVERSION� s
�
z CTCPTests.test_noNumbersVERSIONc C sl d| j _d| j _d| j _| j �ddd� dtjtjtj | j j| j j| j jd� }| j �
� }| � ||� dS ) z�
The response to a CTCP VERSION query includes the version number and
environment information, as specified by L{IRCClient.versionNum} and
L{IRCClient.versionEnv}.
rZ z1.2g�ZorkOSr[ r\ NzANOTICE nick :%(X)cVERSION %(vname)s:%(vnum)s:%(venv)s%(X)c%(EOL)s)r� rQ r] �vnum�venv) rJ r^ �
versionNum�
versionEnvr_ r rR r� rS rG rU r r` r r r �test_fullVERSION� s
���
zCTCPTests.test_fullVERSIONc sZ � fdd�}d� _ |� j_� j�dddjtjd�g� � �� j� � d� � �
� j d � d
S ) zL
Duplicated CTCP messages are ignored and no reply is made.
c s � j d7 _ d S �NrV )�called��user�channel�datar' r r �testCTCP� s z8CTCPTests.test_noDuplicateCTCPDispatch.<locals>.testCTCPr �foo!bar@baz.quux�#chanz {X}TESTTHIS{X}foo{X}TESTTHIS{X}�r� r" rV N) rj rJ �ctcpQuery_TESTTHIS� irc_PRIVMSGr� r rR r rG rU r )r ro r r' r � test_noDuplicateCTCPDispatch� s �z&CTCPTests.test_noDuplicateCTCPDispatchc s� � fdd�}d� _ � �� jd|� � j�dddjtjd�g� � �� j� � d � � �
� j d
� � �
� j d � � j�ddd jtjd�g� � �
� j d
� dS )zQ
The fallback handler is invoked for unrecognized CTCP messages.
c s | |||f� _ � jd7 _d S ri )�
calledWithrj )rl rm �tagrn r' r r � unknownQuery� s z6CTCPTests.test_noDefaultDispatch.<locals>.unknownQueryr �ctcpUnknownQueryrp rq z
{X}NOTREAL{X}rr r" )rp rq �NOTREALNrV z {X}NOTREAL{X}foo{X}NOTREAL{X}rW N) rj �patchrJ rt r� r rR r rG rU r rv )r rx r r' r �test_noDefaultDispatch� s ���z CTCPTests.test_noDefaultDispatchN)
r r r rS rO rY rb rh ru r| r r r r rF � s rF c
s� e Zd Zi dd�dd�dd�dd�dd�d d
�d d �d
d�dd�dd�dd�dd�dd�dd�dd�dd�d d �d d d d d!d"d#d$d%��Zd&d'� Z� fd(d)�Zd*d+� Z� ZS ),�NoticingClient�created��when�yourHost��info�myInfo��
servername�version�umodes�cmodes� luserClient�bounce�isupport)�options�
luserChannels)�channels�luserOp)�ops�luserMe� receivedMOTD)�motd�privmsg�rl rm �message�joined)rm �left�noticed� modeChanged)rl rm �set�modesr� �pong)rl �secs�signedOnr )rm �kickerr� ��nick)rl rm )rl rm r� r� rk )rl rm �newTopic)�oldname�newname)�
kickedFrom� nickChanged�
userJoined�userLeft�
userKicked�action� topicUpdated� userRenamedc O �
g | _ d S rC )�calls)r r� �kwr r r �__init__" s
zNoticingClient.__init__c s` |� d�r|�d�rt� �|�S z
t� �d�| }W n ty) t� �|� Y S w | �||�S )N�__�methods)�
startswith�endswith�super�__getattribute__�KeyError�
makeMethod)r r� r� �� __class__r r r� * s � z NoticingClient.__getattribute__c s � ��fdd�}|S )Nc s� t | �t � �krtd�t � �t | �f ��t� | �D ]\}}||v r*td�|f ��|||< qt |�t � �krDtd�t � �t | �f ���j��|f� d S )Nz-TypeError: %s() takes %d arguments (%d given)z=TypeError: %s() got multiple values for keyword argument '%s')�len� TypeError�zipr� �append)r� r� r� r� �r� �fnamer r r �method5 s* ����
��z)NoticingClient.makeMethod.<locals>.methodr )r r� r� r� r r� r r� 4 s zNoticingClient.makeMethod)r r r r� r� r� r� �
__classcell__r r r� r r} s` �������� �
� � �
������
r} c C s. z| | }W n t y | Y S w | |= |S rC )r� )�dict�keyr r� r r r �popM s �r� c @ s e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� 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'd(�Zd@d)d*�Zd@d+d,�Zd-d.� Zd/d0� Zd1d2� Zd3d4� Zd5d6� Z d7d8� Z d9d:� Z d;d<� Z d=d>� Z d&S )A�ClientImplementationTestsc C s@ t � | _t� | _| j�| j� | �| jj� | �| jjd � d S rC )r rI r} rJ rK rL rM rN r' r r r rO X s
z ClientImplementationTests.setUpc K sv t |dd�}t |dd�}t |dd�}d| d | d | d | d | d
}| j�|� | �| jj||fg� d S ) N�host� server.hostr� �nicknamer� r" �:� � :�
�r� rJ rT r r� ) r �code�msg�funcr� r� r� r� r� r r r �_serverTestImpl` s *� z)ClientImplementationTests._serverTestImplc C � d}| j d|d|d� d S )NzHYour host is some.host[blah.blah/6667], running version server-version-3�002r� r� �r� �r r� r r r � testYourHostl � z&ClientImplementationTests.testYourHostc C r� )Nz=This server was cobbled together Fri Aug 13 18:00:25 UTC 2004�003r~ r r� r� r r r � testCreatedp r� z%ClientImplementationTests.testCreatedc C s d}| j d|dddddd� d S ) Nz(server.host server-version abcDEF bcdEHI�004r� r� zserver-version�abcDEF�bcdEHIr� r� r� r r r �
testMyInfot s
�z$ClientImplementationTests.testMyInfoc C r� )Nz4There are 9227 victims and 9542 hiding on 24 servers�251r� r� r� r� r r r �testLuserClient� r� z)ClientImplementationTests.testLuserClientc C s$ d}d}| j d|d|g d�d� d S )Nz�MODES=4 CHANLIMIT=#:20 NICKLEN=16 USERLEN=10 HOSTLEN=63 TOPICLEN=450 KICKLEN=450 CHANNELLEN=30 KEYLEN=23 CHANTYPES=# PREFIX=(ov)@+ CASEMAPPING=ascii CAPAB IRCD=dancer� are available on this server�005r� )r� zCHANLIMIT=#:20z
NICKLEN=16z
USERLEN=10z
HOSTLEN=63z TOPICLEN=450z KICKLEN=450z
CHANNELLEN=30z KEYLEN=23z CHANTYPES=#z
PREFIX=(ov)@+zCASEMAPPING=ascii�CAPABz IRCD=dancer�r� r� r� �r r� r� r r r �
_sendISUPPORT� s �
�z'ClientImplementationTests._sendISUPPORTc C s | � � dS )zq
The client parses ISUPPORT messages sent by the server and calls
L{IRCClient.isupport}.
N)r� r' r r r �
test_ISUPPORT� s z'ClientImplementationTests.test_ISUPPORTc C r� )Nz Try server some.host, port 321�010r� r� r� r� r r r �
testBounce� r� z$ClientImplementationTests.testBouncec C �$ d}d}| j d|d|t|�d� d S )N�7116zchannels formed�254r� )r� r� �r� �intr� r r r �testLuserChannels� � z+ClientImplementationTests.testLuserChannelsc C r� )N�34zflagged staff members�252r� )r� r� r� r� r r r � testLuserOp� r� z%ClientImplementationTests.testLuserOpc C r� )Nz!I have 1937 clients and 0 servers�255r� r� r� r� r r r � testLuserMe� r� z%ClientImplementationTests.testLuserMec C sd g d�}|D ]}| � | jjg � | j�|d � q| � | jjddddgifg� | �| jjd� dS )z�
Lines received in I{RPL_MOTDSTART} and I{RPL_MOTD} are delivered to
L{IRCClient.receivedMOTD} when I{RPL_ENDOFMOTD} is received.
)z9:host.name 375 nickname :- host.name Message of the Day -�/:host.name 372 nickname :- Welcome to host.name�.:host.name 376 nickname :End of /MOTD command.r� r� r� z host.name Message of the Day -�Welcome to host.nameN)r rJ r� rT �assertIdenticalr� �r �lines�Lr r r �test_receivedMOTD� s �����z+ClientImplementationTests.test_receivedMOTDc C sB ddg}|D ]
}| j �|d � q| �| j jdddgifg� dS )z�
If L{IRCClient} receives I{RPL_MOTD} and I{RPL_ENDOFMOTD} without
receiving I{RPL_MOTDSTART}, L{IRCClient.receivedMOTD} is still
called with a list of MOTD lines.
r� r� r� r� r� r� N)rJ rT r r� r� r r r �test_withoutMOTDSTART� s ��z/ClientImplementationTests.test_withoutMOTDSTARTc K s~ t |dd�}t |dd�}|d | d | } d| d | d | d | d }
| j�|
� | �| jj||fg� g | j_d S ) N�identr� r r� r� r� r� r� r� ) r �sender�group�typer� r� r� r r� � wholeUserr� r r r �_clientTestImpl� s $ z)ClientImplementationTests._clientTestImplc C sD d}| j ddd|ddddd|d �
| j dd
d|ddddd
|d �
d S ) NzTooty toot toot.r z#groupr& r� r r� zsender!ident@host)r r� rl rm r� � recipient)r r� r r r � testPrivmsg� s2 �
�z%ClientImplementationTests.testPrivmsgc s� t t� j�� �\}}� �|g d�� � �|g d�� � fdd�}|d� t t� j�� �\}}� �|g d�� � �|g d�� |d� t t� j�� �\}}� �|g � � �|g � � �� � �� jj�d�d� dS ) z�
L{IRCClient.getChannelModeParams} uses ISUPPORT information, either
given by the server or defaults, to determine which channel modes
require arguments when being added or removed.
)r r: rN rO r� r� )r r: r� r� c sD d| } d}� j d|d| | gd� � �� jj�| �d � g � j_d S )Nr+ r� r� r� r� )r� r� rJ r� r� r� )r� r� r' r r �
removeFeature s
zJClientImplementationTests.test_getChannelModeParams.<locals>.removeFeaturer )r: r� r� r9 N) �map�sortedrJ �getChannelModeParamsr r� �assertNotIdenticalr� r� )r �add�remover r r' r �test_getChannelModeParams s z3ClientImplementationTests.test_getChannelModeParamsc C s0 t t| j�� �\}}| �|g � | �|g � dS )z�
L{IRCClient.getUserModeParams} returns a list of user modes (modes that
the user sets on themself, outside of channel modes) that require
parameters when added and removed, respectively.
N)r
r rJ �getUserModeParamsr )r r r r r r �test_getUserModeParams8 s z0ClientImplementationTests.test_getUserModeParamsr" Nc C s4 |du rd}d|� d|� d|� d�}| j �|� dS )z@
Build a MODE string and send it to the client.
Nrq z :Wolf!~wolf@yok.utu.fi MODE r� r� )rJ rT )r r� r� �targetr� r r r �_sendModeChangeB s z)ClientImplementationTests._sendModeChangec sp |du rd}t |�D ]+\}}|\}� | �|d� | �� d d� | �� d |� t� fdd�d D ��||< q
|S )
zO
Parse the results, do some test and return the data to check.
Nrq r� rl �Wolf!~wolf@yok.utu.firm c 3 s � | ]}� | V qd S rC r )r r� �rn r r � <genexpr>W s � z=ClientImplementationTests._parseModeChange.<locals>.<genexpr>)r� r� r� )� enumerater �tuple)r �resultsr r? �resultr� r r r �_parseModeChangeK s z*ClientImplementationTests._parseModeChangec C s( | � | jj|�}| �||� g | j_dS )zR
Compare the expected result with the one returned by the client.
N)r rJ r� r )r r� r r r r r �_checkModeChangeZ s z*ClientImplementationTests._checkModeChangec C s | � d� | �dg� dS )z�
Mode strings that do not begin with a directional character, C{'+'} or
C{'-'}, have C{'+'} automatically prepended.
r5 )Tr5 rC N�r r r' r r r �test_modeMissingDirectionb s
z3ClientImplementationTests.test_modeMissingDirectionc C s0 | � d� | �dg� | � d� | �dg� dS )zx
No parameters are passed to L{IRCClient.modeChanged} for modes that
don't take any parameters.
r6 )Fr5 rC z+n)Tr? rC Nr r' r r r �test_noModeParametersj s
z/ClientImplementationTests.test_noModeParametersc C s4 | � dd� | �dg� | � dd� | �dg� dS )zk
Parameters are passed to L{IRCClient.modeChanged} for modes that take
parameters.
�+o�a_user)Tr� �r" z-o)Fr� r# Nr r' r r r �test_oneModeParametert s z/ClientImplementationTests.test_oneModeParameterc C s6 | � dd� | �dg� | � dd� | �ddg� dS )z�
Mixing adding and removing modes that do and don't take parameters
invokes L{IRCClient.modeChanged} with mode characters and parameters
that match up.
z+osvza_user another_user)T�osv)r" N� another_userz+v-os)Tr� r# )F�os)r& NNr r' r r r �test_mixedModes~ s �z)ClientImplementationTests.test_mixedModesc C sJ | � dd� | �g � | �tj�}| �t|�d� | �d|d �� � dS )z�
Passing an argument to modes that take no parameters results in
L{IRCClient.modeChanged} not being called and an error being logged.
r3 r� rV zToo many parametersr N� r r �flushLoggedErrorsr r% r r� �assertSubstring�getErrorMessage�r �errorsr r r �test_tooManyModeParameters� s
z4ClientImplementationTests.test_tooManyModeParametersc C sH | � d� | �g � | �tj�}| �t|�d� | �d|d �� � dS )z�
Passing no arguments to modes that do take parameters results in
L{IRCClient.modeChange} not being called and an error being logged.
r! rV zNot enough parametersr Nr) r- r r r �test_tooFewModeParameters� s
z3ClientImplementationTests.test_tooFewModeParametersc C sb | j j}| jd|d� | jdg|d� dd� }| �| j d|� | jdd|d� | jd g|d� d
S ) z�
A C{MODE} message whose target is our user (the nickname of our user,
to be precise), as opposed to a channel, will be parsed according to
the modes specified by L{IRCClient.getUserModeParams}.
r! )r )Tr� rC c S s ddgS )Nr� r" r r r r r r � s zBClientImplementationTests.test_userMode.<locals>.getUserModeParamsr z+Z�an_arg)Tr� )r1 N)rJ r� r r r{ )r r r r r r �
test_userMode� s z'ClientImplementationTests.test_userModec s� � fdd�}t �� � _� jj� _� �� jd|� � �� jjd� � j� dg � � �
� jjd� � � � jj d� � �
� j�� d� � j�� jj� � �
� j�� d� � j�� � j�d� � � t� j�� �d� � �� jjd� dS ) a
When the I{RPL_WELCOME} message is received a heartbeat is started that
will send a I{PING} message to the IRC server every
L{irc.IRCClient.heartbeatInterval} seconds. When the transport is
closed the heartbeat looping call is stopped too.
c s � � � } � j| _| S rC )�_originalCreateHeartbeat�clock)� heartbeatr' r r �_createHeartbeat� s zBClientImplementationTests.test_heartbeat.<locals>._createHeartbeatr6 Nr. r" z
PING foo
r )r �Clockr4 rJ r6 r3 r{ r� �
_heartbeat�irc_RPL_WELCOMEr
r �hostnamer rI r� �advance�heartbeatIntervalrM rN r� �getDelayedCalls)r r6 r r' r �test_heartbeat� s
z(ClientImplementationTests.test_heartbeatc C s: | � | jjd� d| j_| j�dg � | � | jjd� dS )zo
If L{irc.IRCClient.heartbeatInterval} is set to L{None} then no
heartbeat is created.
Nr. )r� rJ r8 r< r9 r' r r r �test_heartbeatDisabled� s z0ClientImplementationTests.test_heartbeatDisabled)r" NrC )!r r r rO r� r� r� r� r� r� r� r� r� r� r� r� r r r r r r r r r r r$ r( r/ r0 r2 r> r? r r r r r� W s> !
"
!r� c @ s� e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� 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-S ).� BasicServerFunctionalityTestsc C s2 t � | _t�| j�| _t�� | _| j�| j� d S rC ) r �fr rH r= r �IRC�prK r' r r r rO � s
z#BasicServerFunctionalityTests.setUpc C s. | j �� }t|t�r|�d�}| �||� dS )z�
Make sure that the internal buffer equals a specified value.
@param s: the value to compare against buffer
@type s: L{bytes} or L{unicode}
r N)rA rU r r r r )r r5 r r r r �check� s
z#BasicServerFunctionalityTests.checkc C � | j �ddd� | �d� dS )z�
Passing a command and parameters to L{IRC.sendMessage} results in a
query string that consists of the command and parameters, separated by
a space, ending with '
'.
�CMD�param1�param2�CMD param1 param2
N)rC � sendMessagerD r' r r r �test_sendMessage� � z.BasicServerFunctionalityTests.test_sendMessagec C � | j �dd� | �d� dS )aF
Passing a command and parameters to L{IRC.sendCommand} results in a
query string that consists of the command and parameters, separated by
a space, ending with '
'.
The format is described in more detail in
U{RFC 1459 <https://tools.ietf.org/html/rfc1459.html#section-2.3>}.
rF �rG rH rI N�rC � sendCommandrD r' r r r �test_sendCommand� s z.BasicServerFunctionalityTests.test_sendCommandc C rM )zk
Passing unicode parameters to L{IRC.sendCommand} encodes the parameters
in UTF-8.
rF )u param¹u param²s CMD param¹ param²
NrO r' r r r �test_sendUnicodeCommand
s z5BasicServerFunctionalityTests.test_sendUnicodeCommandc C s* | � t| jjddd�}| �t|�d� dS )zf
Passing L{None} as the command to L{IRC.sendMessage} raises a
C{ValueError}.
NrG rH � IRC message requires a command.�r$ r� rC rJ r r �r �errorr r r �test_sendMessageNoCommand s �z7BasicServerFunctionalityTests.test_sendMessageNoCommandc C s* | � t| jjdd�}| �|jd d� dS )zf
Passing L{None} as the command to L{IRC.sendCommand} raises a
C{ValueError}.
NrN r rS �r$ r� rC rP r r� rU r r r �test_sendCommandNoCommand � �z7BasicServerFunctionalityTests.test_sendCommandNoCommandc C s* | � t| jjddd�}| �t|�d� dS )zi
Passing an invalid string command to L{IRC.sendMessage} raises a
C{ValueError}.
r� rG rH zCSomebody screwed up, 'cuz this doesn't look like a command to me: NrT rU r r r � test_sendMessageInvalidCommand& s ��z<BasicServerFunctionalityTests.test_sendMessageInvalidCommandc C s* | � t| jjdd�}| �|jd d� dS )zi
Passing an invalid string command to L{IRC.sendCommand} raises a
C{ValueError}.
r� rN r zInvalid command: " "NrX rU r r r � test_sendCommandInvalidCommand3 rZ z<BasicServerFunctionalityTests.test_sendCommandInvalidCommandc C rE )z�
Passing a command and parameters with a specified prefix to
L{IRC.sendCommand} results in a proper query string including the
specified line prefix.
rF rN �irc.example.com�$ :irc.example.com CMD param1 param2
NrO r' r r r �test_sendCommandWithPrefix= rL z8BasicServerFunctionalityTests.test_sendCommandWithPrefixc C sr dddd�}d}| j �ddd|� | j�� }|�d d
�\}}|d
d� �d �}| �|d � | �t|�t|�� dS )
aM
Passing a command and parameters with a specified prefix and tags
to L{IRC.sendCommand} results in a proper query string including the
specified line prefix and appropriate tags syntax. The query string
should be output as follows:
@tags :prefix COMMAND param1 param2
The tags are a string of IRCv3 tags, preceded by '@'. The rest
of the string is as described in test_sendMessage. For more on
the message tag format, see U{the IRCv3 specification
<https://ircv3.net/specs/core/message-tags-3.2.html>}.
�bbbN�eee)�aaa�ccczexample.com/ddd)� aaa=bbbs cccs example.com/ddd=eeerF rN r] � rV � ;r^ �rC rP rA rU �splitr r �r �sendTags� expectedTags�outMsg� outTagStr�outLine�outTagsr r r �test_sendCommandWithTagsF s
z6BasicServerFunctionalityTests.test_sendCommandWithTagsc C �: dddd�}| � t| jjddd|�}| �|jd d � dS )
zW
Passing empty tag names to L{IRC.sendCommand} raises a C{ValueError}.
r` Nr" )rb rc r" rF rN r] r �A tag name is required.rX �r rj rV r r r �!test_sendCommandValidateEmptyTags_ s �z?BasicServerFunctionalityTests.test_sendCommandValidateEmptyTagsc C rq )
zb
Passing None as a tag name to L{IRC.sendCommand} raises a
C{ValueError}.
r` N�beep)rb rc NrF rN r] r rr rX rs r r r � test_sendCommandValidateNoneTagsn s �z>BasicServerFunctionalityTests.test_sendCommandValidateNoneTagsc C �6 ddi}| � t| jjddd|�}| �|jd d� dS ) zl
Passing a tag name containing spaces to L{IRC.sendCommand} raises a
C{ValueError}.
zaaa bbbrc rF rN r] r � Tag contains invalid characters.NrX rs r r r �&test_sendCommandValidateTagsWithSpaces~ � �zDBasicServerFunctionalityTests.test_sendCommandValidateTagsWithSpacesc C rw ) zx
Passing a tag name containing invalid characters to L{IRC.sendCommand}
raises a C{ValueError}.
zaaa_b^@rc rF rN r] r rx NrX rs r r r �,test_sendCommandValidateTagsWithInvalidChars� rz zJBasicServerFunctionalityTests.test_sendCommandValidateTagsWithInvalidCharsc C sd ddd�}d}| j �ddd|� | j�� }|�dd �\}}|d d
� �d �}| �t|�t|�� d
S ) zr
Tags with values containing invalid characters passed to
L{IRC.sendCommand} are escaped.
r` z
test
\;;)rb rc )rd s ccc=test\r\n\s\\\:\:rF rN r] re rV Nrf rg ri r r r �(test_sendCommandValidateTagValueEscaping� s
zFBasicServerFunctionalityTests.test_sendCommandValidateTagValueEscapingc C � | j �ddd� | �d� d S )N�this-is-sender�
this-is-recipzthis is messagez8:this-is-sender PRIVMSG this-is-recip :this is message
)rC r� rD r' r r r r � � z)BasicServerFunctionalityTests.testPrivmsgc C r} )Nr~ r zthis is noticez6:this-is-sender NOTICE this-is-recip :this is notice
)rC �noticerD r' r r r �
testNotice� r� z(BasicServerFunctionalityTests.testNoticec C r} )Nr~ r zthis is actionz6:this-is-sender ACTION this-is-recip :this is action
)rC r� rD r' r r r �
testAction� r� z(BasicServerFunctionalityTests.testActionc C � | j �dd� | �d� d S )N� this-personz
#this-channelz!:this-person JOIN #this-channel
)rC �joinrD r' r r r �testJoin� � z&BasicServerFunctionalityTests.testJoinc C r� )Nr� z
#that-channelz!:this-person PART #that-channel
)rC �partrD r' r r r �testPart� r� z&BasicServerFunctionalityTests.testPartc C st t t�� d �}| jj}d}d}| j�||dddddd d
|d d g� g d
�}d�|�t||||d� }| �|� dS )zp
Verify that a whois by the client receives the right protocol actions
from the server.
�d zrequesting-nickz target-nickr zhost.comz Target Userz irc.host.comz
A fake serverF� z
#fakeusersz #fakemisc)zA:%(hostname)s 311 %(req)s %(targ)s target host.com * :Target Userz>:%(hostname)s 312 %(req)s %(targ)s irc.host.com :A fake serverzN:%(hostname)s 317 %(req)s %(targ)s 12 %(timestamp)s :seconds idle, signon timez8:%(hostname)s 319 %(req)s %(targ)s :#fakeusers #fakemiscz6:%(hostname)s 318 %(req)s %(targ)s :End of WHOIS list.r" r� )r: � timestamp�req�targN)r� �timerC r: �whoisr� r� rD )r r� r: r� r� r� r� r r r � testWhois� s, �
�z'BasicServerFunctionalityTests.testWhoisN)r r r rO rD rK rQ rR rW rY r[ r\ r_ rp rt rv ry r{ r| r r� r� r� r� r� r r r r r@ � s.
r@ c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd S )
� DummyClientz{
A L{twisted.words.protocols.irc.IRCClient} that stores sent lines in a
C{list} rather than transmitting them.
c C r� rC )r� r' r r r r� � s
zDummyClient.__init__c C s t j�| � g | _d S rC )r r �connectionMader� r' r r r r� � s
zDummyClient.connectionMadec C s |dt jt| j� � S )zE
Truncate an IRC line to the maximum allowed length.
N)r �MAX_COMMAND_LENGTHr� � delimiter�r �liner r r �
_truncateLine� r) zDummyClient._truncateLinec C s | � |�}tj�| |�S rC )r� r r � lineReceivedr� r r r r� � s
zDummyClient.lineReceivedc C s | j �| �|�� d S rC )r� r� r� )r �mr r r �sendLine� s zDummyClient.sendLineN) r r r rS r� r� r� r� r� r r r r r� � s r� c @ r� ) �ClientInviteTestsz(
Tests for L{IRCClient.invite}.
c C s t � | _dS )zO
Create a L{DummyClient} to call C{invite} on in test methods.
N)r� rJ r' r r r rO s zClientInviteTests.setUpc C �$ | j �dd� | �| j jdg� dS )z�
If the channel name passed to L{IRCClient.invite} does not begin with a
channel prefix character, one is prepended to it.
r. r� �INVITE foo #barN�rJ �inviter r� r' r r r �test_channelCorrection � z(ClientInviteTests.test_channelCorrectionc C r� )zs
L{IRCClient.invite} sends an I{INVITE} message with the specified
username and a channel.
r. �#barr� Nr� r' r r r � test_invite r� z ClientInviteTests.test_inviteN)r r r rS rO 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 d'dd�Z
dd� Zdd� Zdd � Zd d � Zd d � Zd!d"� Zd#d$� Zd%d&� ZdS )(�ClientMsgTestszP
Tests for messages sent with L{twisted.words.protocols.irc.IRCClient}.
c C s t � | _| j�� d S rC )r� rJ r� r' r r r rO $ s zClientMsgTests.setUpc C r� )zO
A message containing no newlines is sent in a single command.
r. r� �PRIVMSG foo :barN)rJ r� r r� r' r r r �test_singleLine( s z ClientMsgTests.test_singleLinec C s0 | � t| jjddd� | � t| jjddd� dS )z�
Specifying a C{length} value to L{IRCClient.msg} that is too short to
contain the protocol command to send a message raises C{ValueError}.
r. r� r rX N)r$ r� rJ r� r' r r r �test_invalidMaxLength/ s z$ClientMsgTests.test_invalidMaxLengthc C s8 t d�d d }| j�dd|� | �| jjg d�� dS )z�
Messages longer than the C{length} parameter to L{IRCClient.msg} will
be split and sent in multiple commands.
�
PRIVMSG foo :rX rW r. �barbazbo)r� �PRIVMSG foo :bazzPRIVMSG foo :boN�r� rJ r� r r� )r �maxLenr r r �test_multipleLine7 s �z ClientMsgTests.test_multipleLinec C s� d}t d|� ��d }| j�d||� | �| jjd|� �g� g | j_| j�d||d � | �dt | jj�� g | j_| j�d||d � | �dt | jj�� dS )z�
Messages exactly equal in length to the C{length} parameter to
L{IRCClient.msg} are sent in a single command.
r� r� rW r. rV Nr� )r r� r� r r r �test_sufficientWidthC s z#ClientMsgTests.test_sufficientWidthc C �, g | j _| j �dd� | �| j jdg� dS )zC
An LF at the beginning of the message is ignored.
r. z
barr� N�rJ r� r� r r' r r r �test_newlinesAtStartS � z#ClientMsgTests.test_newlinesAtStartc C r� )z=
An LF at the end of the message is ignored.
r. zbar
r� Nr� r' r r r �test_newlinesAtEnd[ r� z!ClientMsgTests.test_newlinesAtEndc C �. g | j _| j �dd� | �| j jddg� dS )z;
An LF within a message causes a new line.
r. zbar
bazr� r� Nr� r' r r r �test_newlinesWithinMessagec � z)ClientMsgTests.test_newlinesWithinMessagec C r� )z<
Consecutive LFs do not cause a blank line.
r. zbar
bazr� r� Nr� r' r r r �test_consecutiveNewlinesk r� z'ClientMsgTests.test_consecutiveNewlinesNc s� d� | jj| jj| jj�}| jjd||d� g � | �| jd� fdd�� | jjD ]
}| j�|| � q(| � t
� �|� d� dd � � D ��}| � ||� d
S ) z�
Assert that messages sent by L{IRCClient.msg} are split into an
expected number of commands and the original message is transmitted in
its entirety over those commands.
z
:{}!{}@{} r. ��lengthr� c s
� � | �S rC )r� )r� �r� r r �<lambda>� s
z;ClientMsgTests.assertLongMessageSplitting.<locals>.<lambda>r" c s s � | ]\}}}|V qd S rC r )r rl r r� r r r r � s � z<ClientMsgTests.assertLongMessageSplitting.<locals>.<genexpr>N) r� rJ r� �realnamer: r� r{ r� r� r r� r� )r r� �expectedNumCommandsr� �responsePrefixr� �receivedMessager r� r �assertLongMessageSplittings s � z)ClientMsgTests.assertLongMessageSplittingc C s dt jd }| �|d� dS )a]
If a maximum message length is not provided to L{IRCClient.msg} a
best-guess effort is made to determine a safe maximum, messages longer
than this are split into multiple commands with the intent of
delivering long messages without losing data due to message truncation
when the server relays them.
r� rW N�r r� r� �r r� r r r �!test_splitLongMessagesWithDefault� s z0ClientMsgTests.test_splitLongMessagesWithDefaultc C s( dt jd }| j|dt jd d� dS )a
The maximum message length can be specified to L{IRCClient.msg},
messages longer than this are split into multiple commands with the
intent of delivering long messages without losing data due to message
truncation when the server relays them.
r� rW rX r� Nr� r� r r r �"test_splitLongMessagesWithOverride� s z1ClientMsgTests.test_splitLongMessagesWithOverridec C �D dt jd }| j�d|d | � | �| jjd| d| g� dS )zK
IRCClient breaks on newlines before it breaks long lines.
r� rW r. �
r� N�r r� rJ r� r r� �r �longliner r r � test_newlinesBeforeLineBreaking� �
�z.ClientMsgTests.test_newlinesBeforeLineBreakingc C r� )zK
IRCClient prefers to break long lines at word boundaries.
r� rW r. r� r� Nr� r� r r r � test_lineBreakOnWordBoundaries� r� z-ClientMsgTests.test_lineBreakOnWordBoundariesc C sN | � ttjdd� | � ttjdd� | �g t�dd�� | �g t�d�� dS )z�
L{twisted.words.protocols.irc.split} raises C{ValueError} if given a
length less than or equal to C{0} and returns C{[]} when splitting
C{''}.
r. ���r r" rV N)r$ r� r rh r r' r r r �test_splitSanity� s z ClientMsgTests.test_splitSanityc C s< t �dd�}| �g d�|� t �dd�}| �g d�|� dS )z�
L{twisted.words.protocols.irc.split} skips any delimiter (space or
newline) that it finds at the very beginning of the string segment it
is operating on. Nothing should be added to the output list because of
it.
zxx yyzrW )�xx�yy�zzxx
yyzN)r rh r )r �rr r r �test_splitDelimiters� s z#ClientMsgTests.test_splitDelimitersc C s( | � ttjdd� | � ttjdd� dS )z�
L{twisted.words.protocols.irc.split} raises C{ValueError} if given a
length less than or equal to C{0}.
r. r r� N)r$ r� r rh r' r r r �test_splitValidatesLength� s z(ClientMsgTests.test_splitValidatesLengthc C r� )z�
L{IRCClient.say} prepends the channel prefix C{"#"} if necessary and
then sends the message to the server for delivery to that channel.
�
thechannelz the messagez PRIVMSG #thechannel :the messageN)rJ �sayr r� r' r r r �test_say� r� zClientMsgTests.test_sayrC )r r r rS rO r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� 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
dd� Zdd� Zdd� Zd d � Zd d � Zd S )!� ClientTestszs
Tests for the protocol-level behavior of IRCClient methods intended to
be called by application code.
c C sZ t � | _t� | _d| j_| j�| j� | �| j�� d� | �| jj � | �| jj
d� dS )zT
Create and connect a new L{IRCClient} to a new L{StringTransport}.
Fr" N) r rI r r rE rK r r� rL rM rN r' r r r rO � s zClientTests.setUpc C s2 |� � }ttkrt|t�r|�d�}|�d�d S )zF
Return the last IRC message in the transport buffer.
r r� rZ )r� �bytesr r r rh )r rI r� r r r � getLastLine� s
zClientTests.getLastLinec C s: d}| j �|� d|� �dg}| �| j�� �d�|� dS )zU
L{IRCClient.away} sends an AWAY command with the specified message.
zSorry, I'm not here.�AWAY :r" �
N)r �awayr rI r� rh )r r� r� r r r � test_away s � zClientTests.test_awayc C s. | j �� ddg}| �| j�� �d�|� dS )zP
L{IRCClient.back} sends an AWAY command with an empty message.
r� r" r� N)r �backr rI r� rh )r r� r r r � test_back s
� zClientTests.test_backc C s, | j �d� | �| j�� �d�ddg� dS )z;
L{IRCClient.whois} sends a WHOIS message.
�alicer� z WHOIS alicer" N�r r� r rI r� rh r' r r r �
test_whois s �zClientTests.test_whoisc C s. | j �dd� | �| j�� �d�ddg� dS )z�
L{IRCClient.whois} sends a WHOIS message with a server name if a
value is passed for the C{server} parameter.
r� z example.orgr� zWHOIS example.org alicer" Nr� r' r r r �test_whoisWithServer$ s �z ClientTests.test_whoisWithServerc C sh d}d}d}d| j _d| j _| j �|||� d|� �d|||| j jf dg}| �| j�� �d �|� dS )
z�
L{IRCClient.register} sends NICK and USER commands with the
username, name, hostname, server name, and real name specified.
�testuser�testhost�
testserver�testnameN�NICK �USER %s %s %s :%sr" r� �r r� �password�registerr rI r� rh �r �usernamer: r� r� r r r �
test_register. s �� zClientTests.test_registerc C st d}d}d}d| j _d| j _| j �|||� d| j j� �d|� �d|||| j jf d g}| �| j�� �d
�|� d S ) z�
If the C{password} attribute of L{IRCClient} is not L{None}, the
C{register} method also sends a PASS command with it as the
argument.
r� r� r� r� �testpasszPASS r� r� r" r� Nr� r� r r r �test_registerWithPasswordA s �� z%ClientTests.test_registerWithPasswordc C s� d}d}d}d| j _d| j _| j �|||� | j �ddg� | �| j�}| �|d|� �� | j �ddg� | �| j�}| �|d � |d
�� d S ) z�
Verify that the client repeats the L{IRCClient.setNick} method with a
new value when presented with an C{ERR_NICKNAMEINUSE} while trying to
register.
r� r� r� r� r� r� r� r� �NICK {}r� N)
r r� r� r� �irc_ERR_NICKNAMEINUSEr� rI r� r r� )r r� r: r� �lastLiner r r �test_registerWithTakenNickV s z&ClientTests.test_registerWithTakenNickc C sR d}dd� | j _| j �|� | j �ddg� | �| j�}| �|d�|d �� dS ) z�
L{IRCClient.alterCollidedNick} determines how a nickname is altered upon
collision while a user is trying to change to that nickname.
r. c S s | d S )N�***r r� r r r r� q s z<ClientTests.test_overrideAlterCollidedNick.<locals>.<lambda>r� r� r� r� N)r �alterCollidedNickr� r� r� rI r r� )r r� r� r r r � test_overrideAlterCollidedNickk s z*ClientTests.test_overrideAlterCollidedNickc C sj d}d}| j �|� | j �ddg� | j �|� | �| j j|� | j �|� d�|g� | �| j j|� dS )z�
When a NICK command is sent after signon, C{IRCClient.nickname} is set
to the new nickname I{after} the server sends an acknowledgement.
r. r� r� r� z !quux@quxN)r r� r9 �setNickr r� �irc_NICK)r �oldnick�newnickr r r �test_nickChangew s zClientTests.test_nickChangec C s� d}| � | jjd� | j�|� | j�ddg� | �| j�}| � |d| jj� �� | j�ddg� | � | jjd� | j� | jj� | � | jj
| jj� | jj
}| j� |� | j�ddg� | �| j�}| � |d|� �� | � | jj
|� dS )z�
Trying to register an illegal nickname results in the default legal
nickname being set, and trying to change a nickname to an illegal
nickname results in the old nickname being kept.
r. Fr� r� r� TN) r r � _registeredr� �irc_ERR_ERRONEUSNICKNAMEr� rI �erroneousNickFallbackr9 r� r� )r �badnickr� r� r r r �test_erroneousNick� s z ClientTests.test_erroneousNickc C sj d}d}d}| j �||� | j �||� d|� d|� d�d|� d|� d�dg}| �| j�� �d�|� d S )
zd
L{IRCClient.desrcibe} sends a CTCP ACTION message to the target
specified.
r. r� �waveszPRIVMSG z
:ACTION �r" r� N)r �describer rI r� rh )r r rm r� r� r r r �
test_describe� s � zClientTests.test_describec s* � fdd�}|� j _� j �dddg� dS )zh
The default implementation of L{IRCClient.noticed} doesn't invoke
C{privmsg()}
c s � � d� d S )Nz%privmsg() should not have been called)�failr� r' r r r� � s z6ClientTests.test_noticedDoesntPrivmsg.<locals>.privmsg�spamz#greasyspooncafezI don't want any spam!N)r r� �
irc_NOTICE)r r� r r' r �test_noticedDoesntPrivmsg� s z%ClientTests.test_noticedDoesntPrivmsgc C s� | j �d� | �| j�� �d�� | j�� | j �dd� | �| j�� d� | j�� i | j _t | j j
d �D ]}t � � | | j jdt |�f< q7| �t
| j j�| j j
d � | j �dd� | �t
| j j�| j j
� | �| j�� d� dS ) z#
L{IRCClient.ping}
� otherusers PRIVMSG otheruser :PINGz
are you theres) PRIVMSG otheruser :PING are you there
rX zI sent a lot of pingss1 PRIVMSG otheruser :PING I sent a lot of pings
N)r �pingr� rI r� r� �clearr �_pings�range�
_MAX_PINGRINGr� r r� )r �pingNumr r r � test_ping� s( �
�
�zClientTests.test_pingN)r r r rS rO r� r� r� r� r� r� r� r� r� r r r r
r r r r r r� � s"
r� c @ rT )�CollectorClientzQ
A client that saves in a list the names of the methods that got called.
c s8 g � _ d� _|D ]}|f� fdd� }t� ||� � qdS )zx
@param methodsList: list of methods' names that should be replaced.
@type methodsList: C{list}
�Wolfc s � �fdd�}|S )z6
Collects C{method}s.
c s �j �� | f� d S rC )r� r� )r� )r� r r r �inner� s z<CollectorClient.__init__.<locals>.fake_method.<locals>.innerr )r� r r' )r� r � fake_method� s z-CollectorClient.__init__.<locals>.fake_methodN)r� r� �setattr)r � methodsListr� r r r' r r� � s
�zCollectorClient.__init__N)r r r rS r� r r r r r � rb r c @ sx e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� Zdd� Zdd� Zd S ) �DccTestsz%
Tests for C{dcc_*} methods.
c C s" g d�}d| _ d| _t|�| _d S )N)� dccDoSend�dccDoAcceptResume� dccDoResume� dccDoChatr �#twisted�rl rm r rJ �r r� r r r rO s zDccTests.setUpc C s@ | j �| j| jd� | �| j jd| jddddg d�ffg� dS ) zO
L{irc.IRCClient.dcc_SEND} invokes L{irc.IRCClient.dccDoSend}.
�foo.txt 127.0.0.1 1025r � 127.0.0.1� �foo.txtr� �r' r% �1025N)rJ �dcc_SENDrl rm r r� r' r r r � test_dccSend s ����zDccTests.test_dccSendc C s t �� }| �t|jdd� dS )zM
L{irc.IRCClient.dccDoSend} is raises C{NotImplementedError}
r� N)r r r$ �NotImplementedError�dccSend)r rJ r r r �test_dccSendNotImplemented s z#DccTests.test_dccSendNotImplementedc C �0 | � tj| jj| j| jd�}| �t|�d� dS )z{
L{irc.IRCClient.dcc_SEND} raises L{irc.IRCBadMessage} when it is passed
a malformed query string.
r. z#malformed DCC SEND request: ['foo']N� r$ r �
IRCBadMessagerJ r* rl rm r r �r r r r r � test_dccSendMalformedRequest# � �z%DccTests.test_dccSendMalformedRequestc C r/ )z�
L{irc.IRCClient.dcc_SEND} raises L{irc.IRCBadMessage} when it is passed
a query string that doesn't contain a valid address.
zfoo.txt #23 sd@dz Indecipherable address '#23'Nr0 r2 r r r �!test_dccSendIndecipherableAddress- � �z*DccTests.test_dccSendIndecipherableAddressc C r/ )z�
L{irc.IRCClient.dcc_SEND} raises L{irc.IRCBadMessage} when it is passed
a query string that doesn't contain a valid port number.
�foo.txt 127.0.0.1 sd@d�Indecipherable port 'sd@d'Nr0 r2 r r r � test_dccSendIndecipherablePort; r6 z'DccTests.test_dccSendIndecipherablePortc C �8 | j �| j| jd� | �| j jd| jdddffg� dS )zY
L{irc.IRCClient.dcc_ACCEPT} invokes L{irc.IRCClient.dccDoAcceptResume}.
�foo.txt 1025 2r r' r& rW N)rJ �
dcc_ACCEPTrl rm r r� r' r r r �test_dccAcceptI s
�zDccTests.test_dccAcceptc C r/ )z}
L{irc.IRCClient.dcc_ACCEPT} raises L{irc.IRCBadMessage} when it is
passed a malformed query string.
r. z*malformed DCC SEND ACCEPT request: ['foo']N) r$ r r1 rJ r< rl rm r r r2 r r r � test_dccAcceptMalformedRequestS r4 z'DccTests.test_dccAcceptMalformedRequestc C r: )zS
L{irc.IRCClient.dcc_RESUME} invokes L{irc.IRCClient.dccDoResume}.
r; r r' r& rW N)rJ �
dcc_RESUMErl rm r r� r' r r r �test_dccResume] s �zDccTests.test_dccResumec C r/ )z}
L{irc.IRCClient.dcc_RESUME} raises L{irc.IRCBadMessage} when it is
passed a malformed query string.
r. z*malformed DCC SEND RESUME request: ['foo']N) r$ r r1 rJ r? rl rm r r r2 r r r � test_dccResumeMalformedRequestf r4 z'DccTests.test_dccResumeMalformedRequestc
C s@ | j �| j| jd� | �| j jd| j| jddg d�ffg� dS )zO
L{irc.IRCClient.dcc_CHAT} invokes L{irc.IRCClient.dccDoChat}.
r$ r r% r& r( N)rJ �dcc_CHATrl rm r r� r' r r r � test_dccChatp s ����zDccTests.test_dccChatc C r/ )z{
L{irc.IRCClient.dcc_CHAT} raises L{irc.IRCBadMessage} when it is
passed a malformed query string.
r. z#malformed DCC CHAT request: ['foo']N� r$ r r1 rJ rB rl rm r r r2 r r r � test_dccChatMalformedRequest� r4 z%DccTests.test_dccChatMalformedRequestc C r/ )z�
L{irc.IRCClient.dcc_CHAT} raises L{irc.IRCBadMessage} when it is passed
a query string that doesn't contain a valid port number.
r7 r8 NrD r2 r r r � test_dccChatIndecipherablePort� r6 z'DccTests.test_dccChatIndecipherablePortN)r r r rS rO r+ r. r3 r5 r9 r= r> r@ rA rC rE rF r r r r r � s
r c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dS )�ServerToClientTestszL
Tests for the C{irc_*} methods sent from the server to the client.
c C s" d| _ d| _g d�}t|�| _d S )Nr r! ) r� r� r� r� �userQuitr� r� r� r� r" r# r r r rO � s zServerToClientTests.setUpc C �D | j �| j| jg� | j �ddg� | �| j jd| jffdg� dS )z�
L{IRCClient.joined} is called when I join a channel;
L{IRCClient.userJoined} is called when someone else joins.
� Svadilfari!~svadi@yok.utu.fi�#pythonr� )r� ��
SvadilfarirK N)rJ �irc_JOINrl rm r r� r' r r r �
test_irc_JOIN� � �z!ServerToClientTests.test_irc_JOINc C rI )z�
L{IRCClient.left} is called when I part the channel;
L{IRCClient.userLeft} is called when someone else parts.
rJ rK r� )r� rL N)rJ �irc_PARTrl rm r r� r' r r r �
test_irc_PART� rP z!ServerToClientTests.test_irc_PARTc C s: | j �ddg� | j �| jdg� | �| j jddg� dS )zo
L{IRCClient.userQuit} is called whenever someone quits
the channel (myself included).
rJ �Adios.� Farewell.)rH )rM rS )rH )r rT N)rJ �irc_QUITrl r r� r' r r r �
test_irc_QUIT� s ��z!ServerToClientTests.test_irc_QUITc C sP dt jt jt j d� }| j�| j| j|g� | �| jj d| jddffg� dS )zK
L{IRCClient.noticed} is called when a notice is received.
z:%(X)cextended%(X)cdata1%(X)cextended%(X)cdata2%(X)c%(EOL)srP r� r! z data1 data2N)
r rR r� rS rJ r rl rm r r� r� r r r �test_irc_NOTICE� s
��z#ServerToClientTests.test_irc_NOTICEc C sP | j �dg d�� | j �| j| jddg� | �| j jddd| jddffg� dS ) z�
L{IRCClient.kickedFrom} is called when I get kicked from the channel;
L{IRCClient.userKicked} is called when someone else gets kicked.
rJ )rK �WOLF�
shoryuken!rM z hadouken!)r� )rK rM rY r� r N)rJ �irc_KICKrl rm r r� r' r r r �
test_irc_KICK� s ���z!ServerToClientTests.test_irc_KICKc C s8 | j �| j| jdg� | �| j jdd| jdffg� dS )zR
L{IRCClient.topicUpdated} is called when someone sets the topic.
�new topic is newr� r N)rJ � irc_TOPICrl rm r r� r' r r r �test_irc_TOPIC� s
�z"ServerToClientTests.test_irc_TOPICc C s: | j �| jd| jdg� | �| j jdd| jdffg� dS )zc
L{IRCClient.topicUpdated} is called when the topic is initially
reported.
�?r\ r� r N)rJ �
irc_RPL_TOPICrl rm r r� r' r r r �test_irc_RPL_TOPIC
s
�z&ServerToClientTests.test_irc_RPL_TOPICc C s8 | j �| jd| jg� | �| j jdd| jdffg� dS )zP
L{IRCClient.topicUpdated} is called when the topic is removed.
r_ r� r r" N)rJ �irc_RPL_NOTOPICrl rm r r� r' r r r �test_irc_RPL_NOTOPIC
s �z(ServerToClientTests.test_irc_RPL_NOTOPICN)
r r r rS rO rO rR rV rW r[ r^ ra rc r r r r rG � s
rG c @ sP e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dS )�CTCPQueryTestsz/
Tests for the C{ctcpQuery_*} methods.
c C s d| _ d| _tdg�| _d S )Nr r! �
ctcpMakeReplyr" r' r r r rO
s zCTCPQueryTests.setUpc C �4 | j �| j| jd� | �| j jdddgffg� dS )zm
L{IRCClient.ctcpQuery_PING} calls L{IRCClient.ctcpMakeReply} with the
correct args.
rn re r )�PINGrn N)rJ �ctcpQuery_PINGrl rm r r� r' r r r �test_ctcpQuery_PING"
s �z"CTCPQueryTests.test_ctcpQuery_PINGc C �< d| j _| j �| j| jd� | �| j jdddgffg� dS )zo
L{IRCClient.ctcpQuery_FINGER} calls L{IRCClient.ctcpMakeReply} with the
correct args.
rX rn re r )�FINGERrX N)rJ � fingerReply�ctcpQuery_FINGERrl rm r r� r' r r r �test_ctcpQuery_FINGER,
�
�z$CTCPQueryTests.test_ctcpQuery_FINGERc C s> d| j _| j �| j| jd� | �| j jddddgffg� dS )zo
L{IRCClient.ctcpQuery_SOURCE} calls L{IRCClient.ctcpMakeReply} with the
correct args.
�urlrn re r )�SOURCErp )rq NN)rJ � sourceURL�ctcpQuery_SOURCErl rm r r� r' r r r �test_ctcpQuery_SOURCE7
s �z$CTCPQueryTests.test_ctcpQuery_SOURCEc C rj )zq
L{IRCClient.ctcpQuery_USERINFO} calls L{IRCClient.ctcpMakeReply} with
the correct args.
r� rn re r )�USERINFOr� N)rJ �userinfo�ctcpQuery_USERINFOrl rm r r� r' r r r �test_ctcpQuery_USERINFOC
ro z&CTCPQueryTests.test_ctcpQuery_USERINFOc C s\ | j �| j| jd� | j �| j| jd� d}| �| j jddd|fgffdddgffg� dS ) zs
L{IRCClient.ctcpQuery_CLIENTINFO} calls L{IRCClient.ctcpMakeReply} with
the correct args.
r" z PING PONGzEACTION CLIENTINFO DCC ERRMSG FINGER PING SOURCE TIME USERINFO VERSIONre r �
CLIENTINFO)ry NN)rJ �ctcpQuery_CLIENTINFOrl rm r r� )r r� r r r �test_ctcpQuery_CLIENTINFON
s � ��z(CTCPQueryTests.test_ctcpQuery_CLIENTINFOc C s4 | j �| j| jd� | �| j jd d d d� dS )zm
L{IRCClient.ctcpQuery_TIME} calls L{IRCClient.ctcpMakeReply} with the
correct args.
rn r rV r N)rJ �ctcpQuery_TIMErl rm r r� r' r r r �test_ctcpQuery_TIME`
s z"CTCPQueryTests.test_ctcpQuery_TIMEc C rf )zl
L{IRCClient.ctcpQuery_DCC} calls L{IRCClient.ctcpMakeReply} with the
correct args.
rn re r )�ERRMSGz!DCC data :Unknown DCC type 'DATA'N)rJ �
ctcpQuery_DCCrl rm r r� r' r r r �test_ctcpQuery_DCCh
s ���z!CTCPQueryTests.test_ctcpQuery_DCCN) r r r rS rO ri rn rt rx r{ r} r� r r r r rd
s
rd c @ rT )�DccChatFactoryTestsz&
Tests for L{DccChatFactory}.
c C s: d}t �d|�}|�d�}| �|t j� | �|j|� dS )z�
An instance of the L{irc.DccChat} protocol is returned, which has the
factory property set to the factory which created it.
)�fromUserNNNr% )r �DccChatFactory�
buildProtocol�assertIsInstance�DccChatr �factory)r � queryDatar� r r r r �test_buildProtocol~
s
z&DccChatFactoryTests.test_buildProtocolN)r r r rS r� r r r r r� y
rb r� c @ rT )�DccDescribeTestsz#
Tests for L{dccDescribe}.
c C s t �d�}| �|d� dS )z@
L{irc.dccDescribe} supports long IP addresses.
zCHAT arg 3232235522 6666z$CHAT for host 192.168.0.2, port 6666N)r � dccDescriber r2 r r r � test_address�
s
z DccDescribeTests.test_addressN)r r r rS r� r r r r r� �
rb r� c @ sZ e Zd ZdZddd�Zdd� Zdd � Zd
d � Zd d
� Zdd� Z dd� Z
dd� Z dd� Z dS )�DccFileReceiveTestsz&
Tests for L{DccFileReceive}.
r Nc C s0 t j||d�}|r|�d� t� }|�|� |S )a�
Factory helper that returns a L{DccFileReceive} instance
for a specific test case.
@param filename: Path to the local file where received data is stored.
@type filename: L{str}
@param resumeOffset: An integer representing the amount of bytes from
where the transfer of data should be resumed.
@type resumeOffset: L{int}
@param overwrite: A boolean specifying whether the file to write to
should be overwritten by calling L{DccFileReceive.set_overwrite}
or not.
@type overwrite: L{bool}
@return: An instance of L{DccFileReceive}.
@rtype: L{DccFileReceive}
�� resumeOffsetT)r �DccFileReceive�
set_overwriter rK )r �filenamer� � overwriter rI r r r �makeConnectedDccFileReceive�
s
z/DccFileReceiveTests.makeConnectedDccFileReceivec C s |� |� |�d� dS )z�
Arrange the protocol so that it received all data.
@param protocol: The protocol which will receive the data.
@type: L{DccFileReceive}
@param data: The received data.
@type data: L{bytest}
N)rT rN )r r rn r r r �allDataReceivedForProtocol�
s
z.DccFileReceiveTests.allDataReceivedForProtocolc C �F t | �� �}|�d� | j|jdd�}| �|d� | �|�� d� dS )z�
If given a resumeOffset argument, L{DccFileReceive} will attempt to
resume from that number of bytes if the file exists.
s Twisted is awesome!� r� s amazing!� Twisted is amazing!N�r �mktemp�
setContentr� �pathr� r �
getContent�r �fpr r r r �test_resumeFromResumeOffset�
�
z/DccFileReceiveTests.test_resumeFromResumeOffsetc C r� )aG
When resuming from an offset somewhere in the middle of the file,
for example, if there are 50 bytes in a file, and L{DccFileReceive}
is given a resumeOffset of 25, and after that 15 more bytes are
written to the file, then the resultant file should have just 40
bytes of data.
r� r� r� s cool!s Twisted is cool!Nr� r� r r r �:test_resumeFromResumeOffsetInTheMiddleOfAlreadyWrittenData�
s
zNDccFileReceiveTests.test_resumeFromResumeOffsetInTheMiddleOfAlreadyWrittenDatac C sF t | �� �}|�d� | j|jdd�}| �|d� | �|�� d� dS )z�
When local file already exists it can be overwritten using the
L{DccFileReceive.set_overwrite} method.
s I love contributing to Twisted!T)r� s Twisted rocks!Nr� r� r r r �test_setOverwrite�
r� z%DccFileReceiveTests.test_setOverwritec C s8 t | �� �}| �|j�}| �|d� | �|�� d� dS )z~
If the file does not already exist, then L{DccFileReceive} will
create one and write the data to it.
s I <3 TwistedN)r r� r� r� r� r r� r� r r r �test_fileDoesNotExist�
s z)DccFileReceiveTests.test_fileDoesNotExistc C s6 t | �� �}| jt| j|jdd�}| �tj|j� dS )z�
If given a resumeOffset to resume writing to a file that does not
exist, L{DccFileReceive} will raise L{OSError}.
rV r� N) r r� r$ �OSErrorr� r� r �errno�ENOENT)r r� rV r r r � test_resumeWhenFileDoesNotExist�
s
�z3DccFileReceiveTests.test_resumeWhenFileDoesNotExistc C s* t | �� �}|�� | �t| j|j� dS )zn
If the file already exists and overwrite action was not asked,
L{OSError} is raised.
N)r r� �touchr$ r� r� r� �r r� r r r �!test_fileAlreadyExistsNoOverwrite s z5DccFileReceiveTests.test_fileAlreadyExistsNoOverwritec C s( t | �� ��d�}| �t| j|j� dS )zO
L{IOError} is raised when failing to open the requested path.
z child-with-no-existing-parentN)r r� �childr$ �IOErrorr� r� r� r r r �test_failToOpenLocalFile s z,DccFileReceiveTests.test_failToOpenLocalFile)r N)
r r r rS r� r� r� r� r� r� r� r� r� r r r r r� �
s
r� )6rS r� r_ r� �twisted.internetr r �twisted.python.filepathr �twisted.test.proto_helpersr r �twisted.trial.unittestr �twisted.words.protocolsr �twisted.words.protocols.ircr r
rl r r rU rc r� r� r� r� r� �X_QUOTE�M_QUOTEr� r� �_CommandDispatcherMixinr� r� r� rD rF r} r� r� r@ r� r� r� r� r r rG rd r� r� r� r r r r �<module> sn `& �� " d}H
G { $za