o is}@sbddlZddlmZmZddlmZmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZddlmZddlmZddlmZddlmZd d Zd d Zd dZddZ e!Z"ddZ#ddZ$e$ddgdgdgdgdgdgdgdgdgdgd gd!gd"gd#gd$gd%gd&ge%d'd(Z&e$d)ddd*gdd+gd,d-gd.d/gdd0gd1d2gd3d4gd5d6gd7d8gd9dgd:d;gdd?gd@dAgdBdCgdDdEgge%dFd(Z'GdGdHdHZ(GdIdJdJZ)dKZ*GdLdMdMe)Z+[*GdNdOdOe+ej,Z-GdPdQdQe+ej,Z.GdRdSdSe+ej,Z/GdTdUdUe+ej,Z0GdVdWdWej,e)Z1GdXdYdYej,Z2dS)ZN)OptionalType) ValueConstantValues) BLINK CS_ALTERNATECS_ALTERNATE_SPECIAL CS_DRAWINGCS_UKCS_USG0G1 UNDERLINEClientProtocolServerProtocolmodes privateModes)Protocol)StringTransport) iterbytes)unittestcCstt||SN)superMock__getattribute__)mocknamereC:\Users\psant\Downloads\PAF backend\school\venv\Lib\site-packages\twisted/conch/test/test_insults.py_getattrsrcC t|dS)N occurrencesrrrrrr!# r!cCr )Nmethodsr"r#rrrr%'r$r%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). N)bytes bytearray)columnrowrrr_ecmaCodeTableCoordinate2sr.cs,fddt|D}||d<t|tf|S)Ncs:i|]\}}t|D]\}}|r |tt||q qSr) enumeraterr.).0jr-ir colOffsetrr Ksz/_makeControlFunctionSymbols..__doc__)r/typer)rr4namesdocattrsrr3r_makeControlFunctionSymbolsFs r; CSFinalByter))ZICHDCHZHPA)CUUZSSEZHPR)CUDZCPRZREP)CUFZSUZDA)CUBZSDZVPA)ZCNLZNPZVPR)ZCPLZPPZHVP)ZCHAZCTCZTBC)CUPZECHSM)ZCHTZCVTZMC)EDZCBTZHPB)ELZSRSZVPB)ILZPTXRM)DLZSDSSGR)ZEFZSIMDDSR)ZEANZDAQaW 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} )r4r8r9 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|_|dur i}||_|tur||_dSdS)z @param methods: Mapping of names to return values @param callReturnValue: object __call__ should return N)r!r%defaultcallReturnValue)selfr%rSrrr__init__s z Mock.__init__cOs.t|d}|tur t}t|d|||f|S)NrS__call__)rrRrr()rTakwZ returnValuerrrrVs z Mock.__call__cCs:t|d}||vrt||d}nt}t|||f|S)Nr%)rS)rrr()rTrr% attrValuerrrrs zMock.__getattribute__)__name__ __module__ __qualname__rRrSrUrVrrrrrrs  rc@seZdZdifddZdS) MockMixinrc Csb|\}}||||tt|dt|\\}}} } ||d|| ||| ||S)NrV) assertEquallenr!) rTZ occurrence methodNameZexpectedPositionalArgsZexpectedKeywordArgsattrrcallresultargsrXrrr assertCalls    zMockMixin.assertCallN)rZr[r\rfrrrrr]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)rfr!popr_)rT transportprotoparserrdrrr verifyResultssz ByteGroupingsMixin.verifyResults) rZr[r\rhrrr__annotations__wordnexec_byteGroupingTestTemplaterrrrrrrgs   rgc@eZdZeZdZddZdS)ServerArrowKeysTestss cCsht|||||j|j|j|jfD]}|t|dd|df}| t|gq| t|dS)NrkeystrokeReceived) rgrrZUP_ARROWZ DOWN_ARROWZ RIGHT_ARROWZ LEFT_ARROWrfr!rnr_ assertFalse)rTrorprqZarrowrdrrrrrsz"ServerArrowKeysTests.verifyResultsNrZr[r\rrh TEST_BYTESrrrrrrrys ryc@rx)PrintableCharactersTestssabc123ABC!@#abc123cCst||||tdD]}|t|dd|df}|t|gq tdD]}|t|dd||jf}|t|gq)t|}|||ddS)Ns abc123ABC!@#rrzsabc123z should have been []) rgrrrrfr!rnr_ZALTr{)rTrorprqcharrdoccsrrrrrs  z&PrintableCharactersTests.verifyResultsNr|rrrrr~ s r~c@sFeZdZdZeZgZdD] Zedeq d 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%drrz) rgrrrangegetattrrfr!rnr_r{)rTrorprqZfuncNumZfuncArgrdrrrrr<sz%ServerFunctionKeysTests.verifyResultsN) rZr[r\r6rrhZbyteListZ byteCodesr&joinr}rrrrrrr$s  rc@sHeZdZeZdZdZdZdZeeeeeeZ [[[[ddZ dS)ClientCursorMovementTestssssscCs^t||||dD]\}}|t|dd||f}|t|gq |t|dS)N))Downri)Forwardr)Upr^Backwardrirrrcursor)rgrrrfr!rnr_r{)rTrorprqmethodcountrdrrrrrRs z'ClientCursorMovementTests.verifyResultsN) rZr[r\rrhZd2Zr4u1l2r}rrrrrrrGs rc@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)rprrTrrhsz3ClientControlSequencesTests.setUp..rrm) rrorrprrqfactoryrmrfr!rnr{)rTrdrrrsetUpesz!ClientControlSequencesTests.setUpcCst|jdddtdDt|j}dD]}dD]}||dd||f}|t|qq||dS) Nrcss&|]}dD] }d||VqqdS))r2s20s200rNr)r0chrurrr rs zBClientControlSequencesTests.testSimpleCardinals..sBACD)rrrr)r^rirr) rq dataReceivedrrr!rprfrnr{)rTrmethrrdrrrtestSimpleCardinalsps  z/ClientControlSequencesTests.testSimpleCardinalscCsh|jdt|j}||ddd}|t|||ddd}|t|||dS)Ns rsetScrollRegion)rk)NNrqrr!rprfrnr{rTrrdrrrtestScrollRegions  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|q"q||dS)Nrcs(g|]dfddtdDqS)rcsg|]}d|qS)r)r0rugrr szKClientControlSequencesTests.testCharacterSet...sAB012)rr)r0rrrrsz@ClientControlSequencesTests.testCharacterSet..s()rZselectCharacterSet)rqrrrr!rpr r r r r rrrfrnr{)rTrwhichcharsetrdrrrtestCharacterSets*   z,ClientControlSequencesTests.testCharacterSetcCd|jdt|j}||dd}|t|||dd}|t|||dS)NsrshiftInshiftOutrrrrr testShifting  z(ClientControlSequencesTests.testShiftingcCr)NsNOr singleShift2 singleShift3rrrrrtestSingleShiftsrz,ClientControlSequencesTests.testSingleShiftscCr)Ns=>rZapplicationKeypadModeZnumericKeypadModerrrrrtestKeypadModerz*ClientControlSequencesTests.testKeypadModecCr)Ns78rZ saveCursorZ restoreCursorrrrrr testCursorrz&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;cs|]}d|fVqdSs%dNrr0mrrrrz8ClientControlSequencesTests.testModes..hcsrrrrrrrrrlrsetModes resetModes) rqrrrKAMIRMLNMr!rprfrnr{rrrr testModess0 z%ClientControlSequencesTests.testModescCsp|jdt|j}dD]}||d|}|t|q ||ddd}|t|||dS)Ns)eraseToLineEnderaseToLineBeginning eraseLineeraseToDisplayEnderaseToDisplayBeginning eraseDisplayrdeleteCharacter)rjr)rTrrrdrrr testErasures  z'ClientControlSequencesTests.testErasurecCR|jdt|j}dD]}||dd|f}|t|q ||dS)Nsr^rjr deleteLinerrTrargrdrrrtestLineDeletion"  z,ClientControlSequencesTests.testLineDeletioncCr)Nsrr insertLinerrrrrtestLineInsertion+rz-ClientControlSequencesTests.testLineInsertioncCsXdt|jd<|jd||jdt|j}|| dd}||ddS)N)rlreportCursorPositionssr) r%rprqrr_rovaluer!rfrnrrrrtestCursorPosition4s   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!rprqrrfrn)rTrrrrtest_applicationDataBytes?s   z5ClientControlSequencesTests.test_applicationDataBytescCsRt|j}|j||r|j|dg|dR|s ||d|dS)NrzNo other calls should happen: )r!rprqrrfrnr{)rTdataZcallsrrrr_applicationDataTestKs  z0ClientControlSequencesTests._applicationDataTestcC|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_shiftInAfterApplicationDataRzZ#d?d@Z$dAdBZ%dCS)DServerProtocolOutputTestszh Tests for the bytes L{ServerProtocol} writes to its transport when its methods are called. r^ rkcCs"t|_t|_|j|jdSr)rprotocolrrormrrrrrszServerProtocolOutputTests.setUpcC0|jd||j|jdtjjdS)z L{ServerProtocol.cursorUp} writes the control sequence ending with L{CSFinalByte.CUU} to its transport. r^1N)rZcursorUpr_rorrMr<r>rrrr test_cursorUp z'ServerProtocolOutputTests.test_cursorUpcCr)z L{ServerProtocol.cursorDown} writes the control sequence ending with L{CSFinalByte.CUD} to its transport. r^rN)rZ cursorDownr_rorrMr<r?rrrrtest_cursorDownrz)ServerProtocolOutputTests.test_cursorDowncCr)z L{ServerProtocol.cursorForward} writes the control sequence ending with L{CSFinalByte.CUF} to its transport. r^rN)rZ cursorForwardr_rorrMr<r@rrrrtest_cursorForwardrz,ServerProtocolOutputTests.test_cursorForwardcCr)z L{ServerProtocol.cursorBackward} writes the control sequence ending with L{CSFinalByte.CUB} to its transport. r^rN)rrr_rorrMr<rArrrrtest_cursorBackwardrz-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_rorrMr<rBrrrrtest_cursorPositionsz-ServerProtocolOutputTests.test_cursorPositioncC*|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_rorrMr<rBrrrrtest_cursorHome  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_rorESCr.rrrr test_indexs z$ServerProtocolOutputTests.test_indexcCr)zt L{ServerProtocol.reverseIndex} writes the control sequence ending in the L{C1SevenBit.RI}. N)rrr_rorrrKrNrrrrtest_reverseIndex  z+ServerProtocolOutputTests.test_reverseIndexcCs |j||jddS)zM L{ServerProtocol.nextLine} writes C{" "} to its transport. s N)rrr_rorrrrr test_nextLines z'ServerProtocolOutputTests.test_nextLinecCPtjtjtjg}|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}. rcsrrrrrrrrrz:ServerProtocolOutputTests.test_setModes..N) rrrrrrr_rorrMrr<rCrTZ modesToSetrrr test_setModess z'ServerProtocolOutputTests.test_setModescCr)z L{ServerProtocol.setPrivatesModes} writes a control sequence containing the requested private modes and ending in the L{CSFinalByte.SM}. rcsrrrrrrrrrzAServerProtocolOutputTests.test_setPrivateModes..N) rERRORZCOLUMNZORIGINrrr_rorrMrr<rC)rTZprivateModesToSetrrrtest_setPrivateModess z.ServerProtocolOutputTests.test_setPrivateModescCr)zs L{ServerProtocol.resetModes} writes the control sequence ending in the L{CSFinalByte.RM}. rcsrrrrrrrrrz.N) rrrrrrr_rorrMrr<rGrrrrtest_resetModess z)ServerProtocolOutputTests.test_resetModescCr)zp L{ServerProtocol.singleShift2} writes an escape sequence followed by L{C1SevenBit.SS2} N)rrr_rorrrKrPrrrrtest_singleShift2rz+ServerProtocolOutputTests.test_singleShift2cCr)zp L{ServerProtocol.singleShift3} writes an escape sequence followed by L{C1SevenBit.SS3} N)rrr_rorrrKrQrrrrtest_singleShift3rz+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_rorrMr<rIrrrrtest_selectGraphicRendition%s z5ServerProtocolOutputTests.test_selectGraphicRenditioncCr)zz L{ServerProtocol.horizontalTabulationSet} writes the escape sequence ending in L{C1SevenBit.HTS} N)rZhorizontalTabulationSetr_rorrrKrLrrrrtest_horizontalTabulationSet1rz6ServerProtocolOutputTests.test_horizontalTabulationSetcCr)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_rorrMr<rErrrrtest_eraseToLineEnd9  z-ServerProtocolOutputTests.test_eraseToLineEndcC.|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_rorrMr<rErrrrtest_eraseToLineBeginningC $z3ServerProtocolOutputTests.test_eraseToLineBeginningcCr)z L{ServerProtocol.eraseLine} writes the control sequence sequence ending in L{CSFinalByte.EL} and a parameter of 2 (the entire line.) rN)rrr_rorrMr<rErrrrtest_eraseLineM $z(ServerProtocolOutputTests.test_eraseLinecCr)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_rorrMr<rDrrrrtest_eraseToDisplayEndVrz0ServerProtocolOutputTests.test_eraseToDisplayEndcCr)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_rorrMr<rDrrrrtest_eraseToDisplayBeginning`rz6ServerProtocolOutputTests.test_eraseToDisplayBeginningcCr)z L{ServerProtocol.eraseDisplay} writes the control sequence sequence ending in L{CSFinalByte.ED} a parameter of 2 (the entire page) rN)rrr_rorrMr<rDrrrrtest_eraseToDisplayjrz-ServerProtocolOutputTests.test_eraseToDisplaycCr)z L{ServerProtocol.deleteCharacter} writes the control sequence containing the number of characters to delete and ending in L{CSFinalByte.DCH} r)4N)rrr_rorrMr<r=rrrrtest_deleteCharacterss z.ServerProtocolOutputTests.test_deleteCharactercCr)z L{ServerProtocol.insertLine} writes the control sequence containing the number of lines to insert and ending in L{CSFinalByte.IL} rk5N)rrr_rorrMr<rFrrrrtest_insertLine~ $z)ServerProtocolOutputTests.test_insertLinecCr)z L{ServerProtocol.deleteLine} writes the control sequence containing the number of lines to delete and ending in L{CSFinalByte.DL} rl6N)rrr_rorrMr<rHrrrrtest_deleteLinerz)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_rorrMrrrrtest_setScrollRegionNoArgsrz4ServerProtocolOutputTests.test_setScrollRegionNoArgscC.|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;rNr rrrrtest_setScrollRegionJustFirst z7ServerProtocolOutputTests.test_setScrollRegionJustFirstcCr")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;1rNr rrrrtest_setScrollRegionJustLastr%z6ServerProtocolOutputTests.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^ri)r#r&s1;2rNr rrrr test_setScrollRegionFirstAndLasts z:ServerProtocolOutputTests.test_setScrollRegionFirstAndLastcCr)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_rorrMr<rJrrrrtest_reportCursorPositions z3ServerProtocolOutputTests.test_reportCursorPositionN)&rZr[r\r6r.rrMrrrrrrrrrrrrrr r r r rrrrrrrrrr!r$r'r(r)rrrrrzsF                      r)3textwraptypingrrZ constantlyrrZtwisted.conch.insults.insultsrrrr r r r r rrrrrZtwisted.internet.protocolrZtwisted.internet.testingrZtwisted.python.compatrZ twisted.trialrrr!r%r(objectrRr.r;dedentr<rKrr]rwrgZTestCaseryr~rrrrrrrrs<    %""#