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

  5. 

  6. 

  7. class TestDeviantartEngine(SearxTestCase):

  8. 

  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'])

  18. 

  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'])

  25. 

  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, '[]')

  31. 

  32.         response = mock.Mock(text='<html></html>')

  33.         self.assertEqual(deviantart.response(response), [])

  34. 

  35.         response = mock.Mock(status_code=302)

  36.         self.assertEqual(deviantart.response(response), [])

  37. 

  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')

  56. 

  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)