σ Ώb›]c@@ssddlmZddlZddlZddlZddlmZddlmZddlm Z ddlm Z dd lm Z d e fd „ƒYZ e ƒZd „Ze d „ƒZe d„ƒZe d„ƒZe d„ƒZdd„Zde fd„ƒYZddd„Zddd„Zdd„Zde fd„ƒYZde fd„ƒYZeed„ZdS(i(tabsolute_importNi(tconfig(tuses_deprecated(t decoratori(tevent(tpooltConnectionKillercB@seZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z e ƒd „ƒZ e ƒd „ƒZd „ZRS( cC@s.tjƒ|_tjƒ|_tƒ|_dS(N(tweakreftWeakKeyDictionaryt proxy_refsttesting_enginestsettconns(tself((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyt__init__scC@st|j|(texcludetdtnametmod((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyt all_dialectsstReconnectFixturecB@sAeZd„Zd„Zd„Zd„Zed„Zd„ZRS(cC@s||_g|_t|_dS(N(tdbapit connectionsR5t is_stopped(R RW((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyR­s  cC@st|j|ƒS(N(RORW(R tkey((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyt __getattr__²scO@st|jj||Ž}|jr\|j|jƒ|jƒ}|jdƒtsptdƒ‚n|j j |ƒ|SdS(Nsselect 1s%simulated connect failure didn't work( RWRRYR!R4tcursortexecuteR5R6RXtappend(R RFtkwargsR+tcurs((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyR΅s   cC@s6y |ƒWn$tk r1}tjd|ƒnXdS(Ns.ReconnectFixture couldn't close connection: %s(RRR(R RR ((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyR!Γs  cC@s@||_x't|jƒD]}|j|jƒqWg|_dS(N(RYR"RXR!R4(R tstoptc((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pytshutdownΛs cC@s t|_dS(N(R5RY(R ((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pytrestartΦs( R8R9RR[RR!R5RcRd(((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyRV¬s      c@s›|ptjj}tjjj}|s0i}nt|ƒ|dRWRcR5RY((t_disposeR(s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyR3γs( RR2turlR>RWRVttesting_engineR3Rct test_shutdownRdt test_restart(RgR-RWR3((RfRs9lib/python2.7/site-packages/sqlalchemy/testing/engines.pytreconnecting_engineΪs   cC@sšddlm}ddlm}|s/t}n|jdtƒ}|pPtjj}||ƒ}|dkr§tjdks’|j tjjj kržtj }qνi}nFtjdk rν|j tjjj krνtj j ƒ}|j |ƒn|||}t|_t|jtjƒr5d|j_d|j_n|r–tj|jdtjƒtj|jdtjƒtj|jdtjƒtj|ƒn|S( sBProduce an engine configured by --options with optional overrides.i(t create_engine(tmake_urlt use_reaperRRRN(t sqlalchemyRltsqlalchemy.engine.urlRmRtpopRR2RgR#t drivernametdb_optstcopytupdatet _has_eventst isinstanceRt QueuePoolt_timeoutt _max_overflowRtlistenR<RRRR(RgR-RlRmRnt default_optR((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyRhξs2   '  '  c@s©ddlm}|s%tjj}ng‰‡fd†}‡fd†}‡fd†}||dddd |ƒ‰tˆdƒ sŠt‚ˆˆ_|ˆ_|ˆ_ ˆS( s'Provides a mocking engine based on the current testing.db. This is normally used to test DDL generation flow as emitted by an Engine. It should not be used in other cases, as assert_compile() and assert_sql_execution() are much better choices with fewer moving parts. i(Rlc@sˆj|ƒdS(N(R^(tsqltaRG(tbuffer(s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pytexecutor'sc@sJgˆD]!}tjddt|ƒƒ^q}||ksFt|ƒ‚dS(Ns[\n\t]R@(tretsubtstrR6(tstmtststrecv(R(s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyt assert_sql*s.c@s)ˆj‰dj‡fd†ˆjDƒƒS(Ns c3@s'|]}t|jdˆƒƒVqdS(R>N(Rƒtcompile(t.0R…(RR(s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pys 0s(R>tjointmock((R(RRs9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyt print_sql.s s://tstrategyR‹R€( RoRlRR2RSR=R6R‹R‡RŒ(t dialect_nameRlR€R‡RŒ((RRs9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyt mock_engines    tDBAPIProxyCursorcB@s5eZdZd„Zdd„Zd„Zd„ZRS(suProxy a DBAPI cursor. Tests can provide subclasses of this to intercept DBAPI-level cursor operations. cO@s+||_||_|j||Ž|_dS(N(RR*R\(R RR+RFR_((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyRDs  cK@s3|r|jj|||S|jj||SdS(N(R\R](R tstmtt parametersRG((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyR]IscK@s|jj|||S(N(R\t executemany(R R‘tparamsRG((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyR“OscC@st|j|ƒS(N(ROR\(R RZ((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyR[RsN(R8R9t__doc__RR#R]R“R[(((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyR<s    tDBAPIProxyConnectioncB@s2eZdZd„Zd„Zd„Zd„ZRS(s}Proxy a DBAPI connection. Tests can provide subclasses of this to intercept DBAPI-level connection operations. cC@s/|jjƒ|_|_||_||_dS(N(Rt_creatorR+t _sqla_unwrapRt cursor_cls(R RR™((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyR^s cO@s|j|j|j||ŽS(N(R™RR+(R RFR_((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyR\cscC@s|jjƒdS(N(R+R4(R ((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyR4fscC@st|j|ƒS(N(ROR+(R RZ((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyR[is(R8R9R•RR\R4R[(((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyR–Vs    c@s&‡‡fd†}tdi|d6ƒS(sIProduce an engine that provides proxy hooks for common methods. c@sˆtjˆƒS(N(RR2((tconn_clsR™(s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyt mock_connusR-tcreator(Rh(RšR™R›((RšR™s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pytproxying_enginems(t __future__RRRRR@RRtutilRRRtobjectRR<RERHRIRJRKR#RURVRkRhRRR–R(((s9lib/python2.7/site-packages/sqlalchemy/testing/engines.pyts.   T    .& (