
    2i                         d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlZ G d dej                        Z
edk(  r ej                          yy)    N)SerpApiClient)SerpApiClientExceptionc                   |    e Zd Z ej                   ej                  d      dk(  d      d        Zd Zd Z	d Z
d Zy)	TestSerpSearchApiAPI_KEYNzno api_key providedc                     t        dddt        j                  d      d      }|j                         }|j	                  d      d k(  sJ | j                  |d   d   d	          y )
NCoffeezAustin,Texasgoogle_scholarr   qlocationengineapi_keyerrororganic_resultsr   title)r   osgetenvget_jsongetassertIsNotNoneselfsearchdatas      b/var/www/html/marco-python-backend/venv/lib/python3.12/site-packages/tests/test_serp_api_client.pytest_get_jsonzTestSerpSearchApi.test_get_json   si    	yy#	 6 //
4
((7
t
##
#./27;<    c                 l    t        ddddd      }|j                         }| j                  |d          y )Nr	   USAgoogleinvalid_api_keyr   r   )r   r   r   r   s      r   test_invalid_api_keyz&TestSerpSearchApi.test_invalid_api_key   s=    	 	 6 //
4W&r   c                     t        dt        j                  d      d      }t        j                  t
              5  |j                          d d d        y # 1 sw Y   y xY w)Nr	   r   )r   r   )r   r   r   pytestraisesr   r   )r   r   s     r   test_error_missing_enginez+TestSerpSearchApi.test_error_missing_engine$   sN    	yy# 6 	,- 
OO  s   AAc                     t        ddt        j                  d      d      }|j                         }| j	                  |d          | j                  |d   d       y )Nr    r!   r   )r   r   r   r   zMissing query `q`)r   r   r   r   r   assertRegexr   s      r   test_missing_zTestSerpSearchApi.test_missing_,   sX    yy# 6
 //
4W&DM#67r   c                 R    t        j                  d      }|j                  |       y )N   )indent)pprintPrettyPrinter)r   payloadpps      r   debugzTestSerpSearchApi.debug6   s    A&299Wr   )__name__
__module____qualname__unittestskipIfr   r   r   r#   r'   r*   r2    r   r   r   r      sG    8??IBIIi(D03HI	= J	='8r   r   __main__)randomr6   r   r.   serpapir   !serpapi.serp_api_client_exceptionr   r%   TestCaser   r3   mainr8   r   r   <module>r?      sG      	  ! D 
,)) ,\ z(--/ r   