o i2@sxddlZddlZddlZddlZddlmZddlmZddlmZddlm Z d8z compound.not_..csr(r r))r,skiprrrr.9r/r$)rr'rrrnot_6s z compound.not_cCs |tjSr )enabled_for_configr_currentrrrrenabled=s zcompound.enabledcCs&|j|jD] }||rdSqdS)NFTr unionr)rrrrrrr2As zcompound.enabled_for_configcsfdd|j|jDS)Ncsg|] }|r|qSr _as_string)r,rrrr Is z4compound.matching_config_reasons..r5rrrrrmatching_config_reasonsHs  z compound.matching_config_reasonscCs$t|j| o| p|j|Sr )boolr intersection)r include_tags exclude_tagsrrr include_testOszcompound.include_testcCs.|j|j|j|j|j|jdSr )r r%rrr rrr_extendUszcompound._extendcs<t|dr |j|Stfdd}||}|_|S)N_sa_exclusion_extendcsjtj|g|Ri|Sr )_dorr3)fnargskwrrrdecorate_sz#compound.__call__..decorate)hasattrrBrAr)rrDrG decoratedrrr__call__Zs  zcompound.__call__c csnt}|j|j|jzdVWnty.}z|tj|WYd}~dSd}~ww| tjdSr ) r rr%r r6 Exception_expect_failurerr3_expect_success)r all_failsexrrrfail_ifgs zcompound.fail_ifc Os|jD]}||rdt||f}t|qz ||i|}Wnty?}z|j|||jdWYd}~dSd}~ww|j||jd|S)Nz '%s' : %s)name) r rget_current_test_namer8 skip_testrKrL__name__rM) rcfgrDrErFr0msg return_valuerOrrrrCss   z compound._doblockcCsn|jD]&}||r)tjrt|jddd}nt|}td||||fdSqtj|t dddS)Nzutf-8ignore)errorsz%s failed as expected (%s): %s r)with_traceback) rrpy2kunicodeencodestrprintr8raise_sysexc_info)rrrOrQr-str_exrrrrLs  zcompound._expect_failurec sH|jsdS|jD]}|r!td|dfdd|jDfqdS)Nz Unexpected success for '%s' (%s) and c3s|]}|VqdSr r7r+rrrr.s  z+compound._expect_success..)rAssertionErrorjoin)rrrQr-rrrrMs zcompound._expect_successN)rX)rT __module__ __qualname__rr"rr1propertyr4r2r;r@rArJ contextlibcontextmanagerrPrCrLrMrrrrr "s"    r cCs t|gSr )r)tagnamerrr requires_tagr#rncCst}|j||Sr )r rr%)tagnamescomprrrrs rcCt|}tt||Sr )r rr*rrrrronly_ifrscCrqr )r rr*rrrrr succeeds_ifrtruc@s.eZdZed ddZd ddZd ddZdS) PredicateNcs t|tr |j|St|tr|r|jdur||_|St|ttfr0tfdd|D|St|t r9t |St|t j rut d|}|sMtd||d}|d}|drkt dd |dd Dnd}t ||||d St |rt||SJd |)Ncsg|]}|qSr) as_predicate)r,rclsrrr9z*Predicate.as_predicate..z.([\+\w]+)\s*(?:(>=|==|!=|<=|<|>)\s*([\d\.]+))?z(Couldn't locate DB name in predicate: %rrrcsr(r )int)r,drrrr.r/z)Predicate.as_predicate...) descriptionFzunknown predicate type: %s) isinstancer rwr2rvrlistr OrPredicatetuple SpecPredicater string_typesrematch ValueErrorgroupsplitcallableLambdaPredicate)ryrrtokensdbopspecrrxrrws<          zPredicate.as_predicateFcCs\||}|r | }|j|r|jjnd|r|jjnd|r!dnd|r)ddSddS)Nz z zdoesn't supportz does support)driverdatabasedoesnt_support does_support)rrurlget_driver_nameget_backend_name)rrnegatebool_rrr_format_descriptions zPredicate._format_descriptioncCstr )NotImplementedErrorrrrrrrr8zPredicate._as_stringr F)NF)rTrhri classmethodrwrr8rrrrrvs   #rvc@(eZdZd ddZddZd ddZdS) BooleanPredicateNcCs||_|pd||_dS)Nz boolean %s)valuer)rrrrrrrszBooleanPredicate.__init__cCs|jSr )rr:rrrrJrzBooleanPredicate.__call__FcCs|j||dS)Nrrrrrrr8zBooleanPredicate._as_stringr rrTrhrirrJr8rrrrr rc @sPeZdZd ddZejejejejej ej ej dddZ ddZ d d d ZdS)rNcCs||_||_||_||_dSr )rrrr)rrrrrrrrrs zSpecPredicate.__init__cCs||dko ||dkS)Nrrr)valpairrrr zSpecPredicate.)<>z==z!=z<=z>=inbetweencCs|j}d|jvr|jd\}}n|jd}}|r |j|kr dS|dur+|j|kr+dS|jdurQ|dus8Jdt|}t|jdrE|jpJ|j|j}|||jSdS)N+Fz%DBAPI version specs not supported yetrJT) rrrQrr_server_versionrH_opsr)rrenginedialectrversionoperrrrrJs    zSpecPredicate.__call__FcCsb|jdur ||S|jdur|rd|jSd|jS|r'd|j|j|jfSd|j|j|jfS)Nznot %sz%sz not %s %s %sz%s %s %s)rrrrrrrrrr8(s     zSpecPredicate._as_stringNNNr)rTrhriroperatorltgteqnelegecontainsrrJr8rrrrrs  rc@r) rNcsdt}|dsfdd|_n|_|pd|_|pi|_|r$||_dSjr-j|_dSd|_dS)NrcsSr r)rlambda_rrr;sz*LambdaPredicate.__init__..rzcustom function)rrrErFr__doc__)rrrrErFrrrrr8s     zLambdaPredicate.__init__cCrr rr:rrrrJGr#zLambdaPredicate.__call__FcCrr rrrrrr8Jr#zLambdaPredicate._as_stringrrrrrrrr7s rc@r) r*NcC||_||_dSr )rr)rrrrrrrO zNotPredicate.__init__cCs || Sr )rr:rrrrJSs zNotPredicate.__call__FcCs$|jr ||| S|j|| Sr )rrrr8rrrrr8VszNotPredicate._as_stringr rrrrrrr*Nrr*c@s:eZdZd ddZddZdddZd d Zdd d ZdS)rNcCrr ) predicatesr)rrrrrrr^rzOrPredicate.__init__cCs|jD] }||r dSqdS)NTF)r)rrrrrrrJbs zOrPredicate.__call__Fcs*rd}nd}|fdd|jDS)Nrez or c3s|] }|jdVqdS)rNr7)r,prrrrr.ms z(OrPredicate._eval_str..)rgr)rrr conjunctionrrr _eval_strhs zOrPredicate._eval_strcCs&|jdur d||S|j|ddS)NzNot Tr)rrrr:rrr _negation_strqs zOrPredicate._negation_strcCs,|r||S|jdur||S||Sr )rrrrrrrrr8ws     zOrPredicate._as_stringr r)rTrhrirrJrrr8rrrrr]s    rcCst|||tjSr )rrr3)rrrrrr _is_excludedrcCs.|}t|jdd}|durd}||S)z#Return a server_version_info tuple.server_version_infoNr)connectgetattrrclose)rconnrrrrrs rcGstdd|DS)NcSg|]}t|qSrrvrwr,rrrrr9rzzdb_spec..rdbsrrrdb_specrrcCttddS)NFzmark as executerrrrrropenrrcCr)NTzmarked as skiprrrrrclosedrrcCsttd|pdS)NTzexpected to fail)rrrrrrrrrcGstt|dS)NzFuture feature)rr)rDargrrrfuturesrcC t||Sr )rrrrrrfails_onr#rcGsttdd|DS)NcSrrrrrrrr9rzz.fails_on_everything_except..)rurrrrrfails_on_everything_exceptsrcCrr )rrrrrr0r#r0cs ttfddt|DS)Ncsg|]}t|qSrrrrrrr9rzonly_on..)rsrrto_list)rrrrronly_ons rcCstt||||Sr )rr)rrrrrrrexcluderrcGs"|sJdtdd|D|S)Nzno queries sent!cSrrr)r,queryrrrr9rzzagainst..r)rqueriesrrragainsts rr )'rkrrrbrrr util.compatrrrobjectr rnrrsrurvrrrr*rrwr rrrrrrrrrr0rrrrrrrsJ        8 9$