| # -*- coding: utf-8 -*- |
| |
| # Licensed to the Apache Software Foundation (ASF) under one |
| # or more contributor license agreements. See the NOTICE file |
| # distributed with this work for additional information |
| # regarding copyright ownership. The ASF licenses this file |
| # to you under the Apache License, Version 2.0 (the |
| # "License"); you may not use this file except in compliance |
| # with the License. You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, |
| # software distributed under the License is distributed on an |
| # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| # KIND, either express or implied. See the License for the |
| # specific language governing permissions and limitations |
| # under the License. |
| |
| |
| import mock |
| import unittest |
| import urllib |
| |
| from allura.lib.spam.akismetfilter import AKISMET_AVAILABLE, AkismetSpamFilter |
| |
| |
| @unittest.skipIf(not AKISMET_AVAILABLE, "Akismet not available") |
| class TestAkismet(unittest.TestCase): |
| |
| @mock.patch('allura.lib.spam.akismetfilter.akismet') |
| def setUp(self, akismet_lib): |
| self.akismet = AkismetSpamFilter({}) |
| |
| def side_effect(*args, **kw): |
| # side effect to test that data being sent to |
| # akismet can be successfully urlencoded |
| urllib.urlencode(kw.get('data', {})) |
| self.akismet.service.comment_check = mock.Mock(side_effect=side_effect) |
| self.fake_artifact = mock.Mock(**{'url.return_value': 'artifact url'}) |
| self.fake_user = mock.Mock(display_name=u'Søme User', |
| email_addresses=['user@domain']) |
| self.fake_headers = dict( |
| USER_AGENT='some browser', |
| REFERER='some url') |
| self.content = u'spåm text' |
| self.expected_data = dict( |
| comment_content=self.content.encode('utf8'), |
| comment_type='comment', |
| user_ip='some ip', |
| user_agent='some browser', |
| referrer='some url') |
| |
| @mock.patch('allura.lib.spam.akismetfilter.c') |
| @mock.patch('allura.lib.spam.akismetfilter.request') |
| def test_check(self, request, c): |
| request.headers = self.fake_headers |
| request.remote_addr = 'some ip' |
| c.user = None |
| self.akismet.service.comment_check.side_effect({'side_effect': ''}) |
| self.akismet.check(self.content) |
| self.akismet.service.comment_check.assert_called_once_with( |
| self.content, |
| data=self.expected_data, build_data=False) |
| |
| @mock.patch('allura.lib.spam.akismetfilter.c') |
| @mock.patch('allura.lib.spam.akismetfilter.request') |
| def test_check_with_explicit_content_type(self, request, c): |
| request.headers = self.fake_headers |
| request.remote_addr = 'some ip' |
| c.user = None |
| self.akismet.check(self.content, content_type='some content type') |
| self.expected_data['comment_type'] = 'some content type' |
| self.akismet.service.comment_check.assert_called_once_with( |
| self.content, |
| data=self.expected_data, build_data=False) |
| |
| @mock.patch('allura.lib.spam.akismetfilter.c') |
| @mock.patch('allura.lib.spam.akismetfilter.request') |
| def test_check_with_artifact(self, request, c): |
| request.headers = self.fake_headers |
| request.remote_addr = 'some ip' |
| c.user = None |
| self.akismet.check(self.content, artifact=self.fake_artifact) |
| expected_data = self.expected_data |
| expected_data['permalink'] = 'artifact url' |
| self.akismet.service.comment_check.assert_called_once_with( |
| self.content, |
| data=expected_data, build_data=False) |
| |
| @mock.patch('allura.lib.spam.akismetfilter.c') |
| @mock.patch('allura.lib.spam.akismetfilter.request') |
| def test_check_with_user(self, request, c): |
| request.headers = self.fake_headers |
| request.remote_addr = 'some ip' |
| c.user = None |
| self.akismet.check(self.content, user=self.fake_user) |
| expected_data = self.expected_data |
| expected_data.update(comment_author=u'Søme User'.encode('utf8'), |
| comment_author_email='user@domain') |
| self.akismet.service.comment_check.assert_called_once_with( |
| self.content, |
| data=expected_data, build_data=False) |
| |
| @mock.patch('allura.lib.spam.akismetfilter.c') |
| @mock.patch('allura.lib.spam.akismetfilter.request') |
| def test_check_with_implicit_user(self, request, c): |
| request.headers = self.fake_headers |
| request.remote_addr = 'some ip' |
| c.user = self.fake_user |
| self.akismet.check(self.content) |
| expected_data = self.expected_data |
| expected_data.update(comment_author=u'Søme User'.encode('utf8'), |
| comment_author_email='user@domain') |
| self.akismet.service.comment_check.assert_called_once_with( |
| self.content, |
| data=expected_data, build_data=False) |
| |
| @mock.patch('allura.lib.spam.akismetfilter.c') |
| @mock.patch('allura.lib.spam.akismetfilter.request') |
| def test_submit_spam(self, request, c): |
| request.headers = self.fake_headers |
| request.remote_addr = 'some ip' |
| c.user = None |
| self.akismet.submit_spam(self.content) |
| self.akismet.service.submit_spam.assert_called_once_with( |
| self.content, data=self.expected_data, build_data=False) |
| |
| @mock.patch('allura.lib.spam.akismetfilter.c') |
| @mock.patch('allura.lib.spam.akismetfilter.request') |
| def test_submit_ham(self, request, c): |
| request.headers = self.fake_headers |
| request.remote_addr = 'some ip' |
| c.user = None |
| self.akismet.submit_ham(self.content) |
| self.akismet.service.submit_ham.assert_called_once_with( |
| self.content, data=self.expected_data, build_data=False) |