2iddlZddlZddlZddlZejdkDrddlmZnddlmZddlZddlm Z GddejZ e dk(rejyy)N)r)Queue) GoogleSearchc"eZdZdZej ejddk(ddZej ejddk(ddZ ej ejddk(ddZ ej ejddk(dd Z ej ejddk(dd Z ej ejddk(dd Z y) TestExamplecBtjddt_y)NAPI_KEYdemo)osgetenvr SERP_API_KEY)selfs Z/var/www/html/marco-python-backend/venv/lib/python3.12/site-packages/tests/test_example.pysetUpzTestExample.setUps$&IIi$? !r Nzno api_key providedctddd}|j}t|d|dd}t|dz|j|d}t|y)NCoffeegoogle_scholar)qenginesearch_metadataid: get search from archivehtml)rget_jsonprintget_search_archive)rsearchdata search_idraw_htmls r test_get_jsonzTestExample.test_get_jsonscH8HIJ  d$%&*+D1  i556--i@ hr DEBUGAPI_KEYctddd}|jdD]} |d}td|zy#tdY)xYw)Ncoffeisch)rtbmimages_resultsoriginalzlink is found: zlink is not found.)rrr)rr image_resultlinks rtest_search_google_imagesz%TestExample.test_search_google_images,s]GF;<"OO-.>? L #J/'$./   *+s < A ct}tddd}dD]e}td|z||jd<|j }|td|zAtd|d d z|j |gtd td di}|j s|j}|d d }t|d z|j|}t|dz|d dztjd|d drt|dz|ddzn4t|dz|j |tjd|j stdy)Nz Austin,TexasT)locationasync)amdnvidiaintelzexecute async search: q = rzoops data is empty for: z"add search to the queue where id: rrz4wait until all search statuses are cached or successr/rz : status = statuszCached|Successz: search done with q = search_parametersz: requeue searchzall searches completed) rrr params_dictget_dictputemptygetrrertimesleep)r search_queuercompanyrr search_archiveds r test_asynczTestExample.test_async:sw &   0 !G ,w6 7$+&  S !"$  .89 4tQXAYYZ [ ^ _rcdD]}tij|ddd}td|ddd}|j}|j|j d|d dd }t d |zd z|zy) N)znew yorkparisberlinr5rcanonical_namezbest coffee shoprD)rr.rGrIerrororganic_resultsrMztop coffee result for z is: )r get_locationr assertIsNoner:r)rcityr.rr top_results rtest_search_by_locationz#TestExample.test_search_by_locations3 ND#B'44T1=a@AQRH!'$ #F ??$D   dhhw/ 0/03Grpsf   v ! zN(##zNz zHMMOr