Adding tests.

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')

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