Updated tests - removed old ones and created the structure for the ne…

Javinator9889 · Jun 1, 2020 · 84ef305 · 84ef305
1 parent b234aee
commit 84ef305
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/YouTubeMDBot/tests/download_test.py b/YouTubeMDBot/tests/download_test.py
@@ -5,8 +5,6 @@
 from YouTubeMDBot.downloader import M4AYouTubeDownloader
 from YouTubeMDBot.downloader import MultipleYouTubeDownloader
 
-from YouTubeMDBot.utils import CBarrier
-
 log = logging.basicConfig()
 
 
@@ -59,12 +57,12 @@ def _test_download(self, downloader: YouTubeDownloader):
         self.assertEqual(io.read(), data)
 
     def _download_finished_callback(self, data):
-        print("Video download finished")
-        print(type(data))
+        log.info("Video download finished")
+        log.debug(type(data))
         self.finished += 1
 
     def _download_failed_callback(self, err):
-        print(f"Captured error: {err}")
+        log.error(f"Captured error: {err}")
 
 
 if __name__ == '__main__':