File: //opt/alt/python38/lib/python3.8/site-packages/pymysql/tests/__pycache__/base.cpython-38.opt-1.pyc
U
�
V�
� @ sZ d dl Z d dlZd dlZd dlZd dlZd dlZd dlZddlmZ G dd� dej �Z
dS )� N� )�CPYTHONc @ s� e Zd Zej�ej�e�d�Zej� e�rJe
e��Z e �
e �ZW 5 Q R X n ddddddd�ddddd �gZd
d � Zd d
� Zdd� Zddd�Zdd� Zdd� ZdS )�PyMySQLTestCasezdatabases.jsonZ localhost�root� Z test_pymysqlT)�host�user�passwd�dbZ use_unicodeZ local_infileZ
test_pymysql2)r r r r
c C s4 |� � }tdd� t�d|��ddd�D ��}||kS )z�Return 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
c s s" | ]}|d k rt |�ndV qd S )Nr )�int)�.0�dig� r �C/opt/alt/python38/lib/python3.8/site-packages/pymysql/tests/base.py� <genexpr>$ s �z2PyMySQLTestCase.mysql_server_is.<locals>.<genexpr>z(\d+)\.(\d+)\.(\d+)� r � )Zget_server_info�tuple�re�match�group)�selfZconnZ
version_tupleZserver_versionZserver_version_tupler r r �mysql_server_is s
�z PyMySQLTestCase.mysql_server_isc C s6 g | _ | jD ]}| j �tjf |�� q | �| j� d S �N)� connections� databases�append�pymysqlZconnect�
addCleanup�_teardown_connections)r Zparamsr r r �setUp+ s
zPyMySQLTestCase.setUpc C s | j D ] }|�� qd S r )r �close)r �
connectionr r r r 1 s
z%PyMySQLTestCase._teardown_connectionsc C s` |� � }t�� � t�d� |�d|f � W 5 Q R X |�|� |�� |r\| �| j||� dS )z�create 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.
�ignore�drop table if exists `%s`N)�cursor�warnings�catch_warnings� simplefilter�executer! r �
drop_table)r r" � tablenameZddlZcleanupr% r r r �safe_create_table5 s
z!PyMySQLTestCase.safe_create_tablec C sB |� � }t�� � t�d� |�d|f � W 5 Q R X |�� d S )Nr# r$ )r% r& r' r( r) r! )r r" r+ r% r r r r* G s
zPyMySQLTestCase.drop_tablec C s t �� tst �� dS )zfEnsure cycles are collected via gc.
Runs additional times on non-CPython platforms.
N)�gcZcollectr )r r r r �safe_gc_collectN s z PyMySQLTestCase.safe_gc_collectN)T)�__name__�
__module__� __qualname__�os�path�join�dirname�__file__Zfname�exists�open�f�json�loadr r r r r, r* r. r r r r r
s$
� �
r ) r- r: r2 r r& Z unittest2r Z_compatr ZTestCaser r r r r �<module> s