diff --git a/tests/test_submission.py b/tests/test_submission.py index 6af1191..c21434b 100644 --- a/tests/test_submission.py +++ b/tests/test_submission.py @@ -95,6 +95,27 @@ def test_submission_open(submission_page, terminal): assert terminal.open_browser.called +def test_submission_prompt(submission_page, terminal): + + # Prompt for a different subreddit + with mock.patch.object(terminal, 'prompt_input'): + # Valid input + submission_page.active = True + submission_page.selected_subreddit = None + terminal.prompt_input.return_value = 'front/top' + submission_page.controller.trigger('/') + assert not submission_page.active + assert submission_page.selected_subreddit + + # Invalid input + submission_page.active = True + submission_page.selected_subreddit = None + terminal.prompt_input.return_value = 'front/pot' + submission_page.controller.trigger('/') + assert submission_page.active + assert not submission_page.selected_subreddit + + def test_submission_move_top_bottom(submission_page): submission_page.controller.trigger('G') diff --git a/tests/test_subscription.py b/tests/test_subscription.py index a3a97b2..90ce82d 100644 --- a/tests/test_subscription.py +++ b/tests/test_subscription.py @@ -71,6 +71,27 @@ def test_subscription_refresh(subscription_page): assert not curses.flash.called +def test_subscription_prompt(subscription_page, terminal): + + # Prompt for a different subreddit + with mock.patch.object(terminal, 'prompt_input'): + # Valid input + subscription_page.active = True + subscription_page.selected_subreddit = None + terminal.prompt_input.return_value = 'front/top' + subscription_page.controller.trigger('/') + assert not subscription_page.active + assert subscription_page.selected_subreddit + + # Invalid input + subscription_page.active = True + subscription_page.selected_subreddit = None + terminal.prompt_input.return_value = 'front/pot' + subscription_page.controller.trigger('/') + assert subscription_page.active + assert not subscription_page.selected_subreddit + + def test_subscription_move(subscription_page): # Test movement