searx

commit: d0001f10e6c604f3094d728bed703a60baa9ae17
parent 0bfbdff234535f0a190f272d2f03d7a801b74e10
Author: Kirill Isakov <ukwt@ya.ru>
Date:   Wed, 30 Mar 2016 17:30:46 +0600

Add searx user agent to Reddit engine request to comply with API usage terms

Diffstat:
M
searx/engines/reddit.py
4
++++
M
tests/unit/engines/test_reddit.py
2
++

2 files changed, 6 insertions(+), 0 deletions(-)
diff --git a/searx/engines/reddit.py b/searx/engines/reddit.py
@@ -15,6 +15,7 @@ from cgi import escape
 from urllib import urlencode
 from urlparse import urlparse
 from datetime import datetime
+from searx.utils import searx_useragent
 
 # engine dependent config
 categories = ['general', 'images', 'news', 'social media']
@@ -30,6 +31,9 @@ def request(query, params):
                        'limit': page_size})
     params['url'] = search_url.format(query=query)
 
+    # using searx User-Agent
+    params['headers']['User-Agent'] = searx_useragent()
+
     return params
 
 
diff --git a/tests/unit/engines/test_reddit.py b/tests/unit/engines/test_reddit.py
@@ -3,6 +3,7 @@ import mock
 from searx.engines import reddit
 from searx.testing import SearxTestCase
 from datetime import datetime
+from searx.utils import searx_useragent
 
 
 class TestRedditEngine(SearxTestCase):
@@ -14,6 +15,7 @@ class TestRedditEngine(SearxTestCase):
         self.assertTrue('url' in params)
         self.assertTrue(query in params['url'])
         self.assertTrue('reddit.com' in params['url'])
+        self.assertEqual(params['headers']['User-Agent'], searx_useragent())
 
     def test_response(self):
         resp = mock.Mock(text='{}')