Edit File: 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.