wR\8@svddlZddlZddlZddlZddlZddlZddlmZddlm Z m Z e edrej Z ej Z n<ddlZddZ ddZ d d Zeej_e ejd  sejjejjk rDejjej_ejjej_ejjej_ejjej_ndd d Zeej_GdddejZddZddZddZddZddZddZddZddZd d!Zd"d#Zd$d%Zd&d'Z d(d)Z!d*d+Z"e"d,Z#d-d.Z$d/d0Z%d1d2Z&Gd3d4d4e'Z(d5d6Z)dS)7N)wraps)dsnrepl_dsnskipIfcsfdd}|S)Ncs(tfdd}|S)Nc sHr:tj"tjdttjWdQXdS|SdS)Nalways)warningscatch_warnings simplefilter UserWarningwarn)self)condfmsgM/var/www/dbchiro/venv/lib/python3.4/site-packages/psycopg2/tests/testutils.pyskipIf__,s  z)skipIf..skipIf_..skipIf__)r)rr)rr)rrskipIf_+s$ zskipIf..skipIf_r)rrrr)rrrr*s cCs td|S)NT)r)rrrrskip8src Cs4tj"tjdttj|WdQXdS)Nr)rr r r r )r rrrrskipTest;s rassert_cCs/|jt|jt|j|dS)z4Check that two conninfo string have the same contentN) assertEqualsetsplit)r Zdsn1Zdsn2rrrrassertDsnEqualOsrc@seZdZdZddZddZdddZd d Zd d Zd dZ ddZ e e e Z ddZ dS)ConnectingTestCasea5A test case providing connections for tests. A connection for the test is always available as `self.conn`. Others can be created with `self.connect()`. All are closed on tearDown. Subclasses needing to customize setUp and tearDown should remember to call the base class implementations. cCs g|_dS)N)_conns)r rrrsetUp_szConnectingTestCase.setUpcCs.x'|jD]}|js |jq q WdS)N)rclosedclose)r connrrrtearDownbs zConnectingTestCase.tearDownNcs1fdd}|j||||S)z;Compare two quoted strings disregarding eventual E'' quotescsLt|tr"tjdd|SttrDtjdd|S|SdS)Nz\bE''s\bE'') isinstancestrresubbytes)s)firstrrrjs z/ConnectingTestCase.assertQuotedEqual..f)r)r r+secondrrr)r+rassertQuotedEqualhsz$ConnectingTestCase.assertQuotedEqualcKsy |jWn5tk rB}ztd|WYdd}~XnXd|kra|jd}nt}ddl}|j||}|jj||S)Nz7%s (did you forget to call ConnectingTestCase.setUp()?)rr)rAttributeErrorpoprpsycopg2connectappend)r kwargseZconninfor0r!rrrr1ts   zConnectingTestCase.connectcKstdkr|jdSd|kr2t|d.skip_if_no_uuid_)r)rrXr)rrskip_if_no_uuidsrYcs"tfdd}|S)z3Skip a test if the server has tpc support disabled.c sddlm}|j}|j}y|jdWn|k rZ|jdSYnXt|jd}|j|s|jdS|S)Nr)ProgrammingErrorzSHOW max_prepared_transactions;z5server too old: two phase transactions not supported.zfserver not configured for two phase transactions. set max_prepared_transactions to > 0 to run the test) r0rZr1rTrUrintrVr )r rZZcnnrWZmtp)rrrskip_if_tpc_disabled_s     z3skip_if_tpc_disabled..skip_if_tpc_disabled_)r)rr\r)rrskip_if_tpc_disabledsr]cs"tfdd}|S)Nc sDyddlm}Wntk r5|jdSYn X|SdS)Nr) namedtuplez$collections.namedtuple not available) collectionsr^rSr)r r^)rrrskip_if_no_namedtuple_s  z5skip_if_no_namedtuple..skip_if_no_namedtuple_)r)rr`r)rrskip_if_no_namedtuples racs"tfdd}|S)z.Skip a test if io.TextIOBase is not available.c sDyddlm}Wntk r5|jdSYn X|SdS)Nr) TextIOBasezio.TextIOBase not found.)iorbrSr)r rb)rrrskip_if_no_iobase_ s  z-skip_if_no_iobase..skip_if_no_iobase_)r)rrdr)rrskip_if_no_iobases recs.ddtfdd}|S)z3Skip a test on PostgreSQL before a certain version.rcs%tfdd}|S)NcsA|jjtdkr3|jd|jjS|SdS)Nz %d%02d%02dzskipped because PostgreSQL %s)r!server_versionr[r)r )rverrrskip_before_postgres__s zSskip_before_postgres..skip_before_postgres_..skip_before_postgres__)r)rri)rh)rrskip_before_postgres_s!z3skip_before_postgres..skip_before_postgres_)r)len)rhrjr)rhrskip_before_postgress rlcs.ddtfdd}|S)z>Skip a test on PostgreSQL after (including) a certain version.rrfcs%tfdd}|S)NcsA|jjtdkr3|jd|jjS|SdS)Nz %d%02d%02dzskipped because PostgreSQL %s)r!rgr[r)r )rrhrrskip_after_postgres__,s zPskip_after_postgres..skip_after_postgres_..skip_after_postgres__)r)rrm)rh)rrskip_after_postgres_+s!z1skip_after_postgres..skip_after_postgres_)r)rk)rhrnr)rhrskip_after_postgres's rocCs@ddl}|j}|dkr<t||jj}n|S)Nri_)r0Z__libpq_version__minr? libpq_version)r0vrrrrq8s    rqcs.ddtfdd}|S)zESkip a test if libpq we're linked to is older than a certain version.rrfcs%tfdd}|S)Ncs>t}|tdkr0|jd|S|SdS)Nz %d%02d%02dzskipped because libpq %d)rqr[r)r rr)rrhrrskip_before_libpq__Es zJskip_before_libpq..skip_before_libpq_..skip_before_libpq__)r)rrs)rh)rrskip_before_libpq_Ds!z-skip_before_libpq..skip_before_libpq_)r)rk)rhrtr)rhrskip_before_libpq@s rucs.ddtfdd}|S)zESkip a test if libpq we're linked to is newer than a certain version.rrfcs%tfdd}|S)Ncs>t}|tdkr0|jd|S|SdS)Nz %d%02d%02dzskipped because libpq %s)rqr[r)r rr)rrhrrskip_after_libpq__Vs zGskip_after_libpq..skip_after_libpq_..skip_after_libpq__)r)rrv)rh)rrskip_after_libpq_Us!z+skip_after_libpq..skip_after_libpq_)r)rk)rhrwr)rhrskip_after_libpqQs rxcsfdd}|S)z/Skip a test on Python before a certain version.cs%tfdd}|S)Nc sctjdtkrU|jddjtttjdtS|SdS)Nzskipped because Python %s.)sys version_inforkrjoinmapr&)r )rrhrrskip_before_python__es -zMskip_before_python..skip_before_python_..skip_before_python__)r)rr~)rh)rrskip_before_python_ds!z/skip_before_python..skip_before_python_r)rhrr)rhrskip_before_pythonbs rcsfdd}|S)z:Skip a test on Python after (including) a certain version.cs%tfdd}|S)Nc sctjdtkrU|jddjtttjdtS|SdS)Nzskipped because Python %sry)rzr{rkrr|r}r&)r )rrhrrskip_from_python__ts -zGskip_from_python..skip_from_python_..skip_from_python__)r)rr)rh)rrskip_from_python_ss!z+skip_from_python..skip_from_python_r)rhrr)rhrskip_from_pythonqs rcs"tfdd}|S)zDSkip a test if the database user running the test is not a superusercs~ddlm}y|SWnY|k ry}z9ddl}|j|jjkrd|jdnWYdd}~XnXdS)Nr)rZzskipped because not superuser)r0rZZpsycopg2.errorcodesr6Z errorcodesZINSUFFICIENT_PRIVILEGEr)r rZr4r0)rrrskip_if_no_superuser_s z3skip_if_no_superuser..skip_if_no_superuser_)r)rrr)rrskip_if_no_superusers rcsfdd}|S)Ncs%tfdd}|S)Ncs1ddlm}|r#|jS|SdS)Nr)green) testconfigrr)r r)rreasonrrskip_if_green__s z>skip_if_green..skip_if_green_..skip_if_green__)r)rr)r)rrskip_if_green_s!z%skip_if_green..skip_if_green_r)rrr)rr skip_if_greens rz*copy in async mode currently not supportedcs"tfdd}|S)Ncs*ttds|jdS|SdS)N getrefcountzskipped, no sys.getrefcount())r9rzr)r )rrrskip_if_no_getrefcount_s z7skip_if_no_getrefcount..skip_if_no_getrefcount_)r)rrr)rrskip_if_no_getrefcountsrcs"tfdd}|S)zSkip a test if run on windowscs-tjdkr|jdS|SdS)NWindowszNot supported on Windows)platformsystemr)r )rrrskip_if_windows_s z)skip_if_windows..skip_if_windows_)r)rrr)rrskip_if_windowssrc Cstjddkr|Sddl}|jdddd}|j|j|j|j}|jddl }|j dt j dd l m}|d d d d |grtdnt|}z|jSWd|jt j|XdS)z/Convert a script to Python3 syntax if required.rrfNsuffixz.pydeleteFfilename)mainz lib2to3.fixesz --no-diffsz-wz-nzpy3 conversion failed)rzr{tempfileNamedTemporaryFilewriteencodeflushnamer logging basicConfigosdevnullZ lib2to3.mainrrAopenreadremove)scriptrrrrrf2rrr script_to_py3s$       rc@s(eZdZddZddZdS)py3_raises_typeerrorcCsdS)Nr)r rrr __enter__szpy3_raises_typeerror.__enter__cCs-tjddkr)|tks%tdSdS)NrrfT)rzr{ TypeErrorAssertionError)r typeexctbrrr__exit__szpy3_raises_typeerror.__exit__N)rCrDrErrrrrrrs  rcs"tfdd}|S)zDecorator to mark slow tests we may want to skip Note: in order to find slow tests you can run: make check 2>&1 | ts -i "%.s" | sort -n cs)tjjdr|jdS|S)NZPSYCOPG2_TEST_FASTz slow test)renvirongetr)r )rrrslow_s zslow..slow_)r)rrr)rrslowsr)*r'rrzr@rZunittest functoolsrrrrr9rrrrZTestCase assertTruerrrrQrYr]rarerlrorqrurxrrrrZskip_copy_if_greenrrrobjectrrrrrrsX              j