Darren/dpaste
1
0
Fork 0
mirror of https://github.com/DarrenOfficial/dpaste.git synced 2024-11-15 16:12:51 +11:00
Code Issues Projects Releases Packages Wiki Activity

Tests for the management command.

Browse source
This commit is contained in:
Martin Mahner 2013-11-25 23:31:14 +01:00
parent f40508064f
commit 6a7e201a72
1 changed files with 41 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

41
dpaste/tests/test_snippet.py
View file

@ -1,5 +1,8 @@
# -*- encoding: utf-8 -*- # -*- encoding: utf-8 -*-
from datetime import timedelta
from django.core import management
from django.core.urlresolvers import reverse from django.core.urlresolvers import reverse
from django.test.client import Client from django.test.client import Client
from django.test import TestCase from django.test import TestCase
@ -22,6 +25,9 @@ class SnippetTestCase(TestCase):
'expire_options': EXPIRE_DEFAULT, 'expire_options': EXPIRE_DEFAULT,
} }
# -------------------------------------------------------------------------
# New Snippet
# -------------------------------------------------------------------------
def test_empty(self): def test_empty(self):
""" """
The browser sent a content field but with no data. The browser sent a content field but with no data.
@ -69,6 +75,9 @@ class SnippetTestCase(TestCase):
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertEqual(Snippet.objects.count(), 0) self.assertEqual(Snippet.objects.count(), 0)
# -------------------------------------------------------------------------
# Reply
# -------------------------------------------------------------------------
def test_reply(self): def test_reply(self):
data = self.valid_form_data() data = self.valid_form_data()
response = self.client.post(self.new_url, data, follow=True) response = self.client.post(self.new_url, data, follow=True)
@ -85,6 +94,9 @@ class SnippetTestCase(TestCase):
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertEqual(Snippet.objects.count(), 1) self.assertEqual(Snippet.objects.count(), 1)
# -------------------------------------------------------------------------
# Snippet Functions
# -------------------------------------------------------------------------
def test_raw(self): def test_raw(self):
data = self.valid_form_data() data = self.valid_form_data()
self.client.post(self.new_url, data, follow=True) self.client.post(self.new_url, data, follow=True)
@ -94,6 +106,9 @@ class SnippetTestCase(TestCase):
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertContains(response, data['content']) self.assertContains(response, data['content'])
# -------------------------------------------------------------------------
# History
# -------------------------------------------------------------------------
def test_snippet_history(self): def test_snippet_history(self):
response = self.client.get(reverse('snippet_history')) response = self.client.get(reverse('snippet_history'))
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
@ -123,3 +138,29 @@ class SnippetTestCase(TestCase):
response = self.client.get(reverse('snippet_history') + '?delete-all', follow=True) response = self.client.get(reverse('snippet_history') + '?delete-all', follow=True)
self.assertEqual(response.status_code, 200) self.assertEqual(response.status_code, 200)
self.assertEqual(Snippet.objects.count(), 0) self.assertEqual(Snippet.objects.count(), 0)
# -------------------------------------------------------------------------
# Management Command
# -------------------------------------------------------------------------
def test_delete_management(self):
# Create two snippets
data = self.valid_form_data()
self.client.post(self.new_url, data, follow=True)
data = self.valid_form_data()
self.client.post(self.new_url, data, follow=True)
self.assertEqual(Snippet.objects.count(), 2)
# But the management command will only remove snippets past
# its expiration date, so change one to last month
s = Snippet.objects.all()[0]
s.expires = s.expires - timedelta(days=30)
s.save()
# You can call the management command with --dry-run which will
# list snippets to delete, but wont actually do.
management.call_command('cleanup_snippets', dry_run=True)
self.assertEqual(Snippet.objects.count(), 2)
# Calling the management command will delete this one
management.call_command('cleanup_snippets')
self.assertEqual(Snippet.objects.count(), 1)