diff --git a/tests/test_content.py b/tests/test_content.py index 9f6bf27..b1f61f6 100644 --- a/tests/test_content.py +++ b/tests/test_content.py @@ -31,6 +31,7 @@ SUBREDDIT_PROMPTS = OrderedDict([ ('top', ('pics/top', '/r/pics', 'top')), ('rising', ('r/pics/rising', '/r/pics', 'rising')), ('controversial', ('/r/pics/controversial', '/r/pics', 'controversial')), + ('gilded', ('/r/pics/gilded', '/r/pics', 'gilded')) ('top-day', ('/r/pics/top-day', '/r/pics', 'top-day')), ('top-hour', ('/r/pics/top-hour', '/r/pics', 'top-hour')), ('top-month', ('/r/pics/top-month', '/r/pics', 'top-month')), diff --git a/tests/test_submission.py b/tests/test_submission.py index 9ba9e2c..77d32fe 100644 --- a/tests/test_submission.py +++ b/tests/test_submission.py @@ -62,7 +62,8 @@ def test_submission_page_construct(reddit, terminal, config, oauth): '[2]top ' '[3]rising ' '[4]new ' - '[5]controversial').encode('utf-8') + '[5]controversial' + '[6]gilded ').encode('utf-8') window.addstr.assert_any_call(0, 0, menu) # Footer @@ -185,6 +186,8 @@ def test_submission_order(submission_page): assert submission_page.content.order == 'new' submission_page.controller.trigger('5') assert submission_page.content.order == 'controversial' + submission_page.controller.trigger('6') + assert submission_page.content.order == 'gilded' def test_submission_move_top_bottom(submission_page): diff --git a/tests/test_subreddit.py b/tests/test_subreddit.py index 0a9be83..e043c05 100644 --- a/tests/test_subreddit.py +++ b/tests/test_subreddit.py @@ -34,7 +34,8 @@ def test_subreddit_page_construct(reddit, terminal, config, oauth): '[2]top ' '[3]rising ' '[4]new ' - '[5]controversial').encode('utf-8') + '[5]controversial' + '[6]gilded ').encode('utf-8') window.addstr.assert_any_call(0, 0, menu) # Submission @@ -243,6 +244,23 @@ def test_subreddit_order_controversial(subreddit_page, terminal): assert subreddit_page.content.order == 'controversial' +def test_subreddit_order_gilded(subreddit_page, terminal): + + # Sort by controversial + with mock.patch.object(terminal, 'show_notification'): + # Invalid selection + terminal.show_notification.return_value = ord('x') + subreddit_page.controller.trigger('6') + terminal.show_notification.assert_called_with('Invalid option') + assert subreddit_page.content.order is None + + # Valid selection - sort by default + terminal.show_notification.reset_mock() + terminal.show_notification.return_value = ord('\n') + subreddit_page.controller.trigger('6') + assert subreddit_page.content.order == 'gilded' + + def test_subreddit_order_search(subreddit_page, terminal): # Search the current subreddit diff --git a/tests/test_subscription.py b/tests/test_subscription.py index c2e8467..b7dba02 100644 --- a/tests/test_subscription.py +++ b/tests/test_subscription.py @@ -40,7 +40,8 @@ def test_subscription_page_construct(reddit, terminal, config, oauth, '[2]top ' '[3]rising ' # Whitespace is relevant '[4]new ' - '[5]controversial').encode('utf-8') + '[5]controversial' + '[6]gilded ').encode('utf-8') with pytest.raises(AssertionError): window.addstr.assert_any_call(0, 0, menu)