î
wÐR\¯  ã               @   sÐ   d  d l  m Z d  d l  m Z d  d l m Z d  d l m Z m Z d d l Z d  d l m	 Z	 Gd d	 „  d	 e j
 ƒ Z Gd
 d „  d e j e j ƒ Z e e e ƒ d d „  Z e d k rÌ e j ƒ  n  d S)é   )Údbapi20)Údbapi20_tpc)Úskip_if_tpc_disabled)ÚunittestÚdecorate_all_testsé    N)Údsnc               @   sS   e  Z d  Z e Z f  Z i e d 6Z d Z d d „  Z	 d d „  Z
 d d „  Z d	 S)
ÚPsycopg2Testsr   Úlowerc          
   C   s¹   |  j  ƒ  } z› | j ƒ  } |  j r¦ t | d ƒ r¦ | j |  j d	 ƒ | j ƒ  } |  j t | ƒ d d ƒ |  j t | d ƒ d d ƒ |  j | d d d d ƒ n  Wd  | j ƒ  Xd  S)
NÚcallprocÚFOOr   zcallproc produced no result setr   z$callproc produced invalid result setZfooz!callproc produced invalid results)r   )	Z_connectZcursorÚ
lower_funcÚhasattrr   ZfetchallÚassertEqualÚlenÚclose)ÚselfZconÚcurÚr© r   úY/var/www/dbchiro/venv/lib/python3.4/site-packages/psycopg2/tests/test_psycopg2_dbapi20.pyÚtest_callproc)   s    zPsycopg2Tests.test_callprocc             C   s   d  S)Nr   )r   r   r   r   Útest_setoutputsizeA   s    z Psycopg2Tests.test_setoutputsizec             C   s   d  S)Nr   )r   r   r   r   Útest_nextsetE   s    zPsycopg2Tests.test_nextsetN)Ú__name__Ú
__module__Ú__qualname__Úpsycopg2ÚdriverZconnect_argsr   Zconnect_kw_argsr   r   r   r   r   r   r   r   r	   "   s   r	   c               @   s"   e  Z d  Z e Z d d „  Z d S)ÚPsycopg2TPCTestsc             C   s   t  j d t ƒ S)Nr   )r   Úconnectr   )r   r   r   r   r    M   s    zPsycopg2TPCTests.connectN)r   r   r   r   r   r    r   r   r   r   r   J   s   r   c               C   s   t  j ƒ  j t ƒ S)N)r   Z
TestLoaderZloadTestsFromNamer   r   r   r   r   Ú
test_suiteS   s    r!   Ú__main__)Ú r   r   Z	testutilsr   r   r   r   Z
testconfigr   ZDatabaseAPI20Testr	   ZTwoPhaseCommitTestsZTestCaser   r!   r   Úmainr   r   r   r   Ú<module>   s   (