From 3b8bb915e5dfdde88db85f5aab91b6b0e9ef88f5 Mon Sep 17 00:00:00 2001
From: Martin Mahner <martin@mahner.org>
Date: Tue, 21 Jan 2014 12:57:29 +0100
Subject: [PATCH] One time tests and docs.

---
 docs/settings.rst            | 24 ++++++++++++++++++++++--
 dpaste/forms.py              |  4 +++-
 dpaste/tests/test_snippet.py | 25 +++++++++++++++++++++++++
 3 files changed, 50 insertions(+), 3 deletions(-)

diff --git a/docs/settings.rst b/docs/settings.rst
index d3eb319..deb356d 100644
--- a/docs/settings.rst
+++ b/docs/settings.rst
@@ -55,8 +55,28 @@ behavior without touching the code:
                 (3600 * 24 * 30 * 12 * 100, ugettext(u'100 Years')),
             )
 
-        You can keep snippets forever when you set the choice key to ``never``.
-        The management command will ignore these snippets::
+        **One-time snippets** are supported. One time snippets are automatically
+        deleted once a defined view count has reached (Default: ``2``). To
+        enable one-time snippets you have to add a choice ``onetime`` to the
+        expire choices::
+
+            ugettext = lambda s: s
+            DPASTE_EXPIRE_CHOICES = (
+                ('onetime', ugettext(u'One-Time snippet')),
+                (3600, ugettext(u'In one hour')),
+                (3600 * 24 * 7, ugettext(u'In one week')),
+                (3600 * 24 * 30, ugettext(u'In one month')),
+            )
+
+        You can also set the maximum view count after what the snippet gets
+        deleted. The default is ``2``. One view is from the author, one view
+        is from another user::
+
+            DPASTE_ONETIME_LIMIT = 2
+
+        **Infinite snippets** are supported. You can keep snippets forever when
+        you set the choice key to ``never``. The management command will ignore
+        these snippets::
 
             ugettext = lambda s: s
             DPASTE_EXPIRE_CHOICES = (
diff --git a/dpaste/forms.py b/dpaste/forms.py
index 1ad5429..c5cc6db 100644
--- a/dpaste/forms.py
+++ b/dpaste/forms.py
@@ -8,11 +8,13 @@ from dpaste.models import Snippet
 from dpaste.highlight import LEXER_LIST, LEXER_DEFAULT, LEXER_KEYS
 
 EXPIRE_CHOICES = getattr(settings, 'DPASTE_EXPIRE_CHOICES', (
+    ('onetime', _(u'One Time Snippet')),
     (3600, _(u'In one hour')),
     (3600 * 24 * 7, _(u'In one week')),
     (3600 * 24 * 30, _(u'In one month')),
+    # ('never', _(u'Never')),
 ))
-EXPIRE_DEFAULT = getattr(settings, 'DPASTE_EXPIRE_DEFAULT', EXPIRE_CHOICES[2][0])
+EXPIRE_DEFAULT = getattr(settings, 'DPASTE_EXPIRE_DEFAULT', EXPIRE_CHOICES[3][0])
 MAX_CONTENT_LENGTH = getattr(settings, 'DPASTE_MAX_CONTENT_LENGTH', 250*1024*1024)
 
 class SnippetForm(forms.ModelForm):
diff --git a/dpaste/tests/test_snippet.py b/dpaste/tests/test_snippet.py
index 57e1840..732ee38 100644
--- a/dpaste/tests/test_snippet.py
+++ b/dpaste/tests/test_snippet.py
@@ -104,6 +104,31 @@ class SnippetTestCase(TestCase):
         self.assertEqual(response.status_code, 200)
         self.assertEqual(Snippet.objects.count(), 0)
 
+    def test_new_snippet_onetime(self):
+        """
+        One-time snippets get deleted after two views.
+        """
+        # POST data
+        data = self.valid_form_data()
+        data['expires'] = 'onetime'
+
+        # First view, the author gets redirected after posting
+        response = self.client.post(self.new_url, data, follow=True)
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(Snippet.objects.count(), 1)
+        self.assertContains(response, data['content'])
+
+        # Second View, another user looks at the snippet
+        response = self.client.get(response.request['PATH_INFO'], follow=True)
+        self.assertEqual(response.status_code, 200)
+        self.assertEqual(Snippet.objects.count(), 1)
+        self.assertContains(response, data['content'])
+
+        # Third/Further View, another user looks at the snippet but it was deleted
+        response = self.client.get(response.request['PATH_INFO'], follow=True)
+        self.assertEqual(response.status_code, 404)
+        self.assertEqual(Snippet.objects.count(), 0)
+
     # -------------------------------------------------------------------------
     # Reply
     # -------------------------------------------------------------------------