a
    DgZ
                     @   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 documentZ123456)titleZ	file_hashzserialization.doczA boring example document)r   Documentdocumentfilesaver   self r   g/var/www/lab.imftr.de/x/nb_venv/lib/python3.9/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   Zdocument_idparamsr   r   r   get_response   s    z0TestCorrectDownloadUrlSerialization.get_responseredirectdefaultZ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    Zexpected_urlr   r   r   *test_serializer_wagtaildocs_serve_redirect    s    zNTestCorrectDownloadUrlSerialization.test_serializer_wagtaildocs_serve_redirectdirectzhttp://remotestorage.com/media/)r   r   Z	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(   r*   r   r   r   r   r   	   s:   

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