Hopiu/wagtail
1
0
Fork 0
mirror of https://github.com/Hopiu/wagtail.git synced 2026-04-10 18:10:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Ensure example.doc is deleted after running TestServeView

Browse source
This commit is contained in:
Matt Westcott 2017-06-20 13:30:48 +01:00
parent b00377c418
commit deeee4a597
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
wagtail/wagtaildocs/tests/test_serve_view.py
View file

@ -20,6 +20,9 @@ class TestServeView(TestCase):
self.document = models.Document(title="Test document")
self.document.file.save('example.doc', ContentFile("A boring example document"))
def tearDown(self):
self.document.delete()
def get(self):
return self.client.get(reverse('wagtaildocs_serve', args=(self.document.id, self.document.filename)))
@ -78,6 +81,9 @@ class TestServeViewWithSendfile(TestCase):
self.document = models.Document(title="Test document")
self.document.file.save('example.doc', ContentFile("A boring example document"))
def tearDown(self):
self.document.delete()
def get(self):
return self.client.get(reverse('wagtaildocs_serve', args=(self.document.id, self.document.filename)))