logo

searx

My custom branche(s) on searx, a meta-search engine git clone https://hacktivis.me/git/searx.git

test_deviantart.py (4141B)

    

  1. from collections import defaultdict
  2. import mock
  3. from searx.engines import deviantart
  4. from searx.testing import SearxTestCase
  7. class TestDeviantartEngine(SearxTestCase):
  9.     def test_request(self):
  10.         dicto = defaultdict(dict)
  11.         query = 'test_query'
  12.         dicto['pageno'] = 0
  13.         dicto['time_range'] = ''
  14.         params = deviantart.request(query, dicto)
  15.         self.assertTrue('url' in params)
  16.         self.assertTrue(query in params['url'])
  17.         self.assertTrue('deviantart.com' in params['url'])
  19.     def test_no_url_in_request_year_time_range(self):
  20.         dicto = defaultdict(dict)
  21.         query = 'test_query'
  22.         dicto['time_range'] = 'year'
  23.         params = deviantart.request(query, dicto)
  24.         self.assertEqual({}, params['url'])
  26.     def test_response(self):
  27.         self.assertRaises(AttributeError, deviantart.response, None)
  28.         self.assertRaises(AttributeError, deviantart.response, [])
  29.         self.assertRaises(AttributeError, deviantart.response, '')
  30.         self.assertRaises(AttributeError, deviantart.response, '[]')
  32.         response = mock.Mock(text='<html></html>')
  33.         self.assertEqual(deviantart.response(response), [])
  35.         response = mock.Mock(status_code=302)
  36.         self.assertEqual(deviantart.response(response), [])
  38.         html = """
  39.         <div id="page-1-results" class="page-results results-page-thumb torpedo-container">
  40.         <span class="thumb wide" href="http://amai911.deviantart.com/art/Horse-195212845"
  41.         data-super-full-width="900" data-super-full-height="600">
  42.             <a class="torpedo-thumb-link" href="https://url.of.image">
  43.                 <img data-sigil="torpedo-img" src="https://url.of.thumbnail" />
  44.             </a>
  45.         <span class="info"><span class="title-wrap"><span class="title">Title of image</span></span>
  46.         </div>
  47.         """
  48.         response = mock.Mock(text=html)
  49.         results = deviantart.response(response)
  50.         self.assertEqual(type(results), list)
  51.         self.assertEqual(len(results), 1)
  52.         self.assertEqual(results[0]['title'], 'Title of image')
  53.         self.assertEqual(results[0]['url'], 'https://url.of.image')
  54.         self.assertNotIn('content', results[0])
  55.         self.assertEqual(results[0]['thumbnail_src'], 'https://url.of.thumbnail')
  57.         html = """
  58.         <span class="tt-fh-tc" style="width: 202px;">
  59.             <span class="tt-bb" style="width: 202px;">
  60.             </span>
  61.             <span class="shadow">
  62.                 <a class="thumb" href="http://url.of.result/2nd.part.of.url"
  63.                     title="Behoimi BE Animation Test by test-0, Jan 4,
  64.                     2010 in Digital Art &gt; Animation"> <i></i>
  65.                     <img width="200" height="200" alt="Test"
  66.                         src="http://url.of.thumbnail" data-src="http://th08.deviantart.net/test.jpg">
  67.                 </a>
  68.             </span>
  69.             <!-- ^TTT -->
  70.         </span>
  71.         <span class="details">
  72.             <a href="http://test-0.deviantart.com/art/Test" class="t"
  73.                 title="Behoimi BE Animation Test by test-0, Jan 4, 2010">
  74.                 <span class="tt-fh-oe">Title of image</span> </a>
  75.             <small>
  76.             <span class="category">
  77.                 <span class="age">
  78.                     5 years ago
  79.                 </span>
  80.                 in <a title="Behoimi BE Animation Test by test-0, Jan 4, 2010"
  81.                     href="http://www.deviantart.com/browse/all/digitalart/animation/">Animation</a>
  82.             </span>
  83.             <div class="commentcount">
  84.                 <a href="http://test-0.deviantart.com/art/Test#comments">
  85.                 <span class="iconcommentsstats"></span>9 Comments</a>
  86.             </div>
  87.             <a class="mlt-link" href="http://www.deviantart.com/morelikethis/149167425">
  88.             <span class="mlt-icon"></span> <span class="mlt-text">More Like This</span> </a>
  89.         </span>
  90.         </small> <!-- TTT$ -->
  91.         """
  92.         response = mock.Mock(text=html)
  93.         results = deviantart.response(response)
  94.         self.assertEqual(type(results), list)
  95.         self.assertEqual(len(results), 0)