logo

searx

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

test_genius.py (8491B)

    

  1. from collections import defaultdict

  2. import mock

  3. from datetime import datetime

  4. from searx.engines import genius

  5. from searx.testing import SearxTestCase

  6. 

  7. 

  8. class TestGeniusEngine(SearxTestCase):

  9. 

  10.     def test_request(self):

  11.         query = 'test_query'

  12.         dicto = defaultdict(dict)

  13.         dicto['pageno'] = 1

  14.         params = genius.request(query, dicto)

  15.         self.assertTrue('url' in params)

  16.         self.assertTrue(query in params['url'])

  17.         self.assertTrue('genius.com' in params['url'])

  18. 

  19.     def test_response(self):

  20. 

  21.         json_empty = """

  22.         {

  23.         "meta": {

  24.             "status": 200

  25.         },

  26.         "response": {

  27.             "sections": [

  28.             {

  29.                 "type": "top_hit",

  30.                 "hits": []

  31.             },

  32.             {

  33.                 "type": "song",

  34.                 "hits": []

  35.             },

  36.             {

  37.                 "type": "lyric",

  38.                 "hits": []

  39.             },

  40.             {

  41.                 "type": "artist",

  42.                 "hits": []

  43.             },

  44.             {

  45.                 "type": "album",

  46.                 "hits": []

  47.             },

  48.             {

  49.                 "type": "tag",

  50.                 "hits": []

  51.             },

  52.             {

  53.                 "type": "video",

  54.                 "hits": []

  55.             },

  56.             {

  57.                 "type": "article",

  58.                 "hits": []

  59.             },

  60.             {

  61.                 "type": "user",

  62.                 "hits": []

  63.             }

  64.             ]

  65.         }

  66.         }

  67.         """

  68. 

  69.         resp = mock.Mock(text=json_empty)

  70.         self.assertEqual(genius.response(resp), [])

  71. 

  72.         json = """

  73.         {

  74.         "meta": {

  75.             "status": 200

  76.         },

  77.         "response": {

  78.             "sections": [

  79.             {

  80.                 "type": "lyric",

  81.                 "hits": [

  82.                 {

  83.                     "highlights": [

  84.                     {

  85.                         "property": "lyrics",

  86.                         "value": "Sample lyrics",

  87.                         "snippet": true,

  88.                         "ranges": []

  89.                     }

  90.                     ],

  91.                     "index": "lyric",

  92.                     "type": "song",

  93.                     "result": {

  94.                     "_type": "song",

  95.                     "annotation_count": 45,

  96.                     "api_path": "/songs/52916",

  97.                     "full_title": "J't'emmerde by MC Jean Gab'1",

  98.                     "header_image_thumbnail_url": "https://images.genius.com/xxx.300x300x1.jpg",

  99.                     "header_image_url": "https://images.genius.com/ef9f736a86df3c3b1772f3fb7fbdb21c.1000x1000x1.jpg",

  100.                     "id": 52916,

  101.                     "instrumental": false,

  102.                     "lyrics_owner_id": 15586,

  103.                     "lyrics_state": "complete",

  104.                     "lyrics_updated_at": 1498744545,

  105.                     "path": "/Mc-jean-gab1-jtemmerde-lyrics",

  106.                     "pyongs_count": 4,

  107.                     "song_art_image_thumbnail_url": "https://images.genius.com/xxx.300x300x1.jpg",

  108.                     "stats": {

  109.                         "hot": false,

  110.                         "unreviewed_annotations": 0,

  111.                         "pageviews": 62490

  112.                     },

  113.                     "title": "J't'emmerde",

  114.                     "title_with_featured": "J't'emmerde",

  115.                     "updated_by_human_at": 1498744546,

  116.                     "url": "https://genius.com/Mc-jean-gab1-jtemmerde-lyrics",

  117.                     "primary_artist": {

  118.                         "_type": "artist",

  119.                         "api_path": "/artists/12691",

  120.                         "header_image_url": "https://images.genius.com/c7847662a58f8c2b0f02a6e217d60907.960x657x1.jpg",

  121.                         "id": 12691,

  122.                         "image_url": "https://s3.amazonaws.com/rapgenius/Mc-jean-gab1.jpg",

  123.                         "index_character": "m",

  124.                         "is_meme_verified": false,

  125.                         "is_verified": false,

  126.                         "name": "MC Jean Gab'1",

  127.                         "slug": "Mc-jean-gab1",

  128.                         "url": "https://genius.com/artists/Mc-jean-gab1"

  129.                     }

  130.                     }

  131.                 }

  132.                 ]

  133.             },

  134.             {

  135.                 "type": "artist",

  136.                 "hits": [

  137.                 {

  138.                     "highlights": [],

  139.                     "index": "artist",

  140.                     "type": "artist",

  141.                     "result": {

  142.                     "_type": "artist",

  143.                     "api_path": "/artists/191580",

  144.                     "header_image_url": "https://assets.genius.com/images/default_avatar_300.png?1503090542",

  145.                     "id": 191580,

  146.                     "image_url": "https://assets.genius.com/images/default_avatar_300.png?1503090542",

  147.                     "index_character": "a",

  148.                     "is_meme_verified": false,

  149.                     "is_verified": false,

  150.                     "name": "ASDF Guy",

  151.                     "slug": "Asdf-guy",

  152.                     "url": "https://genius.com/artists/Asdf-guy"

  153.                     }

  154.                 }

  155.                 ]

  156.             },

  157.             {

  158.                 "type": "album",

  159.                 "hits": [

  160.                 {

  161.                     "highlights": [],

  162.                     "index": "album",

  163.                     "type": "album",

  164.                     "result": {

  165.                     "_type": "album",

  166.                     "api_path": "/albums/132332",

  167.                     "cover_art_thumbnail_url": "https://images.genius.com/xxx.300x300x1.jpg",

  168.                     "cover_art_url": "https://images.genius.com/xxx.600x600x1.jpg",

  169.                     "full_title": "ASD by A Skylit Drive",

  170.                     "id": 132332,

  171.                     "name": "ASD",

  172.                     "name_with_artist": "ASD (artist: A Skylit Drive)",

  173.                     "release_date_components": {

  174.                         "year": 2015,

  175.                         "month": null,

  176.                         "day": null

  177.                     },

  178.                     "url": "https://genius.com/albums/A-skylit-drive/Asd",

  179.                     "artist": {

  180.                         "_type": "artist",

  181.                         "api_path": "/artists/48712",

  182.                         "header_image_url": "https://images.genius.com/814c1551293172c56306d0e310c6aa89.620x400x1.jpg",

  183.                         "id": 48712,

  184.                         "image_url": "https://images.genius.com/814c1551293172c56306d0e310c6aa89.620x400x1.jpg",

  185.                         "index_character": "s",

  186.                         "is_meme_verified": false,

  187.                         "is_verified": false,

  188.                         "name": "A Skylit Drive",

  189.                         "slug": "A-skylit-drive",

  190.                         "url": "https://genius.com/artists/A-skylit-drive"

  191.                     }

  192.                     }

  193.                 }

  194.                 ]

  195.             }

  196.             ]

  197.         }

  198.         }

  199.         """

  200. 

  201.         resp = mock.Mock(text=json)

  202.         results = genius.response(resp)

  203. 

  204.         self.assertEqual(len(results), 3)

  205.         self.assertEqual(type(results), list)

  206. 

  207.         # check lyric parsing

  208.         r = results[0]

  209.         self.assertEqual(r['url'], 'https://genius.com/Mc-jean-gab1-jtemmerde-lyrics')

  210.         self.assertEqual(r['title'], "J't'emmerde by MC Jean Gab'1")

  211.         self.assertEqual(r['content'], "Sample lyrics")

  212.         self.assertEqual(r['template'], 'videos.html')

  213.         self.assertEqual(r['thumbnail'], 'https://images.genius.com/xxx.300x300x1.jpg')

  214.         created = datetime.fromtimestamp(1498744545)

  215.         self.assertEqual(r['publishedDate'], created)

  216. 

  217.         # check artist parsing

  218.         r = results[1]

  219.         self.assertEqual(r['url'], 'https://genius.com/artists/Asdf-guy')

  220.         self.assertEqual(r['title'], "ASDF Guy")

  221.         self.assertEqual(r['content'], None)

  222.         self.assertEqual(r['template'], 'videos.html')

  223.         self.assertEqual(r['thumbnail'], 'https://assets.genius.com/images/default_avatar_300.png?1503090542')

  224. 

  225.         # check album parsing

  226.         r = results[2]

  227.         self.assertEqual(r['url'], 'https://genius.com/albums/A-skylit-drive/Asd')

  228.         self.assertEqual(r['title'], "ASD by A Skylit Drive")

  229.         self.assertEqual(r['content'], "Released: 2015")

  230.         self.assertEqual(r['template'], 'videos.html')

  231.         self.assertEqual(r['thumbnail'], 'https://images.genius.com/xxx.600x600x1.jpg')