| # -*- coding: utf-8 -*- |
| |
| import mock |
| import unittest |
| import urllib |
| |
| try: |
| from allura.lib.spam.mollomfilter import MollomSpamFilter |
| except ImportError: |
| MollomSpamFilter = None |
| |
| |
| @unittest.skipIf(MollomSpamFilter is None, "Can't import MollomSpamFilter") |
| class TestMollom(unittest.TestCase): |
| @mock.patch('allura.lib.spam.mollomfilter.Mollom') |
| def setUp(self, mollom_lib): |
| self.mollom = MollomSpamFilter({}) |
| def side_effect(*args, **kw): |
| # side effect to test that data being sent to |
| # mollom can be successfully urlencoded |
| urllib.urlencode(kw.get('data', {})) |
| return dict(spam=2) |
| self.mollom.service.checkContent = mock.Mock(side_effect=side_effect, |
| return_value=dict(spam=2)) |
| 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( |
| REMOTE_ADDR='fallback ip', |
| X_FORWARDED_FOR='some ip', |
| USER_AGENT='some browser', |
| REFERER='some url') |
| self.content = u'spåm text' |
| self.expected_data = dict( |
| postBody=self.content.encode('utf8'), |
| authorIP='some ip') |
| |
| @mock.patch('allura.lib.spam.mollomfilter.c') |
| @mock.patch('allura.lib.spam.mollomfilter.request') |
| def test_check(self, request, c): |
| request.headers = self.fake_headers |
| c.user = None |
| self.mollom.check(self.content) |
| self.mollom.service.checkContent.assert_called_once_with(**self.expected_data) |
| |
| @mock.patch('allura.lib.spam.mollomfilter.c') |
| @mock.patch('allura.lib.spam.mollomfilter.request') |
| def test_check_with_user(self, request, c): |
| request.headers = self.fake_headers |
| c.user = None |
| self.mollom.check(self.content, user=self.fake_user) |
| expected_data = self.expected_data |
| expected_data.update(authorName=u'Søme User'.encode('utf8'), |
| authorMail='user@domain') |
| self.mollom.service.checkContent.assert_called_once_with(**self.expected_data) |
| |
| @mock.patch('allura.lib.spam.mollomfilter.c') |
| @mock.patch('allura.lib.spam.mollomfilter.request') |
| def test_check_with_implicit_user(self, request, c): |
| request.headers = self.fake_headers |
| c.user = self.fake_user |
| self.mollom.check(self.content) |
| expected_data = self.expected_data |
| expected_data.update(authorName=u'Søme User'.encode('utf8'), |
| authorMail='user@domain') |
| self.mollom.service.checkContent.assert_called_once_with(**self.expected_data) |
| |
| @mock.patch('allura.lib.spam.mollomfilter.c') |
| @mock.patch('allura.lib.spam.mollomfilter.request') |
| def test_check_with_fallback_ip(self, request, c): |
| self.expected_data['authorIP'] = 'fallback ip' |
| self.fake_headers.pop('X_FORWARDED_FOR') |
| request.headers = self.fake_headers |
| request.remote_addr = self.fake_headers['REMOTE_ADDR'] |
| c.user = None |
| self.mollom.check(self.content) |
| self.mollom.service.checkContent.assert_called_once_with(**self.expected_data) |