­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ ­ U SW @s|ddlmZddlZddlmZddlmZedZGdddeZ Gdd d ej Z Gd d d eZ Gd d d e Z dS)wrapsN)create_model_tables)drop_model_tablespeeweec@s&eZdZd ddZddZddZd S) test_databaseTFcCs>t|tttfstd|||_||_||_||_||_ dS)Nz%r must be a list or tuple.) isinstancelisttupleset ValueErrordbmodels create_tables drop_tables fail_silently)selfr rrrrrG/opt/alt/python38/lib64/python3.8/site-packages/playhouse/test_utils.py__init__ s ztest_database.__init__cCsFg|_|jD]}|j|jj|j|j_q |jrBt|j|jddSN)r) origrappend_metadatabaser rrr)rmrrr __enter__s   ztest_database.__enter__cCsB|jr|jrt|j|jdt|jD]\}}|j||j_q&dSr) rrrrr enumeraterrr)rexc_typeexc_valexc_tbirrrr__exit__s ztest_database.__exit__N)TTF)__name__ __module__ __qualname__rrr"rrrrr s  rc@seZdZddZddZdS)_QueryLogHandlercOsg|_tjj|f||dSN)queriesloggingHandlerr)rargskwargsrrrr%sz_QueryLogHandler.__init__cCs|j|dSr')r(r)rrecordrrremit)sz_QueryLogHandler.emitN)r#r$r%rr.rrrrr&$sr&c@s.eZdZd ddZddZddZdd Zd S) count_queriesFcCs||_d|_dS)Nr) only_selectcount)rr0rrrr.szcount_queries.__init__cCs|jjSr')_handlerr(rrrr get_queries2szcount_queries.get_queriescCs$t|_ttjt|j|Sr')r&r2loggersetLevelr)DEBUG addHandlerr3rrrr5s  zcount_queries.__enter__cCs>t|j|jr,tdd|jjD|_nt|jj|_dS)NcSs g|]}|jddr|qS)rzSELECT )msg startswith).0qrrr >sz*count_queries.__exit__..)r5 removeHandlerr2r0lenr(r1rrrr rrrr";s zcount_queries.__exit__N)F)r#r$r%rr4rr"rrrrr/-s r/cs:eZdZd fdd ZddZddZfdd ZZS) assert_query_countFcstt|j|d||_dS)N)r0)superrArexpected)rrCr0 __class__rrrEszassert_query_count.__init__cstfdd}|S)Nc s&||}W5QRX|Sr') _assert_count)r+kwdsretfrrr decoratedJsz.assert_query_count.__call__..decoratedr)rrJrKrrIr__call__Iszassert_query_count.__call__cCs(d|j|jf}|j|jks$t|dS)Nz%s != %s)r1rCAssertionError)rZ error_msgrrrrFTsz assert_query_count._assert_countcs tt|||||dSr')rBrAr"rFr@rDrrr"Xszassert_query_count.__exit__)F)r#r$r%rrLrFr" __classcell__rrrDrrADs rA) functoolsrr)rrr getLoggerr5objectrr*r&r/rArrrrs