From 5478c299f8ac3c701da65633d989e4c48430cd1d Mon Sep 17 00:00:00 2001
From: woorst <woorst@users.noreply.github.com>
Date: Sun, 17 Jul 2016 23:45:28 -0500
Subject: [PATCH] Add test for subscription page

---
 tests/conftest.py     | 16 ++++++++++++++++
 tests/test_reddits.py | 11 ++++++++---
 2 files changed, 24 insertions(+), 3 deletions(-)

diff --git a/tests/conftest.py b/tests/conftest.py
index 151aaa8..5a453f7 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -228,3 +228,19 @@ def list_reddits_page(reddit, terminal, config, oauth):
     assert terminal.loader.exception is None
     page.draw()
     return page
+
+
+@pytest.fixture()
+def subscription_page(reddit, terminal, config, oauth, refresh_token):
+    # Must be logged in to view your subscriptions
+    config.refresh_token = refresh_token
+    oauth.authorize()
+
+    title = 'My Subreddits'
+    func = lambda : reddit.get_my_subreddits(limit=None)
+
+    with terminal.loader():
+        page = ListRedditsPage(reddit, title, func, terminal, config, oauth)
+    assert terminal.loader.exception is None
+    page.draw()
+    return page
diff --git a/tests/test_reddits.py b/tests/test_reddits.py
index f128c47..f206acc 100644
--- a/tests/test_reddits.py
+++ b/tests/test_reddits.py
@@ -16,7 +16,11 @@ except ImportError:
 
 def test_list_reddits_page_construct(reddit, terminal, config,
                                      oauth, refresh_token):
-    window = terminal.stdscr.subwin
+
+    # Log in
+    config.refresh_token = refresh_token
+    oauth.authorize()
+
     title = 'Popular Subreddits'
     func = reddit.get_popular_subreddits
 
@@ -26,6 +30,7 @@ def test_list_reddits_page_construct(reddit, terminal, config,
 
     page.draw()
 
+    window = terminal.stdscr.subwin
     # Header - Title
     window.addstr.assert_any_call(0, 0, title.encode('utf-8'))
 
@@ -121,7 +126,7 @@ def test_list_reddits_close(list_reddits_page):
     assert list_reddits_page.active is False
 
 
-def test_list_reddits_page_invalid(list_reddits_page):
+def test_subscription_page_invalid(subscription_page):
 
     # Test that other commands don't crash
     methods = [
@@ -132,5 +137,5 @@ def test_list_reddits_page_invalid(list_reddits_page):
     ]
     for ch in methods:
         curses.flash.reset_mock()
-        list_reddits_page.controller.trigger(ch)
+        subscription_page.controller.trigger(ch)
         assert curses.flash.called