From 41726be658474ff667d3eeb3cbd91391a9848644 Mon Sep 17 00:00:00 2001
From: Michael Lazar <lazar.michael22@gmail.com>
Date: Sat, 6 Aug 2016 01:06:48 -0700
Subject: [PATCH] Adding tests.

---
 tests/test_submission.py   | 21 +++++++++++++++++++++
 tests/test_subscription.py | 21 +++++++++++++++++++++
 2 files changed, 42 insertions(+)

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