File: //usr/lib/python3/dist-packages/twisted/words/test/__pycache__/test_service.cpython-38.pyc
U
W[.o � @ s d Z ddlZddlmZmZmZ ddlmZmZm Z ddl
m Z m Z m
Z
mZ ddlmZ ddlmZ ddlmZ dd lmZ dd
lmZmZ dd lmZ G d d
� d
ej �Z G dd� de �Z G dd� dej �Z G dd� dej!�Z"e�#e"ej$� G dd� dej �Z%dS )z%
Tests for L{twisted.words.service}.
� N)�portal� credentials�checkers)�address�defer�reactor)�Deferred� DeferredList�
maybeDeferred�succeed)�unicode)�pb)�
proto_helpers)�unittest)�ewords�service)�ircc @ sT e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd d � Zd
d� Z dd� Z
dd� Z dS )�
RealmTestsc C s$ t �d�}d|�� }t|d|�� �}t|d|�� �}d|�� d }ttd|�� �}ttd|�� �}| �||��} | �|| j� | � ||���
|� t ||d� | �|d |�� ��} | �d |�� | j� | �|d |�� ��}
| � | |
� t ||d
� | � |d |�� ���
|� d S ) N� realmname�test�create�getZ OnRequestZ DuplicateZNoSuchT�newFZanother)
r �InMemoryWordsRealm�lower�getattr�titler �successResultOf� assertEqual�name�failureResultOf�trap�setattr�assertIdentical) �selfZkind�realmr r r �flagZdupExcZ noSuchExc�pZnewp� r( �A/usr/lib/python3/dist-packages/twisted/words/test/test_service.py�_entityCreationTest s"
z RealmTests._entityCreationTestc C s
| � d�S )N�User�r* �r$ r( r( r) �testUserCreation5 s zRealmTests.testUserCreationc C s
| � d�S )N�Groupr, r- r( r( r) �testGroupCreation9 s z RealmTests.testGroupCreationc C sn t �d�}| �|�d��}| �|�d��}| �||� | �|�d��}| �||� | �|�d���t j
� d S )Nr �testuserZ
nosuchuser) r r r �
createUser�getUserr# �
lookupUserr r! r Z
NoSuchUser)r$ r% �user� retrieved�lookedUpr( r( r) �testUserRetrieval= s
�z RealmTests.testUserRetrievalc C sl t �d�}t �d�}| �|�|��}| �||� | �|�d��}| �||� | �|�d��}| �||� d S )Nr r1 )r r r+ r �addUserr# r3 r4 )r$ r% r' r5 r6 r7 r( r( r) �testUserAdditionP s
zRealmTests.testUserAdditionc C sR t �d�}| �|�d��}| �|�d��}| �||� | �|�d���tj � d S )Nr � testgroupZ nosuchgroup)
r r r � createGroup�getGroupr# r r! r Z NoSuchGroup)r$ r% �groupr6 r( r( r) �testGroupRetrievala s
�z RealmTests.testGroupRetrievalc C sD t �d�}t �d�}| �|�|�� | �|�d��}| �||� d S )Nr r; Z testGroup)r r r/ r �addGroupr= r# )r$ r% r' r> r( r( r) �testGroupAdditionn s
z RealmTests.testGroupAdditionc C s. t �d�}| �|�d�� | �|�d�� dS )z�
Try creating a group with the same name as an existing user and
assert that it succeeds, since users and groups should not be in the
same namespace and collisions should be impossible.
r r N)r r r r2 r< )r$ r% r( r( r) �testGroupUsernameCollisionw s
z%RealmTests.testGroupUsernameCollisionc C sb t �d�}| �|�d�� | �|�d�� | �|�� �}dd� |D �}|�� | �|ddg� d S )Nr �groupone�grouptwoc S s g | ]
}|j �qS r( )r )�.0�gr( r( r) �
<listcomp>� s z.RealmTests.testEnumeration.<locals>.<listcomp>)r r r r< �
itergroups�sortr )r$ r% �groups�nr( r( r) �testEnumeration� s
zRealmTests.testEnumerationN) �__name__�
__module__� __qualname__r* r. r0 r8 r: r? rA rB rL r( r( r( r) r s
r c @ s* e Zd Ze�ddd�fdd�Zdd� ZdS ) �TestCaseUserAggZTCP� 127.0.0.1i1� c C sB || _ t�� | _|�|�| _| j| j_| j| j _| j�| j� d S �N)r5 r Z StringTransportWithDisconnection� transportZ
buildProtocol�protocol�mindZmakeConnection)r$ r5 r% �factoryr r( r( r) �__init__� s
zTestCaseUserAgg.__init__c C s | j �|� d S rR )rT Z dataReceived)r$ �stuffr( r( r) �write� s zTestCaseUserAgg.writeN)rM rN rO r Z IPv4AddressrW rY r( r( r( r) rP � s rP c @ s� e Zd Zddddddddd d
d g Zd d
� Zdd� Zd=dd�Zdd� Zd>dd�Zdd� Z dd� Z
dd � Z d d � Z d d � Z
d!d"� Zd#d$� Zd%d&� Zd'd(� Zd)d*� Zd+d,� Zd-d.� Zd/d0� Zd1d2� Zd3d4� Zd5d6� Zd7d8� Zd9d:� Zd;d<� ZdS )?�IRCProtocolTests�useruser� otheruser�someguy� firstuserZusername�userone�usertwo� userthreeZuserfours userfive�someuserc C s� t �d�| _t�� | _t�| j| jg�| _t �| j| j�| _ g }| j
D ]<}t |t �r^|�
d�}|�| j�|�� | j�||d � qFt|�S )Nr �utf-8� _password)r r r% r �'InMemoryUsernamePasswordDatabaseDontUse�checkerr �PortalZ
IRCFactoryrV � STATIC_USERS�
isinstance�bytes�decode�appendr2 r9 r )r$ �c�nickr( r( r) �setUp� s
zIRCProtocolTests.setUpc C sV | � |�}tjtjtjtjg}|D ] \}}}||kr"|�|� q"| �|d|f � dS )z�
The user has been greeted with the four messages that are (usually)
considered to start an IRC session.
Asserts that the required responses were received.
zMissing responses for %rN)� _responser Z RPL_WELCOMEZ RPL_YOURHOSTZ RPL_CREATEDZ
RPL_MYINFO�remove� assertFalse)r$ r5 �responseZexpected�prefixZcommand�argsr( r( r) �_assertGreeting� s
� z IRCProtocolTests._assertGreetingNc C s4 |d kr|d }|� d|f � |� d|f � d S )Nrd z PASS %s
zNICK %s extrainfo
)rY )r$ r5 rn �passwordr( r( r) �_login� s zIRCProtocolTests._loginc C s2 | � | j�|��}t|| j| j�}| �||� |S rR )r r% r4 rP rV rx )r$ r r5 Zaggr( r( r) �
_loggedInUser� s z IRCProtocolTests._loggedInUserc C sp |j �� }ttkr&t|t�r&|�d�}|�� }|j �� g }tt j
|�D ]"}|dks`|d |krH|� |� qH|S )z�
Extracts the user's response, and returns a list of parsed lines.
If messageType is defined, only messages of that type will be returned.
rc N� ) rS �valuerj �strri rk �
splitlines�clear�mapr Zparsemsgrl )r$ r5 Z messageTypers �result�messager( r( r) rp � s
zIRCProtocolTests._responsec C s | � d�}| �|� d S )Nr^ )ry rv )r$ r5 r( r( r) �
testPASSLogin� s
z IRCProtocolTests.testPASSLoginc C s� | � | j�d��}t|| j| j�}|�d� | �|d�}| �t|�d� | �|d d t j
� | �|d d d� | �|d d ddg� |j � � |�d� | �
|� d S )
z�
Sending NICK without PASS will prompt the user for their password.
When the user sends their password to NickServ, it will respond with a
Greeting.
r^ zNICK firstuser extrainfo
�PRIVMSGrz r � z Password?z%PRIVMSG nickserv firstuser_password
N)r r% r4 rP rV rY rp r �lenr ZNICKSERVrS r~ rv �r$ r^ r5 rs r( r( r) �test_nickServLogin� s
z#IRCProtocolTests.test_nickServLoginc C sh | � | j�d��}t|| j| j�}| �|dd� | �|d�}| �t|�d� | �|d d ddg� d S )Nr^ Z wrongpassr� rz r r� zLogin failed. Goodbye.) r r% r4 rP rV rx rp r r� r� r( r( r) �testFailedLogin� s z IRCProtocolTests.testFailedLoginc s^ g � | � | j�d��}t|| j| j�}| �|d� � fdd�|j_|�d� | � � dg� d S )Nr^ c s
� � d�S )NT)rl r( ��logoutr( r) �<lambda> � z-IRCProtocolTests.testLogout.<locals>.<lambda>zQUIT
T)
r r% r4 rP rV rx rT r� rY r )r$ r^ r5 r( r� r) �
testLogout s
zIRCProtocolTests.testLogoutc C s� | � | j�d��}| � | j�d��}d|jd< t|| j| j�}| �|d� |j� � |�
d� | � |�}| � t
|�d� | � |d d d� | � |d d d
� | � |d d d g� | � |d d d
� | � |d d d� | � |d d d� | � |d d d� | �d�}|j� � |j� � |�
d� | � |�}| � |�}| � t
|�d � | � |d d d� | � |d d d
� | � |d d d g� | � |d d d� | � |d d d
� | �|d d ddd dgddd dgg� d S )Nr^ � somechannelzsome random topic�topic�JOIN #somechannel
� r z firstuser!firstuser@realmnamerz ZJOINr� � #somechannelZ353Z366� �332� �333r\ � otheruser!otheruser@realmnamer �=zfirstuser otheruserzotheruser firstuser)r r% r4 r< �metarP rV rx rS r~ rY rp r r� ry ZassertIn)r$ r^ r� r5 rs �other�eventr( r( r) �testJoin sD
�
�zIRCProtocolTests.testJoinc C s� | � | j�d��}| � | j�d�� t|| j| j�}| �|d� |j�� |� d� | �
|�}dd� |D �}| � d|� | � d|� dS ) zo
When a user joins a group without a topic, no topic information is
sent to that user.
r^ r� r� c S s g | ] }|d �qS )rz r( )rE �rr( r( r) rG Z s z7IRCProtocolTests.test_joinTopicless.<locals>.<listcomp>r� r� N) r r% r4 r< rP rV rx rS r~ rY rp Z assertNotIn)r$ r^ r5 rs Z
responseCodesr( r( r) �test_joinTopiclessJ s
z#IRCProtocolTests.test_joinTopiclessc C s | � d�}| �| j�d�� |�d� |j�� | � d�}|�d� |j�� |j�� |�d� | �|�}| �|�}| �t |�d� | �|d d d� | �|d d d � | �|d d
d d g� | �||� |�d� |j�� |j�� |�d
� | �|�}| �|�}| �t |�d� | �|d d d� | �|d d d � | �|d d
d dg� | �||� |�d� |j�� |j�� |�d� | �|�}| �|�}| �t |�d� | �|d d d� | �|d d d � | �|d d
d dg� | �||� d S )Nr[ r� r� r\ zPART #somechannel
rz r �useruser!useruser@realmnameZPARTr� r� Zleavingz(PART #somechannel :goodbye stupidheads
zgoodbye stupidheadss JOIN #somechannel
s) PART #somechannel :goodbye stupidheads1
zgoodbye stupidheads1�
ry r r% r< rY rS r~ rp r r� �r$ r5 r� rs r� r( r( r) � testLeave_ sL
zIRCProtocolTests.testLeavec C sT | � d�}t�d�}d|jd< d|jd< d|jd< | �| j�|�� |j�� |� d � | �
|�}| � |d
d d � | � |d
d
d� | � |d
d dddg� | � |d d
d� | � |d d ddddg� |j�� |� d� | �
|�}| � |d d
d� | � |d d dddg� | � |d
d
d� | � |d
d ddddg� d S )Nr[ r� zThis is a test topic.r� Z some_fellowZ topic_authoriqˢ�
topic_dater� r� r r rz r� r� r� r� r� Z77777777�TOPIC #somechannel
) ry r r/ r� r r% r@ rS r~ rY rp r )r$ r5 r> rs r( r( r) � testGetTopic� s*
z IRCProtocolTests.testGetTopicc C st | � d�}| �| j�d��}|�d� | � d�}|�d� |j�� |j�� |�d� | �|�}| �|�}| �||� | �|d d d� | �|d d d � | �|d d
d d g� |j�� d
|j d< |�d� | �|�}| �|d d d� | �|d d
dd d g� | �|d d d� | �|d d
dd ddg� |j�� |�d� | �|�}| �|d d d� d S )Nr[ r� r� r\ z,TOPIC #somechannel :This is the new topic.
r r� rz ZTOPICr� r� zThis is the new topic.i90 r� r� r� r� Z12345zTOPIC #asdlkjasd
Z403)
ry r r% r< rY rS r~ rp r r� )r$ r5 r� r� rs r� r( r( r) � testSetTopic� s8
�
z IRCProtocolTests.testSetTopicc C s� | � d�}| �| j�d�� |�d� | � d�}|�d� |j�� |j�� |�d� | �|�}| �|�}| �|� | � t
|�d� | � |d d d� | � |d d d d
� | � |d d d d
g� d S )Nr[ r� r� r\ z%PRIVMSG #somechannel :Hello, world.
rz r r� r� ���r� r� z
Hello, world.) ry r r% r< rY rS r~ rp rr r r� r� r( r( r) �testGroupMessage� s
z!IRCProtocolTests.testGroupMessagec C s
| � d�}| � d�}|j�� |j�� |�d� | �|�}| �|�}| �|� | �t|�d� | �|d d d� | �|d d d� | �|d d dd g� |�d
� | �|�}| �t|�d� | �|d d d � | �|d d d � | �|d d dd
dg� d S )Nr[ r\ z#PRIVMSG otheruser :Hello, monkey.
rz r r� r� r� zHello, monkey.z)PRIVMSG nousernamedthis :Hello, monkey.
r Z401ZnousernamedthiszNo such nick/channel.)ry rS r~ rY rp rr r r� r� r( r( r) �testPrivateMessage s$
z#IRCProtocolTests.testPrivateMessagec C sP | � d�}|j�� |�d� | �|�}| �t|�d� | �|d d d� d S )Nr[ zOPER user pass
rz r Z491�ry rS r~ rY rp r r� �r$ r5 rs r( r( r) �testOper! s
zIRCProtocolTests.testOperc C s| | � d�}|j�� |�d� | �|�}| �t|�d� | �|d d d� | �|d d d� | �|d d ddg� d S ) Nr[ zMODE useruser
rz r r Z221r� �+r� r� r( r( r) �testGetUserMode, s
z IRCProtocolTests.testGetUserModec C sP | � d�}|j�� |�d� | �|�}| �t|�d� | �|d d d� d S )Nr[ zMODE useruser +abcd
rz r �472r� r� r( r( r) �testSetUserMode9 s
z IRCProtocolTests.testSetUserModec C sl | � d�}| �| j�d�� |�d� |j�� |�d� | �|�}| �t |�d� | �|d d d� d S )Nr[ r� r� zMODE #somechannel
rz r Z324r� r� r( r( r) �testGetGroupModeD s
z!IRCProtocolTests.testGetGroupModec C sl | � d�}| �| j�d�� |�d� |j�� |�d� | �|�}| �t |�d� | �|d d d� d S )Nr[ � groupname�JOIN #groupname
zMODE #groupname +abcd
rz r r� r� r� r( r( r) �testSetGroupModeS s
z!IRCProtocolTests.testSetGroupModec C s� t �d�}| �| j�|�� g }dD ]&}| �|�}|�|� |d �d� q$|D ]}|j� � qP|d �d� | �
|d �}| � | �
|d �� | � | �
|d �� d d
d g}|d d� D ]�\}} }
| � |d � | � | d
� |
\} }} }
}}}}| � | d � | � |d� | �
| |k� | � |
d � | � |d � |�|� | � |d� | � |d| � q�| � |� |d \}} }
| � |d � | � | d� |
\} }}| � | d � | � |d� | � |d� d S )Nr� )r_ r` ra r� r� r zWHO #groupname
rz r� r_ r` ra r Z352z
#groupname�Hz0 Z315zEnd of /WHO list.)r r/ r r% r@ ry rl rY rS r~ rp rr r Z
assertTruerq )r$ r> Zusersrn �ur5 r� Z wantusersrt �coderX ZmynameZ theirnameZ theirhostZ theirserverZ theirnickr& ZextraZchannelr( r( r) �testWhob sD
zIRCProtocolTests.testWhoc C s� | � d�}|j�� | �| j�d��}dd� |_d|jd< |�d� | � |�}| �
t |�d� |\}}| �
|d d
� | �
|d d � | �
|d d d� | �
|d d d� | �
|d d d
� | �
|d d d� | �
|d d
� | �
|d d� | �
|d d d� | �
|d d d� |j�� |�d� | � |�}| �
t |�d� |\}}| �
|d d � | �
|d d d� | �
|d d d
� | �
|d d d� | �
|d d� d S )Nrb � somegroupc S s t d�S )N� )r r( r( r( r) r� � r� z+IRCProtocolTests.testList.<locals>.<lambda>zthis is the topic woor� zLIST #somegroup
r� r r rz Z322Z17r� Z323z End of /LISTzLIST
) ry rS r~ r r% r< �sizer� rY rp r r� )r$ r5 r� r� Zresp�endZfg1r( r( r) �testList� s:
zIRCProtocolTests.testListc C s� | � d�}t�d�}dd� |_d|_t�� d |_| �| j� |�� |j
� � |� d� | �
|�}| �t|�d� |\}}}}}| �|d d
� | �|d d � | �|d
d d� | �|d
d d� | �|d
d
d� | �|d
d d
� | �|d
d d� | �|d
d d� | �|d d
� | �|d d� | �|d
d d� | �|d
d d� | �|d
d
d
� | �|d
d d� | �|d d
� | �|d d� | �|d
d d� | �|d
d d� | �|d
d
d� | �|d
d d� | �|d
d d� | �|d d
� | �|d d� | �|d
d d� | �|d
d d� | �|d
d
d� | �|d d
� | �|d d� | �|d
d d� | �|d
d d� | �|d
d
d� d S )Nr] �otherguyc S s t t�d�t�d�g�S )NZgroupAZgroupB)�iterr r/ r( r( r( r) r� � s �z,IRCProtocolTests.testWhois.<locals>.<lambda>�
� zWHOIS otherguy
r� r r rz Z311r� r� r� �*Z312zHi mom!Z317Z15Z10zseconds idle, signon timeZ319z#groupA #groupBZ318zEnd of WHOIS list.)ry r r+ rH ZsignOn�timeZ lastMessager r% r9 rS r~ rY rp r r� ) r$ r5 r� r� ZwuserZwserverZidleZchannelsr� r( r( r) � testWhois� sT
zIRCProtocolTests.testWhois)N)N) rM rN rO rh ro rv rx ry rp r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r( r( r( r) rZ � sH �
7<#,
+,rZ c @ s4 e Zd Zdd� Zdd� Zdd� Zdd� Zd d
� Zd S ) �TestMindc O s g | _ g | _g | _g | _d S rR )�joins�parts�messagesr� )r$ �a�kwr( r( r) rW � s zTestMind.__init__c C s | j �||f� d S rR )r� rl )r$ r5 r> r( r( r) �remote_userJoined� s zTestMind.remote_userJoinedc C s | j �|||f� d S rR )r� rl )r$ r5 r> �reasonr( r( r) �remote_userLeft� s zTestMind.remote_userLeftc C s | j �|||f� d S rR )r� rl )r$ ZsenderZ recipientr� r( r( r) �remote_receive s zTestMind.remote_receivec C s | j �||f� d S rR )r� rl )r$ r> r� r( r( r) �remote_groupMetaUpdate s z TestMind.remote_groupMetaUpdateN)rM rN rO rW r� r� r� r� r( r( r( r) r� � s
r� c @ s: e Zd Zdd� Zdd� Zdd� Zdd� Zejd d
� �Z d S ) �PBProtocolTestsc C s� t �d�| _t�� | _t�| j| jg�| _t� | j�| _
| j | j
_ d| j
_
t�� | _d| j_
t�d| j
�| _t�d| j�� j| j�| _d S )Nr Tr rQ )r r r% r re rf r rg r
ZPBServerFactoryZ
serverFactory�_protocolFactoryrT ZunsafeTracebacksZPBClientFactory�
clientFactoryr Z listenTCP�
serverPortZ
connectTCPZgetHostZport�
clientConnr- r( r( r) ro
s"
�
�zPBProtocolTests.setUpc O s t �d�| _| jS )Nr )r
ZBroker�_serverProtocol)r$ ru r� r( r( r) r� s z PBProtocolTests._protocolFactoryc s8 t � � | j�� fdd�� tt| jj�t| jj�� g�S )Nc s
� � d �S rR )�callbackr( �Zd3r( r) r� " r� z*PBProtocolTests.tearDown.<locals>.<lambda>) r r� ZnotifyOnDisconnectr r
r� Z
stopListeningr� Z
disconnectr- r( r� r) �tearDown s
�zPBProtocolTests.tearDownc sX |}t |t�r|�d�}t�||�� �j�||� �j�|�}|� � ��fdd�� |S )N�asciic s �j �� ��S rR )r� Zlogin)Zign�ZcredsrU r$ r( r) r� / r� z1PBProtocolTests._loggedInAvatar.<locals>.<lambda>)
ri r �encoder ZUsernamePasswordrf r9 r% r2 Z addCallback)r$ r rw rU Z nameBytes�dr( r� r) �_loggedInAvatar( s
z PBProtocolTests._loggedInAvatarc c s� t � }| �dd|�V }t � }| �dd|�V }t � }| �dd|�V }| j�d�V | j�d�V |�d�V }|�d�V }|�d�V |�d�V |�d�V |�d d
i�V |�� V |�� V d S ) N�ones p1�twos p2s threes p3Zfoobars barfoos texts hello, monkeys)r� r� r% r< �join�sendZleave) r$ Zmindoner� Zmindtwor� Z mindThreeZthreerC rD r( r( r) �
testGroups3 s
zPBProtocolTests.testGroupsN)
rM rN rO ro r� r� r� r ZinlineCallbacksr� r( r( r( r) r� s r� )&�__doc__r� Z twisted.credr r r Ztwisted.internetr r r Ztwisted.internet.deferr r r
r Ztwisted.python.compatr Ztwisted.spreadr
Z twisted.testr Z
twisted.trialr Z
twisted.wordsr r Ztwisted.words.protocolsr ZTestCaser �objectrP rZ ZPBMindr� ZsetUnjellyableForClassZPBMindReferencer� r( r( r( r) �<module> s( } V