def testInsertSafe(self):
for _ in xrange(5):
funcs.insert_url(self.test_ip, self.test_url)
for _ in xrange(5):
count = funcs.insert_url(self.test_ip, "http://www.testcase.com")
self.assertEqual(10, count)
def testMalformedUrl(self):
r = funcs.insert_url(self.test_ip, "not a url")
self.assertEqual(r, None)
def testSpamStop(self):
for _ in xrange(29):
funcs.insert_url(self.test_ip, self.test_url)
with self.assertRaises(funcs.SpamProtectionStop):
funcs.insert_url(self.test_ip, self.test_url)
def hit():
try:
funcs.insert_url(request.remote_addr, "http://%s" % request.forms.get("url"))
except funcs.SpamProtectionStop:
pass
return u'ok'
