2013-09-19 21:02:28 +00:00
|
|
|
# -*- coding: utf-8 -*-
|
|
|
|
# vim: autoindent shiftwidth=4 expandtab textwidth=120 tabstop=4 softtabstop=4
|
|
|
|
|
|
|
|
###############################################################################
|
|
|
|
# OpenLP - Open Source Lyrics Projection #
|
|
|
|
# --------------------------------------------------------------------------- #
|
2016-12-31 11:01:36 +00:00
|
|
|
# Copyright (c) 2008-2017 OpenLP Developers #
|
2013-09-19 21:02:28 +00:00
|
|
|
# --------------------------------------------------------------------------- #
|
|
|
|
# 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-02-13 19:10:41 +00:00
|
|
|
"""
|
2013-09-19 21:02:28 +00:00
|
|
|
Package to test the openlp.core.ui package.
|
2013-02-13 19:10:41 +00:00
|
|
|
"""
|
|
|
|
import os
|
2014-04-12 16:52:29 +00:00
|
|
|
import time
|
|
|
|
from threading import Lock
|
2013-09-08 10:30:14 +00:00
|
|
|
from unittest import TestCase
|
2017-04-24 05:17:55 +00:00
|
|
|
from unittest.mock import patch
|
|
|
|
|
2015-11-07 00:49:40 +00:00
|
|
|
from PyQt5 import QtGui
|
2013-02-13 19:10:41 +00:00
|
|
|
|
2017-10-07 07:05:07 +00:00
|
|
|
from openlp.core.common.registry import Registry
|
2017-10-10 07:08:44 +00:00
|
|
|
from openlp.core.display.screens import ScreenList
|
|
|
|
from openlp.core.lib.imagemanager import ImageManager, Priority
|
2017-04-24 05:17:55 +00:00
|
|
|
|
2014-03-14 22:08:44 +00:00
|
|
|
from tests.helpers.testmixin import TestMixin
|
2017-12-22 21:20:49 +00:00
|
|
|
from tests.utils.constants import RESOURCE_PATH
|
2013-02-16 06:51:25 +00:00
|
|
|
|
2017-12-22 21:20:49 +00:00
|
|
|
TEST_PATH = str(RESOURCE_PATH)
|
2013-02-13 19:10:41 +00:00
|
|
|
|
2013-02-13 19:22:43 +00:00
|
|
|
|
2014-03-14 22:08:44 +00:00
|
|
|
class TestImageManager(TestCase, TestMixin):
|
2013-02-13 19:10:41 +00:00
|
|
|
|
|
|
|
def setUp(self):
|
|
|
|
"""
|
|
|
|
Create the UI
|
|
|
|
"""
|
|
|
|
Registry.create()
|
2014-10-22 20:43:05 +00:00
|
|
|
self.setup_application()
|
2013-02-13 19:10:41 +00:00
|
|
|
ScreenList.create(self.app.desktop())
|
|
|
|
self.image_manager = ImageManager()
|
2014-04-12 16:52:29 +00:00
|
|
|
self.lock = Lock()
|
|
|
|
self.sleep_time = 0.1
|
2013-02-13 19:10:41 +00:00
|
|
|
|
2013-02-16 15:02:19 +00:00
|
|
|
def tearDown(self):
|
|
|
|
"""
|
|
|
|
Delete all the C++ objects at the end so that we don't have a segfault
|
|
|
|
"""
|
2017-08-12 19:27:43 +00:00
|
|
|
self.image_manager.stop_manager = True
|
|
|
|
self.image_manager.image_thread.wait()
|
2013-02-16 15:02:19 +00:00
|
|
|
del self.app
|
|
|
|
|
2016-05-31 21:40:13 +00:00
|
|
|
def test_basic_image_manager(self):
|
2013-02-13 19:10:41 +00:00
|
|
|
"""
|
|
|
|
Test the Image Manager setup basic functionality
|
|
|
|
"""
|
|
|
|
# GIVEN: the an image add to the image manager
|
2013-11-07 20:13:15 +00:00
|
|
|
full_path = os.path.normpath(os.path.join(TEST_PATH, 'church.jpg'))
|
|
|
|
self.image_manager.add_image(full_path, 'church.jpg', None)
|
2013-02-13 19:10:41 +00:00
|
|
|
|
|
|
|
# WHEN the image is retrieved
|
2013-11-07 20:13:15 +00:00
|
|
|
image = self.image_manager.get_image(full_path, 'church.jpg')
|
2013-02-13 19:10:41 +00:00
|
|
|
|
|
|
|
# THEN returned record is a type of image
|
2013-08-31 18:17:38 +00:00
|
|
|
self.assertEqual(isinstance(image, QtGui.QImage), True, 'The returned object should be a QImage')
|
2013-02-13 19:10:41 +00:00
|
|
|
|
2013-03-12 08:19:47 +00:00
|
|
|
# WHEN: The image bytes are requested.
|
2013-11-07 20:13:15 +00:00
|
|
|
byte_array = self.image_manager.get_image_bytes(full_path, 'church.jpg')
|
2013-03-12 08:19:47 +00:00
|
|
|
|
2013-04-20 18:05:01 +00:00
|
|
|
# THEN: Type should be a str.
|
2013-08-31 18:17:38 +00:00
|
|
|
self.assertEqual(isinstance(byte_array, str), True, 'The returned object should be a str')
|
2013-03-12 08:19:47 +00:00
|
|
|
|
2013-02-13 19:10:41 +00:00
|
|
|
# WHEN the image is retrieved has not been loaded
|
|
|
|
# THEN a KeyError is thrown
|
|
|
|
with self.assertRaises(KeyError) as context:
|
2013-08-31 18:17:38 +00:00
|
|
|
self.image_manager.get_image(TEST_PATH, 'church1.jpg')
|
2017-12-17 15:35:35 +00:00
|
|
|
assert context.exception is not '', 'KeyError exception should have been thrown for missing image'
|
2013-11-07 20:13:15 +00:00
|
|
|
|
2016-05-31 21:40:13 +00:00
|
|
|
def test_different_dimension_image(self):
|
2013-11-07 20:13:15 +00:00
|
|
|
"""
|
|
|
|
Test the Image Manager with dimensions
|
|
|
|
"""
|
|
|
|
# GIVEN: add an image with specific dimensions
|
|
|
|
full_path = os.path.normpath(os.path.join(TEST_PATH, 'church.jpg'))
|
2014-07-15 18:52:59 +00:00
|
|
|
self.image_manager.add_image(full_path, 'church.jpg', None, 80, 80)
|
2013-11-07 20:13:15 +00:00
|
|
|
|
|
|
|
# WHEN: the image is retrieved
|
2014-07-15 18:52:59 +00:00
|
|
|
image = self.image_manager.get_image(full_path, 'church.jpg', 80, 80)
|
2013-11-07 20:13:15 +00:00
|
|
|
|
|
|
|
# THEN: The return should be of type image
|
2017-12-18 17:10:04 +00:00
|
|
|
assert isinstance(image, QtGui.QImage), 'The returned object should be a QImage'
|
2013-11-07 20:13:15 +00:00
|
|
|
|
|
|
|
# WHEN: adding the same image with different dimensions
|
2014-07-15 18:52:59 +00:00
|
|
|
self.image_manager.add_image(full_path, 'church.jpg', None, 100, 100)
|
2013-11-07 20:13:15 +00:00
|
|
|
|
|
|
|
# THEN: the cache should contain two pictures
|
|
|
|
self.assertEqual(len(self.image_manager._cache), 2,
|
2013-12-30 08:35:05 +00:00
|
|
|
'Image manager should consider two dimensions of the same picture as different')
|
2013-11-07 20:13:15 +00:00
|
|
|
|
|
|
|
# WHEN: adding the same image with first dimensions
|
2014-07-15 18:52:59 +00:00
|
|
|
self.image_manager.add_image(full_path, 'church.jpg', None, 80, 80)
|
2013-11-07 20:13:15 +00:00
|
|
|
|
|
|
|
# THEN: the cache should still contain only two pictures
|
2013-12-30 08:35:05 +00:00
|
|
|
self.assertEqual(len(self.image_manager._cache), 2, 'Same dimensions should not be added again')
|
2013-11-07 20:13:15 +00:00
|
|
|
|
|
|
|
# WHEN: calling with correct image, but wrong dimensions
|
|
|
|
with self.assertRaises(KeyError) as context:
|
2014-07-15 18:52:59 +00:00
|
|
|
self.image_manager.get_image(full_path, 'church.jpg', 120, 120)
|
2017-12-17 15:35:35 +00:00
|
|
|
assert context.exception is not '', 'KeyError exception should have been thrown for missing dimension'
|
2013-11-07 20:13:15 +00:00
|
|
|
|
2016-05-31 21:40:13 +00:00
|
|
|
def test_process_cache(self):
|
2014-04-12 16:52:29 +00:00
|
|
|
"""
|
|
|
|
Test the process_cache method
|
|
|
|
"""
|
|
|
|
with patch('openlp.core.lib.imagemanager.resize_image') as mocked_resize_image, \
|
|
|
|
patch('openlp.core.lib.imagemanager.image_to_byte') as mocked_image_to_byte:
|
|
|
|
# GIVEN: Mocked functions
|
|
|
|
mocked_resize_image.side_effect = self.mocked_resize_image
|
|
|
|
mocked_image_to_byte.side_effect = self.mocked_image_to_byte
|
|
|
|
image1 = 'church.jpg'
|
|
|
|
image2 = 'church2.jpg'
|
|
|
|
image3 = 'church3.jpg'
|
|
|
|
image4 = 'church4.jpg'
|
|
|
|
|
|
|
|
# WHEN: Add the images. Then get the lock (=queue can not be processed).
|
|
|
|
self.lock.acquire()
|
|
|
|
self.image_manager.add_image(TEST_PATH, image1, None)
|
|
|
|
self.image_manager.add_image(TEST_PATH, image2, None)
|
|
|
|
|
|
|
|
# THEN: All images have been added to the queue, and only the first image is not be in the list anymore, but
|
|
|
|
# is being processed (see mocked methods/functions).
|
|
|
|
# Note: Priority.Normal means, that the resize_image() was not completed yet (because afterwards the #
|
|
|
|
# priority is adjusted to Priority.Lowest).
|
2017-12-17 15:35:35 +00:00
|
|
|
assert self.get_image_priority(image1) == Priority.Normal, "image1's priority should be 'Priority.Normal'"
|
|
|
|
assert self.get_image_priority(image2) == Priority.Normal, "image2's priority should be 'Priority.Normal'"
|
2014-04-12 16:52:29 +00:00
|
|
|
|
|
|
|
# WHEN: Add more images.
|
|
|
|
self.image_manager.add_image(TEST_PATH, image3, None)
|
|
|
|
self.image_manager.add_image(TEST_PATH, image4, None)
|
|
|
|
# Allow the queue to process.
|
|
|
|
self.lock.release()
|
|
|
|
# Request some "data".
|
2017-10-10 07:08:44 +00:00
|
|
|
self.image_manager.get_image_bytes(TEST_PATH, image4)
|
|
|
|
self.image_manager.get_image(TEST_PATH, image3)
|
2014-04-12 16:52:29 +00:00
|
|
|
# Now the mocked methods/functions do not have to sleep anymore.
|
|
|
|
self.sleep_time = 0
|
|
|
|
# Wait for the queue to finish.
|
|
|
|
while not self.image_manager._conversion_queue.empty():
|
|
|
|
time.sleep(0.1)
|
|
|
|
# Because empty() is not reliable, wait a litte; just to make sure.
|
|
|
|
time.sleep(0.1)
|
|
|
|
# THEN: The images' priority reflect how they were processed.
|
2017-12-17 15:35:35 +00:00
|
|
|
assert self.image_manager._conversion_queue.qsize() == 0, "The queue should be empty."
|
|
|
|
assert self.get_image_priority(image1) == Priority.Lowest, \
|
|
|
|
"The image should have not been requested (=Lowest)"
|
|
|
|
assert self.get_image_priority(image2) == Priority.Lowest, \
|
|
|
|
"The image should have not been requested (=Lowest)"
|
|
|
|
assert self.get_image_priority(image3) == Priority.Low, \
|
|
|
|
"Only the QImage should have been requested (=Low)."
|
|
|
|
assert self.get_image_priority(image4) == Priority.Urgent, \
|
|
|
|
"The image bytes should have been requested (=Urgent)."
|
2014-04-12 16:52:29 +00:00
|
|
|
|
|
|
|
def get_image_priority(self, image):
|
|
|
|
"""
|
|
|
|
This is a help method to get the priority of the given image out of the image_manager's cache.
|
|
|
|
|
|
|
|
NOTE: This requires, that the image has been added to the image manager using the *TEST_PATH*.
|
|
|
|
|
|
|
|
:param image: The name of the image. E. g. ``image1``
|
|
|
|
"""
|
2014-07-15 18:52:59 +00:00
|
|
|
return self.image_manager._cache[(TEST_PATH, image, -1, -1)].priority
|
2014-04-12 16:52:29 +00:00
|
|
|
|
|
|
|
def mocked_resize_image(self, *args):
|
|
|
|
"""
|
|
|
|
This is a mocked method, so that we can control the work flow of the image manager.
|
|
|
|
"""
|
|
|
|
self.lock.acquire()
|
|
|
|
self.lock.release()
|
|
|
|
# The sleep time is adjusted in the test case.
|
|
|
|
time.sleep(self.sleep_time)
|
|
|
|
return QtGui.QImage()
|
|
|
|
|
|
|
|
def mocked_image_to_byte(self, *args):
|
|
|
|
"""
|
|
|
|
This is a mocked method, so that we can control the work flow of the image manager.
|
|
|
|
"""
|
|
|
|
self.lock.acquire()
|
|
|
|
self.lock.release()
|
|
|
|
# The sleep time is adjusted in the test case.
|
|
|
|
time.sleep(self.sleep_time)
|
2014-04-23 19:30:12 +00:00
|
|
|
return ''
|