diff options
Diffstat (limited to 'Biz/PodcastItLater')
| -rw-r--r-- | Biz/PodcastItLater/Web.py | 61 |
1 files changed, 19 insertions, 42 deletions
diff --git a/Biz/PodcastItLater/Web.py b/Biz/PodcastItLater/Web.py index 6134a98..30ebf53 100644 --- a/Biz/PodcastItLater/Web.py +++ b/Biz/PodcastItLater/Web.py @@ -3108,11 +3108,7 @@ class TestAccountPage(BaseWebTest): """Test account page functionality.""" def setUp(self) -> None: -<<<<<<< Updated upstream - """Set up test client with logged-in user.""" -======= """Set up test with user.""" ->>>>>>> Stashed changes super().setUp() self.user_id, _ = Core.Database.create_user( "test@example.com", @@ -3120,23 +3116,34 @@ class TestAccountPage(BaseWebTest): ) self.client.post("/login", data={"email": "test@example.com"}) -<<<<<<< Updated upstream def test_account_page_logged_in(self) -> None: """Account page should render for logged-in users.""" + # Create some usage to verify stats are shown + ep_id = Core.Database.create_episode( + title="Test Episode", + audio_url="https://example.com/audio.mp3", + duration=300, + content_length=1000, + user_id=self.user_id, + author="Test Author", + original_url="https://example.com/article", + original_url_hash=Core.hash_url("https://example.com/article"), + ) + Core.Database.add_episode_to_user(self.user_id, ep_id) + response = self.client.get("/account") + self.assertEqual(response.status_code, 200) - self.assertIn("Account Management", response.text) + self.assertIn("My Account", response.text) self.assertIn("test@example.com", response.text) + self.assertIn("1 / 10", response.text) # Usage / Limit for free tier - def test_account_page_logged_out(self) -> None: - """Account page should redirect logged-out users.""" + def test_account_page_login_required(self) -> None: + """Should redirect to login if not logged in.""" self.client.post("/logout") response = self.client.get("/account", follow_redirects=False) self.assertEqual(response.status_code, 307) - self.assertEqual( - response.headers["location"], - "/?error=login_required", - ) + self.assertEqual(response.headers["location"], "/?error=login_required") def test_logout(self) -> None: """Logout should clear session.""" @@ -3169,36 +3176,6 @@ class TestAccountPage(BaseWebTest): response.headers["location"], "https://billing.stripe.com/test", ) -======= - def test_account_page_shows_usage(self) -> None: - """Account page should show usage stats.""" - # Create some usage - ep_id = Core.Database.create_episode( - title="Test Episode", - audio_url="https://example.com/audio.mp3", - duration=300, - content_length=1000, - user_id=self.user_id, - author="Test Author", - original_url="https://example.com/article", - original_url_hash=Core.hash_url("https://example.com/article"), - ) - Core.Database.add_episode_to_user(self.user_id, ep_id) - - response = self.client.get("/account") - - self.assertEqual(response.status_code, 200) - self.assertIn("1 / 10", response.text) # Usage / Limit for free tier - self.assertIn("My Account", response.text) - self.assertIn("test@example.com", response.text) - - def test_account_page_login_required(self) -> None: - """Should redirect to login if not logged in.""" - self.client.get("/logout") - response = self.client.get("/account", follow_redirects=False) - self.assertEqual(response.status_code, 307) - self.assertEqual(response.headers["location"], "/?error=login_required") ->>>>>>> Stashed changes def test() -> None: |