From a9477c7873d0228ef50c8da0b4856c39985ff840 Mon Sep 17 00:00:00 2001
From: Ritiek Malhotra <ritiekmalhotra123@gmail.com>
Date: Sun, 28 Apr 2019 15:26:18 +0530
Subject: [PATCH] Fix tests

---
 spotdl/patcher.py    | 7 ++++---
 test/test_patcher.py | 3 ++-
 2 files changed, 6 insertions(+), 4 deletions(-)

diff --git a/spotdl/patcher.py b/spotdl/patcher.py
index dde2fa3..d9e8762 100644
--- a/spotdl/patcher.py
+++ b/spotdl/patcher.py
@@ -1,3 +1,4 @@
+from pafy import backend_youtube_dl
 import pafy
 
 from spotdl import internals
@@ -26,7 +27,7 @@ def _process_streams(self):
             self._ydl_info['formats'][format_index]['url'] = self._ydl_info['formats'][format_index]['fragment_base_url']
         except KeyError:
             pass
-    return pafy.backend_youtube_dl.YtdlPafy._old_process_streams(self)
+    return backend_youtube_dl.YtdlPafy._old_process_streams(self)
 
 @classmethod
 def _content_available(cls, url):
@@ -40,5 +41,5 @@ class PatchPafy:
         pafy.backend_shared.BasePafy.getbestthumb = _getbestthumb
 
     def patch_process_streams(self):
-        pafy.backend_youtube_dl.YtdlPafy._old_process_streams = pafy.backend_youtube_dl.YtdlPafy._process_streams
-        pafy.backend_youtube_dl.YtdlPafy._process_streams = _process_streams
+        backend_youtube_dl.YtdlPafy._old_process_streams = backend_youtube_dl.YtdlPafy._process_streams
+        backend_youtube_dl.YtdlPafy._process_streams = _process_streams
diff --git a/test/test_patcher.py b/test/test_patcher.py
index baddd0e..f0f92aa 100644
--- a/test/test_patcher.py
+++ b/test/test_patcher.py
@@ -3,7 +3,8 @@ import pafy
 
 import pytest
 
-patcher.patch_pafy()
+pafy_patcher = patcher.PatchPafy()
+pafy_patcher.patch_getbestthumb()
 
 class TestPafyContentAvailable:
     pass