Your IP : 216.73.216.213
�
j f�I � @ s� d d l Z d d l Z d d l Z d d l Z d d l Z d d l Z d d l m Z d d l Z d d l Z Gd d � d e j
� Z Gd d � d e j � Z
e d k r� e j � n d S)� N)�supportc @ sP e Z d Z d Z d Z d Z Z Z Z d Z d Z
d Z Z d d � Z
d S)�TestableTestProgramNT� � c C s d S)N� )�selfr r �A/opt/alt/python34/lib64/python3.4/unittest/test/test_discovery.py�__init__ s z TestableTestProgram.__init__)�__name__�
__module__� __qualname__�module�exitZ defaultTest�failfast�
catchbreak�buffer� verbosityZprogNameZ
testRunner�
testLoaderr r r r r r
s
r c @ s e Z d Z d d � Z d d � Z d d � Z d d � Z d d
� Z d d � Z d
d � Z d d � Z
d d � Z d d � Z d d � Z
d d � Z d d � Z d d � Z d d � Z d d � Z d! d" � Z d# d$ � Z d% d& � Z d' d( � Z d) S)*�
TestDiscoveryc C se t j � } d | _ | j d � } | j | d � t s> d S| j t � � | j d � Wd QXd S)Nz/fooz/foo/bar/baz.pyzbar.bazz /bar/baz.py)�unittest�
TestLoader�_top_level_dirZ_get_name_from_path� assertEqual� __debug__� assertRaises�AssertionError)r �loader�namer r r �test_get_name_from_path s z%TestDiscovery.test_get_name_from_pathc
s� t j � } t j � � f d d � } t j j � � f d d � } t j j � � f d d � } d d d d
d d d
g d d g g � � f d d � t _ | j | � d d � } | t j _ | j | � d d � } | t j _ | j | � d d � | _ d d � | _ t j j
d � } | | _ t | j
| d � � } d d � d! D� } | j d d � d" D� � | j | | � d S)#Nc s
� t _ d S)N)�os�listdirr )�original_listdirr r �restore_listdir- s z6TestDiscovery.test_find_tests.<locals>.restore_listdirc s � t j _ d S)N)r �path�isfiler )�original_isfiler r �restore_isfile0 s z5TestDiscovery.test_find_tests.<locals>.restore_isfilec s � t j _ d S)N)r r# �isdirr )�original_isdirr r �
restore_isdir3 s z4TestDiscovery.test_find_tests.<locals>.restore_isdirztest2.pyztest1.pyz
not_a_test.pyZtest_dirztest.fooztest-not-a-module.py� another_dirztest4.pyztest3.pyc s
� j d � S)Nr )�pop)r# )�
path_listsr r �<lambda>9 s z/TestDiscovery.test_find_tests.<locals>.<lambda>c S s
| j d � S)N�dir)�endswith)r# r r r r' < s z,TestDiscovery.test_find_tests.<locals>.isdirc S s | j d � o d | k S)Nr. r* )r/ )r# r r r r$ A s z-TestDiscovery.test_find_tests.<locals>.isfilec S s | d S)Nz moduler )r# r r r r- G s c S s | d S)Nz testsr )r
r r r r- H s z/fooztest*.pyc S s g | ] } | d � q S)z
module testsr )�.0r r r r �
<listcomp>P s z1TestDiscovery.test_find_tests.<locals>.<listcomp>�test1�test2c S s g | ] } d | d � q S)z test_dir.%sz
module testsr )r0 r r r r r1 R s �test3�test4)r2 r3 )r4 r5 )r r r r r# r$ r' �
addCleanup�_get_module_from_name�loadTestsFromModule�abspathr �list� _find_tests�extendr )
r r r" r& r) r' r$ Z top_level�suiteZexpectedr )r( r% r! r, r �test_find_tests) s8
z TestDiscovery.test_find_testsc s� t j � } t j � � f d d � } t j j � � f d d � } t j j � � f d d � } d d d g � � g g g g � � f d
d � t _ � j | � d d � t j _ � j | � � f d
d � t j _ � j | � Gd d � d t � � � f d d � | _ � f d d � } | | _
d | _ t | j
d d � � } � j | d d d g � � j � j d d g � � j � j | d d d f g � d S)Nc s
� t _ d S)N)r r r )r! r r r" Z s zCTestDiscovery.test_find_tests_with_package.<locals>.restore_listdirc s � t j _ d S)N)r r# r$ r )r% r r r& ] s zBTestDiscovery.test_find_tests_with_package.<locals>.restore_isfilec s � t j _ d S)N)r r# r' r )r( r r r) ` s zATestDiscovery.test_find_tests_with_package.<locals>.restore_isdirZ a_directory�test_directoryZtest_directory2c s
� j d � S)Nr )r+ )r# )r, r r r- e s z<TestDiscovery.test_find_tests_with_package.<locals>.<lambda>c S s d S)NTr )r# r r r r- h s c s t j j | � � k S)N)r r# �basename)r# )� directoriesr r r- k s c @ s4 e Z d Z g Z g Z d d � Z d d � Z d S)z:TestDiscovery.test_find_tests_with_package.<locals>.Modulec sS | � _ � j j | � t j j | � d k rO � f d d � } | � _ n d S)Nr? c s � j j | | | f � d S)N�
load_tests)�load_tests_args�append)r �tests�pattern)r r r rB v s zWTestDiscovery.test_find_tests_with_package.<locals>.Module.__init__.<locals>.load_tests)r# �pathsrD r r@ rB )r r# rB r )r r r r s
zCTestDiscovery.test_find_tests_with_package.<locals>.Module.__init__c S s | j | j k S)N)r# )r �otherr r r �__eq__{ s zATestDiscovery.test_find_tests_with_package.<locals>.Module.__eq__N)r
r r rG rC r rI r r r r �Modulen s rJ c s
� | � S)Nr )r )rJ r r r- ~ s c s# | r � j d � � n | j d S)Nz+use_load_tests should be False for packagesz
module tests)ZfailureExceptionr# )r
Zuse_load_tests)r r r r8 s zGTestDiscovery.test_find_tests_with_package.<locals>.loadTestsFromModulez/fooztest*rB z
module tests)r r r r r# r$ r' r6 �objectr7 r8 r r: r; r rG rC )r r r" r& r) r8 r= r )rJ rA r( r% r! r, r r � test_find_tests_with_packageV s4
z*TestDiscovery.test_find_tests_with_packagec
s� t j � } t j j � t j j � � f d d � } d d � t j _ | j | � t j d d � � � f d d � } | j | � t j j t j j d � � } | j
t � � | j d d d �Wd QX| j
| j | � | j | t j � d
d � t j _ d d � t j _ � f d d
� } | j | � g � d � f d d � } | | _ t | _ | j d d d � } t j j d � } t j j d � } | j
| d � | j
| j | � | j
� | d f g � | j | t j � d S)Nc s � t j _ d S)N)r r# r$ r )r% r r r& � s z3TestDiscovery.test_discover.<locals>.restore_isfilec S s d S)NFr )r# r r r r- � s z-TestDiscovery.test_discover.<locals>.<lambda>c s � t j d d � <d S)N)�sysr# r )�
orig_sys_pathr r � restore_path� s z1TestDiscovery.test_discover.<locals>.restore_pathz/fooz/foo/bar�
top_level_dirc S s d S)NTr )r# r r r r- � s c S s d S)NTr )r# r r r r- � s c s � t j _ d S)N)r r# r' r )r( r r r) � s z2TestDiscovery.test_discover.<locals>.restore_isdirc s � j | | f � d g S)NrE )rD )� start_dirrF � namespace)�_find_tests_argsr r r; � s z0TestDiscovery.test_discover.<locals>._find_testsz /foo/bar/bazrF z ['tests'])r r r r# r$ r' r6 rM r9 �normpathr � ImportError�discoverr r �assertInr; �str�
suiteClass)
r r r& rO � full_pathr) r; r= rP rQ r )rS rN r( r% r �
test_discover� s:
zTestDiscovery.test_discoverc sx t j � � f d d � t _ t j j � d d � t j _ t j d d � � � � � f d d � } | j | � d S)Nc s � g S)Nr )�_)�fakefiler r r- � s z8TestDiscovery.setup_import_issue_tests.<locals>.<lambda>c S s d S)NTr )r\ r r r r- � s c s, � t j _ � t _ � t j d d � <d S)N)r r# r$ r rM r )r$ r rN r r �restore� s z7TestDiscovery.setup_import_issue_tests.<locals>.restore)r r r# r$ rM r6 )r r] r^ r )r] r$ r rN r �setup_import_issue_tests� s z&TestDiscovery.setup_import_issue_testsc C s� t j � } | j d � | j d � } | j t j � t j � | j | j
� d � t t | � d � d } | j t
� � | j � Wd QXx4 t t j d � D] } t j t j | | � � q� Wd S)Nztest_this_does_not_exist.py�.r r )r r r_ rV rW r �getcwdrM r# r �countTestCasesr: r rU Ztest_this_does_not_exist�range�pickle�HIGHEST_PROTOCOL�loads�dumps)r r r= �test�protor r r �.test_discover_with_modules_that_fail_to_import� s
z<TestDiscovery.test_discover_with_modules_that_fail_to_importc C s� t j � } d d � } | | _ | j d � | j d � } | j | j � d � t j � } | j | � | j t | j
� d � x4 t t j
d � D] } t j t j | | � � q� Wd S)Nc S s t j d � � d S)NZ skipperoo)r ZSkipTest)r r r r r7 � s zeTestDiscovery.test_discover_with_module_that_raises_SkipTest_on_import.<locals>._get_module_from_nameztest_skip_dummy.pyr` r )r r r7 r_ rV r rb Z
TestResultZrun�lenZskippedrc rd re rf rg )r r r7 r= �resultri r r r �8test_discover_with_module_that_raises_SkipTest_on_import� s
zFTestDiscovery.test_discover_with_module_that_raises_SkipTest_on_importc C s� t � } g } | j | _ | j d d g � | j | g g � g | d d � <| j d d d d g � | j | d d g g � d S)N� somethingrV �foo�bar)r rD �
_do_discovery� parseArgsr )r �program�argsr r r �$test_command_line_handling_parseArgs� s z2TestDiscovery.test_command_line_handling_parseArgsc C s� t � } g } | j | _ | j d g � | j | g g � | j | j d � | j | j d � | j | j d � | j | j d � d S)Nrn r F)
r rD rq rr r r �assertIsr r r )r rs rt r r r �.test_command_line_handling_discover_by_default s z<TestDiscovery.test_command_line_handling_discover_by_defaultc C s� t � } g } | j | _ | j d d d d d d g � | j | g g � | j | j d � | j | j d � | j | j d � | j | j d � d S)Nrn z-vz-bz-cz-f� T)
r rD rq rr r r rv r r r )r rs rt r r r �;test_command_line_handling_discover_by_default_with_options s zITestDiscovery.test_command_line_handling_discover_by_default_with_optionsc C s� t � } d | _ t j � �8 } | j t � � } | j d d d d g � Wd QXWd QX| j | j j d � | j
d | j � � d S)NZoneZtwoZthreeZfourrx zusage:)rx ) r r r Zcaptured_stderrr �
SystemExitrq r � exceptionrt rW �getvalue)r rs �stderr�cmr r r �:test_command_line_handling_do_discovery_too_many_arguments s %zHTestDiscovery.test_command_line_handling_do_discovery_too_many_argumentsc C sh t j t j � } | j � Gd d � d t � } | � | _ | j d g � | j | j d g � d S)Nc @ s" e Z d Z g Z d d � Z d S)zYTestDiscovery.test_command_line_handling_do_discovery_uses_default_loader.<locals>.Loaderc S s | j j | | | f � d S)NrE )rt rD )r rQ rF rP r r r rV + s zbTestDiscovery.test_command_line_handling_do_discovery_uses_default_loader.<locals>.Loader.discoverN)r
r r rt rV r r r r �Loader) s r� z-vr` �test*.py)r` r� N) rK �__new__r � TestProgramZ_initArgParsersr rq r rt )r rs r� r r r �;test_command_line_handling_do_discovery_uses_default_loader% s
zITestDiscovery.test_command_line_handling_do_discovery_uses_default_loaderc C s� t � } Gd d � d t � } | j d g d | �| j | j d � | j | j d � | j | j d g � g | _ t � } | j d g d | �| j | j d � | j | j d g � g | _ t � } | j g d | �| j | j d � | j | j d g � g | _ t � } | j d g d | �| j | j d � | j | j d g � g | _ t � } | j d d
g d | �| j | j d � | j | j d g � g | _ t � } | j d d
d g d | �| j | j d � | j | j d g � g | _ t � } | j d d g d | �| j | j d � | j | j d g � g | _ t � } | j d
d g d | �| j | j d � | j | j d g � g | _ t � } | j d d g d | �| j | j d � | j | j d g � | j | j � | j | j � g | _ t � } | j d d
d d d d d g d | �| j | j d � | j | j d g � | j | j d � | j
| j � | j
| j � d S)Nc @ s" e Z d Z g Z d d � Z d S)zRTestDiscovery.test_command_line_handling_do_discovery_calls_loader.<locals>.Loaderc S s | j j | | | f � d S)NrE )rt rD )r rQ rF rP r r r rV 8 s z[TestDiscovery.test_command_line_handling_do_discovery_calls_loader.<locals>.Loader.discoverN)r
r r rt rV r r r r r� 6 s r� z-vrx rE r` �test*.pyz --verbose�fish�eggs�hamz-sz-tz-pz-fz-c)r` r� N)r` r� N)r` r� N)r� r� N)r� r� N)r� r� r� )r� r� N)r` r� r� )r` r� N)r� r� N) r rK rq r r rh rt Z assertFalser r �
assertTrue)r rs r� r r r �4test_command_line_handling_do_discovery_calls_loader3 sr !zBTestDiscovery.test_command_line_handling_do_discovery_calls_loaderc s� Gd d � d t � } | t j d <t j j d � � t j � t j j � t j j � � � � � f d d � } | j | � d d � } d d � } d
d � } | t _ | t j _ | t j _ � S) Nc @ s e Z d Z d Z d S)z0TestDiscovery.setup_module_clash.<locals>.Modulez
bar/foo.pyN)r
r r �__file__r r r r rJ ~ s rJ ro c sQ � t _ � t j _ � t j _ t j d =� t j k rM t j j � � n d S)Nro )r r r# r$ r' rM �modules�remover )rZ r( r% r! r r �cleanup� s
z1TestDiscovery.setup_module_clash.<locals>.cleanupc S s d g S)Nzfoo.pyr )r\ r r r r � s z1TestDiscovery.setup_module_clash.<locals>.listdirc S s d S)NTr )r\ r r r r$ � s z0TestDiscovery.setup_module_clash.<locals>.isfilec S s d S)NTr )r\ r r r r' � s z/TestDiscovery.setup_module_clash.<locals>.isdir)
rK rM r� r r# r9 r r$ r' r6 )r rJ r� r r$ r' r )rZ r( r% r! r �setup_module_clash} s
z TestDiscovery.setup_module_clashc C s� | j � } t j � } t j j d � } t j j d � } t j d | | f � } | j t d | | j
d d d d �| j t j d | � d S) Nrp ro zZ'foo' module incorrectly imported from %r. Expected %r. Is this module globally installed?z^%s$rQ rF zfoo.pyr )
r� r r r r# r9 �re�escapeZassertRaisesRegexrU rV r rM )r rZ r �mod_dir� expected_dir�msgr r r �test_detect_module_clash� s
z&TestDiscovery.test_detect_module_clashc s� | j � } t j j � t j j d � � t j j d � � � f d d � } | j | � � � f d d � } | t j _ t j � } | j d d d d � d S)
Nrp ro c s � t j _ d S)N)r r# �realpathr )�original_realpathr r r� � s z5TestDiscovery.test_module_symlink_ok.<locals>.cleanupc s2 | t j j � d � k r. t j j � d � S| S)Nzfoo.py)r r# �join)r# )r� r� r r r� � s z6TestDiscovery.test_module_symlink_ok.<locals>.realpathrQ rF zfoo.py) r� r r# r� r9 r6 r r rV )r rZ r� r� r r )r� r� r� r �test_module_symlink_ok� s
z$TestDiscovery.test_module_symlink_okc s� t j � } � g � t j j t j j t j j � � � d � _ d � � � f d d � } | | _ | j
d � } � j � j � � j | j
� � d S)NFc s d � _ � j | � � � S)NT)�wasRunr )rQ rF rR )� expectedPathr rE r r r; � s zBTestDiscovery.test_discovery_from_dotted_path.<locals>._find_testsz
unittest.test)r r r r# r9 �dirnamerh r� r� r; rV r� r Z_tests)r r r; r= r )r� r rE r � test_discovery_from_dotted_path� s $ z-TestDiscovery.test_discovery_from_dotted_pathc s� t j � } t j � d d � t _ t j j � t j j � d d � t j _ t j d d � � � � � � f d d � } | j | � | j t
� � } | j d � Wd QX| j t
| j � d � d S)Nc S s d g S)Nztest_this_does_not_exist.pyr )r\ r r r r- � s zOTestDiscovery.test_discovery_from_dotted_path_builtin_modules.<locals>.<lambda>c S s d S)NFr )r\ r r r r- � s c s8 � t j _ � t j _ � t _ � t j d d � <d S)N)r r# r$ r' r rM r )r' r$ r rN r r r^ � s zNTestDiscovery.test_discovery_from_dotted_path_builtin_modules.<locals>.restorerM z2Can not use builtin modules as dotted module names)r r r r r# r$ r' rM r6 r � TypeErrorrV r rX r{ )r r r^ r~ r )r' r$ r rN r �/test_discovery_from_dotted_path_builtin_modules� s
z=TestDiscovery.test_discovery_from_dotted_path_builtin_modulesc s� t j � } t � t j d � � d d g � _ t j d d d d d g � � _ � f d d � } � f d d � } | j | � | t _ g � d � f d
d � } | | _
t | _ | j
d � } | j | d d
g � d S)N�packagez/az/br �submodule_search_locationsc s � t j | <� S)N)rM r� )� packagenamert �kwargs)r� r r �_import� s
zLTestDiscovery.test_discovery_from_dotted_namespace_packages.<locals>._importc s
� t _ d S)N)�builtins�
__import__r )� orig_importr r r� � s zLTestDiscovery.test_discovery_from_dotted_namespace_packages.<locals>.cleanupc s � j | | f � d | g S)Nz%s/tests)rD )rQ rF rR )rS r r r; � s zPTestDiscovery.test_discovery_from_dotted_namespace_packages.<locals>._find_testsz/a/testsz/b/tests)r r r� �types�
ModuleType�__path__�SimpleNamespace�__spec__r6 r� r; r: rY rV r )r r r� r� r; r= r )rS r� r� r �-test_discovery_from_dotted_namespace_packages� s"
z;TestDiscovery.test_discovery_from_dotted_namespace_packagesc
s� t j � } t j d � � t � � f d d � } � f d d � } | j | � | t _ | j t � � } | j d � Wd QX| j
t | j � d j
� � � d S)Nr� c s � t j | <� S)N)rM r� )r� rt r� )r� r r r� s
z>TestDiscovery.test_discovery_failed_discovery.<locals>._importc s
� t _ d S)N)r� r� r )r� r r r� s z>TestDiscovery.test_discovery_failed_discovery.<locals>.cleanupz$don't know how to discover from {!r})r r r� r� r� r6 r� r r� rV r rX r{ �format)r r r� r� r~ r )r� r� r � test_discovery_failed_discovery s
z-TestDiscovery.test_discovery_failed_discoveryN)r
r r r r> rL r[ r_ rj rm ru rw ry r r� r� r� r� r� r� r� r� r� r r r r r s(
- ? .
J r �__main__)r r� rM r� rd r� rh r r Z
unittest.testr� r ZTestCaser r
�mainr r r r �<module> s � �