File: //usr/lib/python3/dist-packages/twisted/words/test/__pycache__/test_ircsupport.cpython-38.pyc
U
W[�) � @ s� d Z ddlmZ ddlmZmZmZ ddlmZm Z ddl
m Z ddl m
Z
G dd� de�ZG d d
� d
e�ZG d d � d e�ZG d
d� de
�ZdS )z+
Tests for L{twisted.words.im.ircsupport}.
� )�StringTransport)�ChatUI� Conversation�GroupConversation)�
IRCAccount�IRCProto)� OfflineError)� IRCTestCasec @ s e Zd Zdd� Zdd� ZdS )�StubConversationc C s d S �N� ��selfr r �D/usr/lib/python3/dist-packages/twisted/words/test/test_ircsupport.py�show s zStubConversation.showc C s || _ || _d S r )�message�metadata)r r r r r r � showMessage s z StubConversation.showMessageN)�__name__�
__module__� __qualname__r r r r r r r
s r
c @ s& e Zd Zdd� Zdd� Zddd�ZdS ) �StubGroupConversationc C s || _ || _d S r )�topic�
topicSetBy)r r Znicknamer r r �setTopic s z StubGroupConversation.setTopicc C s d S r r r
r r r r " s zStubGroupConversation.showNc C s || _ || _|| _ d S r )r �text)r Zsenderr r r r r �showGroupMessage& s z&StubGroupConversation.showGroupMessage)N)r r r r r r r r r r r s r c @ s( e Zd Zedfdd�Zedfdd�ZdS )�
StubChatUIr c C s t �| |||�S r �r �getGroupConversation�r �groupZClassZ
stayHiddenr r r �getConversation. s zStubChatUI.getConversationc C s t �| |||�S r r r r r r r 2 s z StubChatUI.getGroupConversationN)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 )!�
IRCProtoTestsz
Tests for L{IRCProto}.
c C s2 t dddd dd�| _t| jt� d �| _t� | _d S )Nz Some accountF�alicez example.comi )r �accountr r �protor � transportr
r r r �setUp; s �zIRCProtoTests.setUpc C s$ | j �| j� | �| j�� d� dS )z�
When L{IRCProto} is connected to a transport, it sends I{NICK} and
I{USER} commands with the username from the account object.
z1NICK alice
USER alice foo bar :Twisted-IM user
N)r& �makeConnectionr' �assertEqualBufferValue�valuer
r r r �
test_loginB s
�zIRCProtoTests.test_loginc C s, d| j _| j�| j� | �| j�� d� dS )z�
If created with an account with a password, L{IRCProto} sends a
I{PASS} command before the I{NICK} and I{USER} commands.
Zsecretz>PASS secret
NICK alice
USER alice foo bar :Twisted-IM user
N)r% Zpasswordr& r) r' r* r+ r
r r r �test_authenticateN s �z IRCProtoTests.test_authenticatec C s0 ddg| j _| j�| j� | �| j�� d� dS )z�
If created with an account with a list of channels, L{IRCProto}
joins each of those channels after registering.
�#fooz#barzGNICK alice
USER alice foo bar :Twisted-IM user
JOIN #foo
JOIN #bar
N)r% Zchannelsr& r) r' r* r+ r
r r r �
test_channels\ s �zIRCProtoTests.test_channelsc C s4 | j �| j� | j �d� | �d| j j�d�� dS )z�
L{IRCProto} can interpret I{ISUPPORT} (I{005}) messages from the server
and reflect their information in its C{supported} attribute.
z3:irc.example.com 005 alice MODES=4 CHANLIMIT=#:20
� ZMODESN)r& r) r' � dataReceived� assertEqualZ supportedZ
getFeaturer
r r r �
test_isupportk s
�zIRCProtoTests.test_isupportc C sb | j �| j� | j �d� | j �d� | j �d� | j �d� | �d| j j� | �d| j j� dS )zB
IRC NICK command changes the nickname of a user.
z:alice JOIN #group1
�:alice1 JOIN #group1
z:alice1 NICK newnick
z:alice3 NICK newnick3
Znewnick�alice1N�r& r) r' r1 �assertIn� _ingroups� assertNotInr
r r r � test_nickv s zIRCProtoTests.test_nickc C s� | j �| j� | j �d� | �d| j jd � | �d| j jd � | j �d� | j �d� | j �d� | �d| j jd � | �d| j jd � dS ) zF
IRC PART command removes a user from an IRC channel.
r4 �group1r5 �group2z:alice PART #group1
z:alice1 PART #group1
z:alice1 PART #group2
Nr6 r
r r r � test_part� s zIRCProtoTests.test_partc C s� | j �| j� | j �d� | �d| j jd � | �d| j jd � | j �d� | �d| j jd � | j �d� | �t| j jd �dk� | j �d � d
S ) zH
IRC QUIT command removes a user from all IRC channels.
r4 r; r5 r< z:alice1 JOIN #group3
Zgroup3z:alice1 QUIT
r z:alice3 QUIT
N) r& r) r' r1 r7 r8 r9 Z
assertTrue�lenr
r r r � test_quit� s zIRCProtoTests.test_quitc C sR | j �| j� | j �d� | j �d� | j �d�}| �|jd� | �|jd� dS )zH
IRC TOPIC command changes the topic of an IRC channel.
r4 z :alice1 TOPIC #group1 newtopic
r; Znewtopicr5 N)r& r) r' r1 r r2 r r )r �groupConversationr r r �
test_topic� s zIRCProtoTests.test_topicc C s� | j �| j� | j �d� | j j�| j �d��}| �|jd� | j �d� | j � d�}| �|j
d� | j � d� | j �d� | j � d �}| � t
|d
�� | j j�| j �d��}| � t
|d �� d S )
zG
PRIVMSG sends a private message to a user or channel.
z#:alice1 PRIVMSG t2 test_message_1
r5 Ztest_message_1z(:alice1 PRIVMSG #group1 test_message_2
r; Ztest_message_2r$ z$:alice PRIVMSG #foo test_message_3
�foor r N�r& r) r' r1 �chatr" � getPersonr2 r r r ZsetNickZ assertFalse�hasattr�r Z conversationr@ r r r � test_privmsg� s"
�
�zIRCProtoTests.test_privmsgc C s� | j �| j� | j �d� | j j�| j �d��}| �|jd� | j �d� | j � d�}| �|j
d� | j � d� | j �d� | j � d �}| � t
|d
�� | j j�| j �d��}| � t
|d �� d S )
z3
CTCP ACTION to a user or channel.
z):alice1 PRIVMSG alice1 :ACTION smiles
r5 �smilesz*:alice1 PRIVMSG #group1 :ACTION laughs
r; Zlaughsr$ z(:alice PRIVMSG #group1 :ACTION cries
rB r r NrC rG r r r � test_action� s"
�
�zIRCProtoTests.test_actionc C s� | j �| j� | j �d� dgdgdgdgdgd�}ddddddgi}| �|| j j� | �|| j j� | j �d � | �i | j j� | j �d�}| �|d |j� d
S ) z�
RPL_NAMREPLY server response (353) lists all the users in a channel.
RPL_ENDOFNAMES server response (363) is sent at the end of RPL_NAMREPLY
to indicate that there are no more names.
z>:example.com 353 z3p = #bnl :pSwede Dan- SkOyg @MrOp +MrPlus
Zbnl)�Dan-�pSwede�SkOyg�MrOp�MrPlusrL rK rM rN rO z1:example.com 366 alice #bnl :End of /NAMES list
N) r& r) r' r1 r2 r8 Z _namrepliesr �members)r ZexpectedInGroupsZexpectedNamRepliesr@ r r r �test_rplNamreply� s* �� �� z IRCProtoTests.test_rplNamreplyc C s2 | j �| j� | j �d� | �d| j jd � dS )z[
RPL_TOPIC server response (332) is sent when a channel's topic is changed
z1:example.com 332 alice, #foo :Some random topic
zSome random topicrB N)r& r) r' r1 r2 Z_topicsr
r r r �
test_rplTopic� s
�zIRCProtoTests.test_rplTopicc C s� | j �| j� | j �d�}| �t|jd� | j |j_| j� � |�d� | �
| jj � � d� | j� � |�dddi� | �
| jj � � d� d S )
z*
L{IRCPerson.sendMessage}
r$ � Some message�Some message 2s PRIVMSG alice :Some message 2
rI �style�emotes PRIVMSG alice :ACTION smiles
N)
r& r) r' rE � assertRaisesr Z sendMessager% �client�clearr2 �io�getvalue)r Zpersonr r r �test_sendMessage� s
�
�z IRCProtoTests.test_sendMessagec C s� | j �| j� | j j�d| j �}| �t|jd� | j |j_ | j�
� |�d� | � | jj �
� d� | j�
� |�dddi� | � | jj �
� d� d S )
z.
L{IRCGroup.sendGroupMessage}
r. rS rT s PRIVMSG #foo :Some message 2
rI rU rV s PRIVMSG #foo :ACTION smiles
N)r& r) r' rD ZgetGrouprW r ZsendGroupMessager% rX rY r2 rZ r[ )r r! r r r �test_sendGroupMessage s
�
�z#IRCProtoTests.test_sendGroupMessageN)r r r �__doc__r( r, r- r/ r3 r: r= r? rA rH rJ rQ rR r\ r] r r r r r# 7 s
r# N)r^ Ztwisted.test.proto_helpersr Ztwisted.words.im.basechatr r r Ztwisted.words.im.ircsupportr r Ztwisted.words.im.localsr Ztwisted.words.test.test_ircr r
r r r# r r r r �<module> s