Skip to content
Permalink
Browse files

Some testing for send_file

  • Loading branch information
benjaoming committed Jan 8, 2020
1 parent b73e233 commit 5c5d7dc888cd6fbfa692651818492a572cfe7721
Showing with 18 additions and 0 deletions.
  1. +18 −0 tests/core/test_basic.py
@@ -1,5 +1,9 @@
import tempfile
from datetime import datetime

from django.test import TestCase
from wiki.conf import settings as wiki_settings
from wiki.core.http import send_file
from wiki.forms import Group
from wiki.models import Article, ArticleRevision, URLPath

@@ -36,3 +40,17 @@ def test_manager(self):
article.add_revision(ArticleRevision(title="Root", content="Hello\nworld"),
save=True)
self.assertEqual("Hello\r\nworld", article.current_revision.content)


class HttpTests(TestCase):
def test_send_file(self):
fabricate_request = self.client.get("/").wsgi_request
fobject = tempfile.NamedTemporaryFile("r")
response = send_file(fabricate_request, fobject.name, filename="test.pdf")
assert response.has_header("Content-Disposition")
assert "inline" in response.get("Content-Disposition")
response = send_file(fabricate_request, fobject.name, filename="test.jpeg")
assert response.has_header("Content-Disposition")
response = send_file(fabricate_request, fobject.name, filename="test.jpeg", last_modified=datetime.now())
assert response.has_header("Content-Disposition")
fobject.close()

0 comments on commit 5c5d7dc

Please sign in to comment.
You can’t perform that action at this time.