Vh&dZddlmZmZddlmZddlmZddlm Z m Z m Z ddl m Z Gdde ZGd d e Zd ZGd d e Zy)z5 Tests for IRC portions of L{twisted.words.service}. )checkersportal) proto_helpers)irc)InMemoryWordsRealm IRCFactoryIRCUser) IRCTestCasecLeZdZdZdZdZdZdZdZdZ dZ d Z d Z d Z y ) IRCUserTestsz' Isolated tests for L{IRCUser} ctd|_tj|_t j |j|jg|_|jjddt|j|j |_ |jjd|_ tj|_|jj|jy)zm Sets up a Realm, Portal, Factory, IRCUser, Transport, and Connection for our tests. example.comjohnpassN)rrealmr'InMemoryUsernamePasswordDatabaseDontUsecheckerrPortaladdUserrfactory buildProtocolircUserrStringTransportstringTransportmakeConnectionselfs S/home/dcms/DCMS/lib/python3.12/site-packages/twisted/words/test/test_irc_service.pysetUpzIRCUserTests.setUps ( 6 GGI mmDJJ?  VV,!$**dkk: ||11$7 ,<<> ##D$8$89c|jjddg|jj|jj d|j |jj dy)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"sa b8*- ""$   ' ##  & & (*G r cTdj}|jjddjg|jj |jj dj|j |jj|y)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 <BBD b#7#>#>#@"AB ""$   !;!B!B!DE ##D$8$8$>$>$@.Qr ch|jjddg|jty)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"78 ,'r c$|jjj}|jjj|j d}|j }|Dcgc]}t j|c}Scc}w)zC Grabs our responses and then clears the transport zutf-8)r transportr)r&decode splitlinesrparsemsg)rresponsers rr8zIRCUserTests.responseDsj<<))//1 $$&??7+&&()12A Q222s.B c#PKt|D]\}}|d|k(s||fyw)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)rr8 messageTypenmessages r scanResponsezIRCUserTests.scanResponseNs6$H- !JAwqz[(j  !s& &cj|jjddg|j}t|j |t j }t|j |t j}|j|dddddgffg|j|ddddd gffgy ) 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%r8listr@r RPL_MOTDSTART RPL_ENDOFMOTD assertEqual)rr8startends rtest_sendNickSendsGreetingz'IRCUserTests.test_sendNickSendsGreeting]s b8*-==?T&&x1B1BCD4$$Xs/@/@AB  %!#HI    1}eh8O-PQRS r c|jjddg|jjddg|jjddgdj |j j d}dj |j j d}|j|jd d dd gfd d dd gfd dddgfd dd|gfd dd|gfd ddd |j j dddgfgy)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 {} creationDaterrBrCrDrE001zconnected to Twisted IRC002003004wr>N) rirc_USERirc_PASSr%formatr _serverInforIr8)rversioncreations rtest_fullLoginzIRCUserTests.test_fullLoginxs( b:,/ b6(+ b6(+@GG LL $ $%5 6 399 LL $ $^ 4   MMO0U'VW0G'HI0J'KL'89'9:!% 001AB     r c|jjddg|j}|jjj |j |ddt j|jjddg|j}|jjj |j |ddt j|jjdddg|j}|jjj |j |ddt j|jjdddg|j}|jjj |j |ddt jy ) z irc_PART testuserr#rr; somechannels somechannelsboogaboogaN) rr%r8r4r&rIrrGirc_JOINERR_NOSUCHCHANNELirc_PARTERR_NOTONCHANNELrr8s r test_PARTzIRCUserTests.test_PARTs^ j8*5==? $$& !Q):):; j=/:==? $$& !Q)>)>? j>8*DE==? $$& !Q)=)=> j='*BC==? $$& !Q)=)=>r cf|jjddg|jjddg|jjj |jj ddg|j }|j|ddtjy)z irc_NAMES r"r]r^rr;N) rr%r`r4r& irc_NAMESr8rIrRPL_ENDOFNAMESrds r test_NAMESzIRCUserTests.test_NAMESs b:,/ b=/2 $$& rM?3==? !Q);); 0 h./wi@ 11E7;r cD|jdtjy)zk Tests that irc_JOIN sends ERR_NOSUCHCHANNEL if the channel name can't be decoded. JOINNrrrars r test_JOINz!IRCUserBadEncodingTests.test_JOIN ##FC,A,ABr cD|jdtjy)zl Tests that irc_NAMES sends ERR_NOSUCHCHANNEL if the channel name can't be decoded. NAMESNrrs rriz"IRCUserBadEncodingTests.test_NAMES ##GS-B-BCr cD|jdtjy)zl Tests that irc_TOPIC sends ERR_NOSUCHCHANNEL if the channel name can't be decoded. TOPICNrrs r test_TOPICz"IRCUserBadEncodingTests.test_TOPICrr cD|jdtjy)zk Tests that irc_LIST sends ERR_NOSUCHCHANNEL if the channel name can't be decoded. LISTNrrs r test_LISTz!IRCUserBadEncodingTests.test_LISTrr cD|jdtjy)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 cD|jdtjy)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/A/ABr cD|jdtjy)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 cD|jdtjy)zi Tests that irc_PART sends ERR_NOTONCHANNEL if the target name can't be decoded. PARTN)rrrcrs rrez!IRCUserBadEncodingTests.test_PARTs ##FC,@,@Ar cD|jdtjy)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+;+;rsN*&'II3n=;n=b&G& ^=k^=r