openlp/tests/functional/openlp_core_lib/test_screen.py

"""
Package to test the openlp.core.lib.screenlist package.
"""
import copy
from unittest import TestCase

from mock import MagicMock
from PyQt4 import QtGui, QtCore

from openlp.core.lib import ScreenList


SCREEN = {
    u'primary': False,
    u'number': 1,
    u'size': QtCore.QRect(0, 0, 1024, 768)
}


class TestScreenList(TestCase):

    def setUp(self):
        """
        Set up the components need for all tests.
        """
        self.application = QtGui.QApplication.instance()
        self.screens = ScreenList.create(self.application.desktop())

    def tearDown(self):
        """
        Delete QApplication.
        """
        del self.application

    def add_desktop_test(self):
        """
        Test the ScreenList class' screen_count_changed method to check if new monitors are detected by OpenLP.
        """
        # GIVEN: The screen list.
        old_screens = copy.deepcopy(self.screens.screen_list)
        # Mock the attributes.
        self.screens.desktop.primaryScreen = MagicMock(return_value=SCREEN[u'primary'])
        self.screens.desktop.screenCount = MagicMock(return_value=SCREEN[u'number'] + 1)
        self.screens.desktop.screenGeometry = MagicMock(return_value=SCREEN[u'size'])

        # WHEN: Add a new screen.
        self.screens.screen_count_changed(len(old_screens))

        # THEN: The screen should have been added.
        new_screens = self.screens.screen_list
        assert len(old_screens) + 1 == len(new_screens), u'The new_screens list should be bigger.'

        # THEN: The screens should be identically.
        assert SCREEN == new_screens.pop(), u'The new screen should be identically to the screen defined above.'