File: //opt/alt/python37/lib/python3.7/site-packages/alembic/testing/__pycache__/fixtures.cpython-37.pyc
B
�M!V� � @ s
d dl Z d dlZd dlmZmZmZ d dlZddlmZ ddl m
Z
ddlm Z m Z ddl
mZ ddlmZ dd lmZ d d
lmZ d d lmZ d d
lmZmZ d dl mZ e�� Z e � dg� e
j s�G dd� de �Z n d dl!m Z dd� Z"i Z#edd� �Z$ddd�Z%dS ) � N)�
create_engine�text�MetaData� )� configparser)�util)� string_types� text_type)�MigrationContext)�EnvironmentContext)�
Operations)�contextmanager� )�SkipTest)� _get_dialect�eq_)�mockztest.cfgc @ s: e Zd ZdZdZdZdZdZd dd�Zdd� Z dd� Z
dS )
�TestBase� Nc C s |s t |��d S )N)�AssertionError)�self�val�msgr r �I/opt/alt/python37/lib/python3.7/site-packages/alembic/testing/fixtures.py�assert_+ s zTestBase.assert_c C s t | d�r| �� d S )N�setUp)�hasattrr )r r r r �setup/ s
zTestBase.setupc C s t | d�r| �� d S )N�tearDown)r r )r r r r �teardown3 s
zTestBase.teardown)N) �__name__�
__module__� __qualname__Z
__whitelist__� __requires__Z__unsupported_on__Z __only_on__Z __skip_if__r r r r r r r r s
r )r c s( g � � �fdd�} t dd| d���� fS )Nc s � � t| j�jd��� d S )N)�dialect)�append�str�compiler$ )�sqlZ multiparams�params)�buf�enginer r �dump= s zcapture_db.<locals>.dumpz
postgresql://r )Zstrategy�executor)r )r, r )r* r+ r �
capture_db: s r. c + sh �� dd�rt�� }nt�� }��d|d�� tj� � �fdd�}tj� td|�� |V W d Q R X d S )NZbytes_ioFZsqlite)Z dialect_name�
output_bufferc s |j f �� � | |�S )N)�update)�arg�opt)�conf�kwr r � configureR s z)capture_context_buffer.<locals>.configurer5 )
�pop�io�BytesIO�StringIOr0 r r5 r �patch�object)r4 r* r5 r )r3 r4 r �capture_context_bufferE s
r< �defaultFc
s� i }|r$t jstd��t|d�|d< G � fdd�dt�}|� �G �fdd�dt�}� r^� |d< |rj||d < t| ��� s���fd
d �}tj�|d �}n �|d
< d }|�||�} t | �t
j _ | S )Nz8naming_convention feature requires sqla 0.9.2 or greater)�naming_conventionZtarget_metadatac s( e Zd Zdd� Z� fdd�Zdd� ZdS )zop_fixture.<locals>.buffer_c S s
g | _ d S )N)�lines)r r r r �__init__g s z$op_fixture.<locals>.buffer_.__init__c sF |� � }t�dd|�}� r6t�dd|�}t�dd|�}| j�|� d S )Nz[\n\t]� z z\;
*$)�strip�re�subr? r% )r r )�as_sqlr r �writej s z!op_fixture.<locals>.buffer_.writec S s d S )Nr )r r r r �flushu s z!op_fixture.<locals>.buffer_.flushN)r r! r" r@ rF rG r )rE r r �buffer_f s rH c s0 e Zd Z� fdd�Z� fdd�Z� fdd�ZdS )zop_fixture.<locals>.ctxc s g � j d d �< d S )N)r? )r )r* r r �clear_assertions{ s z(op_fixture.<locals>.ctx.clear_assertionsc s t � jt|�� d S )N)r r? �list)r r( )r* r r r ~ s z op_fixture.<locals>.ctx.assert_c s6 x0� j D ]}||krd S qW ds2td|� j f ��d S )NFz"Could not locate fragment %r in %r)r? r )r r( �stmt)r* r r �assert_contains� s z'op_fixture.<locals>.ctx.assert_containsN)r r! r" rI r rL r )r* r r �ctxz s rM rE �
literal_bindsc s: t | t�rt| �} | js t�t| j�d��}� �|� d S )N)r$ )�
isinstancer r Zsupports_executionr r r' rF )rK Z
multiparam�paramr( )r* � ctx_dialectr r �execute� s
zop_fixture.<locals>.execute)r$ rR r/ )
r Zsqla_092r r r; r
r r ZMockr �alembic�op�_proxy)
r$ rE r> rN �optsrH rM rR Z
connection�contextr )rE r* rQ r �
op_fixtureZ s2 rX )r= FNF)&r7 rC Z
sqlalchemyr r r rS Z util.compatr rA r r r Z migrationr
� environmentr Z
operationsr �
contextlibr
Zplugin.plugin_baser Z
assertionsr r r � ConfigParserZtesting_config�readZsqla_094r; r Zsqlalchemy.testing.fixturesr. Z_engsr< rX r r r r �<module> s.