a ù!fÂ<ã@s>ddlZGdd„dejƒZGdd„deƒZGdd„deƒZdS)éNc@s¶eZdZdZdZdZdZdZdZdd„Zd d „Z d-d d „Z dd„Z dd„Z dd„Z dd„Zdd„Zdd„Zdd„Zdd„Zdd„Zd d!„Zd"d#„Zd$d%„Zd&d'„Zd(d)„Zd*d+„Zd,S).ÚTestQueryResultsÚprojectZjob_nameztest-synchronous-queryÚqueryzselect count(*) from personsÚTOKENcCsddlm}|S)Nr)Ú QueryResults)Zgcloud.bigquery.queryr)Úselfr©rú[/var/www/html/python-backend/venv/lib/python3.9/site-packages/gcloud/bigquery/test_query.pyÚ_getTargetClasss z TestQueryResults._getTargetClasscOs| ¡|i|¤ŽS©N)r )rÚargsÚkwrrr Ú_makeOneszTestQueryResults._makeOneFc Csª|j|jdœ|gdddddœddddœgid œ}|r¦d |d <d d did digid d did digid d did digid d did digig|d<|j|d<d|d<d|d<|S)N©Ú projectIdÚjobIdÚfieldsÚ full_nameÚSTRINGÚREQURED©ÚnameÚtypeÚmodeÚageÚINTEGER)Ú jobReferenceÚ jobCompleteÚerrorsÚschemaZ1000Ú totalRowsÚfÚvúPhred Phlyntstoneé úBharney Rhubbleé!úWylma PhlyntstoneéúBhettye RhubbleéÚrowsÚ pageTokeni †ÚtotalBytesProcessedFÚcacheHit)ÚPROJECTÚJOB_NAMEr)rÚcompleteÚresourcerrr Ú _makeResource!sDþ  þÿùþþþþó zTestQueryResults._makeResourcecCsÐddlm}d|vr¼|dd}| t|jƒt|ƒ¡t|j|ƒD]v\}}| t||ƒ¡| |j|d¡| |j |d¡| |j |d¡| |j |  d¡¡| |j |  d¡¡qBn| |jdu¡dS) Nr)Ú SchemaFieldrrrrrÚ description)Zgcloud.bigquery.tabler4Ú assertEqualÚlenrÚzipÚ assertTrueÚ isinstancerZ field_typerr5Úgetr)rrr2r4rÚfoundÚexpectedrrr Ú _verifySchemaKs  ÿzTestQueryResults._verifySchemacCsr| d¡}|dur"| |jg¡nL|j}| t|ƒt|ƒ¡t||ƒD]&\}}| |tdd„|dDƒƒ¡qFdS)Nr+cSsg|] }|d‘qS)r"r)Ú.0Úcellrrr Ú dóz0TestQueryResults._verifyRows..r!)r;r6r+r7r8Útuple)rrr2r=r<Zf_rowZe_rowrrr Ú _verifyRows[s ÿzTestQueryResults._verifyRowscCsÄ| |j| d¡¡| |j| d¡¡| |j| d¡¡| |j| d¡¡| |j| d¡¡| |j| d¡¡d|vr˜| |j|dd¡n|  |jdu¡|  ||¡|  ||¡dS) Nr.rrr,r r-rr) r6Ú cache_hitr;r1rÚ page_tokenÚ total_rowsÚtotal_bytes_processedrr9r>rD)rrr2rrr Ú_verifyResourcePropertiesfsÿ z*TestQueryResults._verifyResourcePropertiescCst|jƒ}| |j|¡}| |j|j¡| |j|u¡| |jdu¡| |j du¡| |j du¡| |j du¡| |j du¡| |j g¡| |jdu¡| |jdu¡| |jdu¡| |jdu¡| |jdu¡| |jdu¡| |jdu¡| |jdu¡dSr )Ú_Clientr/rÚQUERYr6rr9Ú_clientrEr1rrrFr+rrGrHÚdefault_datasetÚ max_resultsÚpreserve_nullsÚuse_query_cacheÚuse_legacy_sql©rÚclientrrrr Ú test_ctorws$ zTestQueryResults.test_ctorcCs,t|jƒ}| |j|¡}| |jdu¡dSr )rJr/rrKr9ÚjobrRrrr Útest_job_wo_jobids z"TestQueryResults.test_job_wo_jobidcCs–ddlm}d}t|jƒ}| |j|¡}|j|dœ|jd<|j}| t ||ƒ¡|  |j |j¡| |j |u¡|  |j |¡|j}| ||u¡dS)Nr)ÚQueryJobÚSERVER_GENERATEDrr)Úgcloud.bigquery.jobrWrJr/rrKZ _propertiesrUr9r:r6rrLr)rrWrXrSrrUZ fetched_laterrrr Útest_job_w_jobid’s  þ z!TestQueryResults.test_job_w_jobidcCs^t|jƒ}| |j|¡}| |i¡ddddddœddddœgii}| |¡| ||¡dS) Nrrrrrrrr)rJr/rrKrIÚ_set_properties)rrSrr2rrr Ú test_schema£s    þÿÿ zTestQueryResults.test_schemacCsºd|j}|jdd}t|ƒ}t|j|d}| |j|¡}| |jg¡| ¡| t |j ƒd¡|j d}| |dd¡| |d d |¡d |ji}| |d |¡|  ||¡dS) Núprojects/%s/queriesF©r1©rÚ connectionérÚmethodÚPOSTÚpathú/%srÚdata) r/r3Ú _ConnectionrJrrKr6Ú udf_resourcesÚrunr7Ú _requestedrI)rÚPATHÚRESOURCEÚconnrSrÚreqÚSENTrrr Útest_run_w_bound_client²s    z(TestQueryResults.test_run_w_bound_clientc Cs d|j}|jdd}d}tƒ}t|j|d}t|ƒ}t|j|d}| |j|¡}| |¡|_d|_d|_ d|_ d|_ d|_ d|_ |j|d | t|jƒd ¡| t|jƒd ¡|jd } | | d d ¡| | dd|¡|j|j|dœdddddddœ} | | d| ¡| ||¡dS)Nr]Tr^Z test_datasetr_édé NF©rSrrarbrcrdre)rZ datasetId)rZdefaultDatasetZdryRunÚ maxResultsZ preserveNullsÚ timeoutMsZ useQueryCacheZ useLegacySqlrf)r/r3rgrJrrKÚdatasetrMrNrOÚ timeout_msrPrQÚdry_runrir6r7rjrI) rrkrlZDATASETÚconn1Úclient1Úconn2Úclient2rrnrorrr Útest_run_w_alternate_clientÃsD     þõ z,TestQueryResults.test_run_w_alternate_clientc CsÒddlm}d}d|j}|jdd}t|ƒ}t|j|d}| |j|¡}|d|ƒg|_|  ¡|  t |j ƒd ¡|j d}|  |d d ¡|  |d d |¡|jd|igdœ} |  |d| ¡|  ||¡dS)Nr©Ú UDFResourceúvar someCode = "here";r]Fr^r_Ú inlineCoderarbrcrdre©rZuserDefinedFunctionResourcesrf©rYrr/r3rgrJrrKrhrir6r7rjrI) rrÚINLINE_UDF_CODErkrlrmrSrrnrorrr Útest_run_w_inline_udfìs$    þz&TestQueryResults.test_run_w_inline_udfc CsÒddlm}d}d|j}|jdd}t|ƒ}t|j|d}| |j|¡}|d|ƒg|_|  ¡|  t |j ƒd ¡|j d}|  |d d ¡|  |d d |¡|jd|igdœ} |  |d| ¡|  ||¡dS)Nrr~úgs://some-bucket/js/lib.jsr]Fr^r_Ú resourceUrirarbrcrdrer‚rfrƒ) rrÚ RESOURCE_URIrkrlrmrSrrnrorrr Útest_run_w_udf_resource_uris$    þz,TestQueryResults.test_run_w_udf_resource_uric Csddlm}d}d}d|j}|jdd}t|ƒ}t|j|d}| |j|¡}|d |ƒ|d |ƒg|_|  ¡|  t |j ƒd ¡|j d} |  | d d ¡|  | dd|¡|  |j|d |ƒ|d |ƒg¡|jd |id |igdœ} |  | d| ¡|  ||¡dS)Nrr~r†r€r]Fr^r_r‡rrarbrcrdrer‚rfrƒ) rrrˆr„rkrlrmrSrrnrorrr Útest_run_w_mixed_udfss8   ÿ ÿÿþÿz&TestQueryResults.test_run_w_mixed_udfscCs4tƒ}t|j|d}| |j|¡}| t|j¡dS)Nr_)rgrJr/rrKÚ assertRaisesÚ ValueErrorÚ fetch_data)rrmrSrrrr Ú!test_fetch_data_query_not_yet_run4sz2TestQueryResults.test_fetch_data_query_not_yet_runc Cs2d|j|jf}|jdd}|jdd}|d=t|ƒ}t|j|d}| |j|¡}| |¡| |j ¡|  ¡\}}} |  |j ¡|  t |ƒd¡|  |dd ¡|  |d d ¡|  |d d ¡|  |dd¡|  |d¡|  | |d¡|  t |jƒd ¡|jd} |  | dd¡|  | dd|¡dS)Núprojects/%s/queries/%sFr^Tr r_ér©r#r$ra©r%r&é©r'r(é©r)r*r,rbÚGETrdre)r/r0r3rgrJrrKr[Ú assertFalser1rr9r6r7rj) rrkÚBEFOREÚAFTERrmrSrr+rGrFrnrrr Útest_fetch_data_w_bound_client:s,       z/TestQueryResults.test_fetch_data_w_bound_clientcCsd|j|jf}d}d}d}d}|jdd}|jdd}tƒ}t|j|d } t|ƒ} t|j| d } | |j| ¡} |  |¡| | j ¡| j | ||||d \} }}|  | j ¡|  t | ƒd ¡|  | d d ¡|  | dd¡|  | dd¡|  | dd¡|  |t|dƒ¡|  ||d¡|  t |jƒd ¡|  t | jƒd¡| jd }|  |dd¡|  |dd|¡|  |d||||dœ¡dS)Nré riÑrrFr^Tr_)rSrNrFÚ start_indexrwrrr‘rar’r“r”r•r–r r,rbr—rdreZ query_params)rtr,Z startIndexru)r/r0r3rgrJrrKr[r˜r1rr9r6r7Úintrj)rrkÚMAXrÚSTARTÚTIMEOUTr™ršryrzr{r|rr+rGrFrnrrr Ú"test_fetch_data_w_alternate_clientVsL    þ    ýÿz3TestQueryResults.test_fetch_data_w_alternate_clientN)F)Ú__name__Ú __module__Ú __qualname__r/r0ZJOB_TYPErKrr rr3r>rDrIrTrVrZr\rpr}r…r‰rŠrŽr›r¢rrrr rs0 * )rc@seZdZddd„Zdd„ZdS)rJrNcCs||_||_dSr r_)rrr`rrr Ú__init__‚sz_Client.__init__cCsddlm}|||dS)Nr)ÚDatasetrs)Zgcloud.bigquery.datasetr§)rrr§rrr rv†s z_Client.dataset)rN)r£r¤r¥r¦rvrrrr rJ€s rJc@seZdZdd„Zdd„ZdS)rgcGs||_g|_dSr )Ú _responsesrj)rÚ responsesrrr r¦sz_Connection.__init__cKs,|j |¡|jd|jdd…}|_|S)Nrra)rjÚappendr¨)rr Úresponserrr Ú api_request‘s z_Connection.api_requestN)r£r¤r¥r¦r¬rrrr rg‹srg)Z unittest2ZTestCaserÚobjectrJrgrrrr Ús p