openlp/tests/functional/openlp_plugins/presentations/test_mediaitem.py

134 lines
6.7 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 #
# --------------------------------------------------------------------------- #
2016-12-31 11:05:48 +00:00
# Copyright (c) 2008-2017 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-07-17 13:59:35 +00:00
"""
This module contains tests for the lib submodule of the Presentations plugin.
"""
from unittest import TestCase
2013-12-13 17:44:05 +00:00
from openlp.core.common import Registry
2013-07-17 13:59:35 +00:00
from openlp.plugins.presentations.lib.mediaitem import PresentationMediaItem
2015-03-10 23:46:37 +00:00
from tests.functional import patch, MagicMock, call
2014-03-14 22:08:44 +00:00
from tests.helpers.testmixin import TestMixin
2013-07-17 13:59:35 +00:00
2014-03-14 22:08:44 +00:00
class TestMediaItem(TestCase, TestMixin):
2013-07-17 13:59:35 +00:00
"""
Test the mediaitem methods.
"""
def setUp(self):
"""
Set up the components need for all tests.
"""
Registry.create()
2013-08-31 18:17:38 +00:00
Registry().register('service_manager', MagicMock())
Registry().register('main_window', MagicMock())
with patch('openlp.plugins.presentations.lib.mediaitem.MediaManagerItem._setup'), \
patch('openlp.plugins.presentations.lib.mediaitem.PresentationMediaItem.setup_item'):
self.media_item = PresentationMediaItem(None, MagicMock, MagicMock())
self.setup_application()
2013-07-17 13:59:35 +00:00
def build_file_mask_string_test(self):
"""
Test the build_file_mask_string() method
"""
# GIVEN: Different controllers.
impress_controller = MagicMock()
impress_controller.enabled.return_value = True
2013-08-31 18:17:38 +00:00
impress_controller.supports = ['odp']
impress_controller.also_supports = ['ppt']
2013-07-17 13:59:35 +00:00
presentation_controller = MagicMock()
presentation_controller.enabled.return_value = True
2013-08-31 18:17:38 +00:00
presentation_controller.supports = ['ppt']
2013-07-17 13:59:35 +00:00
presentation_controller.also_supports = []
presentation_viewer_controller = MagicMock()
presentation_viewer_controller.enabled.return_value = False
2013-12-29 19:46:04 +00:00
pdf_controller = MagicMock()
pdf_controller.enabled.return_value = True
pdf_controller.supports = ['pdf']
pdf_controller.also_supports = ['xps', 'oxps']
2013-07-17 13:59:35 +00:00
# Mock the controllers.
self.media_item.controllers = {
2013-08-31 18:17:38 +00:00
'Impress': impress_controller,
'Powerpoint': presentation_controller,
2013-12-29 19:46:04 +00:00
'Powerpoint Viewer': presentation_viewer_controller,
'Pdf': pdf_controller
2013-07-17 13:59:35 +00:00
}
# WHEN: Build the file mask.
with patch('openlp.plugins.presentations.lib.mediaitem.translate') as mocked_translate:
mocked_translate.side_effect = lambda module, string_to_translate: string_to_translate
self.media_item.build_file_mask_string()
# THEN: The file mask should be generated correctly
2014-03-04 20:18:14 +00:00
self.assertIn('*.odp', self.media_item.on_new_file_masks, 'The file mask should contain the odp extension')
self.assertIn('*.ppt', self.media_item.on_new_file_masks, 'The file mask should contain the ppt extension')
self.assertIn('*.pdf', self.media_item.on_new_file_masks, 'The file mask should contain the pdf extension')
self.assertIn('*.xps', self.media_item.on_new_file_masks, 'The file mask should contain the xps extension')
self.assertIn('*.oxps', self.media_item.on_new_file_masks, 'The file mask should contain the oxps extension')
2015-03-10 23:46:37 +00:00
def clean_up_thumbnails_test(self):
"""
2015-03-18 22:04:30 +00:00
Test that the clean_up_thumbnails method works as expected when files exists.
2015-03-10 23:46:37 +00:00
"""
# GIVEN: A mocked controller, and mocked os.path.getmtime
mocked_controller = MagicMock()
mocked_doc = MagicMock()
mocked_controller.add_document.return_value = mocked_doc
mocked_controller.supports = ['tmp']
self.media_item.controllers = {
'Mocked': mocked_controller
}
presentation_file = 'file.tmp'
2015-03-18 22:04:30 +00:00
with patch('openlp.plugins.presentations.lib.mediaitem.os.path.getmtime') as mocked_getmtime, \
patch('openlp.plugins.presentations.lib.mediaitem.os.path.exists') as mocked_exists:
2015-03-10 23:46:37 +00:00
mocked_getmtime.side_effect = [100, 200]
2015-03-18 22:04:30 +00:00
mocked_exists.return_value = True
2015-03-10 23:46:37 +00:00
# WHEN: calling clean_up_thumbnails
self.media_item.clean_up_thumbnails(presentation_file, True)
# THEN: doc.presentation_deleted should have been called since the thumbnails mtime will be greater than
# the presentation_file's mtime.
mocked_doc.assert_has_calls([call.get_thumbnail_path(1, True), call.presentation_deleted()], True)
2015-03-18 22:04:30 +00:00
def clean_up_thumbnails_missing_file_test(self):
"""
Test that the clean_up_thumbnails method works as expected when file is missing.
"""
# GIVEN: A mocked controller, and mocked os.path.exists
mocked_controller = MagicMock()
mocked_doc = MagicMock()
mocked_controller.add_document.return_value = mocked_doc
mocked_controller.supports = ['tmp']
self.media_item.controllers = {
'Mocked': mocked_controller
}
presentation_file = 'file.tmp'
with patch('openlp.plugins.presentations.lib.mediaitem.os.path.exists') as mocked_exists:
mocked_exists.return_value = False
# WHEN: calling clean_up_thumbnails
self.media_item.clean_up_thumbnails(presentation_file, True)
# THEN: doc.presentation_deleted should have been called since the presentation file did not exists.
mocked_doc.assert_has_calls([call.get_thumbnail_path(1, True), call.presentation_deleted()], True)