126 lines
4.0 KiB
Python
126 lines
4.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import unicode_literals
|
|
|
|
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
|
|
page.controller.trigger('?')
|
|
message = '[Basic Commands]'.encode('utf-8')
|
|
terminal.stdscr.subwin.addstr.assert_any_call(1, 1, message)
|
|
|
|
# Sort content
|
|
page.controller.trigger('1')
|
|
page.refresh_content.assert_called_with(order='hot')
|
|
page.controller.trigger('2')
|
|
page.refresh_content.assert_called_with(order='top')
|
|
page.controller.trigger('3')
|
|
page.refresh_content.assert_called_with(order='rising')
|
|
page.controller.trigger('4')
|
|
page.refresh_content.assert_called_with(order='new')
|
|
page.controller.trigger('5')
|
|
page.refresh_content.assert_called_with(order='controversial')
|
|
|
|
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() |