B ‰KLcF)ã@sÄdZddlZddlZddlZddlZddlZddlZddlZddlm Z ddl m Z ddl m Z y ddl Z Wnek r„dZ YnXdZdadd „ZGd d „d eƒZdd d„Zejddd„ƒZdS)z§Profiling support for unit and performance tests. These are special purpose profiling methods which operate in a more fine-grained way than nose's profiling plugin. éNé)Úconfig)Ú gc_collecté)Úhas_compiled_extcCs|atjrt ¡dS)N)Ú _current_testÚ_profile_statsÚ force_writeÚ reset_count)Zid_©r ú;lib/python3.7/site-packages/sqlalchemy/testing/profiling.pyÚ_start_current_test2sr c@s^eZdZdZddd„Zedd„ƒZdd „Zd d „Zd d „Z dd„Z dd„Z dd„Z dd„Z dS)ÚProfileStatsFilezøStore per-platform/fn profiling results in a file. There was no json module available when this was written, but now the file format which is very deterministically line oriented is kind of handy in any case for diffs and merges. Ú cumulativeNcCs†tjdk otjj|_|jp*tjdk o*tjj|_tj |¡|_ tj  |j ¡d|_ t   dd„¡|_||_||_| ¡|jr‚| ¡dS)NéÿÿÿÿcSs t t¡S)N)Ú collectionsÚ defaultdictÚdictr r r r ÚMóz+ProfileStatsFile.__init__..)rZoptionsZforce_write_profilesr Zwrite_profilesÚwriteÚosÚpathÚabspathÚfnameÚsplitÚ short_fnamerrÚdataÚdumpÚsortÚ_readÚ_write)ÚselfÚfilenamerrr r r Ú__init__Cs zProfileStatsFile.__init__cCs´tjjdtjj}tjjdkr:tjj tjj¡r:|d7}d dd„tj dd…Dƒ¡}t   ¡t   ¡  ¡t  ¡  ¡||g}| tjjjrŒd nd ¡tƒ}| |r¤d p¦d ¡d |¡S) NÚ_ZsqliteZ_fileÚ.cSsg|] }t|ƒ‘qSr )Ústr)Ú.0Úvr r r ú bsz1ProfileStatsFile.platform_key..rrZ nativeunicodeZ dbapiunicodeZ cextensionsZ nocextensions)rZdbÚnameZdriverZdialectZ_is_url_file_dbZurlÚjoinÚsysÚ version_infoÚplatformÚmachineÚsystemÚlowerZpython_implementationÚappendZconvert_unicoder)r"Z dbapi_keyZ py_versionZplatform_tokensZ _has_cextr r r Ú platform_keyWs    zProfileStatsFile.platform_keycCst}||jko|j|j|kS)N)rrr4)r"Útest_keyr r r Ú has_statsuszProfileStatsFile.has_statsc Cs¦t}|j|}||j}d|kr.g|d<}n|d}d|krLd|d<}n|d}t|ƒ|k}|s‚| |¡|jr|| ¡d}n|d||f}|dd7<|S)NÚcountsÚ current_countrÚlinenor)rrr4Úlenr3rr!) r"Ú callcountr5Úper_fnÚ per_platformr7r8Z has_countÚresultr r r r>{s$    zProfileStatsFile.resultcCsPt}||jkrdS|j|}|j|kr*dS||j}d|krLg|ddd…<dS)Nr7)rrr4)r"r5r<r=r r r r –s    zProfileStatsFile.reset_countcCs\t}|j|}||j}|d}|d}|t|ƒkrB|||d<n||d<|jrX| ¡dS)Nr7r8rr)rrr4r:rr!)r"r;r5r<r=r7r8r r r Úreplace£s   zProfileStatsFile.replacecCs d|jS)Na # %s # This file is written out on a per-environment basis. # For each test in aaa_profiling, the corresponding function and # environment is located within this file. If it doesn't exist, # the test is skipped. # If a callcount does exist, it is compared to what we received. # assertions are raised if the counts do not match. # # To add a new callcount test, apply the function_call_count # decorator and re-run the tests using the --write-profiles # option - this file will be rewritten including the new count. # )r)r"r r r Ú_header°szProfileStatsFile._headerc Cs®yt|jƒ}Wntk r"dSXx|t|ƒD]p\}}| ¡}|r.| d¡rNq.| ¡\}}}|j|}||}dd„| d¡Dƒ} | |d<|d|d<d|d <q.W| ¡dS) Nú#cSsg|] }t|ƒ‘qSr )Úint)r(Úcountr r r r*Ðsz*ProfileStatsFile._read..ú,r7rr9rr8) ÚopenrÚIOErrorÚ enumerateÚstripÚ startswithrrÚclose) r"Ú profile_fr9Úliner5r4r7r<r=Úcr r r r Ãs    zProfileStatsFile._readcCs¦td|jƒt|jdƒ}| | ¡¡xpt|jƒD]b}|j|}| d|¡xDt|ƒD]8}||}d dd„|dDƒ¡}| d|||f¡qZWq4W| ¡dS) NzWriting profile file %sÚwz # TEST: %s rDcss|]}t|ƒVqdS)N)r')r(rCr r r ú àsz*ProfileStatsFile._write..r7z %s %s %s ) ÚprintrrErr@Úsortedrr,rJ)r"rKr5r<r4r=rMr r r r!Ös  zProfileStatsFile._write)rN)Ú__name__Ú __module__Ú __qualname__Ú__doc__r$Úpropertyr4r6r>r r?r@r r!r r r r r:s    rçš™™™™™©?csNddlm}ddlm‰ddlm‰ddlm‰|‡‡‡‡‡‡fdd„ƒ}|S)a©Assert a target for a test case's function call count. The main purpose of this assertion is to detect changes in callcounts for various functions - the actual number is not as important. Callcounts are stored in a file keyed to Python version and OS platform information. This file is generated automatically for new tests, and versioned so that unexpected changes in callcounts will be detected. r)Ú decorator)Ú deprecations)Úrow)Úmockc s’ˆj ˆdd¡xˆj ˆjdˆj¡XxtˆƒD]}|||Žq2Wtˆƒ}tˆdx|D]}|||Ž}q^W|SQRXWdQRXWdQRXdS)NZSQLALCHEMY_WARN_20FZ_default_key_style)Úvariance)ZpatchÚobjectZ LegacyRowZKEY_OBJECTS_NO_WARNÚrangeÚcount_functions)ÚfnÚargsÚkwZwarmZ timerangeZtimeÚrv)rYr[rZÚtimesr\Úwarmupr r Úwrapùs   z!function_call_count..wrap)Zsqlalchemy.utilrXrYZsqlalchemy.enginerZZsqlalchemy.testingr[)r\rdrerXrfr )rYr[rZrdr\rer Úfunction_call_countås     rgc csttdkrt d¡‚t ¡s0tjs0t dtj¡tƒt  ¡}|  ¡dV|  ¡t j |tjd}|j}t |¡}|dkr‚d}n|\}}td||fƒ|jt dtj¡Ž| ¡tjrtj tj¡\}}t d¡d} d|| |pêd f} | | ¡td | ƒtjrt |¡nV|rpt ||ƒ} t!||ƒ| k} | rptjrVt |¡nt"d ||d |tj#fƒ‚dS) NzcProfile is not installedzŒNo profiling stats available on this platform for this function. Run tests with --write-profiles to add statistics to %s for this platform.)ÚstreamzPstats calls: %d Expected %sz[, ]r&rz%s_%s%sz.profilezDumped stats to file %sz†Adjusted function call count %s not within %s%% of expected %s, platform %s. Rerun with --write-profiles to regenerate this callcount.éd)$ÚcProfilerZ_skip_test_exceptionrr6rZ skip_testrrZProfileÚenableÚdisableÚpstatsZStatsr-ÚstdoutZ total_callsr>rPZ sort_statsÚrerrZ print_statsrrrÚsplitextrZ dump_statsr r?rBÚabsÚAssertionErrorr4) r\ZprZstatsr;ZexpectedZexpected_countZline_noÚbaseZextZ test_nameZdumpfileZdevianceZfailedr r r r_ sP        r_)rWrr)rW)rUrÚ contextlibrr/rmror-ÚrÚutilrrrjÚ ImportErrorrrr r]rrgÚcontextmanagerr_r r r r Ú s,      , (