diff options
Diffstat (limited to 'Biz/PodcastItLater/Web.py')
| -rw-r--r-- | Biz/PodcastItLater/Web.py | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/Biz/PodcastItLater/Web.py b/Biz/PodcastItLater/Web.py index 14bca1b..767ab84 100644 --- a/Biz/PodcastItLater/Web.py +++ b/Biz/PodcastItLater/Web.py @@ -1837,20 +1837,22 @@ class TestRSSFeed(BaseWebTest): ) # Create test episodes - Core.Database.create_episode( + ep1_id = Core.Database.create_episode( "Episode 1", "https://example.com/ep1.mp3", 300, 5000, self.user_id, ) - Core.Database.create_episode( + ep2_id = Core.Database.create_episode( "Episode 2", "https://example.com/ep2.mp3", 600, 10000, self.user_id, ) + Core.Database.add_episode_to_user(self.user_id, ep1_id) + Core.Database.add_episode_to_user(self.user_id, ep2_id) def test_feed_generation(self) -> None: """Generate valid RSS XML.""" @@ -1874,13 +1876,14 @@ class TestRSSFeed(BaseWebTest): user2_id, _ = Core.Database.create_user( "other@example.com", ) - Core.Database.create_episode( + other_ep_id = Core.Database.create_episode( "Other Episode", "https://example.com/other.mp3", 400, 6000, user2_id, ) + Core.Database.add_episode_to_user(user2_id, other_ep_id) # Get first user's feed response = self.client.get(f"/feed/{self.token}.xml") @@ -2170,6 +2173,7 @@ class TestEpisodeDetailPage(BaseWebTest): author="Test Author", original_url="https://example.com/article", ) + Core.Database.add_episode_to_user(self.user_id, self.episode_id) self.episode_sqid = encode_episode_id(self.episode_id) def test_episode_page_loads(self) -> None: |