From 4fee73fc37f9d06b70c4310ac50a27ca71755edf Mon Sep 17 00:00:00 2001
From: Andreas Preikschat <googol@lavabit.com>
Date: Wed, 17 Jul 2013 15:59:35 +0200
Subject: [PATCH] added test

---
 openlp/plugins/presentations/lib/mediaitem.py |  2 +-
 .../presentations/test_mediaitem.py           | 71 +++++++++++++++++++
 .../openlp_plugins/songs/test_mediaitem.py    |  2 +-
 3 files changed, 73 insertions(+), 2 deletions(-)
 create mode 100644 tests/functional/openlp_plugins/presentations/test_mediaitem.py

diff --git a/openlp/plugins/presentations/lib/mediaitem.py b/openlp/plugins/presentations/lib/mediaitem.py
index 6e5f20646..c9f11a5d8 100644
--- a/openlp/plugins/presentations/lib/mediaitem.py
+++ b/openlp/plugins/presentations/lib/mediaitem.py
@@ -66,7 +66,7 @@ class PresentationMediaItem(MediaManagerItem):
         Registry().register_function(u'mediaitem_presentation_rebuild', self.populate_display_types)
         Registry().register_function(u'mediaitem_suffixes', self.build_file_mask_string)
         # Allow DnD from the desktop
-        self.list_view.activateDnD()
+    #    self.list_view.activateDnD()
 
     def retranslateUi(self):
         """
diff --git a/tests/functional/openlp_plugins/presentations/test_mediaitem.py b/tests/functional/openlp_plugins/presentations/test_mediaitem.py
new file mode 100644
index 000000000..488e76608
--- /dev/null
+++ b/tests/functional/openlp_plugins/presentations/test_mediaitem.py
@@ -0,0 +1,71 @@
+"""
+This module contains tests for the lib submodule of the Presentations plugin.
+"""
+import os
+from tempfile import mkstemp
+from unittest import TestCase
+
+from mock import patch, MagicMock
+
+from PyQt4 import QtGui
+
+from openlp.core.lib import Registry
+
+from openlp.plugins.presentations.lib.mediaitem import PresentationMediaItem
+
+
+class TestMediaItem(TestCase):
+    """
+    Test the mediaitem methods.
+    """
+    def setUp(self):
+        """
+        Set up the components need for all tests.
+        """
+        Registry.create()
+        Registry().register(u'service_manager', MagicMock())
+        Registry().register(u'main_window', MagicMock())
+
+        with patch('openlp.core.lib.mediamanageritem.MediaManagerItem.__init__'),  \
+                patch('openlp.core.lib.mediamanageritem.ListWidgetWithDnD'):
+            self.media_item = PresentationMediaItem(MagicMock(), MagicMock(), MagicMock(), MagicMock())
+
+        self.application = QtGui.QApplication.instance()
+
+    def tearDown(self):
+        """
+        Delete all the C++ objects at the end so that we don't have a segfault
+        """
+        del self.application
+
+    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
+        impress_controller.supports = [u'odp']
+        impress_controller.also_supports = [u'ppt']
+        presentation_controller = MagicMock()
+        presentation_controller.enabled.return_value = True
+        presentation_controller.supports = [u'ppt']
+        presentation_controller.also_supports = []
+        presentation_viewer_controller = MagicMock()
+        presentation_viewer_controller.enabled.return_value = False
+        # Mock the controllers.
+        self.media_item.controllers = {
+            u'Impress': impress_controller,
+            u'Powerpoint': presentation_controller,
+            u'Powerpoint Viewer': presentation_viewer_controller
+        }
+
+        # 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.
+        assert self.media_item.on_new_file_masks == u'Presentations (*.odp *.ppt )'
+
+
diff --git a/tests/functional/openlp_plugins/songs/test_mediaitem.py b/tests/functional/openlp_plugins/songs/test_mediaitem.py
index 21616e959..c55a45693 100644
--- a/tests/functional/openlp_plugins/songs/test_mediaitem.py
+++ b/tests/functional/openlp_plugins/songs/test_mediaitem.py
@@ -26,7 +26,7 @@ class TestMediaItem(TestCase):
         Registry().register(u'service_list', MagicMock())
         Registry().register(u'main_window', MagicMock())
         with patch('openlp.core.lib.mediamanageritem.MediaManagerItem.__init__'), \
-             patch('openlp.plugins.songs.forms.editsongform.EditSongForm.__init__'):
+                patch('openlp.plugins.songs.forms.editsongform.EditSongForm.__init__'):
             self.media_item = SongMediaItem(MagicMock(), MagicMock())
 
         fd, self.ini_file = mkstemp(u'.ini')