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'