← Back
Editing: test_adbapi.cpython-38.pyc
U ��W[$f � @ s� d Z ddlmZ ddlZddlZddlmZmZ ddlmZm Z ddl m Z m Z m Z ddlmZ ddlmZ d ZG d d � d e�ZG d d � d e�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�ZG dd� de�Z d d � Z e ed e � � e ed e � � G d d!� d!e�Z G d"d#� d#ej �Z!G d$d%� d%ej �Z"G d&d'� d'e�Z#G d(d)� d)e�Z$G d*d+� d+e�Z%G d,d-� d-ej �Z&dS ).z& Tests for twisted.enterprise.adbapi. � )�unittestN)�ConnectionPool�ConnectionLost)� Connection� Transaction)�reactor�defer� interfaces)�Failure�� requireModulez% CREATE TABLE simple ( x integer ) c @ s� e Zd ZdZi Ze�ed�dkr$dZdd� Z dd� Z dd � Z d2d d �Z d d � Z dd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Z d,d-� Z d.d/� Z d0d1� Z dS )3�ADBAPITestBasez, Test the asynchronous DB-API code. N�5ADB-API requires threads, no way to test without themc C s& | � � | j| jd�| _| j�� dS )zR Set up the database and create a connection pool pointing at it. )Z cp_openfunN)�startDB�makePool�openfun�dbpool�start��self� r �:/usr/lib/python3/dist-packages/twisted/test/test_adbapi.py� extraSetUp% s zADBAPITestBase.extraSetUpc s4 � j �d�}|�� fdd�� |�� fdd�� |S )N�DROP TABLE simplec s � j �� S �N�r �close��resr r r �<lambda>0 � z)ADBAPITestBase.tearDown.<locals>.<lambda>c s � � � S r ��stopDBr r r r r 1 r �r � runOperation� addCallback�r �dr r r �tearDown. s zADBAPITestBase.tearDownc C s d| j |< d S )NT)�openfun_called�r �connr r r r 5 s zADBAPITestBase.openfunc C s$ |s| � | j� n| �|| j� d S r )� assertTruer) ZassertInr* r r r �checkOpenfunCalled9 s z!ADBAPITestBase.checkOpenfunCalledc s� � j �t�}� jrT|�� j� |�� j� |�� j� |�� j� |�� fdd�� |�� j � |�� j � |�� j � |�� j � |�� j � |�� j� |�� j� |�� j� |S )Nc s � � � S r )�flushLoggedErrorsr r r r r G r z*ADBAPITestBase.test_pool.<locals>.<lambda>)r r$ �simple_table_schema� test_failuresr% � _testPool_1_1� _testPool_1_2� _testPool_1_3� _testPool_1_4� _testPool_2� _testPool_3� _testPool_4� _testPool_5� _testPool_6� _testPool_7� _testPool_8� _testPool_9r& r r r � test_pool@ s zADBAPITestBase.test_poolc s, t �� jjd�}|�� fdd�dd� � |S )N�select * from NOTABLEc s � � d�S �Nz no exception�Zfailr r r r r U r z.ADBAPITestBase._testPool_1_1.<locals>.<lambda>c S s d S r r ��fr r r r V r �r � maybeDeferredr �runQuery� addCallbacks�r r r' r r r r1 S s �z ADBAPITestBase._testPool_1_1c s, t �� jjd�}|�� fdd�dd� � |S )Nzdeletexxx from NOTABLEc s � � d�S r? r@ r r r r r ] r z.ADBAPITestBase._testPool_1_2.<locals>.<lambda>c S s d S r r rA r r r r ^ r )r rD r r$ rF rG r r r r2 Z s ��z ADBAPITestBase._testPool_1_2c s. t �� jj� j�}|�� fdd�dd� � |S )Nc s � � d�S r? r@ r r r r r e r z.ADBAPITestBase._testPool_1_3.<locals>.<lambda>c S s d S r r rA r r r r f r )r rD r �runInteraction�bad_interactionrF rG r r r r3 b s ��z ADBAPITestBase._testPool_1_3c s. t �� jj� j�}|�� fdd�dd� � |S )Nc s � � d�S r? r@ r r r r r m r z.ADBAPITestBase._testPool_1_4.<locals>.<lambda>c S s d S r r rA r r r r n r )r rD r �runWithConnection�bad_withConnectionrF rG r r r r4 j s ��z ADBAPITestBase._testPool_1_4c s* d}� j �|�}� fdd�}|�|� |S )N�select count(1) from simplec s( � � t| d d �dkd� � �� d S )Nr zInteraction not rolled back)r, �intr- ��rowr r r �_checkv s z*ADBAPITestBase._testPool_2.<locals>._check�r rE r% �r r �sqlr' rP r r r r5 r s zADBAPITestBase._testPool_2c sl d}g }t � j�D ] }d| }|�� j�|�� qt�|�}� fdd�}|�|� � fdd�}|�|� |S )NrL z insert into simple(x) values(%d)c s d}� j �|�}|S )N� select x from simple order by x)r rE )r rS r' r r r �_select� s z+ADBAPITestBase._testPool_3.<locals>._selectc sZ � � t| �� jkd� t� j�D ]4}� � t| | �dkd� � � | | d |kd� q d S )NzWrong number of rows� �Wrong size rowr zValues not returned.)r, �len�num_iterations�range)Zrows�ir r r rP � s �z*ADBAPITestBase._testPool_3.<locals>._check)rZ rY �appendr r$ r Z gatherResultsr% )r r rS Zinsertsr[ r' rU rP r r r r6 } s zADBAPITestBase._testPool_3c s$ � j �� j�}|�� fdd�� |S )Nc s � � | d�S �N�done�� assertEqualr r r r r � r z,ADBAPITestBase._testPool_4.<locals>.<lambda>)r rH � interactionr% rG r r r r7 � s zADBAPITestBase._testPool_4c s$ � j �� j�}|�� fdd�� |S )Nc s � � | d�S r] r_ r r r r r � r z,ADBAPITestBase._testPool_5.<locals>.<lambda>)r rJ �withConnectionr% rG r r r r8 � s zADBAPITestBase._testPool_5c C s | j �| j�}|S r )r rJ �close_withConnectionrG r r r r9 � s zADBAPITestBase._testPool_6c sV g }t � j�D ] }d| }|�� j�|�� qtj|dd�}� fdd�}|�|� |S )Nz!select x from simple where x = %dT�ZfireOnOneErrbackc s4 t � j�D ]$}� �| | d d d |kd� q d S )NrV r zValue not returned)rZ rY r, )�resultr[ r r r rP � s z*ADBAPITestBase._testPool_7.<locals>._check)rZ rY r\ r rE r � DeferredListr% )r r �dsr[ rS �dlistrP r r r r: � s zADBAPITestBase._testPool_7c C s@ g }t | j�D ] }d| }|�| j�|�� qtj|dd�}|S )Nz delete from simple where x = %dTrd )rZ rY r\ r r$ r rf )r r rg r[ rS rh r r r r; � s zADBAPITestBase._testPool_8c s* d}� j �|�}� fdd�}|�|� |S )NrL c s( � � t| d d �dkd� � �� d S )Nr z)Didn't successfully delete table contents)r, rM � checkConnectrN r r r rP � s �z*ADBAPITestBase._testPool_9.<locals>._checkrQ rR r r r r< � s zADBAPITestBase._testPool_9c C s� | j �� }| �|� |�� }|�d� |�d� |�� }| �t|�d� | �t|d �d� | �|d d d� |�d� |�d� | �t|�� �d� |�� | j � |� dS )z/Check the connect/disconnect synchronous calls.z insert into simple(x) values(1)zselect x from simplerV r zdelete from simpleN) r �connectr- �cursor�executeZfetchallr` rX r � disconnect)r r+ �cursr r r r ri � s zADBAPITestBase.checkConnectc C s^ |� d� t| j�D ]4}|�� }| �t|�dkd� | �|d |kd� q| �|�� d� dS )NrT rV rW r �Value not returned.z Too many rowsr^ )rl rZ rY �fetchoner, rX Z assertIsNone)r � transactionr[ rO r r r ra � s zADBAPITestBase.interactionc C s | j r|�d� |�d� d S )Nz insert into simple(x) values(0)r> )� can_rollbackrl )r rq r r r rI � s z ADBAPITestBase.bad_interactionc C sf |� � }zN|�d� t| j�D ]4}|�� }| �t|�dkd� | �|d |kd� q W 5 |�� X dS )NrT rV rW r ro r^ )rk r rl rZ rY rp r, rX )r r+ rn r[ rO r r r rb � s z ADBAPITestBase.withConnectionc C s |� � d S r )r r* r r r rc s z#ADBAPITestBase.close_withConnectionc C s& |� � }z|�d� W 5 |�� X d S )Nr> )rk r rl )r r+ rn r r r rK s z!ADBAPITestBase.bad_withConnection)N) �__name__� __module__� __qualname__�__doc__r) r �IReactorThreadsr �skipr r( r r- r= r1 r2 r3 r4 r5 r6 r7 r8 r9 r: r; r< ri ra rI rb rc rK r r r r r s6 r c @ sd e Zd ZdZe�ed�dkr dZdd� Zdd� Z dd � Z d d � Z d d � Z dd� Z dd� Zdd� ZdS )�ReconnectTestBasez; Test the asynchronous DB-API code with reconnect. Nr c C sF | j dkrt�d��| �� | jdd| j d�| _| j�� | j�t�S )z� Skip the test if C{good_sql} is unavailable. Otherwise, set up the database, create a connection pool pointed at it, and set up a simple schema in it. Nz no good sql for reconnect testrV T)�cp_maxZ cp_reconnectZ cp_good_sql) �good_sqlr �SkipTestr r r r r$ r/ r r r r r s � z ReconnectTestBase.extraSetUpc s4 � j �d�}|�� fdd�� |�� fdd�� |S )Nr c s � j �� S r r r r r r r ' r z,ReconnectTestBase.tearDown.<locals>.<lambda>c s � � � S r r! r r r r r ( r r# r& r r r r( % s zReconnectTestBase.tearDownc C sP t �d �}|�| j� |�| j� | js4|�| j� |�| j� |�| j� |S r ) r Zsucceedr% � _testPool_1r5 �early_reconnectr6 r7 r8 r& r r r r= , s zReconnectTestBase.test_poolc s* d}� j �|�}� fdd�}|�|� |S )NrL c s � � t| d d �dkd� d S �Nr zTable not empty�r, rM rN r r r rP : s z-ReconnectTestBase._testPool_1.<locals>._checkrQ rR r r r r} 7 s z ReconnectTestBase._testPool_1c C s t | jj�� �d �� d S )Nr )�listr Z connections�valuesr )r r r r r r5 @ s z ReconnectTestBase._testPool_2c s0 d}t �� jj|�}|�� fdd�dd� � |S )NrL c s � � d�S r? r@ r r r r r H r z/ReconnectTestBase._testPool_3.<locals>.<lambda>c S s d S r r rA r r r r I r rC �r r rS r' r r r r6 E s �z ReconnectTestBase._testPool_3c s* d}� j �|�}� fdd�}|�|� |S )NrL c s � � t| d d �dkd� d S r r� rN r r r rP P s z-ReconnectTestBase._testPool_4.<locals>._checkrQ rR r r r r7 M s z ReconnectTestBase._testPool_4c s< � � � d}t�� jj|�}|�� fdd�� fdd�� |S )Nr> c s � � d�S r? r@ r r r r r Z r z/ReconnectTestBase._testPool_5.<locals>.<lambda>c s � � | �t��S r )� assertFalseZcheckr rA r r r r [ r )r. r rD r rE rF r� r r r r8 V s �z ReconnectTestBase._testPool_5)rs rt ru rv r rw r rx r r( r= r} r5 r6 r7 r8 r r r r ry s ry c @ sz e Zd ZdZdZdZdZdZdZdZ dZ dZ dZ dZ ejZdZdZdZdd� Zdd � Zd d � Zd d � Zdd� Zdd� ZdS )�DBTestConnectora> A class which knows how to test for the presence of and establish a connection to a relational database. To enable test cases which use a central, system database, you must create a database named DB_NAME with a user DB_USER and password DB_PASS with full access rights to database DB_NAME. NZ twisted_testT�2 c C s6 | � � | _t�| j� | �� s.t�d| j ��| �� S )Nz%s: Cannot access db) Zmktemp�DB_DIR�os�mkdir� can_connectr r| � TEST_PREFIXr r r r r �setUp} s zDBTestConnector.setUpc C s t � �dS )zXReturn true if this database is present on the system and can be used in a test.N��NotImplementedErrorr r r r r� � s zDBTestConnector.can_connectc C s dS )z+Take any steps needed to bring database up.Nr r r r r r � s zDBTestConnector.startDBc C s dS )z Bring database down, if needed.Nr r r r r r" � s zDBTestConnector.stopDBc K s( | � � \}}|�� }|�|� t||�S )z;Create a connection pool with additional keyword arguments.)� getPoolArgs�copy�updater )r Znewkw�args�kwr r r r � s zDBTestConnector.makePoolc C s t � �dS )z�Return a tuple (args, kw) of list and keyword arguments that need to be passed to ConnectionPool to create a connection to this database.Nr� r r r r r� � s zDBTestConnector.getPoolArgs)rs rt ru rv r� �DB_NAME�DB_USER�DB_PASSr� Znulls_ok�trailing_spaces_okrr r0 �escape_slashesr r{ r~ � can_clearrY r� r� r r" r r� r r r r r� ` s* r� c @ s4 e Zd ZdZdZdZdZdd� Zdd� Zd d � Z d S ) �SQLite3ConnectorzB Connector that uses the stdlib SQLite3 database support. ZSQLite3FrV c C s t d�d krdS dS d S )N�sqlite3FTr r r r r r� � s z SQLite3Connector.can_connectc C s2 t j�| j| j�| _t j�| j�r.t �| j� d S r �r� �path�joinr� r� �database�exists�unlinkr r r r r � s zSQLite3Connector.startDBc C s d}| j ddd�}||fS )N)r� rV F�r� rz Zcheck_same_thread�r� �r r� r� r r r r� � s �z SQLite3Connector.getPoolArgsN� rs rt ru rv r� r� rY r� r r� r r r r r� � s r� c @ s4 e Zd ZdZdZdZdZdd� Zdd� Zd d � Z d S ) �PySQLite2ConnectorzA Connector that uses pysqlite's SQLite database support. Z pysqlite2FrV c C s t d�d krdS dS d S )N�pysqlite2.dbapi2FTr r r r r r� � s z PySQLite2Connector.can_connectc C s2 t j�| j| j�| _t j�| j�r.t �| j� d S r r� r r r r r � s zPySQLite2Connector.startDBc C s d}| j ddd�}||fS )N)r� rV Fr� r� r� r r r r� � s �z PySQLite2Connector.getPoolArgsNr� r r r r r� � s r� c @ s e Zd ZdZdd� Zdd� ZdS )�PyPgSQLConnectorZPyPgSQLc C sX zddl m} W n Y dS X z$|j| j| j| jd�}|�� W dS Y dS X d S )Nr )�PgSQLF�r� �user�passwordT)ZpyPgSQLr� rj r� r� r� r )r r� r+ r r r r� � s �z PyPgSQLConnector.can_connectc C s d}| j | j| jdd�}||fS )N)z pyPgSQL.PgSQLr �r� r� r� Zcp_min�r� r� r� r� r r r r� � s �z PyPgSQLConnector.getPoolArgsN�rs rt ru r� r� r� r r r r r� � s r� c @ s e Zd ZdZdd� Zdd� ZdS )�PsycopgConnectorZPsycopgc C sT z dd l }W n Y dS X z$|j| j| j| jd�}|�� W dS Y dS X d S )Nr Fr� T)�psycopgrj r� r� r� r )r r� r+ r r r r� � s �z PsycopgConnector.can_connectc C s d}| j | j| jdd�}||fS )N)r� r r� r� r� r r r r� s �z PsycopgConnector.getPoolArgsNr� r r r r r� � s r� c @ s, e Zd ZdZdZdZdZdd� Zdd� ZdS )�MySQLConnectorZMySQLFc C sT z dd l }W n Y dS X z$|j| j| j| jd�}|�� W dS Y dS X d S )Nr F�Zdbr� ZpasswdT)�MySQLdbrj r� r� r� r )r r� r+ r r r r� s �zMySQLConnector.can_connectc C s d}| j | j| jd�}||fS )N)r� r� r� r� r r r r� " s zMySQLConnector.getPoolArgsN) rs rt ru r� r� rr r~ r� r� r r r r r� s r� c @ sD e Zd ZdZdZdZdZdZdZdd� Z dd� Z d d � Z d d � Z dS ) �FirebirdConnectorZFirebirdFN� c C s: t d�d krdS z| �� | �� W dS Y dS X d S )N� kinterbasdbFT)r r r" r r r r r� 4 s z FirebirdConnector.can_connectc C sf dd l }tj�| jtj�| _t�| jtj tj tj � d}|| j| j | j f; }|�|�}|�� d S )Nr z,create database "%s" user "%s" password "%s")r� r� r� r� r� r� r� �chmod�stat�S_IRWXU�S_IRWXG�S_IRWXOr� r� Zcreate_databaser )r r� rS r+ r r r r ? s zFirebirdConnector.startDBc C s d}| j d| j| jd�}||fS )N)r� � 127.0.0.1�r� Zhostr� r� r� r� r r r r� I s �z FirebirdConnector.getPoolArgsc C s, dd l }|j| jd| j| jd�}|�� d S )Nr r� r� )r� rj r� r� r� Z drop_database)r r� r+ r r r r" P s �zFirebirdConnector.stopDB) rs rt ru r� r0 r� r{ r� rY r� r r� r" r r r r r� ) s r� c s� t tttttg}i }|D ]`� � j| }G � fdd�d� | tj�}||_ t � d�rpd� � j � � dd� |g �|_ |||< q|�|� dS )aQ Make a test case for every db connector which can connect. @param base: Base class for test case. Additional base classes will be a DBConnector subclass and unittest.TestCase @param suffix: A suffix used to create test case names. Prefixes are defined in the DBConnector subclasses. c s e Zd Z� jZdS )z makeSQLTests.<locals>.testcaseN)rs rt ru r �Z connclassr r �testcaseh s r� ru �.r ���N)r� r� r� r� r� r� r� r �TestCasers �hasattrr� ru �splitr� )�base�suffix�globalsZ connectorsZtests�namer� r r� r � makeSQLTestsY s � � r� Z ADBAPITestsZReconnectTestsc @ s0 e Zd ZdZdZdZdd� Zdd� Zdd� Zd S ) �FakePoolz� A fake L{ConnectionPool} for tests. @ivar connectionFactory: factory for making connections returned by the C{connect} method. @type connectionFactory: any callable Tc C s || _ d S r ��connectionFactory)r r� r r r �__init__� s zFakePool.__init__c C s | � � S )zB Return an instance of C{self.connectionFactory}. r� r r r r rj � s zFakePool.connectc C s dS )z Do nothing. Nr )r � connectionr r r rm � s zFakePool.disconnectN) rs rt ru rv � reconnectZnoisyr� rj rm r r r r r� s r� c @ s e Zd ZdZdd� ZdS )�ConnectionTestsz, Tests for the L{Connection} class. c C sd G dd� dt �}t|�}t|�}| �t|j� | �t�}| �t |�d� | �|d j j d d� dS )z| If an error happens during rollback, L{ConnectionLost} is raised but the original error is logged. c @ s e Zd Zdd� ZdS )zIConnectionTests.test_rollbackErrorLogged.<locals>.ConnectionRollbackRaisec S s t d��d S �N�problem!�� RuntimeErrorr r r r �rollback� s zRConnectionTests.test_rollbackErrorLogged.<locals>.ConnectionRollbackRaise.rollbackN)rs rt ru r� r r r r �ConnectionRollbackRaise� s r� rV r r� N) �objectr� r Z assertRaisesr r� r. r� r` rX �valuer� )r r� �poolr� �errorsr r r �test_rollbackErrorLogged� s z(ConnectionTests.test_rollbackErrorLoggedN)rs rt ru rv r� r r r r r� � s r� c @ s e Zd ZdZdd� ZdS )�TransactionTestsz- Tests for the L{Transaction} class. c C sb G dd� dt �}td�}t||� �}|�� | �t�}| �t|�d� | �|d jj d d� dS )z� If the cursor creation raises an error in L{Transaction.reopen}, it reconnects but log the error occurred. c @ s e Zd ZdZdd� Zdd� ZdS )zNTransactionTests.test_reopenLogErrorIfReconnect.<locals>.ConnectionCursorRaiser c S s d S r r r r r r r� � s zXTransactionTests.test_reopenLogErrorIfReconnect.<locals>.ConnectionCursorRaise.reconnectc S s$ | j dkr | j d7 _ td��d S )Nr rV r� )�countr� r r r r rk � s zUTransactionTests.test_reopenLogErrorIfReconnect.<locals>.ConnectionCursorRaise.cursorN)rs rt ru r� r� rk r r r r �ConnectionCursorRaise� s r� NrV r r� ) r� r� r Zreopenr. r� r` rX r� r� )r r� r� rq r� r r r � test_reopenLogErrorIfReconnect� s z/TransactionTests.test_reopenLogErrorIfReconnectN)rs rt ru rv r� r r r r r� � s r� c @ s e Zd Zdd� ZdS )� NonThreadPoolc O s@ d}z|||�}W n t k r0 d}t� }Y nX |||� d S )NTF)� Exceptionr )r ZonResultrB �ar� Zsuccessre r r r �callInThreadWithCallback� s z&NonThreadPool.callInThreadWithCallbackN)rs rt ru r� r r r r r� � s r� c @ s e Zd ZdZe� Zdd� ZdS )�DummyConnectionPoolz' A testable L{ConnectionPool}; c C s t | _dS )z* Don't forward init call. N)r Z_reactorr r r r r� � s z DummyConnectionPool.__init__N)rs rt ru rv r� Z threadpoolr� r r r r r� � s r� c @ s0 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd S ) � EventReactora Partial L{IReactorCore} implementation with simple event-related methods. @ivar _running: A C{bool} indicating whether the reactor is pretending to have been started already or not. @ivar triggers: A C{list} of pending system event triggers. c C s || _ g | _d S r )�_running�triggers)r �runningr r r r� � s zEventReactor.__init__c C s | j r|� n| �dd|�S d S )N�after�startup)r� �addSystemEventTrigger)r Zfunctionr r r �callWhenRunning� s z EventReactor.callWhenRunningc C s |||f}| j �|� |S r )r� r\ )r ZphaseZeventZtrigger�handler r r r� s z"EventReactor.addSystemEventTriggerc C s | j �|� d S r )r� �remove)r r� r r r �removeSystemEventTrigger s z%EventReactor.removeSystemEventTriggerN)rs rt ru rv r� r� r� r� r r r r r� � s r� c @ s8 e Zd ZdZdd� Zdd� Zdd� Zdd � Zd d � Zd S ) �ConnectionPoolTestsz+ Unit tests for L{ConnectionPool}. c sT G dd� dt �}dd� }t� }||_|�|�}� �|t�}� fdd�}|�|� |S )z� If rollback fails, L{ConnectionPool.runWithConnection} raises the original exception and log the error of the rollback. c @ s e Zd Zdd� Zdd� ZdS )z]ConnectionPoolTests.test_runWithConnectionRaiseOriginalError.<locals>.ConnectionRollbackRaisec S s d S r r �r r� r r r r� s zfConnectionPoolTests.test_runWithConnectionRaiseOriginalError.<locals>.ConnectionRollbackRaise.__init__c S s t d��d S r� r� r r r r r� s zfConnectionPoolTests.test_runWithConnectionRaiseOriginalError.<locals>.ConnectionRollbackRaise.rollbackN�rs rt ru r� r� r r r r r� s r� c S s t d��d S �NZfoo�� ValueError)r� r r r �raisingFunction s zUConnectionPoolTests.test_runWithConnectionRaiseOriginalError.<locals>.raisingFunctionc s6 � � t�}� �t|�d� � �|d jjd d� d S �NrV r r� �r. r� r` rX r� r� �Zignoredr� r r r �cbFailed$ s zNConnectionPoolTests.test_runWithConnectionRaiseOriginalError.<locals>.cbFailed)r� r� r� rJ � assertFailurer� r% )r r� r� r� r' r� r r r �(test_runWithConnectionRaiseOriginalError s z<ConnectionPoolTests.test_runWithConnectionRaiseOriginalErrorc C sX G dd� dt �}t� }|�|� � | �t�}| �t|�d� | �|d jjd d� dS )z; L{ConnectionPool._close} logs exceptions. c @ s e Zd Zdd� ZdS )zDConnectionPoolTests.test_closeLogError.<locals>.ConnectionCloseRaisec S s t d��d S r� r� r r r r r 1 s zJConnectionPoolTests.test_closeLogError.<locals>.ConnectionCloseRaise.closeN)rs rt ru r r r r r �ConnectionCloseRaise0 s r� rV r r� N) r� r� Z_closer. r� r` rX r� r� )r r� r� r� r r r �test_closeLogError, s z&ConnectionPoolTests.test_closeLogErrorc sj G dd� dt �}G dd� dt �}dd� }t� }||_||_|�|�}� �|t�}� fdd�}|�|� |S ) z� If rollback fails, L{ConnectionPool.runInteraction} raises the original exception and log the error of the rollback. c @ s e Zd Zdd� Zdd� ZdS )z^ConnectionPoolTests.test_runWithInteractionRaiseOriginalError.<locals>.ConnectionRollbackRaisec S s d S r r r� r r r r� B s zgConnectionPoolTests.test_runWithInteractionRaiseOriginalError.<locals>.ConnectionRollbackRaise.__init__c S s t d��d S r� r� r r r r r� E s zgConnectionPoolTests.test_runWithInteractionRaiseOriginalError.<locals>.ConnectionRollbackRaise.rollbackNr� r r r r r� A s r� c @ s e Zd Zdd� ZdS )zWConnectionPoolTests.test_runWithInteractionRaiseOriginalError.<locals>.DummyTransactionc S s d S r r )r r� r� r r r r� I s z`ConnectionPoolTests.test_runWithInteractionRaiseOriginalError.<locals>.DummyTransaction.__init__N)rs rt ru r� r r r r �DummyTransactionH s r c S s t d��d S r� r� )rq r r r r� L s zVConnectionPoolTests.test_runWithInteractionRaiseOriginalError.<locals>.raisingFunctionc s6 � � t�}� �t|�d� � �|d jjd d� d S r� r� r� r r r r� U s zOConnectionPoolTests.test_runWithInteractionRaiseOriginalError.<locals>.cbFailed)r� r� r� ZtransactionFactoryrH r� r� r% )r r� r r� r� r' r� r r r �)test_runWithInteractionRaiseOriginalError< s z=ConnectionPoolTests.test_runWithInteractionRaiseOriginalErrorc C sD t d�}td|d�}| �|jdd|jfg� |�� | �|j� dS )z� If L{ConnectionPool.close} is called without L{ConnectionPool.start} having been called, the pool's startup event is cancelled. F�twisted.test.test_adbapi�Z cp_reactorr� r� N)r� r r` r� Z_startr r� �r r r� r r r �test_unstartedClose] s z'ConnectionPoolTests.test_unstartedClosec C sD t d�}td|d�}| �|jdd|jfg� |�� | �|j� dS )z� If L{ConnectionPool.close} is called after it has been started, but not by its shutdown trigger, the shutdown trigger is cancelled. Tr r ZduringZshutdownN)r� r r` r� Z finalCloser r� r r r r �test_startedClosek s �z%ConnectionPoolTests.test_startedCloseN) rs rt ru rv r� r� r r r r r r r r� s !r� )'rv Z twisted.trialr r� r� Ztwisted.enterprise.adbapir r r r Ztwisted.internetr r r Ztwisted.python.failurer Ztwisted.python.reflectr r/ r� r ry r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r� r r r r �<module> s: sSE 0 !
Save File
Cancel