o b:'@sdZddlmZmZddlmZddlmZddlm Z m Z m Z ddl m Z Gddde ZGd d d e Zd ZGd d d e ZdS)z5 Tests for IRC portions of L{twisted.words.service}. )checkersportal) proto_helpers)irc)InMemoryWordsRealm IRCFactoryIRCUser) IRCTestCasec@s`eZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ dS) IRCUserTestsz' Isolated tests for L{IRCUser} cCsptd|_t|_t|j|jg|_|jddt|j|j|_ |j d|_ t |_|j |jdS)zm Sets up a Realm, Portal, Factory, IRCUser, Transport, and Connection for our tests. example.comjohnpassN)rrealmr'InMemoryUsernamePasswordDatabaseDontUsecheckerrPortaladdUserrfactory buildProtocolircUserrStringTransportstringTransportmakeConnectionselfrE/usr/lib/python3/dist-packages/twisted/words/test/test_irc_service.pysetUps   zIRCUserTests.setUpcCs<|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 sendMessageassertEqualBufferValuevaluerrrrtest_sendMessage"s    zIRCUserTests.test_sendMessagecCsLd}|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 тест ник ruникuтестN)encoderr!rr"r#r$r%)rexpectedResultrrrtest_utf8Messages.s  zIRCUserTests.test_utf8MessagescCs|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. rsN)rr! assertRaises UnicodeErrorrrrrtest_invalidEncodingNick;sz%IRCUserTests.test_invalidEncodingNickcCsJ|jj}|jjttkrt|tr|d}|}dd|DS)zC Grabs our responses and then clears the transport zutf-8cSsg|]}t|qSr)rparsemsg).0rrrr Msz)IRCUserTests.response..) r transportr%r"bytesstr isinstancedecode splitlinesrresponserrrr8Ds   zIRCUserTests.responseccs.t|D]\}}|d|kr||fVqdS)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 messageTypenmessagerrr scanResponseOs   zIRCUserTests.scanResponsecCsx|jddg|}t||tj}t||tj}||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. rrrr 375#- example.com Message of the Day - r9376End of /MOTD command.N) rr!r8listr>r RPL_MOTDSTART RPL_ENDOFMOTD assertEqual)rr8startendrrrtest_sendNickSendsGreeting^s" z'IRCUserTests.test_sendNickSendsGreetingcCs|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. rzjohn doer r z,Your host is example.com, running version {}serviceVersionzThis server was created on {} creationDater r?r@rArB001zconnected to Twisted IRC002003004wr<N) rirc_USERirc_PASSr!formatr _serverInforFr8)rversioncreationrrrtest_fullLoginys6        zIRCUserTests.test_fullLogincCs|jddg|}|jj||ddtj|jddg|}|jj||ddtj |j dddg|}|jj||ddtj |j dddg|}|jj||ddtj d S) z irc_PART testuserrrr9 somechannels somechannelsboogaboogaN) rr!r8r1r"rFrrDirc_JOINERR_NOSUCHCHANNELirc_PARTERR_NOTONCHANNELr7rrr test_PARTs     zIRCUserTests.test_PARTcCs^|jddg|jddg|jj|jddg|}||ddtj dS)z irc_NAMES rrXrYrr9N) rr!r[r1r" irc_NAMESr8rFrRPL_ENDOFNAMESr7rrr test_NAMESs  zIRCUserTests.test_NAMESN)__name__ __module__ __qualname____doc__rr&r)r,r8r>rIrWr_rbrrrrr s    ' r c@seZdZddZddZdS) MocksyIRCUsercCstd|_g|_dS)Nr )rr mockedCodesrrrr__init__s  zMocksyIRCUser.__init__cOs|j|dSN)rhappend)rcode___rrrr#szMocksyIRCUser.sendMessageN)rcrdrerir#rrrrrgs rgc@sheZdZdZddZddZddZdd Zd d Zd d Z ddZ ddZ ddZ ddZ ddZdS)IRCUserBadEncodingTestszw Verifies that L{IRCUser} sends the correct error messages back to clients when given indecipherable bytes cCs t|_dSrj)rgrrrrrrs zIRCUserBadEncodingTests.setUpcCs.t|jd|dtg||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)getattrrBADTEXTrFrh)rirc_xerrorrrrassertChokesOnBadBytess z.IRCUserBadEncodingTests.assertChokesOnBadBytescC|dtjdS)zk Tests that irc_JOIN sends ERR_NOSUCHCHANNEL if the channel name can't be decoded. JOINNrurr\rrrr test_JOINz!IRCUserBadEncodingTests.test_JOINcCrv)zl Tests that irc_NAMES sends ERR_NOSUCHCHANNEL if the channel name can't be decoded. NAMESNrxrrrrrbrzz"IRCUserBadEncodingTests.test_NAMEScCrv)zl Tests that irc_TOPIC sends ERR_NOSUCHCHANNEL if the channel name can't be decoded. TOPICNrxrrrr test_TOPICrzz"IRCUserBadEncodingTests.test_TOPICcCrv)zk Tests that irc_LIST sends ERR_NOSUCHCHANNEL if the channel name can't be decoded. LISTNrxrrrr test_LISTrzz!IRCUserBadEncodingTests.test_LISTcCrv)zg Tests that irc_MODE sends ERR_NOSUCHNICK if the target name can't be decoded. MODENrurERR_NOSUCHNICKrrrr test_MODErzz!IRCUserBadEncodingTests.test_MODEcCrv)zj Tests that irc_PRIVMSG sends ERR_NOSUCHNICK if the target name can't be decoded. PRIVMSGNrrrrr test_PRIVMSG rzz$IRCUserBadEncodingTests.test_PRIVMSGcCrv)zh Tests that irc_WHOIS sends ERR_NOSUCHNICK if the target name can't be decoded. WHOISNrrrrr test_WHOISrzz"IRCUserBadEncodingTests.test_WHOIScCrv)zi Tests that irc_PART sends ERR_NOTONCHANNEL if the target name can't be decoded. PARTN)rurr^rrrrr_rzz!IRCUserBadEncodingTests.test_PARTcCrv)zo Tests that irc_WHO immediately ends the WHO list if the target name can't be decoded. WHON)rur RPL_ENDOFWHOrrrrtest_WHO&rzz IRCUserBadEncodingTests.test_WHON)rcrdrerfrruryrbr}rrrrr_rrrrrrps  rpN)rf twisted.credrr twisted.testrtwisted.words.protocolsrtwisted.words.servicerrrtwisted.words.test.test_ircr r rgrrrprrrrs   3