U
|
¸ý°d`9 ã @ s d dl mZ d dlmZ d dlZd dlZd dlmZ d dlZd dl Z d dl m
|
Z
|
d dl Z ddl m Z ddl mZ d d
|
l mZ d d lmZ d d lmZ d d lmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dlmZ d dl m Z d dl m Z d dl m Z d dl m Z e s0dd Z!e! Z"Z#nej$Z"dd Z#d d Z%dJd d Z&d d! Z'G d"d# d#e(Z)d$d% Z*d&d' Z+d(d) Z,d*d+ Z-d,d- Z.d.d/ Z/e d0d1 Z0d2d3 Z1d4d5 Z2d6d7 Z3dKd9d:Z4d;d< Z5G d=d> d>e6Z7d?d@ Z8dLdBdCZ9dDdE Z:dFdG Z;dHdI Z<dS )Mé )Ú annotations)ÚdequeN)Úchain)Ú getsizeofé )Úconfig)Úmocké )Úinspect)Ú
|
Connection)ÚColumn)ÚDropConstraint)Ú DropTable)ÚForeignKeyConstraint)ÚMetaData)ÚTable©Úschema)ÚInteger©Ú decorator)Ú defaultdict)Úhas_refcount_gc)Úinspect_getfullargspecc G s t ¡ t ¡ d S ©N)ÚgcÚcollect)Úargs© r úNd:\z\workplace\vscode\pyvenv\venv\Lib\site-packages\sqlalchemy/testing/util.pyÚnon_refcount_gc_collect) s r c C s d S r r r r r r Úlazy_gc2 s r! c # sR t } dd l}| |¡ | D ]0 td|jd D ] j fddfV q0q d S )Nr éþÿÿÿr c s | ¡S r )Údumps©Úd©Zpickle_Úprotocolr r Ú<lambda>? ó zpicklers.<locals>.<lambda>)ÚsetÚpickleÚaddÚrangeÚHIGHEST_PROTOCOLÚloads)Úpicklersr+ r r& r r0 6 s
|
r0 c C s t j| |dS )N)Úk)ÚrandomÚchoices)Ú
|
populationr1 r r r Úrandom_choicesB s r5 c C s< t | trt| |S | t dd| ¡ tj¡td| S )NÚ1Ú0é
|
)Ú
|
isinstanceÚfloatÚroundÚdecimalÚDecimalÚ to_integralÚ ROUND_FLOORÚpow)ÚvalueÚprecr r r Ú round_decimalF s
|
|
ÿþrC c @ s< e Zd Zdd Zdd Zdd Zdd Zd d
|
Zd d Zd S )Ú RandomSetc C s t t | ¡}t |¡ t|S r )Úlistr* Ú__iter__r2 ÚshuffleÚiter)ÚselfÚlr r r rF Q s
|
zRandomSet.__iter__c C s4 t dt| d ¡}tt | ¡| }| |¡ |S )Nr r )r2 ÚrandintÚlenrE r* rF Úremove)rI ÚindexÚitemr r r ÚpopV s
|
z RandomSet.popc C s t t | |¡S r )rD r* Úunion©rI Úotherr r r rQ \ s zRandomSet.unionc C s t t | |¡S r )rD r* Ú
|
differencerR r r r rT _ s zRandomSet.differencec C s t t | |¡S r )rD r* Ú intersectionrR r r r rU b s zRandomSet.intersectionc C s t | S r )rD ©rI r r r Úcopye s zRandomSet.copyN) Ú__name__Ú
|
__module__Ú __qualname__rF rP rQ rT rU rW r r r r rD P s rD c C s` t t}| D ]\}}|| |¡ q t|D ].\}}||d
D ]}||| kr@ dS q@q,dS )zATrue if the given sorting conforms to the given partial ordering.NFT)r r* r, Ú enumerate)ÚtuplesZsorted_elementsÚdepsÚparentÚchildÚiÚnodeÚnr r r Úconforms_partial_orderingi s rc c s> t t| D ]\}}| |¡ q fdd t |S )Nc 3 s^ t | dkrt| V nB| D ]<}t| |g¡}| | ¡s |D ]}|g| V qFq d S )Nr )rL rE r* rT rU )ÚelementsÚelemZsubsetZ sub_ordering©Ú_all_orderingsÚedgesr r rg | s z-all_partial_orderings.<locals>._all_orderings)r r* r, rH )r\ rd r^ r_ r rf r Úall_partial_orderingsw s
|
ri c
|
C s> z
|
|| _ W n. tk
|
r8 t | j| j|| j| j¡} Y nX | S )ac Return a function with a given __name__.
|
|
Will assign to __name__ and return the original function if possible on
|
the Python implementation, otherwise a new function will be constructed.
|
|
This function should be phased out as much as possible
|
in favor of @decorator. Tests that "generate" many named tests
|
should be modernized.
|
|
)rX Ú TypeErrorÚtypesÚ FunctionTypeÚ__code__Ú __globals__Ú __defaults__Ú __closure__)ÚfnÚnamer r r Úfunction_named s
|
ÿ
|
rs c O s` | ¡ }z$||f||}| ddd¡ |W S t ¡ }| j| }|sN n| Y S Y nX dS )a Run the given function under the given contextmanager,
|
simulating the behavior of 'with' to support older
|
Python versions.
|
|
This is not necessary anymore as we have placed 2.6
|
as minimum Python version, however some tests are still using
|
this structure.
|
|
N)Ú __enter__Ú__exit__ÚsysÚexc_info)Úctxrq ÚargÚkwÚobjÚresultrw Zraise_r r r Úrun_as_contextmanager s
|
r} c C s dd | D S )zConverts the results of sql execution into a plain set of column tuples.
|
|
Useful for asserting the results of an unordered query.
|
c S s h | ] }t |qS r ©Útuple)Ú.0Úrowr r r Ú <setcomp>½ s zrowset.<locals>.<setcomp>r )Úresultsr r r Úrowset· s r c C s ds t | d S )NF)ÚAssertionError)Úmsgr r r ÚfailÀ s r c O s ddl m} t ¡ }|d }t|dd}||_z| ||W ¢S | ¡ |jrht||j |j ¡
|
¡ n t|t j ||_X dS )ztProvide bound MetaData for a single test, dropping afterwards.
|
|
Legacy; use the "metadata" pytest fixture.
|
|
r )Úfixturesr ÚmetadataN) Ú r r r Úgetattrr Z_close_all_sessionsZ _connection_fixture_connectionÚ drop_all_tables_from_metadataZget_transactionÚcommitr Údb)rq r rz r r rI Z prev_metar r r Úprovide_metadataÄ s ÿ r c sT t | D ]} |¡ q
|
t tj fdd| D ddt d ¡dS )a A facade around @testing.combinations() oriented towards boolean
|
keyword-based arguments.
|
|
Basically generates a nice looking identifier based on the keywords
|
and also sets up the argument names.
|
|
E.g.::
|
|
@testing.flag_combinations(
|
dict(lazy=False, passive=False),
|
dict(lazy=True, passive=False),
|
dict(lazy=False, passive=True),
|
dict(lazy=False, passive=True, raiseload=True),
|
)
|
|
|
would result in::
|
|
@testing.combinations(
|
('', False, False, False),
|
('lazy', True, False, False),
|
('lazy_passive', True, True, False),
|
('lazy_passive', True, True, True),
|
id_='iaaa',
|
argnames='lazy,passive,raiseload'
|
)
|
|
c s< g | ]4 d fddD ¡ft fddD qS )Ú_c 3 s | ]} |d ¡r|V qdS ©FN©Úget©r r1 r$ r r Ú <genexpr> s z/flag_combinations.<locals>.<listcomp>.<genexpr>c 3 s | ]} |d ¡V qdS r r r r$ r r r s )Újoinr )r ©Úkeysr$ r Ú
|
<listcomp> s þÿz%flag_combinations.<locals>.<listcomp>r` Úaú,)Zid_Úargnames)r* ÚupdateÚsortedr Ú combinationsrL r )r r% r r r Úflag_combinationsñ s
|
ýùr c sL t }dd |d D }fdd tj fddtt|D |S )Nc S s g | ] }t ¡ qS r )r ZMock©r ry r r r r $ s z'lambda_combinations.<locals>.<listcomp>r c s fdd}d |_ |S )Nc s f | S r r ©rz )Úlambda_arg_setsÚposr r Úfixture' s z<lambda_combinations.<locals>.create_fixture.<locals>.fixturez fixture_%3.3d)rX )r¤ r¥ )r£ )r¤ r Úcreate_fixture& s
|
z+lambda_combinations.<locals>.create_fixturec s g | ]} |fqS r r )r r` )r¦ r r r . s )r r r r- rL )r£ rz r Zarg_setsr )r¦ r£ r Úlambda_combinations! s ÿr§ c sJ t | d } fdd|D }t| j}| ¡ t | j|¡}|f |S )zÝGiven a no-arg lambda and a namespace, return a new lambda that
|
has all the values filled in.
|
|
This is used so that we can have module-level fixtures that
|
refer to instance-level variables using lambdas.
|
|
r c s i | ]}| |¡qS r )rP r¡ r¢ r r Ú
|
<dictcomp>< s z"resolve_lambda.<locals>.<dictcomp>)r Údictrn r rk rl rm )Z__fnrz Zpos_argsZ pass_pos_argsZglbZnew_fnr r¢ r Úresolve_lambda2 s
|
|
rª Úfunctionc s fdd}|S )z&Provide MetaData for a pytest fixture.c s fdd}t jd|S )Nc 3 sB t ¡ }| _z | |}| tj¡ |V W 5 | tj¡ X d S r )r r r Údrop_allr r Z
|
create_all)rI r r| ©rq r r Úrun_ddlG s
|
|
z3metadata_fixture.<locals>.decorate.<locals>.run_ddl)Zscope)r r¥ )rq r® ©Úddlr r ÚdecorateF s z"metadata_fixture.<locals>.decorater )r° r± r r¯ r Úmetadata_fixtureC s r² c s t fdd}|S )ziForce the given table names to be dropped after test complete,
|
isolating for foreign key cycles
|
|
c s, z| ||W ¢S t tjttj d X d S )N)Ú include_names)Údrop_all_tablesr r r
|
)rq r rz ©Únamesr r Úgo^ s z force_drop_names.<locals>.gor )r¶ r· r rµ r Úforce_drop_namesX s r¸ c @ s$ e Zd ZdZdd Zdd ZeZdS )Úadictz,Dict keys available as attributes. Shadows.c C s0 z
|
| | W S t k
|
r* t | |¡ Y S X d S r )ÚKeyErrorr© Ú__getattribute__)rI Úkeyr r r r» l s
|
zadict.__getattribute__c s t fdd|D S )Nc s g | ] } | qS r r )r r¼ rV r r r s s z"adict.__call__.<locals>.<listcomp>r~ )rI r r rV r Ú__call__r s zadict.__call__N)rX rY rZ Ú__doc__r» r½ Úget_allr r r r r¹ i s r¹ c sN ddl m fdd}t|tsB| ¡ }|| W 5 Q R X n|| d S )Nr )Úenginesc sT j | ¡ | jjsFddlm} |jddd | ¡ W 5 Q R X n
|
| ¡ d S )Nr )Ú
|
assertionszCan't sort tablesF)Zassert_)Ztesting_reaperZprepare_for_drop_tablesÚdialectÚsupports_alterr rÁ Zexpect_warningsr¬ )Ú
|
connectionrÁ ©rÀ r r r r· { s ÿz)drop_all_tables_from_metadata.<locals>.go)r rÀ r9 r Úbegin)r Zengine_or_connectionr· rÄ r rÅ r r x s
|
|
r r c C s |d k rt |}|d k r.|dks(td|f}| ¡ Ü}t|j|dD ]Ä\}}|r|d k rj|d |krjqH| tt|d t |d d¡ qH|rH| j j
|
sqH|D ]j\}} |d k r¾|d |kr¾q t|d t t dt t dt |d d}
|
| t t|
|
jjg|
|
jjg| d ¡ q qHW 5 Q R X d S )
|
Nr z2consider_schemas and schema are mutually exclusive)Úconsider_schemasr r r ÚxÚy)rr )r* r
rÆ ÚreversedZ%sort_tables_on_foreign_key_dependencyÚexecuter r r r rà r r r r ÚcrÈ rÉ ) ÚengineZ inspectorr rÇ r³ ÚconnZ table_keyZfkcsZt_keyZfkcÚtbr r r r´ sZ
|
þ
|
ÿÿ ÿ
|
þÿÿ ÿ
|
þûÿÿr´ c s t fdd}|S )Nc s z| ||W ¢S ¡ X d S r )Ú_clear)rq ry rz ©Ú event_clsr r r± É s z!teardown_events.<locals>.decorater )rÒ r± r rÑ r Úteardown_eventsÈ s rÓ c sL dd }t tttttt|tttti t td fdd| S )zReturns the approximate memory footprint an object and all of its
|
contents.
|
|
source: https://code.activestate.com/recipes/577504/
|
|
|
c S s t | ¡ ¡S r )r Ú from_iterableÚitemsr$ r r r Ú dict_handlerÜ s z total_size.<locals>.dict_handlerr c sb t | krdS t | ¡ t| } ¡ D ],\}}t| |r0|tt|| 7 } q^q0|S )Nr )Úidr, r rÕ r9 ÚsumÚmap)ÚoÚsÚtypÚhandler©Z all_handlersZ default_sizeÚseenÚsizeofr r rà ê s
|
|
ztotal_size.<locals>.sizeof)r rH rE r r© r* Ú frozensetr )rÚ rÖ r rÞ r Ú
|
total_sizeÓ s$ ú râ c C sV | g}t }d}|rR| d¡}||kr0|d7 }qt|tr|rt||g | }q|S )zgiven a cache key tuple, counts how many instances of actual
|
tuples are found.
|
|
used to alert large jumps in cache key complexity.
|
|
r r )ÚobjectrP r9 r rE )ÚtupÚstackÚsentinelZ num_elementsre r r r Úcount_cache_key_tuplesù s
|
|
|
rç )r )r« )Nr N)=Ú
|
__future__r Ú collectionsr r< r Ú itertoolsr r2 rv r rk r r r r
|
rÍ r r r r r r r r ZsqlZ sql.sqltypesr Úutilr r r r r Z
|
gc_collectr! r r0 r5 rC r* rD rc ri rs r} r r r r r§ rª r² r¸ r© r¹ r r´ rÓ râ rç r r r r Ú<module>
|
sn
|
|
|
|
,0
|
û
|
9 &
|
