5idZddlZddlmZddlmZddlZddlZddlZddl Z ddl Z ddl Z ddl m Z mZmZdefdZdZd d d Zy) z Private utilities for testing N)NativeResource)join_all_native_threads)ClientBootstrapDefaultHostResolverEventLoopGroupreturnc*tjS)a Returns number of bytes currently allocated by awscrt's native code. `AWS_CRT_MEMORY_TRACING `environment variable must be set before module is loaded, or 0 will always be returned. Legal values are: * `AWS_CRT_MEMORY_TRACING=0`: No tracing * `AWS_CRT_MEMORY_TRACING=1`: Only track allocation sizes and total allocated * `AWS_CRT_MEMORY_TRACING=2`: Capture callstacks for each allocation )_awscrtnative_memory_usagen/home/marpiech/ifpan-abm-pgxpred/analysis/marpiech-gwas-test/venv/lib/python3.12/site-packages/awscrt/_test.pyr r s  & & ((r c*tjS)a4 If there are outstanding allocations from awscrt's native code, dump them to log, along with any information gathered based on the tracing level. In order to see the dump, logging must initialized at `LogLevel.Trace` and the `AWS_CRT_MEMORY_TRACING` environment variable must be non-zero when module is loaded. Legal values are: * `AWS_CRT_MEMORY_TRACING=0`: No tracing * `AWS_CRT_MEMORY_TRACING=1`: Only track allocation sizes and total allocated * `AWS_CRT_MEMORY_TRACING=2`: Capture callstacks for each allocation )r native_memory_dumpr r rdump_native_memoryr!s  % % ''r g$@ timeout_secc tjtjtjt j ddk7r t dtjs t dtj|z}tj|krktjstdkDsytjtdtjdtj|krkt!tj}|rmt j d d k( r t#d n t#d  fd }tjD]#}|d|t%j&|dz }g}tj(|D]}}t+|t,j.rPt1j2|}|j4j7dxs|j4j7d} | rm|j9|t#d|t#dt!||D]B}t+|t,j.r|dt1j2|:|d|D&t} | dkDr$t#dj;| t=t dj;|| )a Checks that all awscrt resources have been freed after a test. If any resources still exist, debugging info is printed and an exception is raised. Requirements: * `awscrt.NativeResource._track_lifetime = True`: must be set before test begins to ensure accurate tracking. * `AWS_CRT_MEMORY_TRACING=2`: environment variable that must be set before any awscrt modules are imported, to ensure accurate native leak checks. * `AWS_CRT_MEMORY_PRINT_SECRETS_OK=1`: optional environment variable that will print the full contents of leaked python objects. DO NOT SET THIS if the test results will be made public as it may result in secrets being leaked. AWS_CRT_MEMORY_TRACING2zRenvironment variable AWS_CRT_MEMORY_TRACING=2 must be set for accurate leak checkszOawscrt.NativeResource._track_lifetime=True must be set for accurate leak checksrNg?rAWS_CRT_MEMORY_PRINT_SECRETS_OK1zLeaking NativeResources:zaLeaking NativeResources (set AWS_CRT_MEMORY_PRINT_SECRETS_OK=1 env var for more detailed report):cr.t|}t|dkDr|dddz}t||yt|t|y)Niz...TRUNCATED PRINT)strlenprinttype)prefixobjsleak_secrets_oks r _printobjz"check_for_leaks.._printobjcsCHq6D=%4#77Afa fd3i(r -z_weakrefset.pyzawscrt/_test.pyz sys.getrefcount():z gc.referrers():z -zALeaking {} bytes native memory (enable Trace logging to see more)zKawscrt leak check failed. {} NativeResource objects. {} bytes native memory)rrelease_static_defaultrrosgetenv RuntimeErrorr_track_lifetimetime_livingr gccollectrsleeprrsys getrefcount get_referrers isinstancetypes FrameTypeinspect getframeinfofilenameendswithappendformatr) r wait_untilnum_living_resourcesr"irefcount referrersr frameinfo our_fault mem_bytesr!s @rcheck_for_leaksrD3sf$**,))+..0 yy)*c1opp  ) )lmm{*J ))+ "%%.A.Ca.G  C0 3 ))+ "~556))$EF#M  , - u v ) '' (A c1 q)A-HI%%a( $a1 ' 4 4Q 7I!*!3!3!rJsI!1  JJ )S ) ($$(`*r