Yet another test

tests/functional/openlp_core_api/test_poll.py

@@ -0,0 +1,104 @@
+# -*- 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 OpenLPPoll
+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 = OpenLPPoll()
+
+    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')