File: //opt/alt/python38/lib/python3.8/site-packages/pymysql/tests/__pycache__/base.cpython-38.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.json� localhost�root� Z test_pymysqlT)�host�user�passwd�db� use_unicode� 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 � )�get_server_info�tuple�re�match�group)�self�conn�
version_tuple�server_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�pymysql�connect�
addCleanup�_teardown_connections)r �paramsr 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+ � tablenameZddl�cleanupr. 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/ r0 r1 r2 r* )r r+ r4 r. r r r r3 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__�fname�exists�open�f�json�loadr" r r) r' r6 r3 r8 r r r r r
s$
� �
r ) r7 rE r< r r/ Z unittest2r$ �_compatr ZTestCaser r r r r �<module> s