ó [³XMc@sZdZddlmZddlZddlZddlZddlmZmZm Z ddlm Z m Z ddlm Z ddl mZmZmZddlmZd Zd d/d „ƒYZd d0d „ƒYZdd1d„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZdefd„ƒYZd„Zeedeƒƒeedeƒƒdejfd „ƒYZ d!e!fd"„ƒYZ"d#ejfd$„ƒYZ#d%ejfd&„ƒYZ$d'e!fd(„ƒYZ%d)efd*„ƒYZ&d+e!fd,„ƒYZ'd-ejfd.„ƒYZ(dS(2s& Tests for twisted.enterprise.adbapi. iÿÿÿÿ(tunittestN(tConnectionPooltConnectionLosttsafe(t Connectiont Transaction(t_unreleasedVersion(treactortdefert interfaces(tFailures% CREATE TABLE simple ( x integer ) tADBAPITestBasecBseZdZiZejedƒdkr3dZnd„Z d„Z d„Z dd„Z d„Z d„Zd„Zd „Zd „Zd „Zd „Zd „Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„ZRS(s"Test the asynchronous DB-API code.s5ADB-API requires threads, no way to test without themcCs3|jƒ|jd|jƒ|_|jjƒdS(sR Set up the database and create a connection pool pointing at it. t cp_openfunN(tstartDBtmakePooltopenfuntdbpooltstart(tself((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyt extraSetUp#s csBˆjjdƒ}|j‡fd†ƒ|j‡fd†ƒ|S(NsDROP TABLE simplecs ˆjjƒS(N(Rtclose(tres(R(s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyt.scs ˆjƒS(N(tstopDB(R(R(s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR/s(Rt runOperationt addCallback(Rtd((Rs</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyttearDown,scCst|j|((Rs</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR)hs  cs”d}g}x:tˆjƒD])}d|}|jˆjj|ƒƒqWtj|ƒ}‡fd†}|j|ƒ‡fd†}|j|ƒ|S(Nsselect count(1) from simples insert into simple(x) values(%d)csd}ˆjj|ƒ}|S(Nsselect x from simple order by x(RR6(RR?R(R(s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyt_select{scs{ˆjt|ƒˆjkdƒxUtˆjƒD]D}ˆjt||ƒdkdƒˆj||d|kdƒq/WdS(NsWrong number of rowsisWrong size rowisValues not returned.(Rtlentnum_iterationstrange(trowsti(R(s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR>‚s  (RCRBtappendRRRt gatherResultsR(RRR?tinsertsRERR@R>((Rs</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR*rs   cs/ˆjjˆjƒ}|j‡fd†ƒ|S(Ncsˆj|dƒS(Ntdone(t assertEquals(R(R(s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRs(RR8t interactionR(RRR((Rs</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR+Œscs/ˆjjˆjƒ}|j‡fd†ƒ|S(Ncsˆj|dƒS(NRI(RJ(R(R(s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR•s(RR:twithConnectionR(RRR((Rs</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR,’scCs|jj|jƒ}|S(N(RR:tclose_withConnection(RRR((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR-˜scsxg}x:tˆjƒD])}d|}|jˆjj|ƒƒqWtj|dtƒ}‡fd†}|j|ƒ|S(Ns!select x from simple where x = %dtfireOnOneErrbackcsDx=tˆjƒD],}ˆj||ddd|kdƒqWdS(NiisValue not returned(RCRBR(tresultRE(R(s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR>¤s( RCRBRFRR6Rt DeferredListRR(RRtdsRER?tdlistR>((Rs</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR.s  cCs\g}x:t|jƒD])}d|}|j|jj|ƒƒqWtj|dtƒ}|S(Nsdelete from simple where x = %dRN(RCRBRFRRRRPR(RRRQRER?RR((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR/ªs  cs8d}ˆjj|ƒ}‡fd†}|j|ƒ|S(Nsselect count(1) from simplecs2ˆjt|ddƒdkdƒˆjƒdS(Nis)Didn't successfully delete table contents(RR<t checkConnect(R=(R(s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR>·s(RR6R(RRR?RR>((Rs</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR0³s  cCsê|jjƒ}|j|ƒ|jƒ}|jdƒ|jdƒ|jƒ}|jt|ƒdƒ|jt|dƒdƒ|j|dddƒ|jdƒ|jdƒ|jt|jƒƒdƒ|jƒ|jj |ƒdS(s/Check the connect/disconnect synchronous calls.sinsert into simple(x) values(1)sselect x from simpleiisdelete from simpleN( RtconnectR!tcursortexecutetfetchalltfailUnlessEqualRARt disconnect(RRtcursR((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRS¾s        cCsm|jdƒxYt|jƒD]H}|jƒ}|jt|ƒdkdƒ|j|d|kdƒqWdS(Nsselect x from simple order by xisWrong size rowisValue not returned.RI(RVRCRBtfetchoneRRA(Rt transactionRER=((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRKÏs   cCs*|jr|jdƒn|jdƒdS(Nsinsert into simple(x) values(0)sselect * from NOTABLE(t can_rollbackRV(RR\((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR9Ùs cCs‹|jƒ}zm|jdƒxYt|jƒD]H}|jƒ}|jt|ƒdkdƒ|j|d|kdƒq,WWd|jƒXdS(Nsselect x from simple order by xisWrong size rowisValue not returned.RI(RURVRCRBR[RRAR(RRRZRER=((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRLßs   " cCs|jƒdS(N(R(RR((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRMíscCs/|jƒ}z|jdƒWd|jƒXdS(Nsselect * from NOTABLE(RURVR(RRRZ((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR;ðs N( t__name__t __module__t__doc__RR tIReactorThreadsRR3tskipRRRR!R1R%R&R'R(R)R*R+R,R-R.R/R0RSRKR9RLRMR;(((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR s6                 tReconnectTestBasecBsweZdZejed ƒd kr-dZnd„Zd„Z d„Z d„Z d„Z d„Z d„Zd „ZRS( s1Test the asynchronous DB-API code with reconnect.s5ADB-API requires threads, no way to test without themcCsl|jdkr!tjdƒ‚n|jƒ|jdddtd|jƒ|_|jjƒ|jj t ƒS(s¹ Skip the test if C{good_sql} is unavailable. Otherwise, set up the database, create a connection pool pointed at it, and set up a simple schema in it. sno good sql for reconnect testtcp_maxit cp_reconnectt cp_good_sqlN( tgood_sqlR3RtSkipTestR RRRRRR#(R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRþs  csBˆjjdƒ}|j‡fd†ƒ|j‡fd†ƒ|S(NsDROP TABLE simplecs ˆjjƒS(N(RR(R(R(s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRscs ˆjƒS(N(R(R(R(s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRs(RRR(RR((Rs</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR scCsotjdƒ}|j|jƒ|j|jƒ|jsK|j|jƒn|j|jƒ|j|j ƒ|S(N( RtsucceedR3Rt _testPool_1R)tearly_reconnectR*R+R,(RR((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR1s cs8d}ˆjj|ƒ}‡fd†}|j|ƒ|S(Nsselect count(1) from simplecs(ˆjt|ddƒdkdƒdS(NisTable not empty(RR<(R=(R(s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR> s(RR6R(RRR?RR>((Rs</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRjs  cCs|jjjƒdjƒdS(Ni(Rt connectionstvaluesR(RR((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR)%scs>d}tjˆjj|ƒ}|j‡fd†d„ƒ|S(Nsselect count(1) from simplecs ˆjdƒS(Ns no exception(R2(R(R(s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR,scSsdS(N(R3(R4((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR-s(RR5RR6R7(RRR?R((Rs</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR*)s  cs8d}ˆjj|ƒ}‡fd†}|j|ƒ|S(Nsselect count(1) from simplecs(ˆjt|ddƒdkdƒdS(NisTable not empty(RR<(R=(R(s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR>3s(RR6R(RRR?RR>((Rs</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR+0s  csNˆjƒd}tjˆjj|ƒ}|j‡fd†‡fd†ƒ|S(Nsselect * from NOTABLEcs ˆjdƒS(Ns no exception(R2(R(R(s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR<scsˆj|jtƒƒS(N(tfailIftcheckR(R4(R(s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR=s(R"RR5RR6R7(RRR?R((Rs</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR,8s  N(R^R_R`R RaRR3RbRRR1RjR)R*R+R,(((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRcøs       tDBTestConnectorcBs›eZdZd ZdZdZdZd Ze Z e Z e Z e Z e ZejZe Ze ZdZd„Zd„Zd„Zd„Zd„Zd„ZRS( s9A class which knows how to test for the presence of and establish a connection to a relational database. To enable test cases which use a central, system database, you must create a database named DB_NAME with a user DB_USER and password DB_PASS with full access rights to database DB_NAME. t twisted_testi2cCsN|jƒ|_tj|jƒ|jƒsDtjd|jƒ‚n|jƒS(Ns%s: Cannot access db( tmktemptDB_DIRtostmkdirt can_connectRRht TEST_PREFIXR(R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pytsetUp^s  cCs tƒ‚dS(sXReturn true if this database is present on the system and can be used in a test.N(tNotImplementedError(R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRvescCsdS(s+Take any steps needed to bring database up.N((R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR jscCsdS(sBring database down, if needed.N((R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRnscKs8|jƒ\}}|jƒ}|j|ƒt||ŽS(s;Create a connection pool with additional keyword arguments.(t getPoolArgstcopytupdateR(Rtnewkwtargstkw((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRrs  cCs tƒ‚dS(s™Return a tuple (args, kw) of list and keyword arguments that need to be passed to ConnectionPool to create a connection to this database.N(Ry(R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRzysN(R^R_R`R3RwtDB_NAMEtDB_USERtDB_PASSRsRtnulls_okttrailing_spaces_okR]R$tescape_slashesRRgRkt can_clearRBRxRvR RRRz(((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRpAs*      tGadflyConnectorcBsGeZdZeZeZeZdZdZd„Z d„Z d„Z RS(tGadflysselect * from simple where 1=0icCs@yddl}WntSXt|ddƒs<|j|_ntS(NiÿÿÿÿRT(tgadflytFalsetgetattrR3RTR(RR‰((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRv‰scCs_ddl}|jƒ}|j|j|jƒ|jƒ}|jdƒ|jƒ|jƒdS(Niÿÿÿÿscreate table x (x integer)(R‰tstartupR€RsRURVtcommitR(RR‰RRU((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR s     cCs,d|j|jf}idd6}||fS(NR‰iRd(R€Rs(RR~R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRz›s ( R^R_RwRŠRƒR]R…RgRBRvR Rz(((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR‡s  tSQLiteConnectorcBs5eZdZeZdZd„Zd„Zd„ZRS(tSQLiteicCsyddl}WntSXtS(Niÿÿÿÿ(tsqliteRŠR(RR((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRv§s cCsJtjj|j|jƒ|_tjj|jƒrFtj|jƒndS(N(RttpathtjoinRsR€tdatabasetexiststunlink(R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR ¬scCs'd}i|jd6dd6}||fS(NRR“iRd(ssqlite(R“(RR~R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRz±s( R^R_RwRŠR…RBRvR Rz(((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRŽ s   tPyPgSQLConnectorcBs eZdZd„Zd„ZRS(tPyPgSQLcCsgyddlm}WntSXy9|jd|jd|jd|jƒ}|jƒtSWntSXdS(Niÿÿÿÿ(tPgSQLR“tusertpassword( tpyPgSQLR˜RŠRTR€RR‚RR(RR˜R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRv¹s  cCs;d}i|jd6|jd6|jd6dd6}||fS(Ns pyPgSQL.PgSQLR“R™Ršitcp_min(s pyPgSQL.PgSQL(R€RR‚(RR~R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRzÄs(R^R_RwRvRz(((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR–¶s tPsycopgConnectorcBs eZdZd„Zd„ZRS(tPsycopgcCscyddl}WntSXy9|jd|jd|jd|jƒ}|jƒtSWntSXdS(NiÿÿÿÿR“R™Rš(tpsycopgRŠRTR€RR‚RR(RRŸR((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRvÍs  cCs;d}i|jd6|jd6|jd6dd6}||fS(NRŸR“R™RšiRœ(spsycopg(R€RR‚(RR~R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRzØs(R^R_RwRvRz(((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRÊs tMySQLConnectorcBs2eZdZeZeZeZd„Zd„ZRS(tMySQLcCscyddl}WntSXy9|jd|jd|jd|jƒ}|jƒtSWntSXdS(NiÿÿÿÿtdbR™tpasswd(tMySQLdbRŠRTR€RR‚RR(RR¤R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRvås  cCs4d}i|jd6|jd6|jd6}||fS(NR¤R¢R™R£(sMySQLdb(R€RR‚(RR~R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRzðs$( R^R_RwRŠR„R]RkRvRz(((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR Þs  tFirebirdConnectorcBsPeZdZeZeZdZeZdZ d„Z d„Z d„Z d„Z RS(tFirebirdicCsFyddl}WntSXy|jƒ|jƒtSWntSXdS(Niÿÿÿÿ(t kinterbasdbRŠR RR(RR§((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRvÿs  cCsddl}tjj|jtjƒ|_tj|jtj tj tj ƒd}||j|j |j f;}|j|ƒ}|jƒdS(Niÿÿÿÿs,create database "%s" user "%s" password "%s"(R§RtR‘R’RsRpR€tchmodtstattS_IRWXUtS_IRWXGtS_IRWXORR‚tcreate_databaseR(RR§R?R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR s $cCs;d}i|jd6dd6|jd6|jd6}||fS(NR§R“s 127.0.0.1thostR™Rš(s kinterbasdb(R€RR‚(RR~R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRzsc CsGddl}|jd|jddd|jd|jƒ}|jƒdS(NiÿÿÿÿR“R®s 127.0.0.1R™Rš(R§RTR€RR‚t drop_database(RR§R((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyRs   N(R^R_RwRŠR$R…R3RgR†RBRvR RzR(((s</usr/lib/python2.7/dist-packages/twisted/test/test_adbapi.pyR¥õs cCshttttttg}xI|D]A}|j|}tj|||t j f|j ƒ}|||s8$ÝI>!,  !