openlp/tests/functional/openlp_plugins/remotes/test_router.py

381 lines
16 KiB
Python
Raw Normal View History

# -*- coding: utf-8 -*-
# vim: autoindent shiftwidth=4 expandtab textwidth=120 tabstop=4 softtabstop=4
###############################################################################
# OpenLP - Open Source Lyrics Projection #
# --------------------------------------------------------------------------- #
2015-01-18 13:39:21 +00:00
# Copyright (c) 2008-2015 OpenLP Developers #
# --------------------------------------------------------------------------- #
# This program is free software; you can redistribute it and/or modify it #
# under the terms of the GNU General Public License as published by the Free #
# Software Foundation; version 2 of the License. #
# #
# This program is distributed in the hope that it will be useful, but WITHOUT #
# ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or #
# FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for #
# more details. #
# #
# You should have received a copy of the GNU General Public License along #
# with this program; if not, write to the Free Software Foundation, Inc., 59 #
# Temple Place, Suite 330, Boston, MA 02111-1307 USA #
###############################################################################
2013-03-29 08:25:33 +00:00
"""
This module contains tests for the lib submodule of the Remotes plugin.
"""
import os
import urllib.request
2013-03-29 08:25:33 +00:00
from unittest import TestCase
2014-03-18 20:58:52 +00:00
from openlp.core.common import Settings, Registry
2015-01-08 18:07:36 +00:00
from openlp.core.ui import ServiceManager
2013-09-14 21:00:58 +00:00
from openlp.plugins.remotes.lib.httpserver import HttpRouter
from urllib.parse import urlparse
2013-12-18 21:54:56 +00:00
from tests.functional import MagicMock, patch, mock_open
2014-03-14 22:08:44 +00:00
from tests.helpers.testmixin import TestMixin
2013-03-29 08:25:33 +00:00
__default_settings__ = {
2013-08-31 18:17:38 +00:00
'remotes/twelve hour': True,
'remotes/port': 4316,
'remotes/https port': 4317,
'remotes/https enabled': False,
'remotes/user id': 'openlp',
'remotes/password': 'password',
'remotes/authentication enabled': False,
'remotes/ip address': '0.0.0.0'
2013-03-29 08:25:33 +00:00
}
TEST_PATH = os.path.abspath(os.path.dirname(__file__))
2013-03-29 08:25:33 +00:00
2013-12-18 21:54:56 +00:00
2014-03-14 22:08:44 +00:00
class TestRouter(TestCase, TestMixin):
2013-03-29 08:25:33 +00:00
"""
Test the functions in the :mod:`lib` module.
"""
def setUp(self):
"""
Create the UI
"""
2015-01-08 18:07:36 +00:00
Registry.create()
self.setup_application()
2014-03-14 22:08:44 +00:00
self.build_settings()
2013-03-29 08:25:33 +00:00
Settings().extend_default_settings(__default_settings__)
2015-01-08 18:07:36 +00:00
self.service_manager = ServiceManager()
2013-03-29 09:06:43 +00:00
self.router = HttpRouter()
2013-03-29 08:25:33 +00:00
def tearDown(self):
"""
Delete all the C++ objects at the end so that we don't have a segfault
"""
2014-03-14 22:08:44 +00:00
self.destroy_settings()
2013-03-29 08:25:33 +00:00
2013-09-14 21:00:58 +00:00
def password_encrypter_test(self):
2013-03-29 09:32:46 +00:00
"""
2013-09-14 21:00:58 +00:00
Test hash userid and password function
2013-03-29 09:32:46 +00:00
"""
# GIVEN: A default configuration
2013-09-14 21:00:58 +00:00
Settings().setValue('remotes/user id', 'openlp')
Settings().setValue('remotes/password', 'password')
2013-03-29 09:32:46 +00:00
# WHEN: called with the defined userid
2013-09-14 21:00:58 +00:00
router = HttpRouter()
router.initialise()
test_value = 'b3BlbmxwOnBhc3N3b3Jk'
2013-03-29 09:32:46 +00:00
# THEN: the function should return the correct password
2013-09-14 21:00:58 +00:00
self.assertEqual(router.auth, test_value,
2013-12-18 21:54:56 +00:00
'The result for make_sha_hash should return the correct encrypted password')
2013-03-29 09:32:46 +00:00
2013-03-29 08:25:33 +00:00
def process_http_request_test(self):
"""
2013-03-29 09:06:43 +00:00
Test the router control functionality
2013-03-29 08:25:33 +00:00
"""
2013-03-29 09:06:43 +00:00
# GIVEN: A testing set of Routes
mocked_function = MagicMock()
test_route = [
2013-09-14 21:00:58 +00:00
(r'^/stage/api/poll$', {'function': mocked_function, 'secure': False}),
2013-03-29 09:06:43 +00:00
]
2014-05-07 16:05:27 +00:00
self.router.routes = test_route
2013-03-29 09:06:43 +00:00
# WHEN: called with a poll route
2014-05-07 16:05:27 +00:00
function, args = self.router.process_http_request('/stage/api/poll', None)
2013-03-29 08:25:33 +00:00
2013-03-29 09:06:43 +00:00
# THEN: the function should have been called only once
2013-12-18 21:54:56 +00:00
self.assertEqual(mocked_function, function['function'], 'The mocked function should match defined value.')
self.assertFalse(function['secure'], 'The mocked function should not require any security.')
2014-10-27 20:22:57 +00:00
def process_secure_http_request_test(self):
"""
Test the router control functionality
"""
# GIVEN: A testing set of Routes
mocked_function = MagicMock()
test_route = [
(r'^/stage/api/poll$', {'function': mocked_function, 'secure': True}),
]
self.router.routes = test_route
self.router.settings_section = 'remotes'
Settings().setValue('remotes/authentication enabled', True)
self.router.path = '/stage/api/poll'
self.router.auth = ''
self.router.headers = {'Authorization': None}
self.router.send_response = MagicMock()
self.router.send_header = MagicMock()
self.router.end_headers = MagicMock()
self.router.wfile = MagicMock()
# WHEN: called with a poll route
self.router.do_post_processor()
# THEN: the function should have been called only once
self.router.send_response.assert_called_once_with(401)
2015-02-14 09:12:35 +00:00
self.assertEqual(self.router.send_header.call_count, 5, 'The header should have been called five times.')
2014-10-27 20:22:57 +00:00
def get_content_type_test(self):
"""
2013-11-08 17:43:36 +00:00
Test the get_content_type logic
"""
# GIVEN: a set of files and their corresponding types
2014-04-02 19:35:09 +00:00
headers = [
['test.html', 'text/html'], ['test.css', 'text/css'],
2013-10-28 15:16:22 +00:00
['test.js', 'application/javascript'], ['test.jpg', 'image/jpeg'],
['test.gif', 'image/gif'], ['test.ico', 'image/x-icon'],
['test.png', 'image/png'], ['test.whatever', 'text/plain'],
2013-11-08 17:43:36 +00:00
['test', 'text/plain'], ['', 'text/plain'],
2013-12-18 21:54:56 +00:00
[os.path.join(TEST_PATH, 'test.html'), 'text/html']]
# WHEN: calling each file type
for header in headers:
2013-11-08 17:43:36 +00:00
ext, content_type = self.router.get_content_type(header[0])
2013-12-18 21:54:56 +00:00
# THEN: all types should match
2013-11-08 17:43:36 +00:00
self.assertEqual(content_type, header[1], 'Mismatch of content type')
2013-10-28 02:33:28 +00:00
2014-05-07 16:05:27 +00:00
def main_poll_test(self):
"""
Test the main poll logic
"""
# GIVEN: a defined router with two slides
Registry.create()
2014-05-07 16:05:27 +00:00
Registry().register('live_controller', MagicMock)
router = HttpRouter()
router.send_response = MagicMock()
router.send_header = MagicMock()
router.end_headers = MagicMock()
router.live_controller.slide_count = 2
# WHEN: main poll called
results = router.main_poll()
# THEN: the correct response should be returned
self.assertEqual(results.decode('utf-8'), '{"results": {"slide_count": 2}}',
'The resulting json strings should match')
def serve_file_without_params_test(self):
"""
Test the serve_file method without params
"""
# GIVEN: mocked environment
self.router.send_response = MagicMock()
self.router.send_header = MagicMock()
self.router.end_headers = MagicMock()
self.router.wfile = MagicMock()
self.router.html_dir = os.path.normpath('test/dir')
self.router.template_vars = MagicMock()
2013-12-18 21:54:56 +00:00
# WHEN: call serve_file with no file_name
self.router.serve_file()
2013-12-18 21:54:56 +00:00
# THEN: it should return a 404
self.router.send_response.assert_called_once_with(404)
2013-12-18 21:54:56 +00:00
self.assertEqual(self.router.end_headers.call_count, 1, 'end_headers called once')
def serve_file_with_valid_params_test(self):
"""
Test the serve_file method with an existing file
"""
# GIVEN: mocked environment
self.router.send_response = MagicMock()
self.router.send_header = MagicMock()
self.router.end_headers = MagicMock()
self.router.wfile = MagicMock()
self.router.html_dir = os.path.normpath('test/dir')
self.router.template_vars = MagicMock()
with patch('openlp.core.lib.os.path.exists') as mocked_exists, \
2014-07-11 13:35:44 +00:00
patch('builtins.open', mock_open(read_data='123')):
mocked_exists.return_value = True
2013-12-18 21:54:56 +00:00
# WHEN: call serve_file with an existing html file
self.router.serve_file(os.path.normpath('test/dir/test.html'))
# THEN: it should return a 200 and the file
self.router.send_response.assert_called_once_with(200)
2013-12-18 21:54:56 +00:00
self.router.send_header.assert_called_once_with('Content-type', 'text/html')
self.assertEqual(self.router.end_headers.call_count, 1, 'end_headers called once')
2013-10-28 02:33:28 +00:00
def serve_thumbnail_without_params_test(self):
"""
Test the serve_thumbnail routine without params
"""
2015-02-14 09:12:35 +00:00
# GIVEN: mocked environment
2013-10-28 02:33:28 +00:00
self.router.send_response = MagicMock()
self.router.send_header = MagicMock()
self.router.end_headers = MagicMock()
self.router.wfile = MagicMock()
2015-02-14 09:12:35 +00:00
# WHEN: I request a thumbnail
2013-10-28 02:33:28 +00:00
self.router.serve_thumbnail()
2015-02-14 09:12:35 +00:00
# THEN: The headers should be set correctly
2013-10-28 02:33:28 +00:00
self.router.send_response.assert_called_once_with(404)
2015-02-16 20:56:39 +00:00
self.assertEqual(self.router.send_response.call_count, 1, 'Send response should be called once')
self.assertEqual(self.router.end_headers.call_count, 1, 'end_headers should be called once')
2013-10-28 02:33:28 +00:00
def serve_thumbnail_with_invalid_params_test(self):
"""
Test the serve_thumbnail routine with invalid params
"""
# GIVEN: Mocked send_header, send_response, end_headers and wfile
self.router.send_response = MagicMock()
self.router.send_header = MagicMock()
self.router.end_headers = MagicMock()
self.router.wfile = MagicMock()
2013-12-30 08:35:05 +00:00
2013-10-28 02:33:28 +00:00
# WHEN: pass a bad controller
2013-12-30 08:35:05 +00:00
self.router.serve_thumbnail('badcontroller', 'tecnologia 1.pptx/slide1.png')
2013-10-28 02:33:28 +00:00
# THEN: a 404 should be returned
2015-02-16 20:56:39 +00:00
self.assertEqual(len(self.router.send_header.mock_calls), 4, 'header should be called 4 times')
2013-12-30 08:35:05 +00:00
self.assertEqual(len(self.router.send_response.mock_calls), 1, 'One response')
self.assertEqual(len(self.router.wfile.mock_calls), 1, 'Once call to write to the socket')
2013-10-28 02:33:28 +00:00
self.router.send_response.assert_called_once_with(404)
2013-12-30 08:35:05 +00:00
2013-10-28 02:33:28 +00:00
# WHEN: pass a bad filename
self.router.send_response.reset_mock()
2013-12-30 08:35:05 +00:00
self.router.serve_thumbnail('presentations', 'tecnologia 1.pptx/badfilename.png')
2013-10-28 02:33:28 +00:00
# THEN: return a 404
self.router.send_response.assert_called_once_with(404)
2013-12-30 08:35:05 +00:00
2013-10-28 02:33:28 +00:00
# WHEN: a dangerous URL is passed
self.router.send_response.reset_mock()
2013-12-30 08:35:05 +00:00
self.router.serve_thumbnail('presentations', '../tecnologia 1.pptx/slide1.png')
2013-10-28 02:33:28 +00:00
# THEN: return a 404
self.router.send_response.assert_called_once_with(404)
def serve_thumbnail_with_valid_params_test(self):
"""
Test the serve_thumbnail routine with valid params
"""
# GIVEN: Mocked send_header, send_response, end_headers and wfile
self.router.send_response = MagicMock()
self.router.send_header = MagicMock()
self.router.end_headers = MagicMock()
self.router.wfile = MagicMock()
2013-11-07 20:13:15 +00:00
mocked_image_manager = MagicMock()
2013-12-30 08:35:05 +00:00
Registry().register('image_manager', mocked_image_manager)
file_name = 'another%20test/slide1.png'
2014-07-11 13:35:44 +00:00
full_path = os.path.normpath(os.path.join('thumbnails', file_name))
width = 120
height = 90
2013-10-28 02:33:28 +00:00
with patch('openlp.core.lib.os.path.exists') as mocked_exists, \
2014-07-11 13:35:44 +00:00
patch('builtins.open', mock_open(read_data='123')), \
patch('openlp.plugins.remotes.lib.httprouter.AppLocation') as mocked_location, \
patch('openlp.plugins.remotes.lib.httprouter.image_to_byte') as mocked_image_to_byte:
2013-10-28 02:33:28 +00:00
mocked_exists.return_value = True
mocked_image_to_byte.return_value = '123'
2013-10-28 02:33:28 +00:00
mocked_location.get_section_data_path.return_value = ''
2013-12-30 08:35:05 +00:00
2013-10-28 02:33:28 +00:00
# WHEN: pass good controller and filename
2015-02-14 09:12:35 +00:00
self.router.serve_thumbnail('presentations', '{0}x{1}'.format(width, height), file_name)
2013-12-30 08:35:05 +00:00
2013-10-28 02:33:28 +00:00
# THEN: a file should be returned
2013-12-30 08:35:05 +00:00
self.assertEqual(self.router.send_header.call_count, 1, 'One header')
self.assertEqual(self.router.send_response.call_count, 1, 'Send response called once')
self.assertEqual(self.router.end_headers.call_count, 1, 'end_headers called once')
mocked_exists.assert_called_with(urllib.parse.unquote(full_path))
self.assertEqual(mocked_image_to_byte.call_count, 1, 'Called once')
2015-11-25 21:47:56 +00:00
mocked_image_manager.add_image.assert_any_call(os.path.normpath(os.path.join('thumbnails', 'another test',
'slide1.png')),
'slide1.png', None, width, height)
mocked_image_manager.get_image.assert_any_call(os.path.normpath(os.path.join('thumbnails', 'another test',
'slide1.png')),
'slide1.png', width, height)
2015-01-08 18:07:36 +00:00
def remote_next_test(self):
"""
Test service manager receives remote next click properly (bug 1407445)
"""
# GIVEN: initial setup and mocks
self.router.routes = [(r'^/api/service/(.*)$', {'function': self.router.service, 'secure': False})]
self.router.request_data = False
mocked_next_item = MagicMock()
self.service_manager.next_item = mocked_next_item
with patch.object(self.service_manager, 'setup_ui'), \
patch.object(self.router, 'do_json_header'):
self.service_manager.bootstrap_initialise()
self.app.processEvents()
2015-01-09 18:53:36 +00:00
2015-01-08 18:07:36 +00:00
# WHEN: Remote next is received
self.router.service(action='next')
self.app.processEvents()
2015-01-09 18:53:36 +00:00
2015-01-08 18:07:36 +00:00
# THEN: service_manager.next_item() should have been called
self.assertTrue(mocked_next_item.called, 'next_item() should have been called in service_manager')
def remote_previous_test(self):
"""
Test service manager receives remote previous click properly (bug 1407445)
"""
# GIVEN: initial setup and mocks
self.router.routes = [(r'^/api/service/(.*)$', {'function': self.router.service, 'secure': False})]
self.router.request_data = False
mocked_previous_item = MagicMock()
self.service_manager.previous_item = mocked_previous_item
with patch.object(self.service_manager, 'setup_ui'), \
patch.object(self.router, 'do_json_header'):
self.service_manager.bootstrap_initialise()
self.app.processEvents()
2015-01-09 18:53:36 +00:00
2015-01-08 18:07:36 +00:00
# WHEN: Remote next is received
self.router.service(action='previous')
self.app.processEvents()
2015-01-09 18:53:36 +00:00
2015-01-08 18:07:36 +00:00
# THEN: service_manager.next_item() should have been called
self.assertTrue(mocked_previous_item.called, 'previous_item() should have been called in service_manager')
2015-11-08 21:07:48 +00:00
def remote_stage_personal_html_test(self):
"""
Test the stage url with a parameter after loaded a url/stage.html file
"""
# GIVEN: initial route
self.router.config_dir = ''
self.router.send_response = MagicMock()
self.router.send_header = MagicMock()
self.router.end_headers = MagicMock()
self.router.wfile = MagicMock()
self.router._process_file = MagicMock()
# WHEN: I call stage with a suffix
self.router.stages('stages', 'trb')
# THEN: we should use the specific stage file instance
2015-11-08 21:23:15 +00:00
self.router._process_file.assert_called_with(os.path.join('trb', 'stage.html'))
2015-11-08 21:07:48 +00:00
def remote_stage_personal_css_test(self):
"""
Test the html with reference stages/trb/trb.css then loaded a stages/trb/trb.css file
"""
# GIVEN: initial route
self.router.config_dir = ''
self.router.send_response = MagicMock()
self.router.send_header = MagicMock()
self.router.end_headers = MagicMock()
self.router.wfile = MagicMock()
self.router._process_file = MagicMock()
# WHEN: I call stage with a suffix
self.router.stages('stages', 'stages/trb/trb.css')
# THEN: we should use the specific stage file instance
2015-11-08 21:23:15 +00:00
self.router._process_file.assert_called_with(os.path.join('trb', 'trb.css'))