etjB<ddlZddlZddlmZddlmZddlmZddlmZddl m Z ddl m Z d d lm Z d d lm Z d d lmZd d lmZd dlmZd dlmZd dlmZd dlmZGddejZGddejZGddejejZdS)N)engines)fixtures)eq_) requirements)Column)Table)DateTime)func)Integer)select)sql)String)testingtextc|eZdZdZedZedZdZdZdZ dZ e j dZ d Zd S) RowFetchTestTc td|tdtdtdtdtd|tdtdtdtdS) Nplain_pkidT) primary_keydata2 has_datestoday)r rr rr clsmetadatas c/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/sqlalchemy/testing/suite/test_results.py define_tableszRowFetchTest.define_tabless   4d 3 3 3 66":: & &       4d 3 3 3 7H % %      c ||jjdddddddddg||jjdt jdd d d d d d gdS) Nd1)rrrd2r d3 r)rr)executetablesrinsertrdatetime)r connections r! insert_datazRowFetchTest.insert_data.s J  & & ( ($''$''$''      J ' ' ) ) 1$2r1a H H I I J     r#c.||jj|jjjj}t|jdt|j ddSNr%r&) r,r-rrorder_bycrfirstrrselfr0rows r! test_via_attrzRowFetchTest.test_via_attr>sy  K ' ' ) ) 2 24;3G3I3L M M  %''  CFA CHdr#cF||jj|jjjj}t|j ddt|j dddS)Nrr%rr&) r,r-rrr4r5rr6r_mappingr7s r!test_via_stringzRowFetchTest.test_via_stringFs  K ' ' ) ) 2 24;3G3I3L M M  %''  CL """ CL $'''''r#c2||jj|jjjj}t|ddt|dddS)Nrr%r&) r,r-rrr4r5rr6rr7s r! test_via_intzRowFetchTest.test_via_intNs}  K ' ' ) ) 2 24;3G3I3L M M  %''  CFA CFDr#c||jj|jjjj}t|j |jjjjdt|j |jjjj ddSr3) r,r-rrr4r5rr6rr<rr7s r!test_via_col_objectz RowFetchTest.test_via_col_objectVs  K ' ' ) ) 2 24;3G3I3L M M  %''  CL-/2 3Q777 CL-/4 5t<<<< > > > :&&&&&r#c|jj}||ddd||dddddddddg||dSrbrh)r8r0r]s r!test_executemany_roundtripz1PercentSchemaNamesTest.test_executemany_roundtrips 1   " "B$O$O      " ";;;;;;     :&&&&&r#c|jj}||ddd|||jd|jdddddd ddddg}t |gd ||dS) Nr*r+rcr[r\rdrerfrg)rdrerfrgrerf) r-r]r,r. returningr5rallri)r8r0r]rEs r!$test_executemany_returning_roundtripz;PercentSchemaNamesTest.test_executemany_returning_roundtrips 1   " "B$O$O   ##  " " , , + 67   ;;;;;;     FJJLL555666 :&&&&&r#c |jj}|jj}||||fD]}t t |||j dgdt t || |j d ddg|j dddg|||j d }t |j ddt |j dd t |j |j ddt |j |j dd |||j dd it t |||j dgd dS) Nr[))r*r+rorprqr\rfrgrprqr*r+))r*rv)rdrv)rfrv)rerv)r-r]r`rJrlistr,rr4r5wherein_r6r<updatevalues)r8connr]r`r^r9s r!riz$PercentSchemaNamesTest._assert_tables 1 $(K$I!     ! ! % % + + - -   C CE LL!8!89L!M!MNN544     LL uw'=>BBAr7KKLL!%'*"566'"   ,, '' (;<<egg   Z(! , , ,  34b 9 9 9  UWZ011 5 5 5  UW%;<=r B B B B  " " ) )!78"=      !((**33% 3   2 1 1 r#N)rPrQrR__doc__ __requires__rSrTr"rkrmrinsert_executemany_returningrtrirVr#r!rXrXzs-LK   [   ' ' ' ' ' '.''/.'&4 4 4 4 4 r#rXceZdZdZdZdZdZdZej ddddfddd dfd de d dfd d dd fdd e d  ddfdde d  d d fdde d  dfddddej ddgfdd dd fdd ddfdd dZdZd Zd!Zd"Zd#Zd$S)%ServerSideCursorsTest)server_side_cursorsTcj|jjjdkrt|jS|jjjdkr)t djj}t||S|jjjdvr|j S|jjjdkr)t djj}t||S|jjjdkr|j S|jjjdkr(d t|j vS|jjjd vr|j S|jjjd krt|d d S|jjjdkrtt|dd S|jjjdkrt|d d Sd S)Npsycopg2pymysqlzpymysql.cursors)aiomysqlasyncmyaioodbcmysqldbzMySQLdb.cursorsmariadbconnectormysqlconnectorbuffered)asyncpg aiosqlitepg8000 server_sideFpsycopgnameoracledb)enginedialectdriverboolr __import__cursorsSSCursor isinstancerrtyperPlowergetattr)r8cursorsscursors r!_is_server_sidez%ServerSideCursorsTest._is_server_sides ;  % 3 3 $$ $ [ '9 4 4!"344<EHfh// / [ '+M M M% % [ '9 4 4!"344<EHfh// / [ '+= = =& & [ '+; ; ;T&\\%:%@%@%B%BB B [ '+C C C% % [ '8 3 36=%88 8 [ '9 4 46677 7 [ ': 5 56=%88 85r#c|rItjd5tjd|i|_dddn #1swxYwYntjd|i|_|jS)NzThe create_engine.server_side_cursors parameter is deprecated and will be removed in a future release. Please use the Connection.execution_options.stream_results parameter.r)options)rexpect_deprecatedrtesting_enginer)r8rs r!_fixturezServerSideCursorsTest._fixtures  *   &424GH                "0.0CDDK{sAAAcjtjdtjd|S)Nz SELECT (\d+)c ttt|dt jS)Nr%)strrintgroupcompilerdb)ms r!z1ServerSideCursorsTest.stringify../s6c&QWWQZZ1199'*EEFFr#)rerIsub)r8str_s r! stringifyzServerSideCursorsTest.stringify-s0z/240044 F F   r# global_stringc|dSNselect 1rVrs r!rzServerSideCursorsTest.3s))J2G2Gr# global_textc2t|dSrrrs r!rzServerSideCursorsTest.7d99Z#8#899r# global_exprr%global_off_explicitFc2t|dSrrrs r!rzServerSideCursorsTest.>rr# stmt_optionstream_resultsstmt_option_disabledfor_update_exprfor_update_stringc|dS)NzSELECT 1 FOR UPDATErVrs r!rzServerSideCursorsTest.Rsii(=>>r#sqlitemssql text_no_ssc2t|dS)N select 42rrs r!rzServerSideCursorsTest.Ysd99[#9#9::r#text_ss_optioncZt|ddS)NrTr)rexecution_optionsrs r!rzServerSideCursorsTest._s1d99[#9#9::LL#Mr#iaaaz*engine_ss_arg, statement, cursor_ss_status)id_argnamesc||}|5}t|rtj||j}t |tr||}n| |}t| |j || ddddS#1swxYwYdS)Nr)rbegincallablerresolve_lambdarrrexec_driver_sqlr,rrrclose)r8 engine_ss_arg statementcursor_ss_statusrr|rEs r!test_ss_cursor_statusz+ServerSideCursorsTest.test_ss_cursor_status2sp}-- \\^^ t "" #2 )S)) 1--i88i00 $$V]335E F F F LLNNN                  sB'CC"%C"cb|d}|5}|d|d}||jsJ|ddddS#1swxYwYdS)NFTrr)rconnectrrrrrr)r8rr|rEs r!test_conn_optionz&ServerSideCursorsTest.test_conn_optionxsu%% ^^   ++#,odnnZ8899 '' 66 6 66 LLNNN'                  sA-B$$B(+B(c\|d}tdd}|5}|d|}||jrJ ddddS#1swxYwYdS)NFr%Tr)rrrrr,rr)r8rrMr|rEs r!&test_stmt_enabled_conn_option_disabledzzBServerSideCursorsTest.test_roundtrip_fetchmany..s'???Q8a<(((???r#r%r*cg|] }|d|zf SrrVrs r!rzBServerSideCursorsTest.test_roundtrip_fetchmany..s!888q!X\"888r#rgcg|] }|d|zf SrrVrs r!rzBServerSideCursorsTest.test_roundtrip_fetchmany..s!999q!X\"999r#cg|] }|d|zf SrrVrs r!rzBServerSideCursorsTest.test_roundtrip_fetchmany..s!#M#M#M!Q1 $5#M#M#Mr#)r rr rr rrrr,r.rangerr4r5rr fetchmanyr)r8r rrrr0rEs r!test_roundtrip_fetchmanyz.ServerSideCursorsTest.test_roundtrip_fetchmanys# ]t$$   g4$    66":: & &   \\^^ Oz   jT  : : :   !!##??%2,,???     ''!!##,,Z\_==F   ##88E!QKK888      $$99E!RLL999    !!#M#MuR}}#M#M#M N N N' O O O O O O O O O O O O O O O O O Os4E G  GGN)rPrQrRr~rSrrrr combinationsrrwith_for_updateskip_ifrrrrrr rVr#r!rrs,LK8"   W $ G GN   9 9   ffQii. !  9 9      F1II ' 't ' < <    #  F1II ' 'u ' = =   D&&))";";"="=tD  > >  GOXw/ 0 0     : :           =g444jk44j"0;;;((((T O O O O Or#r)r/rrr assertionsrconfigrschemarr r r r rrrrr TablesTestrrXTestBaseAssertsExecutionResultsrrVr#r!rs !!!!!![E[E[E[E[E8&[E[E[E|| | | | | X0| | | ~OOOOO w6OOOOOr#