3
(hP                 @   s   d dl Z yd dlmZ d dlZW n4 ek
rT   e jjd d dlmZ d dlZY nX G dd dejZ	dgZ
edkrd dlZej  dS )    N)basez
../pymysqlc               @   s   e Zd Zdd ZdS )TestSSCursorc       	      C   s8  d}| j d }d!d"d#d$d%d&d'd(d)d*g
}z|jtjj}|jd |j  x*|D ]"}|jd| | j|j dd qPW |j	  d}|jd x`|j
 }|d krP |d7 }| j|j|dt|  | j|j|dt|  | j||kdd qW |jd | jt|j t|d |jd | jt|jddd x|j
 }|d kr6P q6W |jddg |j	  | j|jt|dtt|  |jd| | j|jt|dtt|  |jd | jt|d+g | j|j  | jt|d,g | j|j  | jt|d-g | j|j  W d |jd  |j  X d S ).Nl    r   America America/JamaicaAmerica/Los_AngelesAmerica/LimaAmerica/New_YorkAmerica/MenomineeAmerica/HavanaAmerica/El_SalvadorAmerica/Costa_RicaAmerica/DenverAmerica/DetroitzKCREATE TABLE tz_data (region VARCHAR(64),zone VARCHAR(64),name VARCHAR(64))z'INSERT INTO tz_data VALUES (%s, %s, %s)   zaffected_rows does not matchzSELECT * FROM tz_datazcursor.rowcount != %sTzRow not found in source dataz.fetchall failed. Number of rows does not match   z/fetchmany failed. Number of rows does not matchzUPDATE tz_data SET zone = %sZFooz"Update failed. affected_rows != %sz)executemany failed. cursor.rowcount != %szSELECT 1; SELECT 2; SELECT 3   zDROP TABLE tz_data)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   )connectionscursorpymysqlZcursorsZSSCursorexecutebeginZassertEqualaffected_rowscommitZfetchoneZrowcountstrZ	rownumberlenZfetchallZ	fetchmanyZexecutemanyZassertListEquallistZ
assertTrueZnextsetZassertFalseclose)	selfr   conndatar   iiterrowres r%   @/tmp/pip-install-q3hcpn_q/PyMySQL/pymysql/tests/test_SSCursor.pytest_SSCursor   sv    











zTestSSCursor.test_SSCursorN)__name__
__module____qualname__r'   r%   r%   r%   r&   r      s   r   __main__)sysZpymysql.testsr   Zpymysql.cursorsr   	ExceptionpathappendZPyMySQLTestCaser   __all__r(   Zunittestmainr%   r%   r%   r&   <module>   s   ^