From d7055fac1d039bdb52ab8bd314d024bd72268673 Mon Sep 17 00:00:00 2001
From: Paldin Bet Eivaz <beteivap@oregonstate.edu>
Date: Wed, 17 Nov 2021 16:24:10 -0800
Subject: [PATCH] added test for track_for_id method

---
 test/test_spotify.py | 61 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 61 insertions(+)

diff --git a/test/test_spotify.py b/test/test_spotify.py
index 41217a9fd..87186d718 100644
--- a/test/test_spotify.py
+++ b/test/test_spotify.py
@@ -127,6 +127,67 @@ class SpotifyPluginTest(_common.TestCase, TestHelper):
         self.assertIn('album:Despicable Me 2', query)
         self.assertEqual(params['type'], ['track'])
 
+    @responses.activate
+    def test_track_for_id(self):
+        """Tests if plugin is able to fetch a track by its Spotify ID"""
+
+        # Mock the Spotify 'Get Track' call
+        json_file = os.path.join(
+            _common.RSRC, b'spotify', b'track_info.json'
+        )
+        with open(json_file, 'rb') as f:
+            response_body = f.read()
+
+        responses.add(
+            responses.GET,
+            spotify.SpotifyPlugin.track_url + '6NPVjNh8Jhru9xOmyQigds',
+            body=response_body,
+            status=200,
+            content_type='application/json',
+        )
+
+        # Mock the Spotify 'Get Album' call
+        json_file = os.path.join(
+            _common.RSRC, b'spotify', b'album_info.json'
+        )
+        with open(json_file, 'rb') as f:
+            response_body = f.read()
+
+        responses.add(
+            responses.GET,
+            spotify.SpotifyPlugin.album_url + '5l3zEmMrOhOzG8d8s83GOL',
+            body=response_body,
+            status=200,
+            content_type='application/json',
+        )
+
+        # Mock the Spotify 'Search' call
+        json_file = os.path.join(
+            _common.RSRC, b'spotify', b'track_request.json'
+        )
+        with open(json_file, 'rb') as f:
+            response_body = f.read()
+
+        responses.add(
+            responses.GET,
+            spotify.SpotifyPlugin.search_url,
+            body=response_body,
+            status=200,
+            content_type='application/json',
+        )
+
+        track_info = self.spotify.track_for_id('6NPVjNh8Jhru9xOmyQigds')
+        item = Item(
+            mb_trackid=track_info.track_id,
+            albumartist=track_info.artist,
+            title=track_info.title,
+            length=track_info.length
+        )
+        item.add(self.lib)
+
+        results = self.spotify._match_library_tracks(self.lib, "Happy")
+        self.assertEqual(1, len(results))
+        self.assertEqual("6NPVjNh8Jhru9xOmyQigds", results[0]['id'])
 
 def suite():
     return unittest.TestLoader().loadTestsFromName(__name__)