B KLc4@sxddlZddlZddlZddlZddlmZddlmZddlmZddlm Z dddZd?ddZGddde ZGdddeZGdddeZGdddeZGdddeZGd d!d!eZejZd"d#Zd$d%Zd&d'Zd(d)Zd*d+Zd@d,d-Zed.d/Z dAd0d1Z!d2d3Z"dBd4d5Z#dCd6d7Z$dDd8d9Z%d:d;Z&dS)EN)config)util) decorator)inspect_getfullargspeccCs t}t||}|j||S)N)compound _as_predicateskipsadd) predicatereasonrulepredr?sz compound.not_..css|]}t|VqdS)N)r)rskiprrrr!@s)rrrr r)rrrrrnot_=s z compound.not_cCs |tjS)N)enabled_for_configr_current)rrrrenabledDszcompound.enabledcCs,x&|j|jD]}||rdSqWdSdS)NFT)r unionr)rrr rrrr$Hszcompound.enabled_for_configcsfdd|j|jDS)Ncsg|]}|r|qSr) _as_string)rr )rrr Qsz4compound.matching_config_reasons..)r r'r)rrr)rrmatching_config_reasonsOs z compound.matching_config_reasonscCs$t|j| o | p |j|S)N)boolr intersection)rZ include_tagsZ exclude_tagsrrr include_testVszcompound.include_testcCs.|j|j|j|j|j|jdS)N)r rrr)rrrrr_extend\szcompound._extendcs<t|dr|j|Stfdd}||}|_|S)N_sa_exclusion_extendcsjtj|f||S)N)_dorr%)fnargskw)rrrdecoratefsz#compound.__call__..decorate)hasattrr/r.r)rr1r4Z decoratedr)rr__call__as  zcompound.__call__c csht}|j|j|jy dVWn0tk rV}z|tj|Wdd}~XYnX| tjdS)N) rrrr r' Exception_expect_failurerr%_expect_success)rZ all_failsexrrrfail_ifns  zcompound.fail_ifc Osx6|jD],}||rdt||f}t|qWy|||}Wn4tk rz}z|j|||jdWdd}~XYnX|j||jd|SdS)Nz '%s' : %s)name) r rZget_current_test_namer(Z skip_testr7r8__name__r9) rZcfgr1r2r3r"msgZ return_valuer:rrrr0zs $z compound._doblockcCsnxh|jD]H}||rtjr.t|jddd}nt|}td||||fPqWtj|t dddS)Nzutf-8ignore)errorsz%s failed as expected (%s): %s r)with_traceback) rrZpy2kZunicodeencodestrprintr(Zraise_sysexc_info)rrr:r<r Zstr_exrrrr8s  zcompound._expect_failurec sL|js dSx<|jD]2}|rtd|dfdd|jDfqWdS)Nz Unexpected success for '%s' (%s)z and c3s|]}|VqdS)N)r()rr )rrrr!sz+compound._expect_success..)rAssertionErrorjoin)rrr<r r)rrr9s zcompound._expect_successN)r?)r?)r= __module__ __qualname__rrrr r#propertyr&r$r*r-r.r6 contextlibcontextmanagerr;r0r8r9rrrrr"s     rcCs t|gS)N)r)Ztagnamerrr requires_tagsrOcCst}|j||S)N)rrr)Ztagnamescomprrrrs rcCst|}tt||S)N)r rr)r r rrronly_ifsrQcCst|}tt||S)N)r rr)r r rrr succeeds_ifsrRc@s.eZdZed ddZd ddZd ddZdS) PredicateNcst|tr|j|St|tr:|r6|jdkr6||_|St|ttfr`tfdd|D|St|t rrt |St|t j rt d|}|std||d}|d}|drt dd |dd Dnd}t ||||d St|rt||Sd std |dS)Ncsg|]}|qSr) as_predicate)rr)clsrrr)sz*Predicate.as_predicate..z.([\+\w]+)\s*(?:(>=|==|!=|<=|<|>)\s*([\d\.]+))?z(Couldn't locate DB name in predicate: %rrrcss|]}t|VqdS)N)int)rdrrrr!sz)Predicate.as_predicate...) descriptionFzunknown predicate type: %s) isinstancerrTr$rSrZlistr OrPredicatetuple SpecPredicaterZ string_typesrematch ValueErrorgroupsplitcallableLambdaPredicaterH)rUr rZtokensdbopspecr)rUrrTs2       (  zPredicate.as_predicateFcCsV||}|r| }|j|r&|jjnd|r8|jjnd|rBdnd|rLdnddS)Nz z zdoesn't supportz does support)driverZdatabaseZdoesnt_supportZ does_support)rZrhZurlZget_driver_nameZget_backend_name)rrnegateZbool_rrr_format_descriptions zPredicate._format_descriptioncCs tdS)N)NotImplementedError)rrrlrrrr(szPredicate._as_string)N)F)NF)r=rJrK classmethodrTrmr(rrrrrSs # rSc@s(eZdZd ddZddZd ddZdS) BooleanPredicateNcCs||_|pd||_dS)Nz boolean %s)valuerZ)rrqrZrrrrszBooleanPredicate.__init__cCs|jS)N)rq)rrrrrr6szBooleanPredicate.__call__FcCs|j||dS)N)rl)rm)rrrlrrrr(szBooleanPredicate._as_string)N)F)r=rJrKrr6r(rrrrrps rpc @sPeZdZd ddZejejejejej ej ej dddZ ddZ d d d ZdS)r_NcCs||_||_||_||_dS)N)rhrirjrZ)rrhrirjrZrrrrszSpecPredicate.__init__cCs||dko||dkS)Nrrr)valZpairrrrzSpecPredicate.)<>z==z!=z<=z>=inZbetweencCs|dkr dS|j}d|jkr.|jd\}}n |jd}}|rL|j|krLdS|dk rb|j|krbdS|jdk r|dks|tdt|}t|jdr|jp|j|j}|||j SdSdS)NF+z%DBAPI version specs not supported yetr6T) rhrdr<rkrirH_server_versionr5_opsrj)rrenginedialectrkversionZoperrrrr6s     zSpecPredicate.__call__FcCsh|jdk r||S|jdkr8|r,d|jSd|jSn,|rPd|j|j|jfSd|j|j|jfSdS)Nznot %sz%sz not %s %s %sz%s %s %s)rZrmrirhrj)rrrlrrrr(4s     zSpecPredicate._as_string)NNN)F)r=rJrKroperatorltgteqnelegecontainsrzr6r(rrrrr_s  r_c@s(eZdZd ddZddZd ddZdS) rfNcs`t}|ds fdd|_n|_|p,d|_|p6i|_|rF||_njrVj|_nd|_dS)NrcsS)Nr)rh)lambda_rrrsGrtz*LambdaPredicate.__init__..rzcustom function)rrr2r3rZ__doc__)rrrZr2r3rjr)rrrDs   zLambdaPredicate.__init__cCs ||S)N)r)rrrrrr6SszLambdaPredicate.__call__FcCs ||S)N)rm)rrrlrrrr(VszLambdaPredicate._as_string)NNN)F)r=rJrKrr6r(rrrrrfCs rfc@s(eZdZd ddZddZd ddZdS) rNcCs||_||_dS)N)r rZ)rr rZrrrr[szNotPredicate.__init__cCs || S)N)r )rrrrrr6_szNotPredicate.__call__FcCs(|jr||| S|j|| SdS)N)rZrmr r()rrrlrrrr(bszNotPredicate._as_string)N)F)r=rJrKrr6r(rrrrrZs rc@s:eZdZd ddZddZdddZd d Zdd d ZdS)r]NcCs||_||_dS)N) predicatesrZ)rrrZrrrrjszOrPredicate.__init__cCs x|jD]}||rdSqWdS)NTF)r)rrrrrrr6ns zOrPredicate.__call__Fcs*r d}nd}|fdd|jDS)Nz and z or c3s|]}|jdVqdS))rlN)r()rp)rrlrrr!zsz(OrPredicate._eval_str..)rIr)rrrlZ conjunctionr)rrlr _eval_strts zOrPredicate._eval_strcCs*|jdk rd||S|j|ddSdS)NzNot T)rl)rZrmr)rrrrr _negation_str}s zOrPredicate._negation_strcCs0|r||S|jdk r"||S||SdS)N)rrZrmr)rrrlrrrr(s    zOrPredicate._as_string)N)F)F)r=rJrKrr6rrr(rrrrr]is   r]cCst|||tjS)N)r_rr%)rhrirjrrr _is_excludedsrcCs.|}t|jdd}|dkr"d}||S)z#Return a server_version_info tuple.Zserver_version_infoNr)Zconnectgetattrr|close)r{Zconnr}rrrrys rycGstdd|DS)NcSsg|]}t|qSr)rSrT)rrhrrrr)szdb_spec..)r])dbsrrrdb_specsrcCsttddS)NFzmark as execute)rrprrrropensrcCsttddS)NTzmarked as skip)rrprrrrclosedsrcCsttd|p dS)NTzexpected to fail)rrp)r rrrrsrcGstt|dS)NzFuture feature)rrf)r1argrrrfuturesrcCs t||S)N)r)rhr rrrfails_onsrcGsttdd|DS)NcSsg|]}t|qSr)rSrT)rrhrrrr)sz.fails_on_everything_except..)rRr])rrrrfails_on_everything_exceptsrcCs t||S)N)r)rhr rrrr"sr"cs ttfddt|DS)Ncsg|]}t|qSr)rSrT)rrh)r rrr)szonly_on..)rQr]rZto_list)rr r)r ronly_onsrcCstt||||S)N)rr_)rhrirjr rrrexcludesrcGs"|s tdtdd|D|S)Nzno queries sent!cSsg|]}t|qSr)rSrT)rZqueryrrrr)szagainst..)rHr])rZqueriesrrragainsts r)N)N)N)N)N)N)N)N)N)'rMr~r`rFrrrZ util.compatrrrobjectrrOrrQrRrSrpr_rfrr]rTr rryrrrrrrrr"rrrrrrr sF        8 <$