Current 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 � /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__Z __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 multiparamsZparams)�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>.configurer3 )
�pop�io�BytesIO�StringIOr. r r3 r Zpatch�object)r2 r( r3 r )r1 r2 r �capture_context_bufferE s
r9 �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= rC rD r )rB r r �buffer_f s rE 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" rF r rI r )r( r r �ctxz s rJ rB �
literal_bindsc s: t | t�rt| �} | js t�t| j�d��}� �|� d S )N)r# )�
isinstancer r Zsupports_executionr r r& rC )rH Z
multiparamZparamr' )r( � ctx_dialectr r �execute� s
zop_fixture.<locals>.execute)r# rN r- )
r Zsqla_092r r r8 r
r r ZMockr �alembic�op�_proxy)
r# rB r; rK ZoptsrE rJ rN Z
connection�contextr )rB r( rM r �
op_fixtureZ s2 rS )r: FNF)&r5 r@ Z
sqlalchemyr r r rO Z util.compatr r> r r r Z migrationr
Z environmentr Z
operationsr �
contextlibr
Zplugin.plugin_baser Z
assertionsr r r Z ConfigParserZtesting_config�readZsqla_094r8 r Zsqlalchemy.testing.fixturesr, Z_engsr9 rS r r r r �<module> s.
Mini Shell