From 70b426d770423641246332816281af2e1aaf2cd2 Mon Sep 17 00:00:00 2001
From: Martin Zimmermann <info@posativ.org>
Date: Wed, 13 Nov 2013 16:12:47 +0100
Subject: [PATCH] add unittest for spam guard

---
 specs/test_guard.py | 80 +++++++++++++++++++++++++++++++++++++++++++++
 specs/test_vote.py  |  5 +--
 2 files changed, 81 insertions(+), 4 deletions(-)
 create mode 100644 specs/test_guard.py

diff --git a/specs/test_guard.py b/specs/test_guard.py
new file mode 100644
index 0000000..e759f26
--- /dev/null
+++ b/specs/test_guard.py
@@ -0,0 +1,80 @@
+# -*- encoding: utf-8 -*-
+
+import unittest
+
+import os
+import json
+import tempfile
+
+from werkzeug.test import Client
+from werkzeug.wrappers import Response
+
+from isso import Isso, core
+from isso.utils import http
+
+from fixtures import curl, FakeIP
+http.curl = curl
+
+
+class TestGuard(unittest.TestCase):
+
+    data = json.dumps({"text": "Lorem ipsum."})
+
+    def setUp(self):
+        self.path = tempfile.NamedTemporaryFile().name
+
+    def makeClient(self, ip, ratelimit=2, direct_reply=3):
+
+        conf = core.Config.load(None)
+        conf.set("general", "dbpath", self.path)
+        conf.set("guard", "enabled", "true")
+        conf.set("guard", "ratelimit", str(ratelimit))
+        conf.set("guard", "direct-reply", str(direct_reply))
+
+        class App(Isso, core.Mixin):
+            pass
+
+        app = App(conf)
+        app.wsgi_app = FakeIP(app.wsgi_app, ip)
+
+        return Client(app, Response)
+
+    def testRateLimit(self):
+
+        bob = self.makeClient("127.0.0.1", 2)
+
+        for i in range(2):
+            rv = bob.post('/new?uri=test', data=self.data)
+            assert rv.status_code == 201
+
+        rv = bob.post('/new?uri=test', data=self.data)
+
+        assert rv.status_code == 403
+        assert "ratelimit exceeded" in rv.data
+
+        alice = self.makeClient("1.2.3.4", 2)
+        for i in range(2):
+            assert alice.post("/new?uri=test", data=self.data).status_code == 201
+
+        bob.application.db.execute([
+            "UPDATE comments SET",
+            "    created = created - 60",
+            "WHERE remote_addr = '127.0.0.0'"
+        ])
+
+        assert bob.post("/new?uri=test", data=self.data).status_code == 201
+
+    def testDirectReply(self):
+
+        client = self.makeClient("127.0.0.1", 15, 3)
+
+        for url in ("foo", "bar", "baz", "spam"):
+            for _ in range(3):
+                rv = client.post("/new?uri=%s" % url, data=self.data)
+                assert rv.status_code == 201
+
+        for url in ("foo", "bar", "baz", "spam"):
+            rv = client.post("/new?uri=%s" % url, data=self.data)
+
+            assert rv.status_code == 403
+            assert "direct responses to" in rv.data
diff --git a/specs/test_vote.py b/specs/test_vote.py
index 874ed39..89e37ed 100644
--- a/specs/test_vote.py
+++ b/specs/test_vote.py
@@ -19,10 +19,7 @@ http.curl = curl
 class TestVote(unittest.TestCase):
 
     def setUp(self):
-        fd, self.path = tempfile.mkstemp()
-
-    def tearDown(self):
-        os.unlink(self.path)
+        self.path = tempfile.NamedTemporaryFile().name
 
     def makeClient(self, ip):