
    2ig                         d dl Z d dlZd dlZd dlmZ  G d dej
                        Zedk(  r ej                          yy)    N)GoogleSearchc                   j    e Zd Zd Z ej
                   ej                  d      dk(  d      d        Zy)TestSearchArchiveApic                 B    t        j                  dd      t        _        y )NAPI_KEYdemo)osgetenvr   SERP_API_KEY)selfs    e/var/www/html/marco-python-backend/venv/lib/python3.12/site-packages/tests/test_search_archive_api.pysetUpzTestSearchArchiveApi.setUp   s    $&IIi$?!    r   Nzno api_key providedc                    t        ddd      }|j                         }|j                  d      j                  d      }t        i       j                  |d      }| j	                  |j                  d      j                  d      |       t        i       j                  |d      }| j                  t        |      d       y )	NCoffeezAustin,Texas)qlocationsearch_metadataidjsonhtmli'  )r   get_dictionarygetget_search_archiveassertEqualassertGreaterlen)r   searchsearch_result	search_idarchived_search_resulthtml_buffers         r   test_get_search_archivez,TestSearchArchiveApi.test_get_search_archive   s    H.IJ--/!%%&78<<TB	!-b!1!D!DYPV!W/334EFJJ4PR[\"2&99)VL3{+U3r   )	__name__
__module____qualname__r   unittestskipIfr	   r
   r#    r   r   r   r      s;    @ X__ibii	*d25JK4 L4r   r   __main__)	randomr'   r	   serpapir   TestCaser   r$   mainr)   r   r   <module>r/      s=      	  48,, 4 zHMMO r   