154 lines
4.7 KiB
Python
154 lines
4.7 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import unicode_literals
|
|
|
|
import curses
|
|
|
|
import pytest
|
|
|
|
from rtv.page import Page, PageController, logged_in
|
|
|
|
try:
|
|
from unittest import mock
|
|
except ImportError:
|
|
import mock
|
|
|
|
|
|
def test_page_logged_in(terminal):
|
|
|
|
page = mock.MagicMock()
|
|
page.term = terminal
|
|
|
|
@logged_in
|
|
def func(_):
|
|
raise RuntimeError()
|
|
|
|
# Logged in runs the function
|
|
page.reddit.is_oauth_session.return_value = True
|
|
with pytest.raises(RuntimeError):
|
|
func(page)
|
|
message = 'Not logged in'.encode('utf-8')
|
|
with pytest.raises(AssertionError):
|
|
terminal.stdscr.subwin.addstr.assert_called_with(1, 1, message)
|
|
|
|
# Logged out skips the function and displays a message
|
|
page.reddit.is_oauth_session.return_value = False
|
|
func(page)
|
|
message = 'Not logged in'.encode('utf-8')
|
|
terminal.stdscr.subwin.addstr.assert_called_with(1, 1, message)
|
|
|
|
|
|
def test_page_unauthenticated(reddit, terminal, config, oauth):
|
|
|
|
page = Page(reddit, terminal, config, oauth)
|
|
page.controller = PageController(page, keymap=config.keymap)
|
|
with mock.patch.object(page, 'refresh_content'), \
|
|
mock.patch.object(page, 'content'), \
|
|
mock.patch.object(page, 'nav'), \
|
|
mock.patch.object(page, 'draw'):
|
|
|
|
# Loop
|
|
def func(_):
|
|
page.active = False
|
|
with mock.patch.object(page, 'controller'):
|
|
page.controller.trigger = mock.MagicMock(side_effect=func)
|
|
page.loop()
|
|
assert page.draw.called
|
|
|
|
# Quit, confirm
|
|
terminal.stdscr.getch.return_value = ord('y')
|
|
with mock.patch('sys.exit') as sys_exit:
|
|
page.controller.trigger('q')
|
|
assert sys_exit.called
|
|
|
|
# Quit, deny
|
|
terminal.stdscr.getch.return_value = terminal.ESCAPE
|
|
with mock.patch('sys.exit') as sys_exit:
|
|
page.controller.trigger('q')
|
|
assert not sys_exit.called
|
|
|
|
# Force quit
|
|
terminal.stdscr.getch.return_value = terminal.ESCAPE
|
|
with mock.patch('sys.exit') as sys_exit:
|
|
page.controller.trigger('Q')
|
|
assert sys_exit.called
|
|
|
|
# Show help
|
|
with mock.patch('subprocess.Popen') as Popen:
|
|
page.controller.trigger('?')
|
|
assert Popen.called
|
|
|
|
logged_in_methods = [
|
|
'a', # Upvote
|
|
'z', # Downvote
|
|
'd', # Delete
|
|
'e', # Edit
|
|
'i', # Get inbox
|
|
]
|
|
for ch in logged_in_methods:
|
|
page.controller.trigger(ch)
|
|
message = 'Not logged in'.encode('utf-8')
|
|
terminal.stdscr.subwin.addstr.assert_called_with(1, 1, message)
|
|
terminal.stdscr.subwin.addstr.reset_mock()
|
|
|
|
|
|
def test_page_authenticated(reddit, terminal, config, oauth, refresh_token):
|
|
|
|
page = Page(reddit, terminal, config, oauth)
|
|
page.controller = PageController(page, keymap=config.keymap)
|
|
config.refresh_token = refresh_token
|
|
|
|
# Login
|
|
page.controller.trigger('u')
|
|
assert reddit.is_oauth_session()
|
|
|
|
# Get inbox - Call the real method
|
|
page.controller.trigger('i')
|
|
|
|
# Get inbox - Simulate no new messages
|
|
reddit.get_unread = mock.Mock(return_value=[])
|
|
page.controller.trigger('i')
|
|
message = 'No New Messages'.encode('utf-8')
|
|
terminal.stdscr.subwin.addstr.assert_called_with(1, 1, message)
|
|
|
|
# Logout
|
|
terminal.stdscr.getch.return_value = ord('y')
|
|
page.controller.trigger('u')
|
|
assert not reddit.is_oauth_session()
|
|
|
|
|
|
def test_page_cycle_theme(reddit, terminal, config, oauth):
|
|
|
|
page = Page(reddit, terminal, config, oauth)
|
|
page.controller = PageController(page, keymap=config.keymap)
|
|
|
|
page.term.set_theme()
|
|
assert page.term.theme.name == 'default'
|
|
|
|
with mock.patch.object(terminal, 'show_notification'), \
|
|
mock.patch.object(page, 'draw'):
|
|
|
|
# Next theme
|
|
page.controller.trigger(curses.KEY_F3)
|
|
assert page.term.theme.name == 'monochrome'
|
|
terminal.show_notification.assert_called_with(
|
|
'monochrome (built-in)', timeout=1)
|
|
|
|
# Previous theme
|
|
page.controller.trigger(curses.KEY_F2)
|
|
assert page.term.theme.name == 'default'
|
|
terminal.show_notification.assert_called_with(
|
|
'default (built-in)', timeout=1)
|
|
|
|
# Previous - will loop to one of the 256 color themes
|
|
page.controller.trigger(curses.KEY_F2)
|
|
assert page.term.theme.source in ('preset', 'installed')
|
|
|
|
# Reset
|
|
page.term.set_theme()
|
|
|
|
# Will skip over any installed themes that aren't supported
|
|
curses.has_colors.return_value = False
|
|
page.controller.trigger(curses.KEY_F2)
|
|
assert page.term.theme.required_colors == 0
|
|
|