2imddlZddlZddlmZddlmZddlmZddlm Z ddl m Z m Z dZ Gdd eZGd d eZd Zy) N)settings)override_settings) force_bytes)is_download_response)encode_basename_asciiencode_basename_utf8c0||_||_||_|S)aMimic ``as_view()``, but returns view instance. Use this function to get view instances on which you can run unit tests, by testing specific methods. This is an early implementation of https://code.djangoproject.com/ticket/20456 ``view`` A view instance, such as ``TemplateView(template_name='dummy.html')``. Initialization arguments are the same you would pass to ``as_view()``. ``request`` A request object, typically built with :class:`~django.test.client.RequestFactory`. ``args`` and ``kwargs`` "URLconf" positional and keyword arguments, the same you would pass to :func:`~django.core.urlresolvers.reverse`. )requestargskwargs)viewr r r s O/var/www/html/acx/venv/lib/python3.12/site-packages/django_downloadview/test.py setup_viewr s,DLDIDK Kc,eZdZdZfdZfdZxZS)temporary_media_roota*Temporarily override settings.MEDIA_ROOT with a temporary directory. The temporary directory is automatically created and destroyed. Use this function as a context manager: >>> from django_downloadview.test import temporary_media_root >>> from django.conf import settings # NoQA >>> global_media_root = settings.MEDIA_ROOT >>> with temporary_media_root(): ... global_media_root == settings.MEDIA_ROOT False >>> global_media_root == settings.MEDIA_ROOT True Or as a decorator: >>> @temporary_media_root() ... def use_temporary_media_root(): ... return settings.MEDIA_ROOT >>> tmp_media_root = use_temporary_media_root() >>> global_media_root == tmp_media_root False >>> global_media_root == settings.MEDIA_ROOT True cptj}||jd<tt|y)zPCreate a temporary directory and use it to override settings.MEDIA_ROOT. MEDIA_ROOTN)tempfilemkdtempoptionssuperrenable)selftmp_dir __class__s rrztemporary_media_root.enableEs/""$%, \" "D02rcptjtjtt |y)zKRemove directory settings.MEDIA_ROOT then restore original setting.N)shutilrmtreerrrrdisable)rrs rr ztemporary_media_root.disableLs%  h))* "D13r)__name__ __module__ __qualname____doc__rr __classcell__)rs@rrr(s8344rrc:eZdZdZdZdZdZdZdZdZ dZ y ) DownloadResponseValidatorz5Utility class to validate DownloadResponse instances.c |j|||jD]\}}t|d|z}|||| y)a Assert that ``response`` is a valid DownloadResponse instance. Optional ``assertions`` dictionary can be used to check additional items: * ``basename``: the basename of the file in the response. * ``content_type``: the value of "Content-Type" header. * ``mime_type``: the MIME type part of "Content-Type" header (without charset). * ``content``: the contents of the file. * ``attachment``: whether the file is returned as attachment or not. z assert_%sN)assert_download_responseitemsgetattr)r test_caseresponse assertionskeyvalue assert_funcs r__call__z"DownloadResponseValidator.__call__VsM$ %%i:$**, 4JC!$ c(9:K  8U 3 4rc8|jt|y)N) assertTruer)rr,r-s rr)z2DownloadResponseValidator.assert_download_responsems1(;>! Jh'<=="    ZL*./     #I;/./  rc.|j|d|yNz Content-Type) assertEqualrr,r-r0s rassert_content_typez-DownloadResponseValidator.assert_content_typesh~6>rcJ|j|dj|yr?)r4 startswithrAs rassert_mime_typez*DownloadResponseValidator.assert_mime_types Xn5@@GHrc|jDcgc] }t|}}|jdj|t|ycc}w)z9Assert value equals response's content (byte comparison).rN)streaming_contentrr@join)rr,r-r0spartss rassert_contentz(DownloadResponseValidator.assert_contentsB)1)C)CDAQDDchhuo{5/ABEsAcn|r|jd|dvy|jd|vxsd|dvy)Nz attachment;r6)r4rAs rassert_attachmentz+DownloadResponseValidator.assert_attachmentsH   (;P2Q!Q R  %X5H 1F(GG rN) r!r"r#r$r2r)r=rBrErKrMrrr'r'Ss*?4.=2?IC rr'c *t}|||fi|S)aMake ``test_case`` assert that ``response`` meets ``assertions``. Optional ``assertions`` dictionary can be used to check additional items: * ``basename``: the basename of the file in the response. * ``content_type``: the value of "Content-Type" header. * ``mime_type``: the MIME type part of "Content-Type" header (without charset). * ``content``: the contents of the file. * ``attachment``: whether the file is returned as attachment or not. )r')r,r-r. validators rr)r)s"*+I Y 7J 77r)rr django.confrdjango.test.utilsrdjango.utils.encodingrdjango_downloadview.middlewaresrdjango_downloadview.responserrrrobjectr'r)rNrrrWsA  /-@T8(4,(4VHHV8r