logo

searx

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

test_spotify.py (4843B)

    

  1. from collections import defaultdict
  2. import mock
  3. from searx.engines import spotify
  4. from searx.testing import SearxTestCase
  7. class TestSpotifyEngine(SearxTestCase):
  9.     def test_request(self):
  10.         query = 'test_query'
  11.         dicto = defaultdict(dict)
  12.         dicto['pageno'] = 0
  13.         params = spotify.request(query, dicto)
  14.         self.assertIn('url', params)
  15.         self.assertIn(query, params['url'])
  16.         self.assertIn('spotify.com', params['url'])
  18.     def test_response(self):
  19.         self.assertRaises(AttributeError, spotify.response, None)
  20.         self.assertRaises(AttributeError, spotify.response, [])
  21.         self.assertRaises(AttributeError, spotify.response, '')
  22.         self.assertRaises(AttributeError, spotify.response, '[]')
  24.         response = mock.Mock(text='{}')
  25.         self.assertEqual(spotify.response(response), [])
  27.         response = mock.Mock(text='{"data": []}')
  28.         self.assertEqual(spotify.response(response), [])
  30.         json = """
  31.         {
  32.           "tracks": {
  33.             "href": "https://api.spotify.com/v1/search?query=nosfell&offset=0&limit=20&type=track",
  34.             "items": [
  35.               {
  36.                 "album": {
  37.                   "album_type": "album",
  38.                   "external_urls": {
  39.                     "spotify": "https://open.spotify.com/album/5c9ap1PBkSGLxT3J73toxA"
  40.                   },
  41.                   "href": "https://api.spotify.com/v1/albums/5c9ap1PBkSGLxT3J73toxA",
  42.                   "id": "5c9ap1PBkSGLxT3J73toxA",
  43.                   "name": "Album Title",
  44.                   "type": "album",
  45.                   "uri": "spotify:album:5c9ap1PBkSGLxT3J73toxA"
  46.                 },
  47.                 "artists": [
  48.                   {
  49.                     "external_urls": {
  50.                       "spotify": "https://open.spotify.com/artist/0bMc6b75FfZEpQHG1jifKu"
  51.                     },
  52.                     "href": "https://api.spotify.com/v1/artists/0bMc6b75FfZEpQHG1jifKu",
  53.                     "id": "0bMc6b75FfZEpQHG1jifKu",
  54.                     "name": "Artist Name",
  55.                     "type": "artist",
  56.                     "uri": "spotify:artist:0bMc6b75FfZEpQHG1jifKu"
  57.                   }
  58.                 ],
  59.                 "disc_number": 1,
  60.                 "duration_ms": 202386,
  61.                 "explicit": false,
  62.                 "external_ids": {
  63.                   "isrc": "FRV640600067"
  64.                 },
  65.                 "external_urls": {
  66.                   "spotify": "https://open.spotify.com/track/2GzvFiedqW8hgqUpWcASZa"
  67.                 },
  68.                 "href": "https://api.spotify.com/v1/tracks/2GzvFiedqW8hgqUpWcASZa",
  69.                 "id": "1000",
  70.                 "is_playable": true,
  71.                 "name": "Title of track",
  72.                 "popularity": 6,
  73.                 "preview_url": "https://p.scdn.co/mp3-preview/7b8ecda580965a066b768c2647f877e43f7b1a0a",
  74.                 "track_number": 3,
  75.                 "type": "track",
  76.                 "uri": "spotify:track:2GzvFiedqW8hgqUpWcASZa"
  77.               }
  78.             ],
  79.             "limit": 20,
  80.             "next": "https://api.spotify.com/v1/search?query=nosfell&offset=20&limit=20&type=track",
  81.             "offset": 0,
  82.             "previous": null,
  83.             "total": 107
  84.           }
  85.         }
  86.         """
  87.         response = mock.Mock(text=json)
  88.         results = spotify.response(response)
  89.         self.assertEqual(type(results), list)
  90.         self.assertEqual(len(results), 1)
  91.         self.assertEqual(results[0]['title'], 'Title of track')
  92.         self.assertEqual(results[0]['url'], 'https://open.spotify.com/track/2GzvFiedqW8hgqUpWcASZa')
  93.         self.assertEqual(results[0]['content'], 'Artist Name - Album Title - Title of track')
  94.         self.assertIn('1000', results[0]['embedded'])
  96.         json = """
  97.         {
  98.           "tracks": {
  99.             "href": "https://api.spotify.com/v1/search?query=nosfell&offset=0&limit=20&type=track",
  100.             "items": [
  101.               {
  102.                 "href": "https://api.spotify.com/v1/tracks/2GzvFiedqW8hgqUpWcASZa",
  103.                 "id": "1000",
  104.                 "is_playable": true,
  105.                 "name": "Title of track",
  106.                 "popularity": 6,
  107.                 "preview_url": "https://p.scdn.co/mp3-preview/7b8ecda580965a066b768c2647f877e43f7b1a0a",
  108.                 "track_number": 3,
  109.                 "type": "album",
  110.                 "uri": "spotify:track:2GzvFiedqW8hgqUpWcASZa"
  111.               }
  112.             ],
  113.             "limit": 20,
  114.             "next": "https://api.spotify.com/v1/search?query=nosfell&offset=20&limit=20&type=track",
  115.             "offset": 0,
  116.             "previous": null,
  117.             "total": 107
  118.           }
  119.         }
  120.         """
  121.         response = mock.Mock(text=json)
  122.         results = spotify.response(response)
  123.         self.assertEqual(type(results), list)
  124.         self.assertEqual(len(results), 0)