Start to reorganize views a little.

[audio.git] / src / youtube / tests.py

diff --git a/src/youtube/tests.py b/src/youtube/tests.py
index 7ce503c..a7587e3 100644 (file)
--- a/src/youtube/tests.py
+++ b/src/youtube/tests.py
@@ -1,3 +1,32 @@
+from unittest import mock

 from django.test import TestCase
 from django.test import TestCase

+from archive.models import Audiobook
+from . import tasks

 
 

-# Create your tests here.
+
+@mock.patch("youtube.models.youtube_call")
+@mock.patch("youtube.utils.subprocess")
+@mock.patch("youtube.models.concat_audio")
+@mock.patch("youtube.models.unlink")
+@mock.patch("youtube.models.YouTube.prepare_thumbnail")
+class YouTubeTests(TestCase):
+    fixtures = ["tests.yaml"]
+
+    def test_youtube_volumes(
+        self, prepare_thumbnail, unlink, concat_audio, subprocess, youtube_call
+    ):
+        youtube_call.return_value = mock.Mock(
+            json=mock.Mock(return_value={"id": "deadbeef"})
+        )
+        audiobooks = Audiobook.objects.all().order_by("index")
+
+        self.assertEqual(audiobooks[0].youtube_volume_count, 3)
+        self.assertEqual([a.youtube_volume_index for a in audiobooks], [1, 2, 2, 3])
+        self.assertEqual(
+            [a.is_youtube_publishable for a in audiobooks], [True, True, False, True]
+        )
+
+        tasks.YouTubeTask().run(None, 2, True)
+
+        # In creating a volume of two audiobooks, we should've called concat with a list of two files.
+        self.assertEqual(len(concat_audio.call_args[0][0]), 2)