Example #1
0
class AkismetFilterStrategyTestCase(unittest.TestCase):

    def setUp(self):
        self.env = EnvironmentStub(enable=[AkismetFilterStrategy])
        self.strategy = AkismetFilterStrategy(self.env)
        self.urlopen = akismet.urllib2.urlopen = DummyURLOpener()

    def test_no_api_key(self):
        req = Mock()
        retval = self.strategy.test(req, 'anonymous', 'foobar')
        self.assertEqual(None, retval)

    def test_bad_api_key(self):
        req = Mock(authname='anonymous', base_url='http://example.org/',
                   remote_addr='127.0.0.1')
        self.env.config.set('spam-filter', 'akismet_api_key', 'INVALID')

        self.urlopen.responses = ['invalid']
        retval = self.strategy.test(req, 'anonymous', 'foobar')
        self.assertEqual(None, retval)
        self.assertEqual(1, len(self.urlopen.requests))

        req = self.urlopen.requests[0]
        self.assertEqual('http://rest.akismet.com/1.1/verify-key', req.url)
        self.assertEqual('blog=http%3A%2F%2Fexample.org%2F&key=INVALID',
                         req.params)
        self.assertEqual(self.strategy.user_agent, req.headers['User-Agent'])

    def test_check_ham(self):
        req = Mock(authname='anonymous', base_url='http://example.org/',
                   remote_addr='127.0.0.1', get_header=lambda x: None)
        self.env.config.set('spam-filter', 'akismet_api_key', 'mykey')

        self.urlopen.responses = ['valid', 'false']
        retval = self.strategy.test(req, 'anonymous', 'foobar')
        self.assertEqual(None, retval)
        self.assertEqual(2, len(self.urlopen.requests))

        req = self.urlopen.requests[1]
        self.assertEqual('http://mykey.rest.akismet.com/1.1/comment-check',
                         req.url)
        self.assertEqual('user_ip=127.0.0.1&referrer=unknown&'
                         'blog=http%3A%2F%2Fexample.org%2F&user_agent=None&'
                         'comment_content=foobar&comment_author=&'
                         'comment_author_email=anonymous', req.params)
        self.assertEqual(self.strategy.user_agent, req.headers['User-Agent'])

    def test_check_spam(self):
        req = Mock(authname='anonymous', base_url='http://example.org/',
                   remote_addr='127.0.0.1', get_header=lambda x: None)
        self.env.config.set('spam-filter', 'akismet_api_key', 'mykey')

        self.urlopen.responses = ['valid', 'true']
        retval = self.strategy.test(req, 'anonymous', 'foobar')
        self.assertEqual((-5, 'Akismet says content is spam'), retval)
        self.assertEqual(2, len(self.urlopen.requests))

    def test_submit_ham(self):
        req = Mock(authname='anonymous', base_url='http://example.org/',
                   remote_addr='127.0.0.1', get_header=lambda x: None)
        self.env.config.set('spam-filter', 'akismet_api_key', 'mykey')

        self.urlopen.responses = ['valid', '']
        self.strategy.train(req, 'anonymous', 'foobar', spam=False)

        req = self.urlopen.requests[1]
        self.assertEqual('http://mykey.rest.akismet.com/1.1/submit-ham',
                         req.url)
        self.assertEqual('user_ip=127.0.0.1&referrer=unknown&'
                         'blog=http%3A%2F%2Fexample.org%2F&user_agent=None&'
                         'comment_content=foobar&comment_author=&'
                         'comment_author_email=anonymous', req.params)
        self.assertEqual(self.strategy.user_agent, req.headers['User-Agent'])

    def test_submit_spam(self):
        req = Mock(authname='anonymous', base_url='http://example.org/',
                   remote_addr='127.0.0.1', get_header=lambda x: None)
        self.env.config.set('spam-filter', 'akismet_api_key', 'mykey')

        self.urlopen.responses = ['valid', '']
        retval = self.strategy.train(req, 'anonymous', 'foobar', spam=True)

        req = self.urlopen.requests[1]
        self.assertEqual('http://mykey.rest.akismet.com/1.1/submit-spam',
                         req.url)
        self.assertEqual('user_ip=127.0.0.1&referrer=unknown&'
                         'blog=http%3A%2F%2Fexample.org%2F&user_agent=None&'
                         'comment_content=foobar&comment_author=&'
                         'comment_author_email=anonymous', req.params)
        self.assertEqual(self.strategy.user_agent, req.headers['User-Agent'])
Example #2
0
class AkismetFilterStrategyTestCase(unittest.TestCase):

    def setUp(self):
        self.env = EnvironmentStub(enable=[AkismetFilterStrategy])
        self.strategy = AkismetFilterStrategy(self.env)
        self.urlopen = akismet.urllib2.urlopen = DummyURLOpener()

    def test_no_api_key(self):
        req = MockRequest(self.env)
        retval = self.strategy.test(req, 'anonymous', 'foobar',
                                    req.remote_addr)
        self.assertEqual(None, retval)

    def test_bad_api_key(self):
        req = MockRequest(self.env)
        self.env.config.set('spam-filter', 'akismet_api_key', 'INVALID')

        self.urlopen.responses = ['invalid']
        retval = self.strategy.test(req, 'anonymous', 'foobar',
                                    req.remote_addr)
        self.assertEqual(None, retval)
        self.assertEqual(1, len(self.urlopen.requests))

        req = self.urlopen.requests[0]
        self.assertEqual('http://rest.akismet.com/1.1/verify-key', req.url)
        self.assertEqual('blog=http%3A%2F%2Fexample.org%2Ftrac.cgi'
                         '&key=INVALID', req.params)
        self.assertEqual(user_agent, req.headers['User-Agent'])

    def test_check_ham(self):
        req = MockRequest(self.env)
        self.env.config.set('spam-filter', 'akismet_api_key', 'mykey')

        self.urlopen.responses = ['valid', 'false']
        retval = self.strategy.test(req, 'anonymous', 'foobar',
                                    req.remote_addr)
        self.assertEqual(None, retval)
        self.assertEqual(2, len(self.urlopen.requests))

        req = self.urlopen.requests[1]
        self.assertEqual('http://mykey.rest.akismet.com/1.1/comment-check',
                         req.url)
        self.assertEqual('user_ip=127.0.0.1&comment_type=trac&'
                         'referrer=unknown&HTTP_ACCEPT_LANGUAGE=en-US&'
                         'blog=http%3A%2F%2Fexample.org%2Ftrac.cgi&'
                         'user_agent=None&comment_content=foobar&'
                         'comment_author=anonymous', req.params)
        self.assertEqual(user_agent, req.headers['User-Agent'])

    def test_check_spam(self):
        req = MockRequest(self.env)
        self.env.config.set('spam-filter', 'akismet_api_key', 'mykey')

        self.urlopen.responses = ['valid', 'true']
        retval = self.strategy.test(req, 'anonymous', 'foobar',
                                    req.remote_addr)
        self.assertEqual((-10, 'Akismet says content is spam'), retval)
        self.assertEqual(2, len(self.urlopen.requests))

    def test_submit_ham(self):
        req = MockRequest(self.env)
        self.env.config.set('spam-filter', 'akismet_api_key', 'mykey')

        self.urlopen.responses = ['valid', '']
        self.strategy.train(req, 'anonymous', 'foobar', req.remote_addr,
                            spam=False)

        req = self.urlopen.requests[1]
        self.assertEqual('http://mykey.rest.akismet.com/1.1/submit-ham',
                         req.url)
        self.assertEqual('user_ip=127.0.0.1&comment_type=trac&'
                         'referrer=unknown&HTTP_ACCEPT_LANGUAGE=en-US&'
                         'blog=http%3A%2F%2Fexample.org%2Ftrac.cgi&'
                         'user_agent=None&comment_content=foobar&'
                         'comment_author=anonymous', req.params)
        self.assertEqual(user_agent, req.headers['User-Agent'])

    def test_submit_spam(self):
        req = MockRequest(self.env)
        self.env.config.set('spam-filter', 'akismet_api_key', 'mykey')

        self.urlopen.responses = ['valid', '']
        self.strategy.train(req, 'anonymous', 'foobar', req.remote_addr,
                            spam=True)

        req = self.urlopen.requests[1]
        self.assertEqual('http://mykey.rest.akismet.com/1.1/submit-spam',
                         req.url)
        self.assertEqual('user_ip=127.0.0.1&comment_type=trac&'
                         'referrer=unknown&HTTP_ACCEPT_LANGUAGE=en-US&'
                         'blog=http%3A%2F%2Fexample.org%2Ftrac.cgi&'
                         'user_agent=None&comment_content=foobar'
                         '&comment_author=anonymous',
                         req.params)
        self.assertEqual(user_agent, req.headers['User-Agent'])