← Back
Editing: test_loader.cpython-38.pyc
U W[�e � @ s� d Z ddlmZmZ ddlZddlZddlZddlm Z ddl m Z m Z ddl mZ ddlmZmZmZ ddlmZ dd lmZ dd lmZ dd lmZ d d � ZG dd� dej�Z G dd� dej �Z G dd� dej �Z G dd� de �Z G dd� dej �Z!dS )z" Tests for loading tests by name. � )�absolute_import�divisionN)�md5)�util�filepath)�packages)�runner�reporter�unittest)� ITestCase)� _iterateTests)� getModule)�_PY3c C s$ g }t | �D ]}|�|�� � q |S )zI Return the id of each test within the given test suite or case. )r �append�id)Ztests�names�test� r �@/usr/lib/python3/dist-packages/twisted/trial/test/test_loader.py� testNames s r c @ sx e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z e r\dZ e e_ e e_ e e_ e e _ dd� Z dd� Zdd� ZdS )� FinderTestsz4 Tests for L{runner.TestLoader.findByName}. c C s t j�| � t�� | _d S �N)r � PackageTest�setUpr � TestLoader�loader��selfr r r r * s zFinderTests.setUpc C s t j�| � d S r )r r �tearDownr r r r r . s zFinderTests.tearDownc C s$ | j �d�}dd l}| �||� d S )N�twistedr )r � findByNamer � assertEqual�r �sample1�sample2r r r �test_findPackage1 s z FinderTests.test_findPackagec C s( | j �d�}ddlm} | �||� d S )N�twisted.trial.test.sampler ��sample)r r �twisted.trial.testr( r! r" r r r �test_findModule6 s zFinderTests.test_findModulec C s4 t �td�}| j�|�}ddlm} | �||� d S )N� sample.pyr r' )r �sibpath�__file__r r r) r( r! )r �pathr# r$ r r r � test_findFile; s zFinderTests.test_findFilec C s* | j �d�}ddlm} | �|j|� d S )N�!twisted.trial.test.sample.FooTestr r' )r r r) r( r! �FooTest)r r# r( r r r �test_findObjectA s zFinderTests.test_findObjectzNot relevant on Python 3c C s | � t| jjd� d S )Nz twisted.trial.test.nonexistent)� assertRaises�AttributeErrorr r r r r r �test_findNonModuleQ s �z FinderTests.test_findNonModulec C s | � t| jjd� d S )NZ nonextant)r3 � ValueErrorr r r r r r �test_findNonPackageV s �z FinderTests.test_findNonPackagec C s" t �td�}| �t| jj|� d S )Nznonexistent.py)r r, r- r3 r6 r r )r r. r r r �test_findNonFile[ s z FinderTests.test_findNonFileN)�__name__� __module__� __qualname__�__doc__r r r% r* r/ r2 r Z _Py3SkipMsg�skipr5 r7 r8 r r r r r & s r c @ sX e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dS )� FileTestsz/ Tests for L{runner.filenameToModule}. c C s$ | � ttjd�}| �t|�d� dS )zo L{runner.filenameToModule} raises a C{ValueError} when a non-existing file is passed. �itz'it' doesn't existN)r3 r6 r �filenameToModuler! �str)r �errr r r � test_notFilee s zFileTests.test_notFilec C s. t �t�td��}ddlm} | �||� dS )z� If the file in question is a module on the Python path, then it should properly import and return that module. r+ r r' N)r r@ r r, r- r) r( r! r" r r r �test_moduleInPathn s zFileTests.test_moduleInPathc C sz | � tjjdd� | �| j� t�tj � | j dd��}| �| j � ddl m} | �tj �|j�d tj �|j�d � dS )a If passed the path to a file containing the implementation of a module within a package which is not on the import path, L{runner.filenameToModule} returns a module object loosely resembling the module defined by that file anyway. � test_sampleN� goodpackageztest_sample.pyr )rE )� addCleanup�sys�modules�pop� mangleSysPath�oldPathr r@ �osr. �join�parent�newPathrF rE r! �splitextr- r" r r r �test_moduleNotInPathx s � �z FileTests.test_moduleNotInPathc C s. t �tj�| jd��}ddl}| �||� dS )z� If the file in question is a package on the Python path, then it should properly import and return that package. rF r N)r r@ rM r. rN rO rF r! �r Zpackage1rF r r r �test_packageInPath� s �z FileTests.test_packageInPathc C sr | � tjjd� | �| j� t�tj � | j d��}| �| j � ddl }| �tj �|j�d tj �|j�d � dS )z� If passed the path to a directory which represents a package which is not on the import path, L{runner.filenameToModule} returns a module object loosely resembling the package defined by that directory anyway. �__init__rF r N)rG rH rI rJ rK rL r r@ rM r. rN rO rP rF r! rQ r- rS r r r �test_packageNotInPath� s � �z FileTests.test_packageNotInPathc C sH t �| j��d�}|�� | �ttj|j �}| � t |�d|j f � dS )z� L{runner.filenameToModule} raises a C{ValueError} when the name of an empty directory is passed that isn't considered a valid Python package because it doesn't contain a C{__init__.py} file. ZemptyDirectoryz %r is not a package directoryN) r �FilePathrO �child�createDirectoryr3 r6 r r@ r. r! rA )r ZemptyDirrB r r r �test_directoryNotPackage� s � �z"FileTests.test_directoryNotPackagec C s2 t �| j��d�}|�d� | �ttj|j � dS )zn L{runner.filenameToModule} raises a C{SyntaxError} when a non-Python file is passed. Z notpythons This isn't pythonN) r rW rO rX � setContentr3 � SyntaxErrorr r@ r. )r �filenamer r r �test_filenameNotPython� s �z FileTests.test_filenameNotPythonc C sT t �| j��d�}|�tj�d�� z t � |j �}| � |j |j � W 5 |�� X dS )zX The C{__file__} attribute of the module should match the package name. zgoodpackage.py�utf8N)r rW rO rX r[ r � testModule�encode�remover r@ r. r! r- )r r] �moduler r r �test_filenameMatchesPackage� s z%FileTests.test_filenameMatchesPackagec C s� t �| j��d�}|�� |�d��d� zHt�|j �}| � |j � d�� t�|j t j j �}| � |j � d�� W 5 |�� X dS )z� Test loader against a filesystem directory containing an empty C{__init__.py} file. It should handle 'path' and 'path/' the same way. Z goodDirectory� __init__.py� N)r rW rO rX rY r[ rb r r@ r. � assertTruer9 �endswithrM �sep)r ZgoodDirrc r r r �test_directory� s zFileTests.test_directoryN) r9 r: r; r< rC rD rR rT rV rZ r^ rd rj r r r r r> a s r> c @ sL e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd d � Z dd� Z dd� Z dd� Z dd� Z dd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Z d4d5� Z d6d7� Z d8d9� Z d:d;� Z d<d=� Z!d>d?� Z"d@dA� Z#dBdC� Z$dDdE� Z%dFdG� Z&e'�rHdHZ(e(e_)e(e _)e(e_)e(e_)[(dIS )J� LoaderTestsz( Tests for L{trial.TestLoader}. c C s t �� | _tj�| � d S r )r r r r �SysPathManglingTestr r r r r r � s zLoaderTests.setUpc s� ddl m} | j�|j�}| �dddgdd� |jD �� dddg}tt|t d���� � fd d �| j_ | j�|j�}| �|d d� |jD �� d S ) Nr r' Ztest_aZtest_bZtest_cc S s g | ] }|j �qS r ��_testMethodName��.0r r r r � <listcomp>� s z.LoaderTests.test_sortCases.<locals>.<listcomp>� c s � � | �� d�S )N���)�getZshortDescription)�x�ZsortDictr r �<lambda>� rf z,LoaderTests.test_sortCases.<locals>.<lambda>c S s g | ] }|j �qS r rm ro r r r rq � s ) r) r( r � loadClassZ AlphabetTestr! �_tests�dict�zip�range�sorter)r r( �suiteZnewOrderr rv r �test_sortCases� s � �zLoaderTests.test_sortCasesc C s> ddl m} | j�|jj�}| �d|�� � | �d|j� d S )Nr r' � �test_foo) r) r( r � loadMethodr1 r� r! �countTestCasesrn �r r( r~ r r r �test_loadMethod s zLoaderTests.test_loadMethodc C sR ddl m} | j�|jj�}t�� }|�|� | � |j d� | � t |j �d� d S )Nr )� erroneousr� ) r) r� r r� ZTestRegularFailZ test_failr � TestResult�runr! �testsRun�len�failures)r r� r~ �resultr r r �test_loadFailingMethod s z"LoaderTests.test_loadFailingMethodc C sB | j �d�}t�� }|�|� | �|jd� | �t|j�d� dS )z{ Loading a test that fails and getting the result of it ends up with one test ran and one failure. z6twisted.trial.test.erroneous.TestRegularFail.test_failr� N) r � loadByNamer r� r� r! r� r� r� )r r~ r� r r r �test_loadFailure s � z LoaderTests.test_loadFailurec C sZ ddl m} | �t| jj|� | �t| jj|j� | �t| jjd� | �t| jjd� d S �Nr r' �string)ZfooZbar)r) r( r3 � TypeErrorr r� r1 �r r( r r r �test_loadNonMethod s � �z LoaderTests.test_loadNonMethodc C sL ddl m} | jj|jj|jdddgd�}| �d|�� � | �d|j� dS ) z� A decorated test method for which the decorator has failed to set the method's __name__ correctly is loaded and its name in the class scope discovered. r r' r( �DecorationTest�test_badDecorator�rO ZqualNamer� N) r) r( r � loadAnythingr� r� r! r� rn r� r r r �test_loadBadDecorator* s �z!LoaderTests.test_loadBadDecoratorc C sL ddl m} | jj|jj|jdddgd�}| �d|�� � | �d|j� dS ) z� A decorated test method for which the decorator has set the method's __name__ correctly is loaded and the only name by which it goes is used. r r' r( r� �test_goodDecoratorr� r� N) r) r( r r� r� r� r! r� rn r� r r r �test_loadGoodDecorator9 s �z"LoaderTests.test_loadGoodDecoratorc C sL ddl m} | jj|jj|jdddgd�}| �d|�� � | �d|j� dS ) z� Load a decorated method which has been copied to a new name inside the class. Thus its __name__ and its key in the class's __dict__ no longer match. r r' r( r� �test_renamedDecoratorr� r� N) r) r( r r� r� r� r! r� rn r� r r r �test_loadRenamedDecoratorG s �z%LoaderTests.test_loadRenamedDecoratorc C sJ ddl m} | j�|j�}| �d|�� � | �ddgdd� |jD �� d S )Nr r' � Ztest_barr� c S s g | ] }|j �qS r rm ro r r r rq [ s z.LoaderTests.test_loadClass.<locals>.<listcomp>)r) r( r rx r1 r! r� ry r� r r r �test_loadClassV s �zLoaderTests.test_loadClassc C s\ ddl m} | �t| jj|� | �t| jj|jj� | �t| jjd� | �t| jjd� d S r� )r) r( r3 r� r rx r1 r� r� r r r �test_loadNonClass^ s � �z LoaderTests.test_loadNonClassc C s$ ddl m} | �t| jj|j� d S �Nr r' )r) r( r3 r6 r rx ZNotATestr� r r r �test_loadNonTestCaseh s �z LoaderTests.test_loadNonTestCasec C s, ddl m} | j�|�}| �d|�� � d S )Nr r' � )r) r( r � loadModuler! r� r� r r r �test_loadModulen s zLoaderTests.test_loadModulec C s^ ddl m} | �t| jj|j� | �t| jj|jj� | �t| jjd� | �t| jjd� d S r� )r) r( r3 r� r r� r1 r� r� r r r �test_loadNonModulet s � � �z LoaderTests.test_loadNonModulec C s( dd l }| j�|�}| �d|�� � d S �Nr � �rF r � loadPackager! r� �r rF r~ r r r �test_loadPackage s z LoaderTests.test_loadPackagec C s^ ddl m} | �t| jj|j� | �t| jj|jj� | �t| jjd� | �t| jjd� d S r� )r) r( r3 r� r r� r1 r� r� r r r �test_loadNonPackage� s � � �z LoaderTests.test_loadNonPackagec C s" ddl m} | �t| jj|� d S r� )r) r( r3 r� r r� r� r r r �test_loadModuleAsPackage� s z$LoaderTests.test_loadModuleAsPackagec C s, dd l }| jj|dd�}| �d|�� � d S �Nr T�Zrecurse� r� r� r r r �test_loadPackageRecursive� s z%LoaderTests.test_loadPackageRecursivec C s: ddl m} | j�|�}| �|j|jd jd jj� d S r� ) r) r( r r� r! r9 ry � __class__r: r� r r r �test_loadAnythingOnModule� s �z%LoaderTests.test_loadAnythingOnModulec C s. ddl m} | j�|j�}| �d|�� � d S )Nr r' r� )r) r( r r� r1 r! r� r� r r r �test_loadAnythingOnClass� s z$LoaderTests.test_loadAnythingOnClassc C s0 ddl m} | j�|jj�}| �d|�� � d S )Nr r' r� )r) r( r r� r1 r� r! r� r� r r r �test_loadAnythingOnMethod� s z%LoaderTests.test_loadAnythingOnMethodc C s< dd l }| j�|�}| �t|| jj�� | �d|�� � d S r� �rF r r� rg � isinstanceZ suiteFactoryr! r� r� r r r �test_loadAnythingOnPackage� s z&LoaderTests.test_loadAnythingOnPackagec C s@ dd l }| jj|dd�}| �t|| jj�� | �d|�� � d S r� r� r� r r r �#test_loadAnythingOnPackageRecursive� s z/LoaderTests.test_loadAnythingOnPackageRecursivec C s | � t| jjd� d S )NrF )r3 r� r r� r r r r �test_loadAnythingOnString� s �z%LoaderTests.test_loadAnythingOnStringc C sx dd l }| jj|dd�}t�� }|�|� | �d|�� � | �dt|j �� dd� |j D �}|� � | �|dd g� d S ) Nr Tr� Fr� c S s g | ]\}}|� � �qS r )r )rp r �errorr r r rq � s z1LoaderTests.test_importErrors.<locals>.<listcomp>zpackage.test_bad_modulezpackage.test_import_module) �packager r� r �Reporterr� r! � wasSuccessfulr� �errors�sort)r r� r~ r� r� r r r �test_importErrors� s �z LoaderTests.test_importErrorsc C sB G dd� dt j�}| j�|�}t�� }|�|� | �|�� � dS )z� L{TestLoader.loadClass} returns a suite with each test method represented by a different instances of the L{TestCase} they are defined on. c @ s e Zd Zdd� Zdd� ZdS )z>LoaderTests.test_differentInstances.<locals>.DistinctInstancesc S s d| _ d S )NZtest1Run)�firstr r r r �test_1� s zELoaderTests.test_differentInstances.<locals>.DistinctInstances.test_1c S s | � t| d�� d S )Nr� )Z assertFalse�hasattrr r r r �test_2� s zELoaderTests.test_differentInstances.<locals>.DistinctInstances.test_2N)r9 r: r; r� r� r r r r �DistinctInstances� s r� N) �pyunitZTestCaser rx r r� r� rg r� )r r� r~ r� r r r �test_differentInstances� s z#LoaderTests.test_differentInstancesc C s@ ddl m} | j�|�}| �d|�� � | �dt|dd�� dS )zp Check that C{test_suite} is used when present and other L{TestCase}s are not included. r )�mockcustomsuite� MyCustomSuite�nameN)r) r� r r� r! r� �getattr)r r� r~ r r r � test_loadModuleWith_test_suite� s z*LoaderTests.test_loadModuleWith_test_suitec C s@ ddl m} | j�|�}| �d|�� � | �dt|dd�� dS )zo Check that C{testSuite} is used when present and other L{TestCase}s are not included. r ��mockcustomsuite2r� r� N)r) r� r r� r! r� r� )r r� r~ r r r � test_loadModuleWith_testSuite� s z)LoaderTests.test_loadModuleWith_testSuitec C s0 ddl m} | j�|�}| �dt|dd�� dS )z� Check that if C{testSuite} and C{test_suite} are both present in a module then C{testSuite} gets priority. r )�mockcustomsuite3� testSuiter� N)r) r� r r� r! r� )r r� r~ r r r � test_loadModuleWithBothCustom� s z)LoaderTests.test_loadModuleWithBothCustomc C sF dd� }ddl m} ||j |_}z| �t| jj|� W 5 ||_X dS )zz Make sure that any C{AttributeError}s raised by C{testSuite} are not swallowed by L{TestLoader}. c S s t d��d S )Nzshould be reraised)r4 r r r r r� s zBLoaderTests.test_customLoadRaisesAttributeError.<locals>.testSuiter r� N)r) r� r� r3 r4 r r� )r r� r� Zoriginalr r r �#test_customLoadRaisesAttributeError s � �z/LoaderTests.test_customLoadRaisesAttributeErrorc C s0 t |�}t |�}|�� |�� | �||� d S r )r r� r! )r Ztest1Ztest2Znames1Znames2r r r �assertSuitesEqual s z LoaderTests.assertSuitesEqualc C s4 d}| j �||gd�}| j �|d�}| �||� dS )z@ Check that loadByNames ignores duplicate names �twisted.trial.test.test_logTN)r � loadByNamesr� r� )r rc �suite1�suite2r r r �test_loadByNamesDuplicate s z%LoaderTests.test_loadByNamesDuplicatec C sH ddddddg}| j �|�}t�t| j j|��}| �t|�t|�� dS )zX L{TestLoader.loadByNames} preserves the order of tests provided to it. �$inheritancepackage.test_x.A.test_foor& rF r� r0 �package.test_moduleN)r r� r � TestSuite�mapr� r! r �r rI r� r� r r r � test_loadByNamesPreservesOrder( s � z*LoaderTests.test_loadByNamesPreservesOrderc C s8 ddg}| j �|�}t�t| j j|��}| �||� dS )zM Check that loadByNames loads all the names that it is given rF r� N�r r� r r� r� r� r� r� r r r �test_loadDifferentNames8 s z#LoaderTests.test_loadDifferentNamesc C s8 ddg}| j �|�}t�t| j j|��}| �||� dS )zu Check that test methods names which are inherited from are all loaded rather than just one. r� z$inheritancepackage.test_x.B.test_fooNr� )r �methodsr� r� r r r �test_loadInheritedMethodsA s � z%LoaderTests.test_loadInheritedMethodszNot possible on Python 3.N)*r9 r: r; r< r r r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r Z_msgr= r r r r rk � sV rk c @ s e Zd Zdd� ZdS )�ZipLoadingTestsc C sN ddl m} t�| � || j| jd � | jd7 _| �| j| jg � d S )Nr )�zipitz.zip)Z twisted.python.test.test_zippathr� rk r rO rK rL )r r� r r r r \ s zZipLoadingTests.setUpN)r9 r: r; r r r r r r� [ s r� c @ s: e Zd Zdd� Zdd� Zejfdd�Zdd� Zd d � Z d S ) �PackageOrderingTestsc C sl t �� | _| �� | _tj�| jd�}t�|� t tj�|d�d�� � t j � | |� | �| j| jg � d S )N� uberpackagere �wb)r r r ZmktempZtopDirrM r. rN �makedirs�open�closer rl r rK rL )r rO r r r r g s zPackageOrderingTests.setUpc c s� t d�}g }|�� D ]$}|j�d�d �d�r|�|� qt||d�}|D ]�}|�d�}|dkrh|V qJg }|�� D ] } t � | �� �rt|�| � qtt||d�} | D ]R} g } | �� D ]$} | j�d�d �d�r�| �| � q�t| |d�}|D ] }|V q�q�qJdS )a� Right now, halfway by accident, trial sorts like this: 1. all modules are grouped together in one list and sorted. 2. within each module, the classes are grouped together in one list and sorted. 3. finally within each class, each test method is grouped together in a list and sorted. This attempts to return a sorted list of testable thingies following those rules, so that we can compare the behavior of loadPackage. The things that show as 'cases' are errors from modules which failed to import, and test methods. Let's gather all those together. r� �.rs Ztest_)�keyNr ) r Z walkModulesr� �split� startswithr �sorted�loadZiterAttributesr Z isTestCase)r r} �pkgZ testModulesr` Z sortedModulesZmodinforc Z testClassesZattribZ sortedClassesZclsinfoZ testMethods�attrZ sortedMethodsZmethinfor r r �_trialSortAlgorithmp s. z(PackageOrderingTests._trialSortAlgorithmc C s� ddl }|| j_| jj|dd�}t�|t�}tt|��}t| � |��}t t ||��D ]0\}\}}| � |j |�� d||j |�� f � qT| � t|�t|�� dS )zG Verify that packages are loaded in the correct order. r NTr� z #%d: %s != %s)r� r r} r� r Zdecorater �listr r� � enumerater{ r! r� r r� ) r r} r� r~ ZresultingTestsZmanifestZnumberZ manifestTestZ actualTestr r r �loadSortedPackages� s" � ��z'PackageOrderingTests.loadSortedPackagesc C s | � � d S r �r� r r r r � test_sortPackagesDefaultOrder� s z2PackageOrderingTests.test_sortPackagesDefaultOrderc C s dd� }| � |� d S )Nc S s t �| �}t|�d���� }|S )Nr_ )r r� r ra Z hexdigest)�s�n�dr r r � sillySorter� s zEPackageOrderingTests.test_sortPackagesSillyOrder.<locals>.sillySorterr� )r r� r r r �test_sortPackagesSillyOrder� s z0PackageOrderingTests.test_sortPackagesSillyOrderN) r9 r: r; r r� r r� r� r� r r r r r r� e s -r� )"r< Z __future__r r rM rH r r� Zhashlibr Ztwisted.pythonr r r) r Z twisted.trialr r Ztwisted.trial.itrialr Ztwisted.trial._asyncrunnerr Ztwisted.python.modulesr Ztwisted.python.compatr r r r rl r> rk r� r� r r r r �<module> s* ; q
Save File
Cancel