
    2i                         d dl Z 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)YoutubeSearchc                       e Zd Zd Z ej
                   ej                  d      dk(  d      d        Z ej
                   ej                  d      dk(  d      d        Z	y)TestYoutubeSearchApic                 B    t        j                  dd      t        _        y )NAPI_KEYdemo)osgetenvr   SERP_API_KEY)selfs    a/var/www/html/marco-python-backend/venv/lib/python3.12/site-packages/tests/test_youtube_search.pysetUpzTestYoutubeSearchApi.setUp	   s    !#9f!=M    r   Nzno api_key providedc                 
   t        ddi      }|j                         }| j                  |j                  d             | j	                  |d   d   d       | j                  |d   d          | j                  |d   d          y )	Nsearch_querychairerrorsearch_metadatastatusSuccessyoutube_urlid)r   get_dictassertIsNonegetassertEqualassertIsNotNoner   searchdatas      r   test_get_jsonz"TestYoutubeSearchApi.test_get_json   s    NG45F??Ddhhw'(T+,X6	B/0?@/067r   c                    t        ddi      }|j                         }| j                  |j                  j                  d       | j                  |j                  j                         | j                  |j                  j                         | j                  |j                  j                  d       | j                  |j                  j                  d       | j                  |j                  j                  d       y )Nr   r   r   youtube
   )r   
get_objectr   r   r   r   r   r   search_parametersr   engineassertGreatersearch_informationtotal_resultsr   s      r   test_get_objectz$TestYoutubeSearchApi.test_get_object   s    NG45FDT))00)<--001--99:T++88'BT++22I>t..<<bAr   )
__name__
__module____qualname__r   unittestskipIfr	   r
   r!   r+    r   r   r   r      so    > 8??IBIIi(D03HI8 J8 8??IBIIi(D03HIB JBr   r   __main__)
randomr/   r	   pprintserpapir   TestCaser   r,   mainr1   r   r   <module>r8      sB      	  !B8,, B2 z(--/ r   