Added unit tests for banner.

2015-12-09 01:34:55 -08:00
parent 9de6056b04
commit 329bfae1ec
3 changed files with 26 additions and 0 deletions
--- a/tests/test_submission.py
+++ b/tests/test_submission.py
@@ -28,6 +28,14 @@ def test_submission_page_construct(reddit, terminal, config, oauth):
    title = url[:terminal.stdscr.ncols-1].encode('utf-8')
    window.addstr.assert_any_call(0, 0, title)

+    # Banner
+    menu = ('[1]hot         '
+            '[2]top         '
+            '[3]rising      '
+            '[4]new         '
+            '[5]controversial').encode('utf-8')
+    window.addstr.assert_any_call(0, 0, menu)
+
    # Submission
    submission_data = page.content.get(-1)
    text = submission_data['title'].encode('utf-8')