Files
tuir/tests/test_subscription.py
2016-07-19 01:09:31 -07:00

143 lines
4.1 KiB
Python

# -*- coding: utf-8 -*-
from __future__ import unicode_literals
import curses
import praw
import pytest
from rtv.reddits import SubscriptionPage
try:
from unittest import mock
except ImportError:
import mock
def test_subscription_page_construct(reddit, terminal, config,
oauth, refresh_token):
# Log in
config.refresh_token = refresh_token
oauth.authorize()
title = 'Popular Subreddits'
func = reddit.get_popular_subreddits
with terminal.loader():
page = SubscriptionPage(reddit, title, func, terminal, config, oauth)
assert terminal.loader.exception is None
page.draw()
window = terminal.stdscr.subwin
# Header - Title
window.addstr.assert_any_call(0, 0, title.encode('utf-8'))
# Header - Name
name = reddit.user.name.encode('utf-8')
assert name in [args[0][2] for args in window.addstr.call_args_list]
# Banner shouldn't be drawn
menu = ('[1]hot '
'[2]top '
'[3]rising '
'[4]new '
'[5]controversial').encode('utf-8')
with pytest.raises(AssertionError):
window.addstr.assert_any_call(0, 0, menu)
# Cursor - 2 lines
window.subwin.chgat.assert_any_call(0, 0, 1, 262144)
window.subwin.chgat.assert_any_call(1, 0, 1, 262144)
# Reload with a smaller terminal window
terminal.stdscr.ncols = 20
terminal.stdscr.nlines = 10
with terminal.loader():
page = SubscriptionPage(reddit, title, func, terminal, config, oauth)
assert terminal.loader.exception is None
page.draw()
def test_subscription_refresh(subscription_page):
# Refresh content - invalid order
subscription_page.controller.trigger('2')
assert curses.flash.called
curses.flash.reset_mock()
# Refresh content
subscription_page.controller.trigger('r')
assert not curses.flash.called
def test_subscription_move(subscription_page):
# Test movement
with mock.patch.object(subscription_page, 'clear_input_queue'):
# Move cursor to the bottom of the page
while not curses.flash.called:
subscription_page.controller.trigger('j')
curses.flash.reset_mock()
assert subscription_page.nav.inverted
assert (subscription_page.nav.absolute_index ==
len(subscription_page.content._reddit_data) - 1)
# And back to the top
for i in range(subscription_page.nav.absolute_index):
subscription_page.controller.trigger('k')
assert not curses.flash.called
assert subscription_page.nav.absolute_index == 0
assert not subscription_page.nav.inverted
# Can't go up any further
subscription_page.controller.trigger('k')
assert curses.flash.called
assert subscription_page.nav.absolute_index == 0
assert not subscription_page.nav.inverted
# Page down should move the last item to the top
n = len(subscription_page._subwindows)
subscription_page.controller.trigger('n')
assert subscription_page.nav.absolute_index == n - 1
# And page up should move back up, but possibly not to the first item
subscription_page.controller.trigger('m')
def test_subscription_page_select(subscription_page):
# Select a subreddit
subscription_page.controller.trigger(curses.KEY_ENTER)
assert subscription_page.reddit_data is not None
assert subscription_page.active is False
def test_subscription_close(subscription_page):
# Close the list of reddits page
subscription_page.reddit_data = None
subscription_page.active = None
subscription_page.controller.trigger('h')
assert subscription_page.reddit_data is None
assert subscription_page.active is False
def test_subscription_page_invalid(subscription_page):
# Test that other commands don't crash
methods = [
'a', # Upvote
'z', # Downvote
'd', # Delete
'e', # Edit
]
for ch in methods:
curses.flash.reset_mock()
subscription_page.controller.trigger(ch)
assert curses.flash.called