
    hL                     Z    d Z ddlmZ ddlmZmZ ddlmZ  G d dej                        Z	y)zTests for the models module.
    )utils)	ThumbnailSourcec                   >     e Zd ZdZ fdZ fdZd Zd Zd Z xZ	S )FileManagerTestzTest for FileManagerc                 >   t         |           t        j                         | _        t        j                  | j                        | _        t        j                  j                  d| j                        | _        | j                  | j                  d      | _        y )NzTest source)namestorage_hashztest.jpg)supersetUptestTemporaryStoragestorager   get_storage_hashr
   r   objectscreatesourcecreate_imagefilenameself	__class__s    Q/home/dcms/DCMS/lib/python3.12/site-packages/easy_thumbnails/tests/test_models.pyr   zFileManagerTest.setUp   st    ,,.!224<<@nn++** , ,
 ))$,,
C    c                 V    | j                   j                          t        |           y )N)r   delete_temporary_storager   tearDownr   s    r   r   zFileManagerTest.tearDown   s    --/r   c                     t         j                  j                  | j                  | j                  d| j
                        }| j                  |j                  | j                         y)z&Create a new Thumbnail in the databaseT)r   r   N)r   r   get_filer   r   r   assertEqualr	   )r   imgs     r   test_create_filez FileManagerTest.test_create_file   sO    ((LLMM;;	 )   	4==1r   c                 6   t         j                  j                  | j                  | j                  | j
                        }t         j                  j                  | j                  | j                  d      }| j                  |       | j                  ||       y)z%Fetch an existing thumb from database)r
   r	   r   F)r   N)
r   r   r   r
   r   r   r   r   
assertTruer    )r   createdfetcheds      r   test_get_filezFileManagerTest.test_get_file&   s    ##****;; +  
 ##,,LLMM - 
 	 '*r   c                    	 t         j                  j                  | j                         | j	                  d       t         j                  j                  | j                  | j                  | j                  d       	 t         j                  j                  | j                         y# t         j
                  $ r Y w xY w# t         j
                  $ r | j	                  d       Y yw xY w)z<Fetch a thumb that is in the storage but not in the database)r	   zThumb should not exist yetT)r   check_cache_misszThumb should exist nowN)	r   r   getr   failDoesNotExistr   r   r   )r   s    r   test_get_file_check_cachez)FileManagerTest.test_get_file_check_cache5   s    	!!t}}!5II23 	""LLMM;;!	 	# 	#	0!!t}}!5 %% 		 %% 	0II./	0s#   ;B* ?*C *C ?C $C*)C*)
__name__
__module____qualname____doc__r   r   r"   r'   r-   __classcell__)r   s   @r   r   r   	   s     
D2+0r   r   N)
r1   easy_thumbnailsr   easy_thumbnails.modelsr   r   easy_thumbnails.testsr   BaseTestr    r   r   <module>r8      s'    " 4 /@0dmm @0r   