Edit File: fixtures.cpython-37.opt-1.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__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 )ZstrategyZexecutor)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>.configurer2 ) �pop�io�BytesIO�StringIOr- r r2 r Zpatch�object)r1 r( r2 r )r0 r1 r �capture_context_bufferE s r8 �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< rB rC r )rA r r �buffer_f s rD 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" rE r rH r )r( r r �ctxz s rI rA � literal_bindsc s: t | t�rt| �} | js t�t| j�d��}� �|� d S )N)r# )� isinstancer r Zsupports_executionr r r& rB )rG Z multiparamZparamr' )r( � ctx_dialectr r �execute� s zop_fixture.<locals>.execute)r# rM r, ) r Zsqla_092r r r7 r r r ZMockr �alembic�op�_proxy) r# rA r: rJ ZoptsrD rI rM Z connection�contextr )rA r( rL r � op_fixtureZ s2 rR )r9 FNF)&r4 r? Z sqlalchemyr r r rN 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_094r7 r Zsqlalchemy.testing.fixturesr+ Z_engsr8 rR r r r r �<module> s.