o
    hZ
                     @   sT   d dl mZ d dlmZ d dlmZmZ d dlmZ d dl	m
Z
 G dd deZdS )	    )settings)ContentFile)TestCaseoverride_settings)reverse)modelsc                   @   sz   e Zd ZdZdd Zdd Zdd Zedi ej	d	d
diidddd Z
edi ej	d	d
diiddddd ZdS )#TestCorrectDownloadUrlSerializationzTest asserts that in case of both `redirect` and `direct`
    WAGTAILDOCS_SERVE_METHOD settings `download_url` field
    is correctly serialized by DocumentDownloadUrlField.c                 C   s(   t jddd| _| jjdtd d S )NzTest document123456)title	file_hashzserialization.doczA boring example document)r   Documentdocumentfilesaver   self r   h/var/www/html/ndineBlogger/venv/lib/python3.10/site-packages/wagtail/documents/tests/test_serializers.pysetUp   s
   z)TestCorrectDownloadUrlSerialization.setUpc                 C   s   | j j  d S )N)r   r   deleter   r   r   r   tearDown   s   z,TestCorrectDownloadUrlSerialization.tearDownc                 K   s   | j td|fd|S )Nzwagtailapi_v2:documents:detail)args)clientgetr   )r   document_idparamsr   r   r   get_response   s   z0TestCorrectDownloadUrlSerialization.get_responseredirectdefaultBACKENDz8wagtail.test.dummy_external_storage.DummyExternalStoragezhttp://example.com/)WAGTAILDOCS_SERVE_METHODSTORAGESWAGTAILAPI_BASE_URLc                 C   s^   |  | jj}| }| d| |d }| d| |d }d| jj d}| || d S )Nmetadownload_urlzhttp://example.com/documents/z/serialization.doc)r   r   idjsonassertInpkassertEqual)r   responsedatar#   r$   expected_urlr   r   r   *test_serializer_wagtaildocs_serve_redirect    s   zNTestCorrectDownloadUrlSerialization.test_serializer_wagtaildocs_serve_redirectdirectzhttp://remotestorage.com/media/)r    r!   	MEDIA_URLr"   c                 C   sN   |  | jj}| }| d| |d }| d| |d }| |d d S )Nr#   r$   z:http://remotestorage.com/media/documents/serialization.doc)r   r   r%   r&   r'   r)   )r   r*   r+   r#   r$   r   r   r   (test_serializer_wagtaildocs_serve_direct6   s   zLTestCorrectDownloadUrlSerialization.test_serializer_wagtaildocs_serve_directN)__name__
__module____qualname____doc__r   r   r   r   r   r!   r-   r0   r   r   r   r   r   	   s<    

r   N)django.confr   django.core.files.baser   django.testr   r   django.urlsr   wagtail.documentsr   r   r   r   r   r   <module>   s    