5dV( @s>ddlZddlZddlZddlmZddlmZyddlZeZWn$e k rddlZdZYnXe edrej Z ej Z n<ddl Z ddZ ddZ d d Zeej_e ejd  s ejjejjk rWejjej_ejjej_ejjej_ejjej_nGd d d ejZddZddZddZddZddZddZddZddZddZd d!Zd"d#Zed$Zd%d&Zd'd(Z d)d*Z!Gd+d,d,e"Z#dS)-N)wraps)dsnskipIfcsfdd}|S)Ncs(tfdd}|S)Ncs%rtjdS|SdS)N)warningswarn)self)condfmsg.skipIf_..skipIf__)r)r r)r r )r r skipIf_0s$zskipIf..skipIf_r )r r rr )r r r r/s cCs td|S)NT)r)r r r r skip;srcCstj|dS)N)rr)rr r r r skipTest>s rZassert_c@saeZdZdZddZddZddZdd Zd d Ze eeZ d S) 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)rr r r setUpYszConnectingTestCase.setUpcCs.x'|jD]}|js |jq q WdS)N)rclosedclose)rconnr r r tearDown\s zConnectingTestCase.tearDowncKsuy |jWn5tk rB}ztd|WYdd}~XnXddl}|jt|}|jj||S)Nz9%s (did you remember calling ConnectingTestCase.setUp()?)r)rAttributeErrorpsycopg2connectrappend)rkwargserrr r r rbs  zConnectingTestCase.connectcCs(t|ds!|j|_n|jS)N _the_conn)hasattrrr)rr r r _get_connoszConnectingTestCase._get_conncCs ||_dS)N)r)rrr r r _set_connuszConnectingTestCase._set_connN) __name__ __module__ __qualname____doc__rrrr!r"propertyrr r r r rPs     rc Gs]xVt|D]H}|jdr x0|D]%}t|||t||q)Wq q WdS)zT Apply all the *decorators* to all the tests defined in the TestCase *cls*. testN)dir startswithsetattrgetattr)cls decoratorsndr r r decorate_all_tests{s r1cs"tfdd}|S)z;Decorator to skip a test if uuid is not supported by Py/PG.csyddl}Wntk r1|jdSYnXz,|jj}|jd|j}Wd|jjX|r|S|jdSdS)Nrz)uuid not available in this Python versionz2select typname from pg_type where typname = 'uuid'z%uuid type not available on the server)uuid ImportErrorrrcursorexecutefetchonerollback)rr2curZhas)r r r skip_if_no_uuid_s   z)skip_if_no_uuid..skip_if_no_uuid_)r)r r9r )r r skip_if_no_uuidsr:cs"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) rr;rr4r5rintr6r)rr;Zcnnr8Zmtp)r r r skip_if_tpc_disabled_s     z3skip_if_tpc_disabled..skip_if_tpc_disabled_)r)r r=r )r r skip_if_tpc_disabledsr>cs"tfdd}|S)Nc sDyddlm}Wntk r5|jdSYn X|SdS)Nr) namedtuplez$collections.namedtuple not available) collectionsr?r3r)rr?)r r r skip_if_no_namedtuple_s  z5skip_if_no_namedtuple..skip_if_no_namedtuple_)r)r rAr )r r skip_if_no_namedtuples rBcs"tfdd}|S)z.Skip a test if io.TextIOBase is not available.c sDyddlm}Wntk r5|jdSYn X|SdS)Nr) TextIOBasezio.TextIOBase not found.)iorCr3r)rrC)r r r skip_if_no_iobase_s  z-skip_if_no_iobase..skip_if_no_iobase_)r)r rEr )r r skip_if_no_iobases rFcs.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)rserver_versionr<r)r)r verr r skip_before_postgres__s zSskip_before_postgres..skip_before_postgres_..skip_before_postgres__)r)r rJ)rI)r r skip_before_postgres_s!z3skip_before_postgres..skip_before_postgres_)r)len)rIrKr )rIr skip_before_postgress rMcs.ddtfdd}|S)z>Skip a test on PostgreSQL after (including) a certain version.rrGcs%tfdd}|S)NcsA|jjtdkr3|jd|jjS|SdS)Nz %d%02d%02dzskipped because PostgreSQL %s)rrHr<r)r)r rIr r skip_after_postgres__s zPskip_after_postgres..skip_after_postgres_..skip_after_postgres__)r)r rN)rI)r r skip_after_postgres_s!z1skip_after_postgres..skip_after_postgres_)r)rL)rIrOr )rIr skip_after_postgress rPcsfdd}|S)z/Skip a test on Python before a certain version.cs%tfdd}|S)Nc sctjdtkrU|jddjtttjdtS|SdS)Nzskipped because Python %s.)sys version_inforLrjoinmapstr)r)r rIr r skip_before_python__s -zMskip_before_python..skip_before_python_..skip_before_python__)r)r rW)rI)r r skip_before_python_s!z/skip_before_python..skip_before_python_r )rIrXr )rIr skip_before_pythons rYcsfdd}|S)z:Skip a test on Python after (including) a certain version.cs%tfdd}|S)Nc sctjdtkrU|jddjtttjdtS|SdS)Nzskipped because Python %srQ)rRrSrLrrTrUrV)r)r rIr r skip_from_python__s -zGskip_from_python..skip_from_python_..skip_from_python__)r)r rZ)rI)r r skip_from_python_s!z+skip_from_python..skip_from_python_r )rIr[r )rIr skip_from_pythons r\cs"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)r;zskipped because not superuser)rr;Zpsycopg2.errorcodesZpgcodeZ errorcodesZINSUFFICIENT_PRIVILEGEr)rr;rr)r r r skip_if_no_superuser_ s z3skip_if_no_superuser..skip_if_no_superuser_)r)r r]r )r r skip_if_no_superuser s r^csfdd}|S)Ncs%tfdd}|S)Ncs1ddlm}|r#|jS|SdS)Nr)green) testconfigr_r)rr_)r reasonr r skip_if_green__s z>skip_if_green..skip_if_green_..skip_if_green__)r)r rb)ra)r r skip_if_green_s!z%skip_if_green..skip_if_green_r )rarcr )rar skip_if_greens rdz*copy in async mode currently not supportedcs"tfdd}|S)Ncs*ttds|jdS|SdS)N getrefcountzskipped, no sys.getrefcount())r rRr)r)r r r skip_if_no_getrefcount_+s z7skip_if_no_getrefcount..skip_if_no_getrefcount_)r)r rfr )r r skip_if_no_getrefcount*srgcs"tfdd}|S)zSkip a test if run on windowscs-tjdkr|jdS|SdS)NWindowszNot supported on Windows)platformsystemr)r)r r r skip_if_windows_5s z)skip_if_windows..skip_if_windows_)r)r rkr )r r skip_if_windows3srlc 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.rrGNsuffixz.pydeleteFfilename)mainz lib2to3.fixesz --no-diffsz-wz-nzpy3 conversion failed)rRrStempfileNamedTemporaryFilewriteencodeflushnamerlogging basicConfigosdevnullZ lib2to3.mainrp Exceptionopenreadremove)scriptrqr rorwrpf2r r r script_to_py3>s$       rc@s(eZdZddZddZdS)py3_raises_typeerrorcCsdS)Nr )rr r r __enter__[szpy3_raises_typeerror.__enter__cCs-tjddkr)|tks%tdSdS)NrrGT)rRrS TypeErrorAssertionError)rtypeexctbr r r __exit__^szpy3_raises_typeerror.__exit__N)r#r$r%rrr r r r rYs  r)$ryrirR functoolsrr`rZ unittest2Zunittestr3r rrrrZTestCaseZ assertTrueZ assertEqualrr1r:r>rBrFrMrPrYr\r^rdZskip_copy_if_greenrgrlrobjectrr r r r sR              +