U P|js}@sddlZddlmZmZddlmZmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZddlmZddlmZddlmZddlmZd d Zd d Zd dZddZ e!Z"ddZ#ddZ$e$dddddgdddgdddgd d!d"gd#d$d%gd&d'd(gd)d*d+gd,d-d.gd/d0d1gd2d3d4gd5d6d7gd8d9d:gd;dd?d@gdAdBdCgdDddEgge%dFdGZ&e$dHdddIgddJgdKdLgdMdNgddOgdPdQgdRdSgdTdUgdVdWgdXdgdYdZgd[d\gd]d^gd_d`gdadbgdcddgge%dedGZ'GdfdgdgZ(GdhdidiZ)djZ*Gdkdldle)Z+[*Gdmdndne+ej,Z-Gdodpdpe+ej,Z.Gdqdrdre+ej,Z/Gdsdtdte+ej,Z0Gdudvdvej,e)Z1Gdwdxdxej,Z2dS)yN)OptionalType) ValueConstantValues) BLINK CS_ALTERNATECS_ALTERNATE_SPECIAL CS_DRAWINGCS_UKCS_USG0G1 UNDERLINEClientProtocolServerProtocolmodes privateModes)Protocol)StringTransport) iterbytes)unittestcCstt||SN)superMock__getattribute__)mocknamerC/tmp/pip-unpacked-wheel-audpp6lv/twisted/conch/test/test_insults.py_getattrsrcCs t|dS)N occurrencesrrrrrr #sr cCs t|dS)Nmethodsr!r"rrrr#'sr#cCst||dSr)r append)robjrrr_append+sr&cCstt|d>|BgS)a Return the byte in 7- or 8-bit code table identified by C{column} and C{row}. "An 8-bit code table consists of 256 positions arranged in 16 columns and 16 rows. The columns and rows are numbered 00 to 15." "A 7-bit code table consists of 128 positions arranged in 8 columns and 16 rows. The columns are numbered 00 to 07 and the rows 00 to 15 (see figure 1)." p.5 of "Standard ECMA-35: Character Code Structure and Extension Techniques", 6th Edition (December 1994). )bytes bytearray)columnrowrrr_ecmaCodeTableCoordinate2sr,cs,fddt|D}||d<t|tf|S)Ncs:i|]2\}}t|D] \}}|r|tt||qqSr) enumeraterr,).0jr+ir colOffsetrr Ks  z/_makeControlFunctionSymbols..__doc__)r-typer)rr2namesdocattrsrr1r_makeControlFunctionSymbolsFs  r9 CSFinalByter'ZICHDCHZHPACUUZSSEZHPRCUDZCPRZREPCUFZSUZDACUBZSDZVPAZCNLZNPZVPRZCPLZPPZHVPZCHAZCTCZTBCCUPZECHSMZCHTZCVTZMCEDZCBTZHPBELZSRSZVPBILZPTXRMDLZSDSSGRZEFZSIMDDSRZEAZDAQaW Symbolic constants for all control sequence final bytes that do not imply intermediate bytes. This happens to cover movement control sequences. See page 11 of "Standard ECMA 48: Control Functions for Coded Character Sets", 5th Edition (June 1991). Each L{ValueConstant} maps a control sequence name to L{bytes} )r2r6r7 C1SevenBitZDCSZPU1ZBPHZPU2ZNBHZSTSZCCHZNELZMWZSSAZSPAZESAZEPAHTSZSOSZHTJZVTSZSCIZPLDCSIZPLUZSTRIOSCSS2ZPMSS3ZAPCa Symbolic constants for all 7 bit versions of the C1 control functions See page 9 "Standard ECMA 48: Control Functions for Coded Character Sets", 5th Edition (June 1991). Each L{ValueConstant} maps a control sequence name to L{bytes} c@s.eZdZeZdefddZddZddZdS)rNcCs*g|_|dkri}||_|tk r&||_dS)z @param methods: Mapping of names to return values @param callReturnValue: object __call__ should return N)r r#defaultcallReturnValue)selfr#rQrrr__init__s z Mock.__init__cOs.t|d}|tkrt}t|d|||f|S)NrQ__call__)rrPrr&)rRakwZ returnValuerrrrTs  z Mock.__call__cCs:t|d}||kr"t||d}nt}t|||f|S)Nr#)rQ)rrr&)rRrr# attrValuerrrrs  zMock.__getattribute__)__name__ __module__ __qualname__rPrQrSrTrrrrrrs rc@seZdZdifddZdS) MockMixinrc Csb|\}}||||tt|dt|\\}}} } ||d|| ||| ||S)NrT) assertEquallenr ) rRZ occurrence methodNameZexpectedPositionalArgsZexpectedKeywordArgsattrrcallresultargsrVrrr assertCalls    zMockMixin.assertCallN)rXrYrZrdrrrrr[sr[adef testByte%(groupName)s(self): transport = StringTransport() proto = Mock() parser = self.protocolFactory(lambda: proto) parser.factory = self parser.makeConnection(transport) bytes = self.TEST_BYTES while bytes: chunk = bytes[:%(bytesPer)d] bytes = bytes[%(bytesPer)d:] parser.dataReceived(chunk) self.verifyResults(transport, proto, parser) c@sNeZdZUdZeeeed<dD]\ZZ e e ee dq"[[ ddZ dS)ByteGroupingsMixinNprotocolFactory))ZPairs)ZTriples)ZQuadsr')ZQuints)ZSexes)Z groupNameZbytesPercCs.|t|dd|f}|t|gdS)NrmakeConnection)rdr popr])rR transportprotoparserrbrrr verifyResultssz ByteGroupingsMixin.verifyResults) rXrYrZrfrrr__annotations__wordnexec_byteGroupingTestTemplaterprrrrres  rec@seZdZeZdZddZdS)ServerArrowKeysTestss cCsht|||||j|j|j|jfD]0}|t|dd|df}| t|gq$| t|dS)NrkeystrokeReceived) rerpZUP_ARROWZ DOWN_ARROWZ RIGHT_ARROWZ LEFT_ARROWrdr rlr] assertFalse)rRrmrnroZarrowrbrrrrps z"ServerArrowKeysTests.verifyResultsNrXrYrZrrf TEST_BYTESrprrrrrvsrvc@seZdZeZdZddZdS)PrintableCharactersTestssabc123ABC!@#abc123cCst||||tdD]0}|t|dd|df}|t|gqtdD]2}|t|dd||jf}|t|gqRt|}|||ddS)Ns abc123ABC!@#rrwsabc123z should have been []) rerprrdr rlr]ZALTrx)rRrmrnrocharrboccsrrrrps"    z&PrintableCharactersTests.verifyResultsNryrrrrr{ sr{c@sFeZdZdZeZgZdD]Zedeqd eZ [[ddZ dS)ServerFunctionKeysTestsz9Test for parsing and dispatching function keys (F1 - F12)) sOPsOQsORsOSs15~s17~s18~s19~s20~s21~s23~s24~Csnt||||tddD]@}t|d|f}|t|dd|df}|t|gq|t|dS)Nr\ zF%drrw) rerprangegetattrrdr rlr]rx)rRrmrnroZfuncNumZfuncArgrbrrrrp<s z%ServerFunctionKeysTests.verifyResultsN) rXrYrZr4rrfZbyteListZ byteCodesr$joinrzrprrrrr~$s r~c@sHeZdZeZdZdZdZdZeeeeeeZ [[[[ddZ dS)ClientCursorMovementTestssssscCs^t||||dD]6\}}|t|dd||f}|t|gq|t|dS)N))Downrg)Forwardr'Upr\Backwardrgrrrcursor)rerprdr rlr]rx)rRrmrnromethodcountrbrrrrpRs  z'ClientCursorMovementTests.verifyResultsN) rXrYrZrrfZd2Zr4u1l2rzrprrrrrGsrc@seZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ ddZ ddZddZddZddZdd Zd!d"Zd#d$Zd%d&Zd'd(Zd)d*Zd+d,Zd-S).ClientControlSequencesTestscsht_t_tfdd_j_jjt j ddjf} t |dS)NcsjSr)rnrrRrrhrz3ClientControlSequencesTests.setUp..rrk) rrmrrnrrofactoryrkrdr rlrx)rRrbrrrsetUpesz!ClientControlSequencesTests.setUpcCst|jdddtdDt|j}dD]6}dD],}||dd||f}|t|q6q.||dS) Nrcss$|]}dD]}d||Vq qdS))r2s20s200rNr)r.chrsrrr rszBClientControlSequencesTests.testSimpleCardinals..sBACD)rrrr)r\rgrr) ro dataReceivedrrr rnrdrlrx)rRr}methrrbrrrtestSimpleCardinalsps  z/ClientControlSequencesTests.testSimpleCardinalscCsh|jdt|j}||ddd}|t|||ddd}|t|||dS)Ns rsetScrollRegion)ri)NNrorr rnrdrlrxrRr}rbrrrtestScrollRegions  z,ClientControlSequencesTests.testScrollRegioncCs|jdt|j}||ddd}|t|||ddd}|t|||dd}|t|||dd}|t|||dS)Ns #3#4#5#6rZdoubleHeightLine)T)FZsingleWidthLineZdoubleWidthLinerrrrrtestHeightAndWidths  z.ClientControlSequencesTests.testHeightAndWidthcCs|jdddtdDt|j}ttfD]>}tt t t t fD]*}| |dd||f}|t|qDq2||dS)Nrcs(g|] dfddtdDqS)rcsg|]}d|qS)r)r.rsgrr szKClientControlSequencesTests.testCharacterSet...sAB012)rr)r.rrrrsz@ClientControlSequencesTests.testCharacterSet..s()rZselectCharacterSet)rorrrr rnr r r r r rrrdrlrx)rRr}whichcharsetrbrrrtestCharacterSets,  z,ClientControlSequencesTests.testCharacterSetcCsd|jdt|j}||dd}|t|||dd}|t|||dS)NsrshiftInshiftOutrrrrr testShiftings  z(ClientControlSequencesTests.testShiftingcCsd|jdt|j}||dd}|t|||dd}|t|||dS)NsNOr singleShift2 singleShift3rrrrrtestSingleShiftss  z,ClientControlSequencesTests.testSingleShiftscCsd|jdt|j}||dd}|t|||dd}|t|||dS)Ns=>rZapplicationKeypadModeZnumericKeypadModerrrrrtestKeypadModes  z*ClientControlSequencesTests.testKeypadModecCsd|jdt|j}||dd}|t|||dd}|t|||dS)Ns78rZ saveCursorZ restoreCursorrrrrr testCursors  z&ClientControlSequencesTests.testCursorcCsD|jdt|j}||dd}|t|||dS)Nscrresetrrrrr testResets   z%ClientControlSequencesTests.testResetcCs|jdt|j}||dd}|t|||dd}|t|||dd}|t|||dS)NsDMErindex reverseIndexnextLinerrrrr testIndexs  z%ClientControlSequencesTests.testIndexc Cs|jddddtjtjtjfDd|jddddtjtjtjfDdt|j}| | dd tjtjtjgf}| t|| | dd tjtjtjgf}| t|| |dS) Nr;css|]}d|fVqdSs%dNrr.mrrrrsz8ClientControlSequencesTests.testModes..hcss|]}d|fVqdSrrrrrrrslrsetModes resetModes) rorrrKAMIRMLNMr rnrdrlrxrrrr testModess8 z%ClientControlSequencesTests.testModescCsp|jdt|j}dD]$}||d|}|t|q||ddd}|t|||dS)Ns)eraseToLineEnderaseToLineBeginning eraseLineeraseToDisplayEnderaseToDisplayBeginning eraseDisplayrdeleteCharacter)rhr)rRr}rrbrrr testErasures  z'ClientControlSequencesTests.testErasurecCsR|jdt|j}dD](}||dd|f}|t|q||dS)Nsr\rhr deleteLinerrRr}argrbrrrtestLineDeletion"s   z,ClientControlSequencesTests.testLineDeletioncCsR|jdt|j}dD](}||dd|f}|t|q||dS)Nsrr insertLinerrrrrtestLineInsertion+s   z-ClientControlSequencesTests.testLineInsertioncCsXdt|jd<|jd||jdt|j}|| dd}||ddS)N)rjreportCursorPositionssr) r#rnrorr]rmvaluer rdrlrrrrtestCursorPosition4s   z.ClientControlSequencesTests.testCursorPositioncCsNt|j}|jd||ddd|jd||ddddS)z Contiguous non-control bytes are passed to a single call to the C{write} method of the terminal to which the L{ClientProtocol} is connected. arwrite)rbc)rN)r rnrorrdrl)rRr}rrrtest_applicationDataBytes?s    z5ClientControlSequencesTests.test_applicationDataBytescCsNt|j}|j||r8|j|df|dq||d|dS)NrzNo other calls should happen: )r rnrorrdrlrx)rRdataZcallsr}rrr_applicationDataTestKs   z0ClientControlSequencesTests._applicationDataTestcCs|dddgdS)z Application data bytes followed by a shift-in command are passed to a call to C{write} before the terminal's C{shiftIn} method is called. sabr)sab)rNrrrrr test_shiftInAfterApplicationDataRszZ#d?d@Z$dAdBZ%dCS)DServerProtocolOutputTestszh Tests for the bytes L{ServerProtocol} writes to its transport when its methods are called. r\ ricCs"t|_t|_|j|jdSr)rprotocolrrmrkrrrrrszServerProtocolOutputTests.setUpcCs0|jd||j|jdtjjdS)z L{ServerProtocol.cursorUp} writes the control sequence ending with L{CSFinalByte.CUU} to its transport. r\1N)rZcursorUpr]rmrrKr:r<rrrr test_cursorUps  z'ServerProtocolOutputTests.test_cursorUpcCs0|jd||j|jdtjjdS)z L{ServerProtocol.cursorDown} writes the control sequence ending with L{CSFinalByte.CUD} to its transport. r\rN)rZ cursorDownr]rmrrKr:r=rrrrtest_cursorDowns  z)ServerProtocolOutputTests.test_cursorDowncCs0|jd||j|jdtjjdS)z L{ServerProtocol.cursorForward} writes the control sequence ending with L{CSFinalByte.CUF} to its transport. r\rN)rZ cursorForwardr]rmrrKr:r>rrrrtest_cursorForwards  z,ServerProtocolOutputTests.test_cursorForwardcCs0|jd||j|jdtjjdS)z L{ServerProtocol.cursorBackward} writes the control sequence ending with L{CSFinalByte.CUB} to its transport. r\rN)rrr]rmrrKr:r?rrrrtest_cursorBackwards  z-ServerProtocolOutputTests.test_cursorBackwardcCs2|jdd||j|jdtjjdS)z L{ServerProtocol.cursorPosition} writes a control sequence ending with L{CSFinalByte.CUP} and containing the expected coordinates to its transport. rs1;1N)rZcursorPositionr]rmrrKr:r@rrrrtest_cursorPositions z-ServerProtocolOutputTests.test_cursorPositioncCs*|j||j|jtjjdS)z L{ServerProtocol.cursorHome} writes a control sequence ending with L{CSFinalByte.CUP} and no parameters, so that the client defaults to (1, 1). N)rZ cursorHomer]rmrrKr:r@rrrrtest_cursorHomes z)ServerProtocolOutputTests.test_cursorHomecCs,|j||j|jtdddS)z L{ServerProtocol.index} writes the control sequence ending in the 8-bit code table coordinates 4, 4. Note that ECMA48 5th Edition removes C{IND}. r'N)rrr]rmrESCr,rrrr test_indexs  z$ServerProtocolOutputTests.test_indexcCs*|j||j|jtjjdS)zt L{ServerProtocol.reverseIndex} writes the control sequence ending in the L{C1SevenBit.RI}. N)rrr]rmrrrIrLrrrrtest_reverseIndexs z+ServerProtocolOutputTests.test_reverseIndexcCs |j||jddS)zM L{ServerProtocol.nextLine} writes C{" "} to its transport. s N)rrr]rmrrrrr test_nextLines z'ServerProtocolOutputTests.test_nextLinecCsPtjtjtjg}|j|||j|j d dd|Dt j jdS)z L{ServerProtocol.setModes} writes a control sequence containing the requested modes and ending in the L{CSFinalByte.SM}. rcss|]}d|fVqdSrrrrrrrsz:ServerProtocolOutputTests.test_setModes..N) rrrrrrr]rmrrKrr:rArRZ modesToSetrrr test_setModess z'ServerProtocolOutputTests.test_setModescCsPtjtjtjg}|j|||j|j d dd|Dt j jdS)z L{ServerProtocol.setPrivatesModes} writes a control sequence containing the requested private modes and ending in the L{CSFinalByte.SM}. rcss|]}d|fVqdSrrrrrrrszAServerProtocolOutputTests.test_setPrivateModes..N) rERRORZCOLUMNZORIGINrrr]rmrrKrr:rA)rRZprivateModesToSetrrrtest_setPrivateModess z.ServerProtocolOutputTests.test_setPrivateModescCsPtjtjtjg}|j|||j|j d dd|Dt j jdS)zs L{ServerProtocol.resetModes} writes the control sequence ending in the L{CSFinalByte.RM}. rcss|]}d|fVqdSrrrrrrrsz.N) rrrrrrr]rmrrKrr:rErrrrtest_resetModess z)ServerProtocolOutputTests.test_resetModescCs*|j||j|jtjjdS)zp L{ServerProtocol.singleShift2} writes an escape sequence followed by L{C1SevenBit.SS2} N)rrr]rmrrrIrNrrrrtest_singleShift2s z+ServerProtocolOutputTests.test_singleShift2cCs*|j||j|jtjjdS)zp L{ServerProtocol.singleShift3} writes an escape sequence followed by L{C1SevenBit.SS3} N)rrr]rmrrrIrOrrrrtest_singleShift3s z+ServerProtocolOutputTests.test_singleShift3cCsB|jtttt||j|jdttft j jdS)z L{ServerProtocol.selectGraphicRendition} writes a control sequence containing the requested attributes and ending with L{CSFinalByte.SGR} s%d;%dN) rZselectGraphicRenditionstrrrr]rmrrKr:rGrrrrtest_selectGraphicRendition%s z5ServerProtocolOutputTests.test_selectGraphicRenditioncCs*|j||j|jtjjdS)zz L{ServerProtocol.horizontalTabulationSet} writes the escape sequence ending in L{C1SevenBit.HTS} N)rZhorizontalTabulationSetr]rmrrrIrJrrrrtest_horizontalTabulationSet1s z6ServerProtocolOutputTests.test_horizontalTabulationSetcCs*|j||j|jtjjdS)a L{ServerProtocol.eraseToLineEnd} writes the control sequence sequence ending in L{CSFinalByte.EL} and no parameters, forcing the client to default to 0 (from the active present position's current location to the end of the line.) N)rrr]rmrrKr:rCrrrrtest_eraseToLineEnd9s z-ServerProtocolOutputTests.test_eraseToLineEndcCs.|j||j|jdtjjdS)a  L{ServerProtocol.eraseToLineBeginning} writes the control sequence sequence ending in L{CSFinalByte.EL} and a parameter of 1 (from the beginning of the line up to and include the active present position's current location.) rN)rrr]rmrrKr:rCrrrrtest_eraseToLineBeginningCs z3ServerProtocolOutputTests.test_eraseToLineBeginningcCs.|j||j|jdtjjdS)z L{ServerProtocol.eraseLine} writes the control sequence sequence ending in L{CSFinalByte.EL} and a parameter of 2 (the entire line.) rN)rrr]rmrrKr:rCrrrrtest_eraseLineMs z(ServerProtocolOutputTests.test_eraseLinecCs*|j||j|jtjjdS)a L{ServerProtocol.eraseToDisplayEnd} writes the control sequence sequence ending in L{CSFinalByte.ED} and no parameters, forcing the client to default to 0 (from the active present position's current location to the end of the page.) N)rrr]rmrrKr:rBrrrrtest_eraseToDisplayEndVs z0ServerProtocolOutputTests.test_eraseToDisplayEndcCs.|j||j|jdtjjdS)a L{ServerProtocol.eraseToDisplayBeginning} writes the control sequence sequence ending in L{CSFinalByte.ED} a parameter of 1 (from the beginning of the page up to and include the active present position's current location.) rN)rrr]rmrrKr:rBrrrrtest_eraseToDisplayBeginning`s z6ServerProtocolOutputTests.test_eraseToDisplayBeginningcCs.|j||j|jdtjjdS)z L{ServerProtocol.eraseDisplay} writes the control sequence sequence ending in L{CSFinalByte.ED} a parameter of 2 (the entire page) rN)rrr]rmrrKr:rBrrrrtest_eraseToDisplayjs z-ServerProtocolOutputTests.test_eraseToDisplaycCs0|jd||j|jdtjjdS)z L{ServerProtocol.deleteCharacter} writes the control sequence containing the number of characters to delete and ending in L{CSFinalByte.DCH} r'4N)rrr]rmrrKr:r;rrrrtest_deleteCharacterss  z.ServerProtocolOutputTests.test_deleteCharactercCs0|jd||j|jdtjjdS)z L{ServerProtocol.insertLine} writes the control sequence containing the number of lines to insert and ending in L{CSFinalByte.IL} ri5N)rrr]rmrrKr:rDrrrrtest_insertLine~s z)ServerProtocolOutputTests.test_insertLinecCs0|jd||j|jdtjjdS)z L{ServerProtocol.deleteLine} writes the control sequence containing the number of lines to delete and ending in L{CSFinalByte.DL} rj6N)rrr]rmrrKr:rFrrrrtest_deleteLines z)ServerProtocolOutputTests.test_deleteLinecCs*|j||j|jdddS)z With no arguments, L{ServerProtocol.setScrollRegion} writes a control sequence with no parameters, but a parameter separator, and ending in C{b'r'}. rrNrrr]rmrrKrrrrtest_setScrollRegionNoArgss z4ServerProtocolOutputTests.test_setScrollRegionNoArgscCs.|jjdd||j|jdddS)z With just a value for its C{first} argument, L{ServerProtocol.setScrollRegion} writes a control sequence with that parameter, a parameter separator, and finally a C{b'r'}. r\)firsts1;r Nr rrrrtest_setScrollRegionJustFirstsz7ServerProtocolOutputTests.test_setScrollRegionJustFirstcCs.|jjdd||j|jdddS)z With just a value for its C{last} argument, L{ServerProtocol.setScrollRegion} writes a control sequence with a parameter separator, that parameter, and finally a C{b'r'}. r\)lasts;1r Nr rrrrtest_setScrollRegionJustLastsz6ServerProtocolOutputTests.test_setScrollRegionJustLastcCs0|jjddd||j|jdddS)z When given both C{first} and C{last} L{ServerProtocol.setScrollRegion} writes a control sequence with the first parameter, a parameter separator, the last parameter, and finally a C{b'r'}. r\rg)r rs1;2r Nr rrrr test_setScrollRegionFirstAndLastsz:ServerProtocolOutputTests.test_setScrollRegionFirstAndLastcCs.|j||j|jdtjjdS)z L{ServerProtocol.reportCursorPosition} writes a control sequence ending in L{CSFinalByte.DSR} with a parameter of 6 (the Device Status Report returns the current active position.) rN)rrr]rmrrKr:rHrrrrtest_reportCursorPositions  z3ServerProtocolOutputTests.test_reportCursorPositionN)&rXrYrZr4r,rrKrrrrrrrrrrrrrrrrrrrrrrrrrrr r rrrrrrrrzsD                       r)3textwraptypingrrZ constantlyrrZtwisted.conch.insults.insultsrrrr r r r r rrrrrZtwisted.internet.protocolrZtwisted.internet.testingrZtwisted.python.compatrZ twisted.trialrrr r#r&objectrPr,r9dedentr:rIrr[rureZTestCaservr{r~rrrrrrrs<    %""#