o >h&@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 buildProtocolircUserrStringTransportstringTransportmakeConnectionselfr/var/www/vedio/testing/chatpythonscript.ninositsolution.com/env/lib/python3.10/site-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_invalidEncodingNickcCs8|jj}|jj|d}|}dd|DS)zC Grabs our responses and then clears the transport zutf-8cSsg|]}t|qSr)rparsemsg).0rrrr Lsz)IRCUserTests.response..)r transportr%r"decode splitlinesrresponserrrr5Ds   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)rr5 messageTypenmessagerrr scanResponseNs   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 - r6376End of /MOTD command.N) rr!r5listr;r RPL_MOTDSTART RPL_ENDOFMOTD assertEqual)rr5startendrrrtest_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=r>r?001zconnected to Twisted IRC002003004wr9N) rirc_USERirc_PASSr!formatr _serverInforCr5)rversioncreationrrrtest_fullLoginxs6        zIRCUserTests.test_fullLogincCs|jddg|}|jj||ddtj|jddg|}|jj||ddtj |j dddg|}|jj||ddtj |j dddg|}|jj||ddtj d S) z irc_PART testuserrrr6 somechannels somechannelsboogaboogaN) rr!r5r1r"rCrrAirc_JOINERR_NOSUCHCHANNELirc_PARTERR_NOTONCHANNELr4rrr test_PARTs     zIRCUserTests.test_PARTcCs^|jddg|jddg|jj|jddg|}||ddtj dS)z irc_NAMES rrUrVrr6N) rr!rXr1r" irc_NAMESr5rCrRPL_ENDOFNAMESr4rrr test_NAMESs  zIRCUserTests.test_NAMESN)__name__ __module__ __qualname____doc__rr&r)r,r5r;rFrTr\r_rrrrr s    ' r c@seZdZddZddZdS) MocksyIRCUsercCstd|_g|_dS)Nr )rr mockedCodesrrrr__init__s  zMocksyIRCUser.__init__cOs|j|dSN)reappend)rcode___rrrr#szMocksyIRCUser.sendMessageN)r`rarbrfr#rrrrrds rdc@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|_dSrg)rdrrrrrrs 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)getattrrBADTEXTrCre)rirc_xerrorrrrassertChokesOnBadBytess z.IRCUserBadEncodingTests.assertChokesOnBadBytescC|dtjdS)zk Tests that irc_JOIN sends ERR_NOSUCHCHANNEL if the channel name can't be decoded. JOINNrrrrYrrrr test_JOINz!IRCUserBadEncodingTests.test_JOINcCrs)zl Tests that irc_NAMES sends ERR_NOSUCHCHANNEL if the channel name can't be decoded. NAMESNrurrrrr_rwz"IRCUserBadEncodingTests.test_NAMEScCrs)zl Tests that irc_TOPIC sends ERR_NOSUCHCHANNEL if the channel name can't be decoded. TOPICNrurrrr test_TOPICrwz"IRCUserBadEncodingTests.test_TOPICcCrs)zk Tests that irc_LIST sends ERR_NOSUCHCHANNEL if the channel name can't be decoded. LISTNrurrrr test_LISTrwz!IRCUserBadEncodingTests.test_LISTcCrs)zg Tests that irc_MODE sends ERR_NOSUCHNICK if the target name can't be decoded. MODENrrrERR_NOSUCHNICKrrrr test_MODErwz!IRCUserBadEncodingTests.test_MODEcCrs)zj Tests that irc_PRIVMSG sends ERR_NOSUCHNICK if the target name can't be decoded. PRIVMSGNr~rrrr test_PRIVMSG rwz$IRCUserBadEncodingTests.test_PRIVMSGcCrs)zh Tests that irc_WHOIS sends ERR_NOSUCHNICK if the target name can't be decoded. WHOISNr~rrrr test_WHOISrwz"IRCUserBadEncodingTests.test_WHOIScCrs)zi Tests that irc_PART sends ERR_NOTONCHANNEL if the target name can't be decoded. PARTN)rrrr[rrrrr\rwz!IRCUserBadEncodingTests.test_PARTcCrs)zo Tests that irc_WHO immediately ends the WHO list if the target name can't be decoded. WHON)rrr RPL_ENDOFWHOrrrrtest_WHO%rwz IRCUserBadEncodingTests.test_WHON)r`rarbrcrrrrvr_rzr|rrrr\rrrrrrms  rmN)rc twisted.credrr twisted.testrtwisted.words.protocolsrtwisted.words.servicerrrtwisted.words.test.test_ircr r rdrormrrrrs   2