File: //lib/python3/dist-packages/twisted/words/test/__pycache__/test_irc_service.cpython-38.pyc
U
W[�% � @ s� d Z ddlmZmZ ddlmZ ddlmZ ddlm Z m
Z
m Z ddl m
Z
G dd� de
�ZG d d
� d
e �Zd ZG d d
� d
e
�ZdS )z5
Tests for IRC portions of L{twisted.words.service}.
� )�checkers�portal)�
proto_helpers)�irc)�InMemoryWordsRealm�
IRCFactory�IRCUser)� IRCTestCasec @ s` e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dS )� IRCUserTestsz'
Isolated tests for L{IRCUser}
c C sp t d�| _t�� | _t�| j| jg�| _| j�dd� t| j| j�| _ | j �
d�| _ t �
� | _| j �| j� dS )zm
Sets up a Realm, Portal, Factory, IRCUser, Transport, and Connection
for our tests.
� example.com�john�passN)r �realmr Z'InMemoryUsernamePasswordDatabaseDontUseZcheckerr ZPortalZaddUserr �factoryZ
buildProtocol�ircUserr ZStringTransport�stringTransportZmakeConnection��self� r �E/usr/lib/python3/dist-packages/twisted/words/test/test_irc_service.py�setUp s
zIRCUserTests.setUpc C s< | j �ddg� | j�� | j �d� | �| j�� d� dS )z�
Sending a message to a user after they have sent NICK, but before they
have authenticated, results in a message from "example.com".
� �mynickZfooz:example.com foo mynick
N)r �irc_NICKr �clear� sendMessage�assertEqualBufferValue�valuer r r r �test_sendMessage" s
z IRCUserTests.test_sendMessagec C sR d� d�}| j�dd� d�g� | j�� | j�d� d�� | �| j�� |� dS )z�
When a UTF8 message is sent with sendMessage and the current IRCUser
has a UTF8 nick and is set to UTF8 encoding, the message will be
written to the transport.
u :example.com тест ник
�utf-8r u никu тестN)�encoder r r r r r r )r ZexpectedResultr r r �test_utf8Messages- s �
z IRCUserTests.test_utf8Messagesc C s | j �ddg� | �t� dS )z�
A NICK command sent with a nickname that cannot be decoded with the
current IRCUser's encoding results in a PRIVMSG from NickServ
indicating that the nickname could not be decoded.
r s ����N)r r Z assertRaises� UnicodeErrorr r r r �test_invalidEncodingNick< s z%IRCUserTests.test_invalidEncodingNickc C sJ | j j�� }| j j�� ttkr4t|t�r4|�d�}|�� }dd� |D �S )zC
Grabs our responses and then clears the transport
r c S s g | ]}t �|��qS r )r Zparsemsg)�.0�rr r r �
<listcomp>O s z)IRCUserTests.response.<locals>.<listcomp>) r � transportr r �bytes�str�
isinstance�decode�
splitlines�r �responser r r r. F s
zIRCUserTests.responsec c s, t |�D ] \}}|d |kr||fV qdS )a8
Gets messages out of a response
@param response: The parsed IRC messages of the response, as returned
by L{IRCUserTests.response}
@param messageType: The string type of the desired messages.
@return: An iterator which yields 2-tuples of C{(index, ircMessage)}
� N)� enumerate)r r. Z messageType�n�messager r r � scanResponseR s zIRCUserTests.scanResponsec C sx | j �ddg� | �� }t| �|tj��}t| �|tj��}| �|dddddgffg� | �|ddddd gffg� d
S ) z�
Receiving NICK without authenticating sends the MOTD Start and MOTD End
messages, which is required by certain popular IRC clients (such as
Pidgin) before a connection is considered to be fully established.
r r r r �375�#- example.com Message of the Day - r/ �376�End of /MOTD command.N) r r r. �listr3 r �
RPL_MOTDSTARTZ
RPL_ENDOFMOTD� assertEqual)r r. �start�endr r r �test_sendNickSendsGreetingb s ��z'IRCUserTests.test_sendNickSendsGreetingc C s� | j �ddg� | j �ddg� | j �ddg� d| jjd f }d| jjd f }| �| �� d d
dd gfd d dd
gfd dddgfd dd|gfd dd|gfd ddd | jjd ddgfg� dS )zz
Receiving USER, PASS, NICK will log in the user, and transmit the
appropriate response messages.
r zjohn doer
r z,Your host is example.com, running version %sZserviceVersionz This server was created on %sZ creationDater r4 r5 r6 r7 Z001zconnected to Twisted IRCZ002Z003Z004�wr1 N)r Zirc_USERZirc_PASSr r Z _serverInfor: r. )r �versionZcreationr r r �test_fullLoginr s2 � �
� ����zIRCUserTests.test_fullLoginc C s� | j �ddg� | �� }| j j�� | �|d d tj� | j �ddg� | �� }| j j�� | �|d d tj � | j �
dddg� | �� }| j j�� | �|d d tj � | j �
dddg� | �� }| j j�� | �|d d tj � d S )
z
irc_PART
�testuserr r r/ � somechannels somechannels boogaZboogaN) r r r. r' r r: r r9 �irc_JOIN�ERR_NOSUCHCHANNELZirc_PART�ERR_NOTONCHANNELr- r r r � test_PART� s zIRCUserTests.test_PARTc C s^ | j �ddg� | j �ddg� | j j�� | j �ddg� | �� }| �|d d tj � dS )z
irc_NAMES
r rA rB r r/ N)
r r rC r' r Z irc_NAMESr. r: r ZRPL_ENDOFNAMESr- r r r �
test_NAMES� s zIRCUserTests.test_NAMESN)�__name__�
__module__� __qualname__�__doc__r r r! r# r. r3 r= r@ rF rG r r r r r
s
r
c @ s e Zd Zdd� Zdd� ZdS )�
MocksyIRCUserc C s t d�| _g | _d S )Nr )r r � mockedCodesr r r r �__init__� s
zMocksyIRCUser.__init__c O s | j �|� d S �N)rM �append)r �code�_�__r r r r � s zMocksyIRCUser.sendMessageN)rH rI rJ rN r r r r r rL � s rL � �c @ sh e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
d � Zd d
� Z dd� Z
dd� Z dd� Z dd� Z
dd� ZdS )�IRCUserBadEncodingTestszw
Verifies that L{IRCUser} sends the correct error messages back to clients
when given indecipherable bytes
c C s t � | _d S rO )rL r r r r r r � s z IRCUserBadEncodingTests.setUpc C s. t | jd| �dtg� | �| jj|g� dS )ak
Asserts that IRCUser sends the relevant error code when a given irc_x
dispatch method is given undecodable bytes.
@param irc_x: the name of the irc_FOO method to test.
For example, irc_x = 'PRIVMSG' will check irc_PRIVMSG
@param error: the error code irc_x should send. For example,
irc.ERR_NOTONCHANNEL
zirc_%sN)�getattrr �BADTEXTr: rM )r Zirc_x�errorr r r �assertChokesOnBadBytes� s z.IRCUserBadEncodingTests.assertChokesOnBadBytesc C s | � dtj� dS )zk
Tests that irc_JOIN sends ERR_NOSUCHCHANNEL if the channel name can't
be decoded.
ZJOINN�rY r rD r r r r � test_JOIN� s z!IRCUserBadEncodingTests.test_JOINc C s | � dtj� dS )zl
Tests that irc_NAMES sends ERR_NOSUCHCHANNEL if the channel name can't
be decoded.
ZNAMESNrZ r r r r rG � s z"IRCUserBadEncodingTests.test_NAMESc C s | � dtj� dS )zl
Tests that irc_TOPIC sends ERR_NOSUCHCHANNEL if the channel name can't
be decoded.
ZTOPICNrZ r r r r �
test_TOPIC� s z"IRCUserBadEncodingTests.test_TOPICc C s | � dtj� dS )zk
Tests that irc_LIST sends ERR_NOSUCHCHANNEL if the channel name can't
be decoded.
ZLISTNrZ r r r r � test_LIST� s z!IRCUserBadEncodingTests.test_LISTc C s | � dtj� dS )zg
Tests that irc_MODE sends ERR_NOSUCHNICK if the target name can't
be decoded.
ZMODEN�rY r ZERR_NOSUCHNICKr r r r � test_MODE� s z!IRCUserBadEncodingTests.test_MODEc C s | � dtj� dS )zj
Tests that irc_PRIVMSG sends ERR_NOSUCHNICK if the target name can't
be decoded.
ZPRIVMSGNr^ r r r r � test_PRIVMSG s z$IRCUserBadEncodingTests.test_PRIVMSGc C s | � dtj� dS )zh
Tests that irc_WHOIS sends ERR_NOSUCHNICK if the target name can't
be decoded.
ZWHOISNr^ r r r r �
test_WHOIS
s z"IRCUserBadEncodingTests.test_WHOISc C s | � dtj� dS )zi
Tests that irc_PART sends ERR_NOTONCHANNEL if the target name can't
be decoded.
ZPARTN)rY r rE r r r r rF s z!IRCUserBadEncodingTests.test_PARTc C s | � dtj� dS )zo
Tests that irc_WHO immediately ends the WHO list if the target name
can't be decoded.
ZWHON)rY r Z RPL_ENDOFWHOr r r r �test_WHO s z IRCUserBadEncodingTests.test_WHON)rH rI rJ rK r rY r[ rG r\ r] r_ r` ra rF rb r r r r rU � s
rU N)rK Z twisted.credr r Z twisted.testr Ztwisted.words.protocolsr Ztwisted.words.servicer r r Ztwisted.words.test.test_ircr r
rL rW rU r r r r �<module> s #