class MergeTests(TestCase): def setUp(self): self.podcast1 = Podcast(urls=['http://example.com/feed.rss']) self.podcast2 = Podcast(urls=['http://test.org/podcast/']) self.podcast1.save() self.podcast2.save() self.episode1 = Episode(podcast=self.podcast1.get_id(), urls = ['http://example.com/episode1.mp3']) self.episode2 = Episode(podcast=self.podcast2.get_id(), urls = ['http://example.com/episode1.mp3']) self.episode1.save() self.episode2.save() self.user = User(username='******') self.user.email = '*****@*****.**' self.user.set_password('secret!') self.user.save() def test_merge_podcasts(self): state1 = self.episode1.get_user_state(self.user) state2 = self.episode2.get_user_state(self.user) action1 = EpisodeAction(action='play', timestamp=datetime.utcnow()) action2 = EpisodeAction(action='download', timestamp=datetime.utcnow()) state1.add_actions([action1]) state2.add_actions([action2]) state1.save() state2.save() pm = PodcastMerger([self.podcast1, self.podcast2]) pm.merge() state1 = self.episode1.get_user_state(self.user) state2 = self.episode2.get_user_state(self.user) self.assertIn(action1, state1.actions) self.assertIn(action2, state1.actions) def tearDown(self): self.podcast1.delete() self.episode1.delete() try: self.podcast2.delete() self.episode2.delete() except: pass self.user.delete()
class SubscriptionAPITests(unittest.TestCase): """ Tests the Subscription API """ def setUp(self): self.password = '******' self.username = '******' self.device_uid = 'test-device' self.user = User(username=self.username, email='*****@*****.**') self.user.set_password(self.password) self.user.save() self.user.is_active = True self.client = Client() self.extra = { 'HTTP_AUTHORIZATION': create_auth_string(self.username, self.password) } self.action_data = { 'add': ['http://example.com/podcast.rss'], } self.url = reverse('subscriptions-api', kwargs={ 'version': '2', 'username': self.user.username, 'device_uid': self.device_uid, }) def tearDown(self): self.user.delete() def test_set_get_subscriptions(self): """ Tests that an upload subscription is returned back correctly """ # upload a subscription response = self.client.post(self.url, json.dumps(self.action_data), content_type="application/json", **self.extra) self.assertEqual(response.status_code, 200, response.content) # verify that the subscription is returned correctly response = self.client.get(self.url, {'since': '0'}, **self.extra) self.assertEqual(response.status_code, 200, response.content) response_obj = json.loads(response.content) self.assertEqual(self.action_data['add'], response_obj['add']) self.assertEqual([], response_obj.get('remove', [])) def test_unauth_request(self): """ Tests that an unauthenticated request gives a 401 response """ response = self.client.get(self.url, {'since': '0'}) self.assertEqual(response.status_code, 401, response.content)
class SimpleWebTests(TestCase): @classmethod def setUpClass(self): self.user = User(username="******", email="*****@*****.**") self.user.set_password("pwd") self.user.save() self.auth_string = create_auth_string("test", "pwd") @classmethod def tearDownClass(self): self.user.delete() def test_access_parameterless_pages(self): pages = [ "history", "suggestions", "tags", "subscriptions", "subscriptions-opml", "subscriptions-download", "favorites", "account", "privacy", "delete-account", "share", "toplist", "episode-toplist", "devices", "device-create", "login", "logout", "home", ] self.access_pages(pages, [], True) def test_access_podcast_pages(self): pages = ["podcast"] def access_pages(self, pages, args, login): if login: self.client.post("/login/", dict(login_username=self.user.username, pwd="pwd")) for page in pages: response = self.client.get(reverse(page, args=args), follow=True) self.assertEquals(response.status_code, 200)
class UnsubscribeMergeTests(TestCase): """ Test if merged podcasts can be properly unsubscribed TODO: this test fails intermittently """ P2_URL = 'http://test.org/podcast/' def setUp(self): self.podcast1 = Podcast(urls=['http://example.com/feed.rss']) self.podcast2 = Podcast(urls=[self.P2_URL]) self.podcast1.save() self.podcast2.save() self.user = User(username='******') self.user.email = '*****@*****.**' self.user.set_password('secret!') self.user.save() self.device = get_device(self.user, 'dev', '') def test_merge_podcasts(self): self.podcast2.subscribe(self.user, self.device) # merge podcast2 into podcast1 pm = PodcastMerger([self.podcast1, self.podcast2], Counter(), []) pm.merge() # seems that setting delayed_commit = false in the CouchDB config, as # well as a delay here fix the intermittent failures. # TODO: further investiation needed import time time.sleep(2) # get podcast for URL of podcast2 and unsubscribe from it p = podcast_for_url(self.P2_URL) p.unsubscribe(self.user, self.device) subscriptions = subscribed_podcast_ids_by_user_id(self.user._id) self.assertEqual(0, len(subscriptions)) def tearDown(self): self.podcast1.delete() self.user.delete()
class MergeTests(TestCase): """ Tests merging of two podcasts, their episodes and states """ def setUp(self): self.podcast1 = Podcast(urls=['http://example.com/feed.rss']) self.podcast2 = Podcast(urls=['http://test.org/podcast/']) self.podcast1.save() self.podcast2.save() self.episode1 = Episode(podcast=self.podcast1.get_id(), urls = ['http://example.com/episode1.mp3']) self.episode2 = Episode(podcast=self.podcast2.get_id(), urls = ['http://example.com/episode1.mp3']) self.episode1.save() self.episode2.save() self.user = User(username='******') self.user.email = '*****@*****.**' self.user.set_password('secret!') self.user.save() def test_merge_podcasts(self): # Create additional data that will be merged state1 = episode_state_for_user_episode(self.user, self.episode1) state2 = episode_state_for_user_episode(self.user, self.episode2) action1 = EpisodeAction(action='play', timestamp=datetime.utcnow(), upload_timestamp=get_timestamp(datetime.utcnow())) action2 = EpisodeAction(action='download', timestamp=datetime.utcnow(), upload_timestamp=get_timestamp(datetime.utcnow())) add_episode_actions(state1, [action1]) add_episode_actions(state2, [action2]) # copy of the object episode2 = episode_by_id(self.episode2._id) # decide which episodes to merge groups = [(0, [self.episode1, self.episode2])] counter = Counter() pm = PodcastMerger([self.podcast1, self.podcast2], counter, groups) pm.merge() state1 = episode_state_for_user_episode(self.user, self.episode1) state2 = episode_state_for_user_episode(self.user, episode2) self.assertIn(action1, state1.actions) self.assertIn(action2, state1.actions) self.assertEqual(state2._id, None) def tearDown(self): self.podcast1.delete() self.episode1.delete() #self.podcast2.delete() #self.episode2.delete() self.user.delete()
class MergeGroupTests(TestCase): """ Tests merging of two podcasts, one of which is part of a group """ def setUp(self): self.podcast1 = Podcast(urls=['http://example.com/feed.rss']) self.podcast2 = Podcast(urls=['http://test.org/podcast/']) self.podcast3 = Podcast(urls=['http://test.org/feed/']) self.podcast1.save() self.podcast2.save() self.podcast3.save() self.episode1 = Episode(podcast=self.podcast1.get_id(), urls = ['http://example.com/episode1.mp3']) self.episode2 = Episode(podcast=self.podcast2.get_id(), urls = ['http://example.com/episode1.mp3']) self.episode3 = Episode(podcast=self.podcast3.get_id(), urls = ['http://example.com/media.mp3']) self.episode1.save() self.episode2.save() self.episode3.save() self.podcast2.group_with(self.podcast3, 'My Group', 'Feed1', 'Feed2') self.user = User(username='******') self.user.email = '*****@*****.**' self.user.set_password('secret!') self.user.save() def test_merge_podcasts(self): podcast1 = podcast_by_id(self.podcast1.get_id()) podcast2 = podcast_by_id(self.podcast2.get_id()) podcast3 = podcast_by_id(self.podcast3.get_id()) # assert that the podcasts are actually grouped self.assertEqual(podcast2._id, podcast3._id) self.assertNotEqual(podcast2.get_id(), podcast2._id) self.assertNotEqual(podcast3.get_id(), podcast3._id) # Create additional data that will be merged state1 = episode_state_for_user_episode(self.user, self.episode1) state2 = episode_state_for_user_episode(self.user, self.episode2) action1 = EpisodeAction(action='play', timestamp=datetime.utcnow(), upload_timestamp=get_timestamp(datetime.utcnow())) action2 = EpisodeAction(action='download', timestamp=datetime.utcnow(), upload_timestamp=get_timestamp(datetime.utcnow())) add_episode_actions(state1, [action1]) add_episode_actions(state2, [action2]) # copy of the object episode2 = episode_by_id(self.episode2._id) # decide which episodes to merge groups = [(0, [self.episode1, self.episode2])] counter = Counter() pm = PodcastMerger([podcast2, podcast1], counter, groups) pm.merge() state1 = episode_state_for_user_episode(self.user, self.episode1) state2 = episode_state_for_user_episode(self.user, episode2) self.assertIn(action1, state1.actions) self.assertIn(action2, state1.actions) self.assertEqual(state2._id, None) episode1 = episode_by_id(self.episode1._id) # episode2 has been merged into episode1, so it must contain its # merged _id self.assertEqual(episode1.merged_ids, [episode2._id]) def tearDown(self): self.podcast2.delete() self.episode1.delete() #self.podcast2.delete() #self.episode2.delete() self.user.delete()
class AdvancedAPITests(unittest.TestCase): def setUp(self): self.password = '******' self.username = '******' self.user = User(username=self.username, email='*****@*****.**') self.user.set_password(self.password) self.user.save() self.user.is_active = True self.client = Client() self.extra = { 'HTTP_AUTHORIZATION': create_auth_string(self.username, self.password) } self.action_data = [ { "podcast": "http://example.com/feed.rss", "episode": "http://example.com/files/s01e20.mp3", "device": "gpodder_abcdef123", "action": "download", "timestamp": "2009-12-12T09:00:00" }, { "podcast": "http://example.org/podcast.php", "episode": "http://ftp.example.org/foo.ogg", "action": "play", "started": 15, "position": 120, "total": 500 } ] def tearDown(self): self.user.delete() def test_episode_actions(self): url = reverse(episodes, kwargs=dict(version='2', username=self.user.username)) # upload actions response = self.client.post(url, json.dumps(self.action_data), content_type="application/json", **self.extra) self.assertEqual(response.status_code, 200, response.content) # get all extra = deepcopy(self.extra) extra['since'] = 0 response = self.client.get(url, **extra) self.assertEqual(response.status_code, 200, response.content) response_obj = json.loads(response.content) actions = response_obj['actions'] self.assertTrue(self.compare_action_list(self.action_data, actions)) def compare_action_list(self, as1, as2): for a1 in as1: found = False for a2 in as2: if self.compare_actions(a1, a2): found = True if not found: raise ValueError('%s not found in %s' % (a1, as2)) return False return True def compare_actions(self, a1, a2): for key, val in a1.items(): if a2.get(key, None) != val: return False return True