Yf&dZddlmZmZddlmZddlmZddlm Z m Z m Z ddl m Z Gdde ZGd d e Zd ZGd d e ZdS)z5 Tests for IRC portions of L{twisted.words.service}. )checkersportal) proto_helpers)irc)InMemoryWordsRealm IRCFactoryIRCUser) IRCTestCasecNeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d S) IRCUserTestsz' Isolated tests for L{IRCUser} ctd|_tj|_t j|j|jg|_|jddt|j|j|_ |j d|_ tj |_|j |jdS)zm Sets up a Realm, Portal, Factory, IRCUser, Transport, and Connection for our tests. example.comjohnpassN)rrealmr'InMemoryUsernamePasswordDatabaseDontUsecheckerrPortaladdUserrfactory buildProtocolircUserrStringTransportstringTransportmakeConnectionselfs U/var/www/html/env/lib/python3.11/site-packages/twisted/words/test/test_irc_service.pysetUpzIRCUserTests.setUps ( 66 GII mDJ??  VV,,,!$*dk:: |11$77 ,<>> ##D$899999c|jddg|j|jd||jddS)z Sending a message to a user after they have sent NICK, but before they have authenticated, results in a message from "example.com". mynickfooz:example.com foo mynick N)rirc_NICKrclear sendMessageassertEqualBufferValuevaluers rtest_sendMessagezIRCUserTests.test_sendMessage"s b8*--- ""$$$   ''' ##  & & ( (*G     r cnd}|jddg|j|jd||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 тест ник r"uникuтестN)encoderr%rr&r'r(r))rexpectedResults rtest_utf8MessageszIRCUserTests.test_utf8Messages.s <BBDD b#7#>#>#@#@"ABBB ""$$$   !;!B!B!D!DEEE ##D$8$>$>$@$@.QQQQQr cr|jddg|tdS)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"sN)rr% assertRaises UnicodeErrorrs rtest_invalidEncodingNickz%IRCUserTests.test_invalidEncodingNick;s: b#6"7888 ,'''''r c|jj}|jj|d}|}d|DS)zC Grabs our responses and then clears the transport zutf-8c6g|]}tj|S)rparsemsg).0rs r z)IRCUserTests.response..Ls 222A Q222r )r transportr)r&decode splitlinesrresponses rr>zIRCUserTests.responseDsg<)//11 $$&&&??7++&&((222222r c#XKt|D]\}}|d|kr||fVdS)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)rr> messageTypenmessages r scanResponsezIRCUserTests.scanResponseNsM$H-- ! !JAwqz[((j    ! !r c|jddg|}t||t j}t||t j}||dddddgffg||ddddd gffgd 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#rr375#- example.com Message of the Day - r@376End of /MOTD command.N) rr%r>listrEr RPL_MOTDSTART RPL_ENDOFMOTD assertEqual)rr>startends rtest_sendNickSendsGreetingz'IRCUserTests.test_sendNickSendsGreeting]s b8*---==??T&&x1BCCDD4$$Xs/@AABB  %!#HI    1}eh8O-PQRS     r c|jddg|jddg|jddgd|jjd}d|jjd}||d d dd gfd d dd gfd dddgfd dd|gfd dd|gfd ddd |jjdddgfgdS)zz Receiving USER, PASS, NICK will log in the user, and transmit the appropriate response messages. r"zjohn doerrz,Your host is example.com, running version {}serviceVersionzThis server was created on {} creationDaterrGrHrIrJ001zconnected to Twisted IRC002003004wrCN) rirc_USERirc_PASSr%formatr _serverInforNr>)rversioncreations rtest_fullLoginzIRCUserTests.test_fullLoginxsE b:,/// b6(+++ b6(+++@GG L $%5 6  399 L $^ 4    MMOO0U'VW0G'HI0J'KL'89'9:!% 01AB         r c|jddg|}|jj||ddt j|jddg|}|jj||ddt j |j dddg|}|jj||ddt j |j dddg|}|jj||ddt j d S) z irc_PART testuserr#rr@ somechannels somechannelsboogaboogaN) rr%r>r:r&rNrrLirc_JOINERR_NOSUCHCHANNELirc_PARTERR_NOTONCHANNELr=s r test_PARTzIRCUserTests.test_PARTs j8*555==?? $$&&& !Q):;;; j=/:::==?? $$&&& !Q)>??? j>8*DEEE==?? $$&&& !Q)=>>> j='*BCCC==?? $$&&& !Q)=>>>>>r cj|jddg|jddg|jj|jddg|}||ddtj dS)z irc_NAMES r"rbrcrr@N) rr%rer:r& irc_NAMESr>rNrRPL_ENDOFNAMESr=s r test_NAMESzIRCUserTests.test_NAMESs b:,/// b=/222 $$&&& rM?333==?? !Q);<<<<rErQr`rirmr5r rr r s : : :     R R R(((333 ! ! !   6% % % N???* = = = = =r r ceZdZdZdZdS) MocksyIRCUserc<td|_g|_dS)Nr)rr mockedCodesrs r__init__zMocksyIRCUser.__init__s' 66 r c:|j|dSN)ruappend)rcode___s rr'zMocksyIRCUser.sendMessages %%%%%r N)rnrorprvr'r5r rrsrss2&&&&&r rscTeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z d Zd S)IRCUserBadEncodingTestszw Verifies that L{IRCUser} sends the correct error messages back to clients when given indecipherable bytes c,t|_dSrx)rsrrs rrzIRCUserBadEncodingTests.setUps$ r ct|jd|zdtg||jj|gdS)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)getattrrBADTEXTrNru)rirc_xerrors rassertChokesOnBadBytesz.IRCUserBadEncodingTests.assertChokesOnBadBytessL 0 h.//wi@@@ 1E7;;;;;r cF|dtjdS)zk Tests that irc_JOIN sends ERR_NOSUCHCHANNEL if the channel name can't be decoded. JOINNrrrfrs r test_JOINz!IRCUserBadEncodingTests.test_JOIN# ##FC,ABBBBBr cF|dtjdS)zl Tests that irc_NAMES sends ERR_NOSUCHCHANNEL if the channel name can't be decoded. NAMESNrrs rrmz"IRCUserBadEncodingTests.test_NAMES# ##GS-BCCCCCr cF|dtjdS)zl Tests that irc_TOPIC sends ERR_NOSUCHCHANNEL if the channel name can't be decoded. TOPICNrrs r test_TOPICz"IRCUserBadEncodingTests.test_TOPICrr cF|dtjdS)zk Tests that irc_LIST sends ERR_NOSUCHCHANNEL if the channel name can't be decoded. LISTNrrs r test_LISTz!IRCUserBadEncodingTests.test_LISTrr cF|dtjdS)zg Tests that irc_MODE sends ERR_NOSUCHNICK if the target name can't be decoded. MODENrrERR_NOSUCHNICKrs r test_MODEz!IRCUserBadEncodingTests.test_MODEs# ##FC,>?????r cF|dtjdS)zj Tests that irc_PRIVMSG sends ERR_NOSUCHNICK if the target name can't be decoded. PRIVMSGNrrs r test_PRIVMSGz$IRCUserBadEncodingTests.test_PRIVMSG s# ##Is/ABBBBBr cF|dtjdS)zh Tests that irc_WHOIS sends ERR_NOSUCHNICK if the target name can't be decoded. WHOISNrrs r test_WHOISz"IRCUserBadEncodingTests.test_WHOISs# ##GS-?@@@@@r cF|dtjdS)zi Tests that irc_PART sends ERR_NOTONCHANNEL if the target name can't be decoded. PARTN)rrrhrs rriz!IRCUserBadEncodingTests.test_PARTs# ##FC,@AAAAAr cF|dtjdS)zo Tests that irc_WHO immediately ends the WHO list if the target name can't be decoded. WHON)rr RPL_ENDOFWHOrs rtest_WHOz IRCUserBadEncodingTests.test_WHO%s# ##E3+;<<<<rs*)))))))&&&&&&''''''IIIIIIIIII333333n=n=n=n=n=;n=n=n=b&&&&&G&&& ^=^=^=^=^=k^=^=^=^=^=r