2 # This file is part of my.gpodder.org.
4 # my.gpodder.org is free software: you can redistribute it and/or modify it
5 # under the terms of the GNU Affero General Public License as published by
6 # the Free Software Foundation, either version 3 of the License, or (at your
7 # option) any later version.
9 # my.gpodder.org is distributed in the hope that it will be useful, but
10 # WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
11 # or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public
12 # License for more details.
14 # You should have received a copy of the GNU Affero General Public License
15 # along with my.gpodder.org. If not, see <http://www.gnu.org/licenses/>.
19 from datetime
import datetime
21 from collections
import Counter
23 from django
.test
import TestCase
, TransactionTestCase
24 from django
.test
.utils
import override_settings
25 from django
.contrib
.auth
import get_user_model
27 from mygpo
.podcasts
.models
import Podcast
, Episode
28 from mygpo
.history
.models
import EpisodeHistoryEntry
29 from mygpo
.maintenance
.merge
import PodcastMerger
36 class SimpleMergeTests(TestCase
):
39 self
.podcast1
= Podcast
.objects
.get_or_create_for_url(
40 'http://example.com/simple-merge-test-feed.rss',
41 defaults
={'title': 'Podcast 1'},
43 self
.podcast2
= Podcast
.objects
.get_or_create_for_url(
44 'http://simple-merge-test.org/podcast/',
45 defaults
={'title': 'Podcast 2'},
48 self
.episode1
= Episode
.objects
.get_or_create_for_url(
49 self
.podcast1
, 'http://example.com/simple-merge-test-episode1.mp3',
51 'title': 'Episode 1 A',
53 self
.episode2
= Episode
.objects
.get_or_create_for_url(
54 self
.podcast2
, 'http://example.com/simple-merge-test-episode1.mp3',
56 'title': 'Episode 1 B',
59 def test_merge_podcasts(self
):
60 # decide which episodes to merge
61 groups
= [(0, [self
.episode1
, self
.episode2
])]
63 pm
= PodcastMerger([self
.podcast1
, self
.podcast2
], counter
, groups
)
67 @override_settings(CACHE
={})
68 class MergeTests(TransactionTestCase
):
69 """ Tests merging of two podcasts, their episodes and states """
72 self
.podcast1
= Podcast
.objects
.get_or_create_for_url(
73 'http://example.com/merge-test-feed.rss',
74 defaults
={'title': 'Podcast 1'},
76 self
.podcast2
= Podcast
.objects
.get_or_create_for_url(
77 'http://merge-test.org/podcast/',
78 defaults
={'title': 'Podcast 2'},
81 self
.episode1
= Episode
.objects
.get_or_create_for_url(
82 self
.podcast1
, 'http://example.com/merge-test-episode1.mp3',
84 'title': 'Episode 1 A',
86 self
.episode2
= Episode
.objects
.get_or_create_for_url(
87 self
.podcast2
, 'http://example.com/merge-test-episode1.mp3',
89 'title': 'Episode 1 B',
92 User
= get_user_model()
93 self
.user
= User(username
='test-merge')
94 self
.user
.email
= 'test-merge-tests@example.com'
95 self
.user
.set_password('secret!')
98 def test_merge_podcasts(self
):
100 action1
= EpisodeHistoryEntry
.create_entry(
103 EpisodeHistoryEntry
.PLAY
,
106 action2
= EpisodeHistoryEntry
.create_entry(
109 EpisodeHistoryEntry
.DOWNLOAD
,
112 # decide which episodes to merge
113 groups
= [(0, [self
.episode1
, self
.episode2
])]
116 pm
= PodcastMerger([self
.podcast1
, self
.podcast2
], counter
, groups
)
119 history
= EpisodeHistoryEntry
.objects
.filter(
120 episode
=self
.episode1
,
124 # both actions must be present for the merged episode
125 self
.assertIn(action1
, history
)
126 self
.assertIn(action2
, history
)
129 self
.episode1
.delete()
130 self
.podcast1
.delete()
134 class MergeGroupTests(TransactionTestCase
):
135 """ Tests merging of two podcasts, one of which is part of a group """
138 self
.podcast1
= Podcast
.objects
.get_or_create_for_url(
139 'http://example.com/group-merge-feed.rss',
141 'title': 'Podcast 1',
144 self
.podcast2
= Podcast
.objects
.get_or_create_for_url(
145 'http://test.org/group-merge-podcast/',
147 'title': 'Podcast 2',
150 self
.podcast3
= Podcast
.objects
.get_or_create_for_url(
151 'http://group-test.org/feed/',
153 'title': 'Podcast 3',
157 self
.episode1
= Episode
.objects
.get_or_create_for_url(
158 self
.podcast1
, 'http://example.com/group-merge-episode1.mp3',
160 'title': 'Episode 1 A',
163 self
.episode2
= Episode
.objects
.get_or_create_for_url(
164 self
.podcast2
, 'http://example.com/group-merge-episode1.mp3',
166 'title': 'Episode 1 B',
169 self
.episode3
= Episode
.objects
.get_or_create_for_url(
170 self
.podcast3
, 'http://example.com/group-merge-media.mp3',
172 'title': 'Episode 2',
176 self
.podcast2
.group_with(self
.podcast3
, 'My Group', 'Feed1', 'Feed2')
178 User
= get_user_model()
179 self
.user
= User(username
='test-merge-group')
180 self
.user
.email
= 'test-merge-group-tests@example.com'
181 self
.user
.set_password('secret!')
184 def test_merge_podcasts(self
):
185 podcast1
= Podcast
.objects
.get(pk
=self
.podcast1
.pk
)
186 podcast2
= Podcast
.objects
.get(pk
=self
.podcast2
.pk
)
187 podcast3
= Podcast
.objects
.get(pk
=self
.podcast3
.pk
)
189 # assert that the podcasts are actually grouped
190 self
.assertEqual(podcast2
.group
, podcast3
.group
)
192 action1
= EpisodeHistoryEntry
.create_entry(
195 EpisodeHistoryEntry
.PLAY
,
198 action2
= EpisodeHistoryEntry
.create_entry(
201 EpisodeHistoryEntry
.DOWNLOAD
,
204 # decide which episodes to merge
205 groups
= [(0, [self
.episode1
, self
.episode2
])]
208 episode2_id
= self
.episode2
.id
210 pm
= PodcastMerger([podcast2
, podcast1
], counter
, groups
)
213 history
= EpisodeHistoryEntry
.objects
.filter(
214 episode
= self
.episode1
,
218 self
.assertIn(action1
, history
)
219 self
.assertIn(action2
, history
)
221 episode1
= Episode
.objects
.get(pk
=self
.episode1
.pk
)
223 # episode2 has been merged into episode1, so it must contain its
225 self
.assertEqual([x
.uuid
for x
in episode1
.merged_uuids
.all()],
229 self
.episode1
.delete()
230 self
.podcast2
.delete()
234 def load_tests(loader
, tests
, ignore
):
235 tests
.addTest(unittest
.TestLoader().loadTestsFromTestCase(MergeTests
))
236 tests
.addTest(unittest
.TestLoader().loadTestsFromTestCase(MergeGroupTests
))