Format: application/x-python-code
1 o
2
3 ��hE� � @ s> d Z ddd� ZdZddlZddlZdZG dd � d ej�ZdS )
4 a� Python DB API 2.0 driver compliance unit test suite.
5
6 This software is Public Domain and may be used without restrictions.
7
8 "Now we have booze and barflies entering the discussion, plus rumours of
9 DBAs on drugs... and I won't tell you what flashes through my mind each
10 time I read the subject line with 'Anal Compliance' in it. All around
11 this is turning out to be a thoroughly unwholesome unit test."
12
13 -- Ian Bicking
14 z$Revision: 1.15.0 $� �����z'Stuart Bishop <stuart@stuartbishop.net>� NFc @ s� e Zd ZdZdZdZi ZdZde Zde Z de Z
15 de Z d Z d
16 d � Z
17 d d
18 � Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� Zd d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Z d*d+� Z d Z d,d-� Z d.d/� Z d0d1� Z!d2d3� Z"d4d5� Z#d6d7� Z$g d8�Z%d9d:� Z&d;d<� Z'd=d>� Z(d?d@� Z)dAdB� Z*dCdD� Z+dEdF� Z,dGdH� Z-dIdJ� Z.dKdL� Z/dMdN� Z0dOdP� Z1dQdR� Z2dSdT� Z3dUdV� Z4dWdX� Z5dYdZ� Z6d[d\� Z7d]d^� Z8d_d`� Z9dadb� Z:dS )c�DatabaseAPI20Testa Test a database self.driver for DB API 2.0 compatibility.
19 This implementation tests Gadfly, but the TestCase
20 is structured so that other self.drivers can subclass this
21 test case to ensure compiliance with the DB-API. It is
22 expected that this TestCase may be expanded in the future
23 if ambiguities or edge conditions are discovered.
24
25 The 'Optional Extensions' are not yet being tested.
26
27 self.drivers should subclass this test, overriding setUp, tearDown,
28 self.driver, connect_args and connect_kw_args. Class specification
29 should be as follows:
30
31 import dbapi20
32 class mytest(dbapi20.DatabaseAPI20Test):
33 [...]
34
35 Don't 'import DatabaseAPI20Test from dbapi20', or you will
36 confuse the unit tester - just 'import dbapi20'.
37 N� Z dbapi20test_z'create table %sbooze (name varchar(20))z<create table %sbarflys (name varchar(20), drink varchar(30))zdrop table %sboozezdrop table %sbarflys�lowerc C � |� | j� d S �N)�execute�ddl1��self�cursorr r �Kc:\Users\shubh\Videos\agent\nova\Lib\site-packages\adodbapi/test/dbapi20.py� executeDDL1| � z DatabaseAPI20Test.executeDDL1c C r r )r �ddl2r r r r � executeDDL2 r z DatabaseAPI20Test.executeDDL2c C s dS )z�self.drivers should override this method to perform required setup
38 if any is necessary, such as creating the database.
39 Nr �r r r r �setUp� s zDatabaseAPI20Test.setUpc C s� z5| � � }z+|�� }| j| jfD ]}z |�|� |�� W q | jjy( Y qw W |�� W dS |�� w t y? Y dS w )z�self.drivers should override this method to perform required cleanup
40 if any is necessary, such as deleting the test database.
41 The default drops the tables that may be created.
42 N)
43 �_connectr
44 �xddl1�xddl2r �commit�driver�Error�close� Exception)r �con�curZddlr r r �tearDown� s
45 �� �zDatabaseAPI20Test.tearDownc C s< z| j j| ji | j��}W |S ty | �d� Y |S w )Nz-No connect method found in self.driver module)r �connect� connect_args�connect_kw_args�AttributeError�fail)r �rr r r r � s � �zDatabaseAPI20Test._connectc C s | � � }|�� d S r )r r �r r r r r � test_connect� s z DatabaseAPI20Test.test_connectc C s: z
46 | j j}| �|d� W d S ty | �d� Y d S w )Nz2.0z Driver doesn't define apilevel)r �apilevel� assertEqualr# r$ )r r( r r r �
47 test_apilevel� s �z DatabaseAPI20Test.test_apilevelc C �< z| j j}| �|dv � W d S ty | �d� Y d S w )N)r � � � z"Driver doesn't define threadsafety)r � threadsafety�
48 assertTruer# r$ )r r/ r r r �test_threadsafety� s �z#DatabaseAPI20Test.test_threadsafetyc C r+ )N)�qmark�numeric�named�format�pyformatz Driver doesn't define paramstyle)r �
49 paramstyler0 r# r$ )r r7 r r r �test_paramstyle� s
50 � �z!DatabaseAPI20Test.test_paramstylec C s� | � t| jjt�� | � t| jjt�� | � t| jj| jj�� | � t| jj| jj�� | � t| jj| jj�� | � t| jj | jj�� | � t| jj
51 | jj�� | � t| jj | jj�� | � t| jj | jj�� d S r )
52 r0 �
53 issubclassr �Warningr r �InterfaceError�
54 DatabaseError�OperationalError�IntegrityError�
55 InternalError�ProgrammingError�NotSupportedErrorr r r r �test_Exceptions� s z!DatabaseAPI20Test.test_Exceptionsc C s� | � � }| j}| �|j|ju � | �|j|ju � | �|j|ju � | �|j|ju � | �|j|ju � | �|j|ju � | �|j |j u � | �|j
56 |j
57 u � | �|j |j u � d S r ) r r r0 r: r r; r< r= r> r? r@ rA )r r �drvr r r �%test_ExceptionsAsConnectionAttributes� s z7DatabaseAPI20Test.test_ExceptionsAsConnectionAttributesc C s* | � � }z |�� W |�� d S |�� w r )r r r r& r r r � test_commit� s
58 z DatabaseAPI20Test.test_commitc C s> | � � }t|d�r z|�� W d S | jjy Y d S w d S )N�rollback)r �hasattrrF r rA r& r r r �
59 test_rollback� s
60 ��z DatabaseAPI20Test.test_rollbackc C s* | � � }z |�� }W |�� d S |�� w r )r r
61 r �r r r r r r � test_cursor� s
62 z DatabaseAPI20Test.test_cursorc C s� | � � }zD|�� }|�� }| �|� |�d| j � |�d| j � |�� }| �t|�d� | �t|d �d� | �|d d d� W |�� d S |�� w )N�.insert into %sbooze values ('Victoria Bitter')�select name from %sboozer, r �Victoria Bitter� r r
63 r r � table_prefix�fetchallr) �lenr )r r Zcur1Zcur2Zboozer r r �test_cursor_isolation� s
64 �z'DatabaseAPI20Test.test_cursor_isolationc C s� | � � }zf|�� }| �|� | �|jd d� |�d| j � | �t|j�dd� | �t|jd �dd� | �|jd d �� dd � | �|jd d | j j
65 d
66 |jd d � | � |� | �|jd d � W |� � d S |� � w ) Nzccursor.description should be none after executing a statement that can return no rows (such as DDL)rL r, z-cursor.description describes too many columnsr � z1cursor.description[x] tuples must have 7 elements�namez0cursor.description[x][0] must return column namez8cursor.description[x][1] must return column type. Got %rzVcursor.description not being set to None when executing no-result statements (eg. DDL))
67 r r
68 r r) � descriptionr rO rQ r r �STRINGr r rI r r r �test_description sJ
69 � � �� ��
70 �z"DatabaseAPI20Test.test_descriptionc C s� | � � }zH|�� }| �|� | �|jdv d� |�d| j � | �|jdv d� |�d| j � | �|jdv d� | �|� | �|jdd � W |� � d S |� � w )
71 N������r zFcursor.rowcount should be -1 or 0 after executing no-result statementsrK �rY r, zccursor.rowcount should == number or rows inserted, or set to -1 after executing an insert statementrL zbcursor.rowcount should == number of rows returned, or set to -1 after executing a select statementrY zJcursor.rowcount not being reset to -1 after executing no-result statements)
72 r r
73 r r0 �rowcountr rO r r) r rI r r r �
74 test_rowcount= s6
75 ����
76 �z DatabaseAPI20Test.test_rowcountc C s� | � � }z[|�� }| jrRt|d�rY|�| jd�}| �t|�d� | �|d d� |�� }| �t|�dd� | �t|d �dd� | �|d d dd � W |�� d S W |�� d S W |�� d S |�� w )
77 N�callproc)�FOOr, r r^ z callproc produced no result setz$callproc produced invalid result setZfooz!callproc produced invalid results) r r
78 �
79 lower_funcrG r] r) rQ rP r �r r r r% r r r �
80 test_callproca s � � z DatabaseAPI20Test.test_callprocc C sv | � � }z
81 |�� }W |�� n|�� w | �| jj| j|� | �| jj|j� tr4| �| jj|j� d S | � d� d S )Nz>Non-idempotent close is considered a bad thing by some people.)
82 r r
83 r � assertRaisesr r r r � TEST_FOR_NON_IDEMPOTENT_CLOSEZskipTestrI r r r �
84 test_closep s
85 �z DatabaseAPI20Test.test_closec C s4 | � � }z|�� }| �|� W |�� d S |�� w r )r r
86 � _paraminsertr rI r r r � test_execute� s
87 z DatabaseAPI20Test.test_executec C s� | � |� |�d| j � | �|jdv � | jjdkr%|�d| j d� nI| jjdkr5|�d| j d� n9| jjdkrG|�d | j d
88 d i� n'| jjd krW|�d
89 | j d� n| jjdkri|�d| j d
90 d i� n| �d� | �|jdv � |�d| j � |�� }| � t
91 |�dd� |d d |d d g}|� � | � |d d d� | � |d dd� d}| � |d d |d|d d �d|��� | � |d d |d|d d �d|��� d S )NzSinsert into %sbarflys values ('Victoria Bitter', 'thi%%s :may ca%%(u)se? troub:1e')rZ r2 zCinsert into %sbarflys values (?, 'thi%%s :may ca%%(u)se? troub:1e')��Cooper'sr3 zDinsert into %sbarflys values (:1, 'thi%%s :may ca%%(u)se? troub:1e')r4 zGinsert into %sbarflys values (:beer, 'thi%%s :may ca%%(u)se? troub:1e')�beerrh r5 zEinsert into %sbarflys values (%%s, 'thi%%s :may ca%%(u)se? troub:1e')r6 zKinsert into %sbarflys values (%%(beer)s, 'thi%%s :may ca%%(u)se? troub:1e')zInvalid paramstylez!select name, drink from %sbarflysr- z%cursor.fetchall returned too few rowsr r, zFcursor.fetchall retrieved incorrect data, or data inserted incorrectlyrM z thi%s :may ca%(u)se? troub:1ezLcursor.fetchall retrieved incorrect data, or data inserted incorrectly. Got=z , Expected=) r r rO r0 r[ r r7 r$ rP r) rQ �sort)r r �res�beersZtroubler r r re � s�
92 �� �� �� �� �� ��
93 ��
94
95 ���
96
97 ���z DatabaseAPI20Test._paraminsertc C s� | � � }z�|�� }| �|� ddg}ddiddig}| jjdkr*|�d| j |� nE| jjdkr:|�d | j |� n5| jjd
98 krJ|�d | j |� n%| jjd krZ|�d
99 | j |� n| jjdkrj|�d| j |� n| �d� | �|j dv d|j � |�
100 d| j � |� � }| � t
101 |�dd� |d d |d d g}|�� | � |d dd|d � | � |d dd� W |�� d S |�� w )Nrg )�Boag'sri rh rm r2 z insert into %sbooze values (?)r3 z insert into %sbooze values (:1)r4 z"insert into %sbooze values (:beer)r5 z insert into %sbooze values (%%s)r6 z&insert into %sbooze values (%%(beer)s)zUnknown paramstyle)rY r- zIinsert using cursor.executemany set cursor.rowcount to incorrect value %rrL r- z2cursor.fetchall retrieved incorrect number of rowsr r, z incorrect data "%s" retrievedzincorrect data retrieved)r r
102 r r r7 Z executemanyrO r$ r0 r[ r rP r) rQ rj r )r r r �largsZmargsrk rl r r r �test_executemany� s\
103
104 �
105 �
106 �
107 � �
108 ��
109 ��z"DatabaseAPI20Test.test_executemanyc C s
110 | � � }z{|�� }| �| jj|j� | �|� | �| jj|j� |�d| j � | � |�� d d� | �
111 |j dv � |�d| j � | �| jj|j� |�d| j � |�� }| � t |�dd� | � |d dd � | � |�� d d
112 � | �
113 |j d v � W |�
114 � d S |�
115 � w ) NrL z?cursor.fetchone should return None if a query retrieves no rowsrX rK r, z2cursor.fetchone should have retrieved a single rowr rM z(cursor.fetchone retrieved incorrect dataz<cursor.fetchone should return None if no more rows availablerZ )r r
116 rb r r �fetchoner r rO r) r0 r[ rQ r r` r r r �
117 test_fetchone s@
118 ��
119 �
120 ��z DatabaseAPI20Test.test_fetchone)z Carlton Coldz
121 Carlton Draftz
122 Mountain GoatZRedbackrM ZXXXXc s � fdd�� j D �}|S )zSReturn a list of sql commands to setup the DB for the fetch
123 tests.
124 c s g | ] }d � j |f �qS )z!insert into %sbooze values ('%s'))rO )�.0�sr r r �
125 <listcomp>C s ��z/DatabaseAPI20Test._populate.<locals>.<listcomp>)�samples)r Zpopulater r r � _populate? s
126 �zDatabaseAPI20Test._populatec C s` | � � }�z%|�� }| �| jj|jd� | �|� | �� D ]}|�|� q |�d| j � |�� }| �
127 t |�dd� d|_ |�d�}| �
128 t |�dd� |�d�}| �
129 t |�dd� |�d�}| �
130 t |�d d
131 � | �
132 |jd v � d|_ |�d| j � |�� }| �
133 t |�dd � |�� }| �
134 t |�d� |�� }| �
135 t |�d � | �
136 |jd v � d
137 |_ |�d| j � |�� }| �
138 |jd v � | �
139 t |�d
140 � | �
141 t |�d
142 � dd� |D �}|�� td d
143 �D ]}| �
144 || | j| d� q�|�� }| �
145 t |�d d� | �
146 |jd v � | �|� |�d| j � |�� }| �
147 t |�d d� | �
148 |jdv � W |�� d S |�� w )N� rL r, zQcursor.fetchmany retrieved incorrect number of rows, default of arraysize is one.�
149 r. z3cursor.fetchmany retrieved incorrect number of rowsr- r zLcursor.fetchmany should return an empty sequence after results are exhausted�rY � z0cursor.arraysize not being honoured by fetchmanyrz c S � g | ]}|d �qS �r r �rr r% r r r rt � � z4DatabaseAPI20Test.test_fetchmany.<locals>.<listcomp>z,incorrect data retrieved by cursor.fetchmanyzfcursor.fetchmany should return an empty sequence if called after the whole result set has been fetched�select name from %sbarflyszKcursor.fetchmany should return an empty sequence if query retrieved no rowsrX )r r
150 rb r r � fetchmanyr rv r rO r) rQ � arraysizer0 r[ rj �rangeru r r )r r r �sqlr% �rows�ir r r �test_fetchmanyI s�
151 �
152
153 �
154
155 �
156 �
157 ���
158 �z DatabaseAPI20Test.test_fetchmanyc C sv | � � }z�|�� }| �| jj|j� | �|� | �� D ]}|�|� q| �| jj|j� |�d| j � |�� }| �
159 |j dt | j
160 �fv � | �t |�t | j
161 �d� dd� |D �}|�� tdt | j
162 ��D ]}| �|| | j
163 | d� qd|�� }| �t |�dd� | �
164 |j dt | j
165 �fv � | �|� |�d | j � |�� }| �
166 |j d
167 v � | �t |�dd � W |�� d S |�� w ) NrL rY z)cursor.fetchall did not retrieve all rowsc S r{ r| r r} r r r rt � r~ z3DatabaseAPI20Test.test_fetchall.<locals>.<listcomp>r z(cursor.fetchall retrieved incorrect rowszacursor.fetchall should return an empty list if called after the whole result set has been fetchedr rX zMcursor.fetchall should return an empty list if a select query returns no rows)r r
168 rb r r rP r rv r rO r0 r[ rQ ru r) rj r� r r )r r r r� r� r� r r r �
169 test_fetchall� sN
170 ���
171 �z DatabaseAPI20Test.test_fetchallc
172 C s< | � � }z�|�� }| �|� | �� D ]}|�|� q|�d| j � |�� }|�d�}|�� }|�� }| � |j
173 dv � | � t |�dd� | � t |�dd� |d g}|�
174 |d d |d d g� |�|d � |�
175 |d d |d d g� |�� tdt | j��D ]} | � || | j| d� q�W |�� d S |�� w ) NrL r- ry z+fetchmany returned incorrect number of rowsz*fetchall returned incorrect number of rowsr r, z$incorrect data retrieved or inserted)r r
176 r rv r rO rp r� rP r0 r[ r) rQ �extend�appendrj r� ru r )
177 r r r r� Zrows1Zrows23Zrows4Zrows56r� r� r r r �test_mixedfetch� s:
178
179
180 �
181 �
182 ��z!DatabaseAPI20Test.test_mixedfetchc C � t d��)z�Should create a procedure called deleteme
183 that returns two result sets, first the
184 number of rows in booze then "name from booze"
185 �Helper not implemented��NotImplementedError�r r r r r �help_nextset_setUp� s z$DatabaseAPI20Test.help_nextset_setUpc C r� )z*If cleaning up is needed after nextSetTestr� r� r� r r r �help_nextset_tearDown � z'DatabaseAPI20Test.help_nextset_tearDownc C r� )Nz"Drivers need to override this testr� r r r r � test_nextset s z DatabaseAPI20Test.test_nextsetc C s< | � � }z|�� }| �t|d�d� W |�� d S |�� w )Nr� z cursor.arraysize must be defined)r r
186 r0 rG r rI r r r �test_arraysize s
187 �z DatabaseAPI20Test.test_arraysizec C s> | � � }z|�� }|�d� | �|� W |�� d S |�� w )N)� )r r
188 Z
189 setinputsizesre r rI r r r �test_setinputsizes s
190 z$DatabaseAPI20Test.test_setinputsizesc C sJ | � � }z|�� }|�d� |�dd� | �|� W |�� d S |�� w )Ni� i� r )r r
191 Z
192 setoutputsizere r rI r r r �test_setoutputsize_basic' s
193 z*DatabaseAPI20Test.test_setoutputsize_basicc C r� )Nz#Driver needed to override this testr� r r r r �test_setoutputsize2 r� z$DatabaseAPI20Test.test_setoutputsizec C s� | � � }zA|�� }| �|� |�d| j � |�d| j � |�� }| �t|�d� | �t|d �d� | �|d d d d� W |�� d S |�� w )Nz!insert into %sbooze values (NULL)rL r, r z NULL value not returned as NonerN r` r r r � test_None6 s
194 zDatabaseAPI20Test.test_Nonec C �& | j �ddd�}| j �t�d��}d S )N�� � r� ) r� r� r� r r r r r r )r �DateZ
195 DateFromTicks�time�mktime)r Zd1Zd2r r r � test_DateD � zDatabaseAPI20Test.test_Datec C r� )N�
196 �- � ) i� r, r, r� r� r� r r r )r ZTimeZ
197 TimeFromTicksr� r� �r �t1�t2r r r � test_TimeJ r� zDatabaseAPI20Test.test_Timec C s, | j �dddddd�}| j �t�d��}d S )Nr� r� r� r� r� r� ) r� r� r� r� r� r� r r r )r � TimestampZTimestampFromTicksr� r� r� r r r �test_TimestampP s �z DatabaseAPI20Test.test_Timestampc C s | j �d�}| j �d�}d S )Ns Something� )r �Binary)r �br r r � test_BinaryX s z DatabaseAPI20Test.test_Binaryc C � | � t| jd�d� d S )NrV z module.STRING must be defined�r0 rG r r r r r � test_STRING\ � z DatabaseAPI20Test.test_STRINGc C r� )N�BINARYz module.BINARY must be defined.r� r r r r � test_BINARY_ � �z DatabaseAPI20Test.test_BINARYc C r� )N�NUMBERz module.NUMBER must be defined.r� r r r r � test_NUMBERd r� z DatabaseAPI20Test.test_NUMBERc C r� )NZDATETIMEz module.DATETIME must be defined.r� r r r r �
198 test_DATETIMEi r� z DatabaseAPI20Test.test_DATETIMEc C r� )NZROWIDz module.ROWID must be defined.r� r r r r �
199 test_ROWIDn r� z DatabaseAPI20Test.test_ROWID);�__name__�
200 __module__� __qualname__�__doc__r r! r" rO r
201 r r r Z lowerfuncr r r r r r' r* r1 r8 rB rD rE rH rJ rR rW r\ r_ ra rd rf re ro rq ru rv 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 V sr
202
203 +"G1.
204 \6! r )r� � __version__�
205 __author__r� Zunittestrc ZTestCaser r r r r �<module> s A