U W[4@sdZddlZddlmZddlmZmZddlmZddl m Z m Z m Z m Z mZddlmZddlmZd d ZeZGd d d ejZGd ddejZGdddejZGdddZGddde ejeZGddde ejeZGddde ejeZGdddejZdS)z% Tests for L{twisted.conch.manhole}. N)unittest)errordefer)StringTransport) _TelnetMixin _SSHMixin _StdioMixinstdiossh)manhole)insultscCs.z ddWntddYSXdS)z Return the string used by Python as the name for code objects which are compiled from interactive input or at the top-level of modules. rN) traceback extract_stackrrA/usr/lib/python3/dist-packages/twisted/conch/test/test_manhole.pydetermineDefaultFunctionNames rc@seZdZdZddZdS)ManholeInterpreterTestsz2 Tests for L{manhole.ManholeInterpreter}. cCs2td}|jddg|||jdS)zR L{ManholeInterpreter.resetBuffer} should empty the input buffer. N12)r ZManholeInterpreterbufferextendZ resetBuffer assertFalse)self interpreterrrrtest_resetBuffer*s z(ManholeInterpreterTests.test_resetBufferN)__name__ __module__ __qualname____doc__rrrrrr&src@seZdZdZddZdS)ManholeProtocolTestsz' Tests for L{manhole.Manhole}. cCsPt}ttj}|||j}|j}|j ddg| | |jdS)zn L{manhole.Manhole.handle_INT} should cause the interpreter input buffer to be reset. rrN) rr ZServerProtocolr ZManholeZmakeConnectionZterminalProtocolrrrZ handle_INTr)rZ transportZterminalZprotocolrrrr%test_interruptResetsInterpreterBuffer9s  z:ManholeProtocolTests.test_interruptResetsInterpreterBufferN)rrr r!r#rrrrr"5sr"c@s\eZdZddZddZddZddZd d Zd d Zd dZ ddZ ddZ ddZ dS) WriterTestscCstddS)z& Colorize an integer. rNr lastColorizedLinerrrr test_IntegerJszWriterTests.test_IntegercCstddS)z7 Colorize an integer in double quotes. z"1"Nr%r'rrrtest_DoubleQuoteStringQsz"WriterTests.test_DoubleQuoteStringcCstddS)z7 Colorize an integer in single quotes. z'1'Nr%r'rrrtest_SingleQuoteStringXsz"WriterTests.test_SingleQuoteStringcCstddS)z7 Colorize an integer in triple quotes. z'''1'''Nr%r'rrrtest_TripleSingleQuotedString_sz)WriterTests.test_TripleSingleQuotedStringcCstddS)zB Colorize an integer in triple and double quotes. z"""1"""Nr%r'rrrtest_TripleDoubleQuotedStringfsz)WriterTests.test_TripleDoubleQuotedStringcCstddS)z1 Colorize a function definition. z def foo():Nr%r'rrrtest_FunctionDefinitionmsz#WriterTests.test_FunctionDefinitioncCstddS)z. Colorize a class definition. z class foo:Nr%r'rrrtest_ClassDefinitiontsz WriterTests.test_ClassDefinitioncCstd}|t|tdS)z, Colorize a Unicode string. иNr r&Z assertTrue isinstancebytesrresrrr test_unicode{s zWriterTests.test_unicodecCstd}|t|tdS)z/ Colorize a UTF-8 byte string. иNr0r3rrr test_bytess zWriterTests.test_bytescCs|tdtddS)zT The output of UTF-8 bytestrings and Unicode strings are identical. r6r/N) assertEqualr r&r'rrrtest_identicalOutputs z WriterTests.test_identicalOutputN) rrr r(r)r*r+r,r-r.r5r7r9rrrrr$Isr$c@seZdZejZddZddZddZddZ d d Z d d Z d dZ ddZ ddZejddZejddZddZddZejddZdS)ManholeLoopbackMixincCs t|S)N)rZwaitForDeferredrdrrrwfdszManholeLoopbackMixin.wfdcs,jd}dfdd}||S)z- Evaluate simple expression. dones 1 + 1 donecsdddgdS)N >>> 1 + 12>>> done _assertBufferignr'rrfinisheds z.finishedrecvlineClientexpect _testwrite addCallbackrZdonerFrr'rtest_SimpleExpressions   z*ManholeLoopbackMixin.test_SimpleExpressioncs,jd}dfdd}||S)z> Evaluate line continuation in triple quotes. r>s ''' ''' donecsddddgdS)Ns>>> '''s... '''s'\n'rArBrDr'rrrFszGManholeLoopbackMixin.test_TripleQuoteLineContinuation..finishedrGrLrr'r test_TripleQuoteLineContinuations   z5ManholeLoopbackMixin.test_TripleQuoteLineContinuationcs,jd}dfdd}||S)z/ Evaluate function definition. r>s'def foo(bar): print(bar) foo(42) donecsddddddgdS)Ns>>> def foo(bar):s... print(bar)... s >>> foo(42)s42rArBrDr'rrrFsz>ManholeLoopbackMixin.test_FunctionDefinition..finishedrGrLrr'rr-s   z,ManholeLoopbackMixin.test_FunctionDefinitioncs,jd}dfdd}||S)z, Evaluate class definition. r>sEclass Foo: def bar(self): print('Hello, world!') Foo().bar() donec sdddddddgdS)Ns>>> class Foo:s... def bar(self):s"... print('Hello, world!')rOs>>> Foo().bar()s Hello, world!rArBrDr'rrrFsz;ManholeLoopbackMixin.test_ClassDefinition..finishedrGrLrr'rr.s   z)ManholeLoopbackMixin.test_ClassDefinitioncs,jd}dfdd}||S)z0 Evaluate raising an exception. r>s#raise Exception('foo bar baz') donecs"dddtdddgdS)Ns">>> raise Exception('foo bar baz')s"Traceback (most recent call last):s File "", line 1, in zutf-8sException: foo bar bazrA)rCdefaultFunctionNameencoderDr'rrrFsz5ManholeLoopbackMixin.test_Exception..finishedrGrLrr'rtest_Exceptions   z#ManholeLoopbackMixin.test_Exceptioncs6jd}dtjdfdd}||S)z4 Evaluate interrupting with CTRL-C. r>cancelled linecsdddgdS)N>>> cancelled lineKeyboardInterruptrArBrDr'rrrFs z4ManholeLoopbackMixin.test_ControlC..finished)rHrIrJr CTRL_CrKrLrr'r test_ControlC s  z"ManholeLoopbackMixin.test_ControlCcsFjd}dfdd}||fdd}|||S)z Sending ^C to Manhole while in a state where more input is required to complete a statement should discard the entire ongoing statement and reset the input prompt to the non-continuation prompt. sthingss( thingscs*ddgjd}tj|S)N>>> ( ... things>>> )rCrHrIrJr rV)ignoredZ interruptedr'rrgotContinuation's  zNManholeLoopbackMixin.test_interruptDuringContinuation..gotContinuationcsddddgdS)NrXrYrUrZrB)r[r'rrgotInterruption0szNManholeLoopbackMixin.test_interruptDuringContinuation..gotInterruptionrG)rZ continuingr\r]rr'r test_interruptDuringContinuations      z5ManholeLoopbackMixin.test_interruptDuringContinuationcs>djd}fdd}fdd}|||S)z2 Evaluate cancelling with CTRL-\. rScs.dgtjjj}|tjS)NrT) rCrJr ZCTRL_BACKSLASHrHonDisconnection assertFailurerConnectionDone)rEr<r'rrgotPartialLineAs  zBManholeLoopbackMixin.test_ControlBackslash..gotPartialLinecsdgdS)NrBrDr'rrgotClearedLineIszBManholeLoopbackMixin.test_ControlBackslash..gotClearedLinerJrHrIrK)rZ partialLinerbrdrr'rtest_ControlBackslash:s     z*ManholeLoopbackMixin.test_ControlBackslashccs|d|jdV|dg|tjd|jdV|dg|d|jdV|tj|jj}||tj VdS) z~ A CTRL+D in the middle of a line doesn't close a connection, but at the beginning of a line it does. s1 + 1\+ 1r? + 1 >>> 1 + 1 + 1 s3 >>> N) rJrHrIrCr ZCTRL_Dr_r`rrar;rrr test_controlDQs     z"ManholeLoopbackMixin.test_controlDccsT|d|jdV|ddg|tjd|jdV|dgdS) a CTRL+L is generally used as a redraw-screen command in terminal applications. Manhole doesn't currently respect this usage of it, but it should at least do something reasonable in response to this event (rather than, say, eating your face). s 1 + 1rgrZr?rhs 1 \+ 1 \+ 1riN)rJrHrIrCr ZCTRL_Lr'rrr test_ControlLgs z"ManholeLoopbackMixin.test_ControlLcs,djd}fdd}||S)zl CTRL-A can be used as HOME - returning cursor to beginning of current line buffer. srint "hello"p print "hello"csdgdSNs>>> print "hello"rBignorer'rrcbsz.ManholeLoopbackMixin.test_controlA..cbrerr<rqrr'r test_controlAzs   z"ManholeLoopbackMixin.test_controlAcs,djd}fdd}||S)zc CTRL-E can be used as END - setting cursor to end of current line buffer. srint "hellop"rmcsdgdSrnrBror'rrrqsz.ManholeLoopbackMixin.test_controlE..cbrerrrr'r test_controlEs   z"ManholeLoopbackMixin.test_controlEc csZ|d|jdV|d|jdV|jdV|ddddd d dgd S) z When a deferred is returned to the manhole REPL, it is displayed with a sequence number, and when the deferred fires, the result is printed. sCfrom twisted.internet import defer, reactor d = defer.Deferred() d s s.c = reactor.callLater(0.1, d.callback, 'Hi!') rZs#Deferred #0 called back: 'Hi!' >>> s/>>> from twisted.internet import defer, reactors>>> d = defer.Deferred()s>>> ds1>>> c = reactor.callLater(0.1, d.callback, 'Hi!')sDeferred #0 called back: 'Hi!'N)rJrHrIrCr'rrr test_deferreds*z"ManholeLoopbackMixin.test_deferredN)rrr r ZColoredManholeserverProtocolr=rMrNr-r.rRrWr^rfrZinlineCallbacksrkrlrsrtrurrrrr:s$    r:c@seZdZdZdS)ManholeLoopbackTelnetTestsz, Test manhole loopback over Telnet. N)rrr r!rrrrrwsrwc@seZdZdZedkrdZdS)ManholeLoopbackSSHTestsz) Test manhole loopback over SSH. Nz!cryptography requirements missing)rrr r!r skiprrrrrxsrxc@s$eZdZdZedkrdZnejZdS)ManholeLoopbackStdioTestsz1 Test manhole loopback over standard IO. NTerminal requirements missing)rrr r!r ryZConsoleManholervrrrrrzsrzc@s$eZdZdZedkrdZddZdS)ManholeMainTestsz= Test the I{main} method from the I{manhole} module. Nr{cCs*|jttjdgd}|d|jddS)z Will raise an exception when called with an argument which is a dotted patch which can not be imported.. z no-such-class)argvzEmpty module namerN)Z assertRaises ValueErrorr mainr8args)rZ exceptionrrrtest_mainClassNotFounds z'ManholeMainTests.test_mainClassNotFound)rrr r!r ryrrrrrr|sr|)r!rZ twisted.trialrZtwisted.internetrrZtwisted.test.proto_helpersrZ twisted.conch.test.test_recvlinerrrr r Z twisted.conchr Ztwisted.conch.insultsr rrPZTestCaserr"r$r:rwrxrzr|rrrrs2     K