File: //usr/lib/python3/dist-packages/trac/__pycache__/test.cpython-310.pyc
o
�k�`�S � @ s d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z d dl
Z
d dl Z d dl Z d dl
Z
d dlZzd dlmZ W n eyO dZY nw e�d�Zd dlZd dlmZ d dlmZmZmZ d dl m Z m Z d dl m Z d dl!m"Z" d d l#m$Z$ d d
l%m&Z& d d l'm(Z(m)Z) d d l*m+Z+m,Z,m-Z- d d
l.m/Z/ d dl0m1Z1 d+dd�Z2G dd� de3�Z4dd� Z5G dd� dej6�Z7G dd� dej8�Z9dd� Z:G d d � d e �Z;d d � Z<d d!� Z=d"d#� Z>d$a?d%d&� Z@eAd'k�rd(e
jBv r�e
jB�Cd(� d)a?ejDd&d*� dS dS ),� N)�Locale�en_US)�
Configuration)�ComponentManager�
ComponentMeta� TracError)�DatabaseManager�parse_connection_uri)� Environment)�PermissionCache)� load_workflow_config_snippet)� translation)�time_now�utc)� _RequestArgs�Request�arg_list_to_args)�Chrome��Session� c
O s� t | t�s| f} i }dd� }| D ]'}t�||�}t�|�D ]\}}||v r&q t |tj�s2t|dd�r6|||< q qt d| |�} | |� }
|�
� D ]
\} } t |
| | � qF|
S )a�
Simple factory for dummy classes that can be used as replacement for the
real implementation in tests.
Base classes for the mock can be specified using the first parameter, which
must be either a tuple of class objects or a single class object. If the
bases parameter is omitted, the base class of the mock will be object.
So to create a mock that is derived from the builtin dict type, you can do:
>>> mock = Mock(dict)
>>> mock['foo'] = 'bar'
>>> mock['foo']
'bar'
Attributes of the class are provided by any additional keyword parameters.
>>> mock = Mock(foo='bar')
>>> mock.foo
'bar'
Objects produces by this function have the special feature of not requiring
the 'self' parameter on methods, because you should keep data at the scope
of the test function. So you can just do:
>>> mock = Mock(add=lambda x,y: x+y)
>>> mock.add(1, 1)
2
To access attributes from the mock object from inside a lambda function,
just access the mock itself:
>>> mock = Mock(dict, do=lambda x: 'going to the %s' % mock[x])
>>> mock['foo'] = 'bar'
>>> mock.do('foo')
'going to the bar'
Because assignments or other types of statements don't work in lambda
functions, assigning to a local variable from a mock function requires some
extra work:
>>> myvar = [None]
>>> mock = Mock(set=lambda x: myvar.__setitem__(0, x))
>>> mock.set(1)
>>> myvar[0]
1
c _ s t ��N)�NotImplementedError)�self�args�kwargsr r �+/usr/lib/python3/dist-packages/trac/test.py�dummyfno � zMock.<locals>.dummyfn�__isabstractmethod__F�Mock) �
isinstance�tuple�types�
MethodType�inspect�
getmembers�abc�abstractproperty�getattr�type�items�setattr)
�bases�initargs�kw�attrsr �base�fn�name�attr�cls�mock�k�vr r r r 9 s(
0
��� r c @ sB e Zd ZdZdZ d dd�ZeZd dd�Z d
d d
�ZeZ dS )�MockPermzWFake permission class. Necessary as Mock can not be used with operator
overloading.� NFc C � dS )NTr )r �action�realm_or_resource�id�versionr r r �has_permission� � zMockPerm.has_permissionc C s | S r r )r r= r> r? r r r �__call__� r zMockPerm.__call__c C � d S r r )r r< r= r> r? �messager r r �require� rA zMockPerm.require)NFF)FF)NFFN)
�__name__�
__module__� __qualname__�__doc__�usernamer@ � __contains__rB rE �assert_permissionr r r r r9 � s
�
�r9 c s� �� d���du rd�t� �nt�����fdd��d�v r.�fdd��d D �� t� ��n t� ����fd d
��� d i ��� D �� �fd d��D �� ��� d
�� dd��� dd��� dd��� dd��� dd��� d���� dd��� dd��� d d �d � }|D ]}t|| t �r�t
|| d �||< q�g � i �t � � �d0��� fd d!� }t
t||�}� |_�|_�|_|j�� fd"d#��fd$d#��fd%d#�t��j�fd&d#��fd'd#��fd(d#��fd)d#��fd*d#��fd+d#�d,d#� d-d#� d.d#� d/�
� |S )1aO Request object for testing. Keyword arguments populate an
`environ` dictionary and the callbacks.
If `authname` is specified in a keyword arguments a `PermissionCache`
object is created, otherwise if `authname` is not specified or is
`None` a `MockPerm` object is used and the `authname` is set to
'anonymous'.
The following keyword arguments are commonly used:
:keyword args: dictionary of request arguments
:keyword authname: the name of the authenticated user, or 'anonymous'
:keyword method: the HTTP request method
:keyword path_info: the request path inside the application
Additionally `cookie`, `format`, `language`, `lc_time`, `locale`,
`remote_addr`, `remote_user`, `script_name`, `server_name`, `server_port`
and `tz` can be specified as keyword arguments.
:since: 1.0.11
�authnameN� anonymousc sN t | t�r tt| ��S t | tj�rt| �S t | ttf�r%� fdd�| D �S | S )Nc s g | ]}� |��qS r r )�.0r8 ��convertr r �
<listcomp>� s z0MockRequest.<locals>.convert.<locals>.<listcomp>)r! �bool�str�int�numbers�Real�listr" )�valrP r r rQ � s
z MockRequest.<locals>.convert�arg_listc s g | ]
\}}|� |�f�qS r r �rO r7 r8 rP r r rR � s z MockRequest.<locals>.<listcomp>c 3 s � | ] \}}|� |�fV qd S r r r[ rP r r � <genexpr>� s � �z MockRequest.<locals>.<genexpr>r c s$ g | ]}� � |�D ]}||f�q qS r )�getlist)rO r3 �value�r r r rR � s
���http�languager: �cookie� path_info�/�method�GET� remote_addrz 127.0.0.1� remote_user� script_namez /trac.cgi� server_name� localhost� server_port�80) z
trac.base_urlzwsgi.url_scheme�HTTP_ACCEPT_LANGUAGE� HTTP_COOKIE� PATH_INFO�REQUEST_METHOD� REMOTE_ADDR� REMOTE_USER� SCRIPT_NAME� SERVER_NAME� SERVER_PORTzutf-8c s �� | � � �t|�� �jS r )�append�update�dict�write)�status�headers�exc_info)� headers_sent�
response_sent� status_sentr r �start_response� s
z#MockRequest.<locals>.start_responsec � � S r r ��req)rZ r r �<lambda>� � z MockRequest.<locals>.<lambda>c r� r r r� r_ r r r� � r� c r� r r r� )rM r r r� � r� c s � � dd�S )N�
form_tokenr ��getr� �r r r r� � � c � � � dt�S )N�lc_time)r� � locale_enr� r� r r r� � r� c s
� � d�S )N�localer� r� r� r r r� � �
c r� r r r� )�permr r r� � r� c s
t � | �S r r r� )�envr r r� � r� c r� )N�tz)r� r r� r� r r r� � r� c S r; )NFr r� r r r r� � r� c S rC r r r� r r r r� � r� c S s g S r r r� r r r r� � r� )
rZ r rM �chromer� r� r� r� �sessionr� �
use_xsendfile�xsendfile_header�configurable_headersr )r� r9 r r r rx r+ �abs_hrefr! �bytesrT �io�BytesIOr r r� r~ r � callbacksr �prepare_request)r� r �environ�keyr� r� r )
rZ r rM rQ r� r~ r r� r r� r � MockRequest� sl
�
�
�
�r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd
S ) � TestSetupzh
Test suite decorator that allows a fixture to be setup for a complete
suite of test cases.
c C r; )z4Sets up the fixture, and sets self.fixture if neededNr �r r r r �setUp� rA zTestSetup.setUpc C r; )zTears down the fixtureNr r� r r r �tearDown rA zTestSetup.tearDownc C sN | � � t| d�r| jD ]
}t|d�r|�| j� q tj�| |� | �� |S )zuSetup the fixture (self.setUp), call .setFixture on all the tests,
and tear down the fixture (self.tearDown).�fixture�
setFixture) r� �hasattr�_testsr� r� �unittest� TestSuite�runr� )r �result�testr r r r� s
�z
TestSetup.runc O sZ | � � t| d�r| jD ]
}t|d�r|�| j� q tjj| g|�R i |�� | �� dS )zLPython 2.7 / unittest2 compatibility - there must be a better
way...r� r� N) r� r� r� r� r� r� r� � _wrapped_runr� )r r r r� r r r r� s
� zTestSetup._wrapped_runN)rF rG rH rI r� r� r� r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )�
TestCaseSetupc C s
|| _ d S r )r� )r r� r r r r� s
zTestCaseSetup.setFixtureN)rF rG rH r� r r r r r� s r� c C s� t j�d�} | rHt| �\}}|dkr(|�di ��d�s(| d| v r"dndd 7 } n&|dkrG|d d
krG|�di ��d �sG| d| v rBdndd 7 } nd}d
| } t�d| � | S )N�TRAC_TEST_DB_URI�postgres�params�schema�?�&zschema=tractest�sqlite�pathz:memory:� synchronouszsynchronous=offz %s::memory:ztrac.db.%s_backend)�osr� r� r � importlib�
import_module)�dburi�scheme�db_propr r r � get_dburi% s ���r� c @ sd e Zd ZdZdZdZ ddd�Zdd� Zed d
� �Z dd d �Z
d
d� Z dd� Z dd� Z
dd� ZdS )�EnvironmentStubz5A stub of the trac.env.Environment class for testing.FTNc C s� |durt |ttf�std��|dur t |ttf�s td��t�| � d| _d| _ddl}|| _ | j du rAt
j � t
j � |j
��| _ t
j �t
j �| j ��| _ td�| _t| jd� | j�ddd� | j�dd d
� |duru| j�d d d
� n| j�d dd� |p�dD ]}| �|�} | j�d | d� q�|p�dD ]}| �|�} | j�d | d
� q�| j�ddd� |p�g D ]}
| jj|
� q�| �� t� | _| j�d dd� | j�dd| j� |s�| �|� | j�ddd� t�t� dS )a� Construct a new Environment stub object.
:param default_data: If True, populate the database with some
defaults.
:param enable: A list of component classes or name globs to
activate in the stub environment.
:param disable: A list of component classes or name globs to
deactivate in the stub environment.
:param path: The location of the environment in the file system.
No files or directories are created when specifying
this parameter.
:param destroying: If True, the database will not be reset. This is
useful for cases when the object is being
constructed in order to call `destroy_db`.
:param config: A list of (section, key, value) configuration
tuples.
Nz(Keyword argument "enable" must be a listz)Keyword argument "disable" must be a listr zbasic-workflow.ini�logging� log_level�DEBUG�log_type�none�
componentsztrac.*�disabledztracopt.versioncontrol.*�enabledr �trac�permission_policiesz/DefaultPermissionPolicy, LegacyAttachmentPolicyz trac.db.*�database�base_urlzhttp://example.org/trac.cgi) r! rX r" � TypeErrorr �__init__�
_old_registry�_old_componentsr� r� r� �abspath�dirname�__file__�normpath�normcaser �configr �set�_component_name� setup_logr� r� �reset_dbr
�activater� ) r � default_data�enable�disabler� �
destroyingr� r� �
name_or_class�
config_key�itemr r r r� > sP
�
�
zEnvironmentStub.__init__c C s� t �d�| _| j�� }tjj�|�}| j�|� t j j
}| jj s8|t j �}t �
| j�}|�|� | j�|� d S t| jj �dkrSt| jj d |�rS| jj d �� d S td��)Nz trac.test� r z!Logger has unexpected handler(s).)r� � getLogger�logr� �upperr� �
LOG_LEVEL_MAPr� �setLevel�handlers�BufferingHandler�sys�maxsize� Formatter�
log_format� setFormatter�
addHandler�lenr! �flushr )r �level� level_as_int� handler_cls� log_handler� formatterr r r r� � s
�zEnvironmentStub.setup_logc C s dd� | j jd jD �S )z*Returns a list of tuples (level, message).c S s g | ] }|j |�� f�qS r )� levelname�
getMessage)rO �recordr r r rR � s �z0EnvironmentStub.log_messages.<locals>.<listcomp>r )r� r� �bufferr� r r r � log_messages� s �z EnvironmentStub.log_messagesc C s� ddl m} g }t| �}z|�� }W n
t| jjfy Y nw ||jkr*|�� }n| � � |s=|�
� | j dkr=|� � |rC|�
� |�|j� dS )z�Remove all data from Trac tables, keeping the tables themselves.
:param default_data: after clean-up, initialize with default data
:return: True upon success
r )�
db_defaultzsqlite::memory:N)r� r� r �get_database_versionr �db_exc�
DatabaseError�
db_version� reset_tables�
destroy_db�init_dbr� �shutdown�insert_default_data�set_database_version)r r� r� �tables�dbmr r r r r� � s$ �
zEnvironmentStub.reset_dbc C s0 z t | ��� W dS t| jjfy Y dS w )zDestroy the database.N)r r r r r r� r r r r � s
�zEnvironmentStub.destroy_dbc C s6 | � � | �� t| jj� | jdur| �� dS dS )z�Performs a complete environment reset in a robust way.
The database is reset, then the connections are shut down, and
finally all environment files are removed from the disk.
N)r� r �rmtreer� r� r� �restore_component_registryr� r r r �reset_db_and_disk� s
�z!EnvironmentStub.reset_db_and_diskc C s� | j �=}|D ]1}t|�dkr|\}}}d}n|\}}}}|d||tt� �f� |d|||f� |d|||f� qW d � dS 1 sCw Y dS )a� Insert a tuple representing a user session to the
`session` and `session_attributes` tables.
The tuple can be length 3 with entries username, name and
email, in which case an authenticated user is assumed. The
tuple can also be length 4, with the last entry specifying
`1` for an authenticated user or `0` for an unauthenticated
user.
� r� z'INSERT INTO session VALUES (%s, %s, %s)z6INSERT INTO session_attribute VALUES (%s,%s,'name',%s)z7INSERT INTO session_attribute VALUES (%s,%s,'email',%s)N)�db_transactionr� rU r )r �users�db�rowrJ r3 �email�
authenticatedr r r � insert_users� s"
����"�z EnvironmentStub.insert_usersc C s | � |��d�r
dS t�| |�S )Nz __main__.T)r� �
startswithr
�is_component_enabled)r r5 r r r r � s z$EnvironmentStub.is_component_enabled)FNNNFNr )rF rG rH rI �required�abstractr� r� �propertyr� r� r r
r r r r r r r� 8 s
�J
r� c C s t jdd�} tj�tj�| ��S )zZCreate a temp directory with prefix `trac-tempenv`
and return the directory name.
z
trac-testdir-)�prefix)�tempfile�mkdtempr� r� r� �realpath)�tmpdirr r r r � s r c C sZ t jdkrdnd}dgt jd �t j� D ]}t j�|| | �}t j�|�r*| S qdS )zRLocates a binary on the path.
Returns the fully-qualified path, or None.
�ntz.exer: �.�PATHN)r� r3 r� �split�pathsepr� �join�exists)r2 � exec_suffix�p�fr r r �locate� s �r* c sB d� fdd� � t jdkrt| t�rt| t�� �} tj| � d� d S )Nr� c s� | t jt jfvr � |d }t|t�rEt �|�j}t �||dB � z| |� W d S tyD |dkr3� t �
d� � | |||d � Y d S w � )Nr� i� �
g�������?) r� �unlink�remover! �PermissionError�stat�st_mode�chmod� Exception�time�sleep)�functionr� �excinfo�retry�e�mode��onerrorr r r;
s
�zrmtree.<locals>.onerrorr r: )r� ) r� r3 r! r� rT r� �getfilesystemencoding�shutilr )r� r r: r r s r Tc C s dd l } dd l} dd l} dd l} dd l} dd l} dd l} dd l} dd l} dd l } dd l
} dd l } dd l } dd l
} dd l}dd l}dd l}dd l}tj�d�rPdat�� }|�| j�� � |�| jj� � � |�| j j� � � |�| j j� � � |�| j j� � � |�| j j� � � |�| j!j� � � |�| j"j� � � |�| j#j� � � |�| j$j� � � |�| j%j� � � |�| j%j&j� � � |�| j'j� � � |�| j(j� � � |�|j)j� � � |�|j!j� � � |�|j%j*j� � � |�|j%j+j� � � |�t,�-t.j/t0 �� t�r|�| j�1� � |S )Nr �SKIP_FUNCTIONAL_TESTSF)2�
trac.tests�trac.admin.tests�
trac.db.tests�trac.mimeview.tests�trac.notification.tests�trac.search.tests�trac.ticket.tests�trac.timeline.tests�trac.upgrades.tests�trac.util.tests�trac.versioncontrol.tests� trac.versioncontrol.web_ui.tests�trac.web.tests�trac.wiki.tests�tracopt.perm.tests�tracopt.ticket.tests� tracopt.versioncontrol.git.tests� tracopt.versioncontrol.svn.testsr� r� r� �INCLUDE_FUNCTIONAL_TESTSr� r� �addTest�tests�
basicSuite�admin�
test_suiter �mimeview� notification�search�ticket�timeline�upgrades�util�versioncontrol�web_ui�web�wikir� �git�svn�doctest� DocTestSuiter� �modulesrF �functionalSuite)r� �tracopt�suiter r r rV * sV rV �__main__z--skip-functional-testsF)� defaultTest)r )Er' rd r� r% r� r� �logging.handlersrV r� r= r� r r3 r# r� �babelr � ImportErrorr� �parse�trac.logr� � trac.configr � trac.corer r r � trac.db.apir r �trac.envr
� trac.permr � trac.ticket.default_workflowr � trac.utilr
�trac.util.datefmtr r � trac.web.apir r r �trac.web.chromer �trac.web.sessionr r �objectr9 r� r� r� �TestCaser� r� r� r r* r rQ rV rF �argvr- �mainr r r r �<module> sh �
Hd% ?
1
�