openlp/tests/functional/openlp_core_api/test_poll.py

# -*- coding: utf-8 -*-
# vim: autoindent shiftwidth=4 expandtab textwidth=120 tabstop=4 softtabstop=4

###############################################################################
# OpenLP - Open Source Lyrics Projection                                      #
# --------------------------------------------------------------------------- #
# Copyright (c) 2008-2016 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                          #
###############################################################################
"""
This module contains tests for the lib submodule of the Remotes plugin.
"""
import json
from unittest import TestCase

from openlp.core.common import Registry, Settings
from openlp.core.api.poll import Poll
from tests.functional import MagicMock
from tests.helpers.testmixin import TestMixin

__default_settings__ = {
    'remotes/twelve hour': True,
    'remotes/port': 4316,
    'remotes/user id': 'openlp',
    'remotes/password': 'password',
    'remotes/authentication enabled': False,
    'remotes/ip address': '0.0.0.0',
    'remotes/thumbnails': True
}


class TestOpenLPPoll(TestCase, TestMixin):
    """
    Test the functions in the :mod:`lib` module.
    """
    def setUp(self):
        """
        Create the UI
        """
        self.build_settings()
        Settings().extend_default_settings(__default_settings__)
        Registry().create()
        self.poll = Poll()

    def tearDown(self):
        """
        Delete all the C++ objects at the end so that we don't have a segfault
        """
        self.destroy_settings()

    def test_main_poll(self):
        """
        Test the main_poll function returns the correct JSON
        """
        # WHEN: the live controller has 5 slides
        mocked_live_controller = MagicMock()
        mocked_live_controller.slide_count = 5
        Registry().register('live_controller', mocked_live_controller)
        # THEN: the live json should be generated
        main_json = self.poll.main_poll()
        self.assertEquals(b'{"results": {"slide_count": 5}}', main_json,
                          'The return value should match the defined json')

    def test_poll(self):
        """
        Test the poll function returns the correct JSON
        """
        # WHEN: the system is configured with a set of data
        mocked_service_manager = MagicMock()
        mocked_service_manager.service_id = 21
        mocked_live_controller = MagicMock()
        mocked_live_controller.selected_row = 5
        mocked_live_controller.service_item = MagicMock()
        mocked_live_controller.service_item.unique_identifier = '23-34-45'
        mocked_live_controller.blank_screen.isChecked.return_value = True
        mocked_live_controller.theme_screen.isChecked.return_value = False
        mocked_live_controller.desktop_screen.isChecked.return_value = False
        Registry().register('live_controller', mocked_live_controller)
        Registry().register('service_manager', mocked_service_manager)
        # THEN: the live json should be generated and match expected results
        poll_json = self.poll.poll()
        data = json.loads(poll_json.decode())
        self.assertTrue(data['results']['blank'], 'The blank return value should be True')
        self.assertFalse(data['results']['theme'], 'The theme return value should be False')
        self.assertFalse(data['results']['display'], 'The display return value should be False')
        self.assertFalse(data['results']['isSecure'], 'The isSecure return value should be False')
        self.assertFalse(data['results']['isAuthorised'], 'The isAuthorised return value should be False')
        self.assertTrue(data['results']['twelve'], 'The twelve return value should be False')
        self.assertEquals(data['results']['version'], 2, 'The version return value should be 2')
        self.assertEquals(data['results']['slide'], 5, 'The slide return value should be 5')
        self.assertEquals(data['results']['service'], 21, 'The version return value should be 21')
        self.assertEquals(data['results']['item'], '23-34-45', 'The item return value should match 23-34-45')