Z d dl mZ d dlZ ej d Z G d dej Z G d de Z G d d e Z dS )
wrapsNpeeweec e Zd Zd Zd ZdS )_QueryLogHandlerc H g | _ t j j | g|R i | d S N)queriesloggingHandler__init__)selfargskwargss t/builddir/build/BUILD/imunify360-venv-2.6.2/opt/imunify360/venv/lib/python3.11/site-packages/playhouse/test_utils.pyr
z_QueryLogHandler.__init__ s2 777777777 c T |j dk r | j | d S d S )Nr )namer
append)r records r emitz_QueryLogHandler.emit
s3 ;( " " L ' ' ' ' ' # "r N)__name__
__module__ __qualname__r
r r r r r s2 8 8 8( ( ( ( (r r c ( e Zd ZddZd Zd Zd ZdS )
count_queriesFc " || _ d| _ d S )Nr ) only_selectcount)r r s r r
zcount_queries.__init__ s &
r c | j j S r )_handlerr
r s r get_querieszcount_queries.get_queries s
}$$r c t | _ t t j t | j | S r )r r" loggersetLevelr DEBUG
addHandlerr# s r __enter__zcount_queries.__enter__ s> (**
&&&$-((( r c t | j | j r*t d | j j D | _ d S t | j j | _ d S )Nc R g | ]$}|j d d "|%S )r zSELECT )msg
startswith).0qs r
z*count_queries.__exit__..$ sG A A AA!"q!4!4Y!?!? Aa A A Ar )r&
removeHandlerr" r lenr
r )r exc_typeexc_valexc_tbs r __exit__zcount_queries.__exit__! sy T]+++ 4 A A )> A A A B BDJJJ T] 233DJJJr NF)r r r r
r$ r* r7 r r r r r sU % % % 4 4 4 4 4r r c 6 e Zd Zd fd Zd Zd Z fdZ xZS )assert_query_countFc h t t | | || _ d S )N)r )superr: r
expected)r r= r __class__s r r
zassert_query_count.__init__+ s0
$''00[0III
r c @ t fd }|S )Nc t 5 | i |}d d d n # 1 swxY w Y |S r )
_assert_count)r kwdsretfr s r decoratedz.assert_query_count.__call__..decorated0 s
'
'a&&&
'
'
'
'
'
'
'
'
'
'
'
'
'
'
'
Js r )r rD rE s `` r __call__zassert_query_count.__call__/ s: q
r c \ | j d| j }| j | j k s
J | d S )Nz != )r r= )r error_msgs r rA z assert_query_count._assert_count: s9 "&***dmm< zT]***I*****r c t t | ||| | d S r )r< r: r7 rA )r r4 r5 r6 r> s r r7 zassert_query_count.__exit__> s>
$''007FKKK r r8 )r r r r
rF rA r7
__classcell__)r> s @r r: r: * st ! ! ! ! ! ! 6 6 6 r r: )
functoolsr r getLoggerr& r r objectr r: r r r rN s
8 $ $( ( ( ( (w ( ( (4 4 4 4 4F 4 4 4. r