o bF@szdZddlZddlZddlZddlZddlmZddlmZm Z m Z m Z m Z m Z mZmZmZmZmZmZmZddlmZddlmZddlmZmZGdd d eZGd d d eZGd d d eZGdddeZe GdddZ GdddeZ!GdddeZ"GdddeZ#GdddeZ$GdddeZ%GdddeZ&GdddeZ'Gd d!d!eZ(Gd"d#d#eZ)dS)$z% Tests for L{twisted.python.compat}. N)skipIf) _PYPY_get_async_param bytesEnvironcmp comparableexecfile intToBytesioType iterbytes lazyByteSlice nativeString networkStringreraise)FilePath)platform)SynchronousTestCaseTestCasec@HeZdZdZddZddZddZdd Zd d Zd d Z ddZ dS) IOTypeTestsz? Test cases for determining a file-like object's type. cC|tttdS)z= An L{io.StringIO} accepts and returns text. N) assertEqualr ioStringIOstrselfr:/usr/lib/python3/dist-packages/twisted/test/test_compat.pytest_3StringIO)zIOTypeTests.test_3StringIOcCr)z= An L{io.BytesIO} accepts and returns bytes. N)rr rBytesIObytesrrrr test_3BytesIO/r zIOTypeTests.test_3BytesIOcCDt|d}|t|tWddS1swYdS)zT A file opened via 'io.open' in text mode accepts and returns text. wN)openmktemprr rrfrrrtest_3openTextMode5"zIOTypeTests.test_3openTextModecCr$)zW A file opened via 'io.open' in binary mode accepts and returns bytes. wbN)r&r'rr r"r(rrrtest_3openBinaryMode<r+z IOTypeTests.test_3openBinaryModecCsFt|d}|t|tWddS1swYdS)z The L{codecs} module, oddly, returns a file-like object which returns bytes when not passed an 'encoding' argument. r,N)codecsr&r'rr r"r(rrrtest_codecsOpenBytesCs"z IOTypeTests.test_codecsOpenBytescCsJtj|ddd}|t|tWddS1swYdS)zY When passed an encoding, however, the L{codecs} module returns unicode. r,utf-8)encodingN)r.r&r'rr rr(rrrtest_codecsOpenTextKs"zIOTypeTests.test_codecsOpenTextcCs|tttdS)zy When passed an object about which no sensible decision can be made, err on the side of unicode. N)rr objectrrrrrtest_defaultToTextRszIOTypeTests.test_defaultToTextN) __name__ __module__ __qualname____doc__rr#r*r-r/r2r4rrrrr$s rc@ eZdZdZddZddZdS) CompatTestszy Various utility functions in C{twisted.python.compat} provide same functionality as modern Python variants. cCst}|d|d|dt|}|||gd|dt|}|||ddg|dddh}||}t|}|||gdd S) zG L{set} should behave like the expected set interface. bca)r=r;r<drs)r=r<r?r@N)setaddlistsortrremovediscardunionrr=r;r>rrrtest_set`s"      zCompatTests.test_setcCsjtddg}|tt|d|t|ddgtddg}||}t|}|||gddS)zS L{frozenset} should behave like the expected frozenset interface. r=r;rBr?r@)r=r;r?r@N) frozenset assertRaisesAttributeErrorgetattrrsortedrGrCrDrHrrrtest_frozensetxs   zCompatTests.test_frozensetN)r5r6r7r8rIrOrrrrr:Zs r:c@s0eZdZdZddZddZddZdd Zd S) ExecfileCompatTestszE Tests for the Python 3-friendly L{execfile} implementation. cCsP|}t|d}||dWdn1swYt|dS)zo Write L{content} to a new temporary file, returning the L{FilePath} for the new file. r,asciiNr0)r'r&writeencoder)rcontentpathr)rrr writeScripts  zExecfileCompatTests.writeScriptcCs2|d}ddi}t|j||d|ddS)zX L{execfile} executes the specified file in the given global namespace. foo += 1 fooNrVrrUr)rscriptglobalNamespacerrrtest_execfileGlobalss  z(ExecfileCompatTests.test_execfileGlobalscCsL|d}ddi}ddi}t|j|||d|d|d|ddS)zk L{execfile} executes the specified file in the given global and local namespaces. rWrX Nr[)rr\r]localNamespacerrrtest_execfileGlobalsAndLocalss z1ExecfileCompatTests.test_execfileGlobalsAndLocalscCs@dD]}|d|}ddi}t|j||d|dqdS)z L{execfile} reads in the specified file using universal newlines so that scripts written on one platform will work on another. )  z z foo = 'okay'rXNokayr[)r lineEndingr\r]rrrtest_execfileUniversalNewliness  z2ExecfileCompatTests.test_execfileUniversalNewlinesN)r5r6r7r8rVr^rcrhrrrrrPs   rPc@seZdZdZddZdS)PYPYTestz! Identification of PyPy. cCs&dtjvr |tdS|tdS)z, On PyPy, L{_PYPY} is True. PyPyN)sysversion assertTruer assertFalserrrr test_PYPYs zPYPYTest.test_PYPYN)r5r6r7r8rorrrrris ric@r9) ComparablezE Objects that can be compared to each other, but not others. cCs ||_dSN)value)rrrrrr__init__s zComparable.__init__cCst|tstSt|j|jSrq) isinstancerpNotImplementedrrr)rotherrrr__cmp__s zComparable.__cmp__N)r5r6r7r8rsrwrrrrrps rpc@@eZdZdZddZddZddZdd Zd d Zd d Z dS)ComparableTestszR L{comparable} decorated classes emulate Python 2's C{__cmp__} semantics. cCs0|tdtdk|tdtdkdS)zo Instances of a class that is decorated by C{comparable} support equality comparisons. rYrZNrmrprnrrrr test_equalityzComparableTests.test_equalitycCs0|tdtdk|tdtdkdS)zq Instances of a class that is decorated by C{comparable} support inequality comparisons. rYrZN)rnrprmrrrrtest_nonEqualityr|z ComparableTests.test_nonEqualitycCs0|tdtdk|tdtdkdS)zs Instances of a class that is decorated by C{comparable} support greater-than comparisons. rZrYrNrzrrrrtest_greaterThanz ComparableTests.test_greaterThancCsF|tdtdk|tdtdk|tdtdkdS)z| Instances of a class that is decorated by C{comparable} support greater-than-or-equal comparisons. rYrZrr~Nrzrrrrtest_greaterThanOrEqualz'ComparableTests.test_greaterThanOrEqualcCs0|tdtdk|tdtdkdS)zp Instances of a class that is decorated by C{comparable} support less-than comparisons. rr~rZNrzrrrr test_lessThanrzComparableTests.test_lessThancCsF|tdtdk|tdtdk|tdtdkdS)zy Instances of a class that is decorated by C{comparable} support less-than-or-equal comparisons. r~rrZNrzrrrrtest_lessThanOrEqualrz$ComparableTests.test_lessThanOrEqualN) r5r6r7r8r{r}rrrrrrrrrys   ryc@rx)Python3ComparableTestsz; Python 3-specific functionality of C{comparable}. cC|tdttdS)z Instances of a class that is decorated by C{comparable} support returning C{NotImplemented} from C{__eq__} if it is returned by the underlying C{__cmp__} call. rYN)rrp__eq__r3rurrrrtest_notImplementedEqualsz0Python3ComparableTests.test_notImplementedEqualscCr)z Instances of a class that is decorated by C{comparable} support returning C{NotImplemented} from C{__ne__} if it is returned by the underlying C{__cmp__} call. rYN)rrp__ne__r3rurrrrtest_notImplementedNotEqualsrz3Python3ComparableTests.test_notImplementedNotEqualscCr)z Instances of a class that is decorated by C{comparable} support returning C{NotImplemented} from C{__gt__} if it is returned by the underlying C{__cmp__} call. rYN)rrp__gt__r3rurrrrtest_notImplementedGreaterThan$rz5Python3ComparableTests.test_notImplementedGreaterThancCr)z Instances of a class that is decorated by C{comparable} support returning C{NotImplemented} from C{__lt__} if it is returned by the underlying C{__cmp__} call. rYN)rrp__lt__r3rurrrrtest_notImplementedLessThan,rz2Python3ComparableTests.test_notImplementedLessThancCr)z Instances of a class that is decorated by C{comparable} support returning C{NotImplemented} from C{__ge__} if it is returned by the underlying C{__cmp__} call. rYN)rrp__ge__r3rurrrr$test_notImplementedGreaterThanEquals4rz;Python3ComparableTests.test_notImplementedGreaterThanEqualscCr)z Instances of a class that is decorated by C{comparable} support returning C{NotImplemented} from C{__le__} if it is returned by the underlying C{__cmp__} call. rYN)rrp__le__r3rurrrr!test_notImplementedLessThanEquals<rz8Python3ComparableTests.test_notImplementedLessThanEqualsN) r5r6r7r8rrrrrrrrrrrs rc@(eZdZdZddZddZddZdS) CmpTestszA L{cmp} should behave like the built-in Python 2 C{cmp}. cCs>|tddd|tddd|tdgdgddS)z5 L{cmp} returns 0 for equal objects. r=rrYNrrrrrr test_equalsJszCmpTests.test_equalscC(|tddd|tddddS)zS L{cmp} returns 1 if its first argument is bigger than its second. rrYzaNrrrrrrRzCmpTests.test_greaterThancCr)zU L{cmp} returns -1 if its first argument is smaller than its second. g?gffffff@rdNrrrrrrYrzCmpTests.test_lessThanN)r5r6r7r8rrrrrrrrEs  rc@r) StringTestsz8 Compatibility functions and types for strings. cCs$|t|||t|tdS)z Raise an exception indicating a failed test if the output of C{nativeString(original)} is unequal to the expected string, or is not a native string. N)rr assertIsInstancer)roriginalexpectedrrrassertNativeStringfszStringTests.assertNativeStringcC|ttddS)zj C{nativeString} raises a C{UnicodeError} if input bytes are not ASCII decodable. NrK UnicodeErrorr rrrrtest_nonASCIIBytesToStringoz&StringTests.test_nonASCIIBytesToStringcCr)zk C{nativeString} raises a C{UnicodeError} if input Unicode is not ASCII encodable. uሴNrrrrrtest_nonASCIIUnicodeToStringvrz(StringTests.test_nonASCIIUnicodeToStringcCs|dddS)z C{nativeString} converts bytes to the native string format, assuming an ASCII encoding if applicable. shellohelloNrrrrrtest_bytesToString}zStringTests.test_bytesToStringcC|dddS)z C{nativeString} converts unicode to the native string format, assuming an ASCII encoding if applicable. zGood dayNrrrrrtest_unicodeToStringrz StringTests.test_unicodeToStringcCr)zJ C{nativeString} leaves native strings as native strings. zHello!Nrrrrrtest_stringToStringszStringTests.test_stringToStringcCr)zu C{nativeString} raises a C{TypeError} if given an object that is not a string of some sort. rYN)rK TypeErrorr rrrrtest_unexpectedTyperzStringTests.test_unexpectedTypeN) r5r6r7r8rrrrrrrrrrrras  rc@r) NetworkStringTestsz% Tests for L{networkString}. cCs|dtddS)zu L{networkString} returns a C{unicode} object passed to it encoded into a C{bytes} instance. sfoorXN)rrrrrrtest_strzNetworkStringTests.test_strcCr)z L{networkString} raises L{UnicodeError} if passed a C{unicode} instance containing characters not encodable in ASCII. u☃N)rKrrrrrrtest_unicodeOutOfRangerz)NetworkStringTests.test_unicodeOutOfRangecCs"|ttt|ttddS)z L{networkString} raises L{TypeError} if passed a non-string object or the wrong type of string object. sbytesN)rKrrr3rrrrtest_nonStringsz!NetworkStringTests.test_nonStringN)r5r6r7r8rrrrrrrrs  rc@r9) ReraiseTestszH L{reraise} re-raises exceptions on both Python 2 and Python 3. c CszddWntyt\}}}Ynwzt|dWn-tyKt\}}}||t||||t |dt |dYdSw| ddS)z Calling L{reraise} with an exception instance and a traceback of L{None} re-raises it with a new traceback. rYrNrThe exception was not raised.) BaseExceptionrkexc_inforrZeroDivisionErrorassertIsassertNotEqual traceback format_tbfailrtyprrtbtyp2value2tb2rrrtest_reraiseWithNones       z!ReraiseTests.test_reraiseWithNonec CszddWntyt\}}}Ynwzt||Wn-tyKt\}}}||t||||t|dt|dYdSw| ddS)z Calling L{reraise} with an exception instance and a traceback re-raises the exception with the given traceback. rYrrrN) rrkrrrrrrrrrrrrtest_reraiseWithTracebacks     &z&ReraiseTests.test_reraiseWithTracebackN)r5r6r7r8rrrrrrrs rc@s8eZdZdZddZddZddZdd Zd d Zd S) Python3BytesTestszB Tests for L{iterbytes}, L{intToBytes}, L{lazyByteSlice}. cCs$d}tt|}||gddS)z When L{iterbytes} is called with a bytestring, the returned object can be iterated over, resulting in the individual bytes of the bytestring. sabcd)rbcrN)rCr r)rinputresultrrrtest_iterations z Python3BytesTests.test_iterationcCs|tdddS)z When L{intToBytes} is called with an integer, the result is an ASCII-encoded string representation of the number. s213N)rr rrrrtest_intToBytesrz!Python3BytesTests.test_intToBytescCsd}|tt||dS)zv L{lazyByteSlice} called with some bytes returns a semantically equal version of these bytes. 123XYZNrr"r rdatarrrtest_lazyByteSliceNoOffsetsz,Python3BytesTests.test_lazyByteSliceNoOffsetcCs&d}|tt|d|dddS)z L{lazyByteSlice} called with some bytes and an offset returns a semantically equal version of these bytes starting at the given offset. rrZNrrrrrtest_lazyByteSliceOffsets"z*Python3BytesTests.test_lazyByteSliceOffsetcCs(d}|tt|dd|dddS)z L{lazyByteSlice} called with some bytes, an offset and a length returns a semantically equal version of these bytes starting at the given offset, up to the given length. rrZr~Nrrrrr!test_lazyByteSliceOffsetAndLength s$z3Python3BytesTests.test_lazyByteSliceOffsetAndLengthN) r5r6r7r8rrrrrrrrrrs  rc@s&eZdZdZeedddZdS)BytesEnvironTestsz$ Tests for L{BytesEnviron}. z+Environment vars are always str on Windows.cCsPt}t}|D]\}}|t||t|q |t|tgdS)zz The output of L{BytesEnviron} should always be a L{dict} with L{bytes} values and L{bytes} keys. N)rrAitemsrBtyperrCr")rrtypeskeyvalrrrtest_alwaysBytess z"BytesEnvironTests.test_alwaysBytesN)r5r6r7r8rr isWindowsrrrrrrs rc@r9)GetAsyncParamTestsz= Tests for L{twisted.python.compat._get_async_param} cCsx|tddd|tddd|tdddiddid|tdddiddid|ttdddidS)z L{twisted.python.compat._get_async_param} uses isAsync by default, or deprecated async keyword argument if isAsync is None. F)isAsyncTrNasyncr)rrrKrrrrrtest_get_async_param-s z'GetAsyncParamTests.test_get_async_paramcCsF|td ddiddid|j|jgd}||ddddS) z L{twisted.python.compat._get_async_param} raises a deprecation warning if async keyword argument is passed. rNrF)offendingFunctionsrmessagez:'async' keyword argument is deprecated, please use isAsyncr)rr flushWarnings test_get_async_param_deprecation)rcurrentWarningsrrrr8s z3GetAsyncParamTests.test_get_async_param_deprecationN)r5r6r7r8rrrrrrr(s r)*r8r.rrkrunittestrtwisted.python.compatrrrrrrr r r r r rrtwisted.python.filepathrtwisted.python.runtimertwisted.trial.unittestrrrr:rPrirpryrrrrrrrrrrrrs2 <  6-0:68.0