3 (h @sZddlZddlZddlZddlZddlZddlZddlZddlmZGdddej Z dS)N)CPYTHONc @seZdZejjejjedZejj erJe eZ e j e ZWdQRXn ddddddddddd d gZd d Zd dZddZdddZddZddZdS)PyMySQLTestCasezdatabases.jsonN localhostrootZ test_pymysqlT)hostuserpasswddbZ use_unicodeZ local_infileZ test_pymysql2)rr r r cCs4|j}tddtjd|jdddD}||kS)zReturn True if the given connection is on the version given or greater. e.g.:: if self.mysql_server_is(conn, (5, 6, 4)): # do something for MySQL 5.6.4 and above css"|]}|dk rt|ndVqdS)Nr)int).0digr7/tmp/pip-install-q3hcpn_q/PyMySQL/pymysql/tests/base.py %sz2PyMySQLTestCase.mysql_server_is..z(\d+)\.(\d+)\.(\d+)r)Zget_server_infotuplerematchgroup)selfconnZ version_tupleZserver_versionZserver_version_tuplerrrmysql_server_iss zPyMySQLTestCase.mysql_server_iscCs:g|_x"|jD]}|jjtjf|qW|j|jdS)N) connections databasesappendpymysqlconnect addCleanup_teardown_connections)rparamsrrrsetUp+s zPyMySQLTestCase.setUpcCsx|jD] }|jqWdS)N)rclose)r connectionrrrr!1s z%PyMySQLTestCase._teardown_connectionsc Cs`|j}tj tjd|jd|fWdQRX|j||j|r\|j|j||dS)zcreate a table. Ensures any existing version of that table is first dropped. Also adds a cleanup rule to drop the table after the test completes. ignorezdrop table if exists `%s`N)cursorwarningscatch_warnings simplefilterexecuter$r drop_table)rr% tablenameZddlcleanupr'rrrsafe_create_table5s   z!PyMySQLTestCase.safe_create_tablec CsB|j}tj tjd|jd|fWdQRX|jdS)Nr&zdrop table if exists `%s`)r'r(r)r*r+r$)rr%r-r'rrrr,Gs   zPyMySQLTestCase.drop_tablecCstjtstjdS)zfEnsure cycles are collected via gc. Runs additional times on non-CPython platforms. N)gcZcollectr)rrrrsafe_gc_collectNszPyMySQLTestCase.safe_gc_collect)T)__name__ __module__ __qualname__ospathjoindirname__file__fnameexistsopenfjsonloadrrr#r!r/r,r1rrrrr s    r) r0r>r5rr(Z unittest2r_compatrZTestCaserrrrrs