openlp/tests/functional/openlp_plugins/songs/test_openlyricsimport.py

195 lines
8.8 KiB
Python
Raw Normal View History

2014-04-10 07:50:30 +00:00
# -*- coding: utf-8 -*-
# vim: autoindent shiftwidth=4 expandtab textwidth=120 tabstop=4 softtabstop=4
###############################################################################
# OpenLP - Open Source Lyrics Projection #
# --------------------------------------------------------------------------- #
2015-12-31 22:46:06 +00:00
# Copyright (c) 2008-2016 OpenLP Developers #
2014-04-10 07:50:30 +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 #
###############################################################################
"""
This module contains tests for the OpenLyrics song importer.
"""
2014-04-15 22:23:29 +00:00
import os
import json
2014-04-10 07:50:30 +00:00
from unittest import TestCase
from lxml import etree, objectify
2014-04-10 07:50:30 +00:00
from tests.functional import MagicMock, patch
from tests.helpers.testmixin import TestMixin
2014-07-04 09:31:06 +00:00
from openlp.plugins.songs.lib.importers.openlyrics import OpenLyricsImport
from openlp.plugins.songs.lib.importers.songimport import SongImport
from openlp.plugins.songs.lib.openlyricsxml import OpenLyrics
from openlp.core.common import Registry, Settings
2014-04-10 07:50:30 +00:00
2014-04-15 22:23:29 +00:00
TEST_PATH = os.path.abspath(os.path.join(os.path.dirname(__file__),
'..', '..', '..', 'resources', 'openlyricssongs'))
SONG_TEST_DATA = {
2014-04-16 21:01:57 +00:00
'What a friend we have in Jesus.xml': {
'title': 'What A Friend We Have In Jesus',
'verses': [
('What a friend we have in Jesus, All ours sins and griefs to bear;\n\
What a privilege to carry, Everything to God in prayer!\n\
O what peace we often forfeit, O what needless pain we bear;\n\
All because we do not carry, Everything to God in prayer!', 'v1'),
('Have we trials and temptations? Is there trouble anywhere?\n\
We should never be discouraged, Take it to the Lord in prayer.\n\
Can we find a friend so faithful? Who will all our sorrows share?\n\
Jesus knows our every weakness; Take it to the Lord in prayer.', 'v2'),
('Are we weak and heavy laden, Cumbered with a load of care?\n\
Precious Saviour still our refuge; Take it to the Lord in prayer.\n\
Do thy friends despise forsake thee? Take it to the Lord in prayer!\n\
In His arms Hell take and shield thee; Thou wilt find a solace there.', 'v3')
2014-04-15 22:23:29 +00:00
]
}
}
2014-09-25 20:31:31 +00:00
start_tags = [{"protected": False, "desc": "z", "start tag": "{z}", "end html": "</strong>", "temporary": False,
"end tag": "{/z}", "start html": "strong>"}]
result_tags = [{"temporary": False, "protected": False, "desc": "z", "start tag": "{z}", "start html": "strong>",
"end html": "</strong>", "end tag": "{/z}"},
{"temporary": False, "end tag": "{/c}", "desc": "c", "start tag": "{c}",
"start html": "<span class=\"chord\" style=\"display:none\"><strong>", "end html": "</strong></span>",
"protected": False}]
author_xml = '<properties>\
<authors>\
<author type="words">Test Author1</author>\
<author type="music">Test Author1</author>\
<author type="words">Test Author2</author>\
</authors>\
</properties>'
2016-01-30 14:13:35 +00:00
songbook_xml = '<properties>\
<songbooks>\
<songbook name="Collection 1" entry="48"/>\
<songbook name="Collection 2" entry="445 A"/>\
</songbooks>\
</properties>'
2014-04-10 07:50:30 +00:00
class TestOpenLyricsImport(TestCase, TestMixin):
2014-04-10 07:50:30 +00:00
"""
Test the functions in the :mod:`openlyricsimport` module.
"""
def setUp(self):
"""
Create the registry
"""
2014-10-22 20:47:47 +00:00
self.setup_application()
Registry.create()
self.build_settings()
def tearDown(self):
"""
Cleanup
"""
self.destroy_settings()
2014-04-10 07:50:30 +00:00
def create_importer_test(self):
"""
Test creating an instance of the OpenLyrics file importer
"""
# GIVEN: A mocked out SongImport class, and a mocked out "manager"
2014-07-04 09:31:06 +00:00
with patch('openlp.plugins.songs.lib.importers.openlyrics.SongImport'):
2014-04-10 07:50:30 +00:00
mocked_manager = MagicMock()
# WHEN: An importer object is created
importer = OpenLyricsImport(mocked_manager, filenames=[])
# THEN: The importer should be an instance of SongImport
self.assertIsInstance(importer, SongImport)
2014-04-15 22:23:29 +00:00
2014-04-16 20:33:06 +00:00
def file_import_test(self):
2014-04-15 22:23:29 +00:00
"""
2014-04-16 20:42:44 +00:00
Test the actual import of real song files
2014-04-15 22:23:29 +00:00
"""
# GIVEN: Test files with a mocked out "manager" and a mocked out "import_wizard"
for song_file in SONG_TEST_DATA:
mocked_manager = MagicMock()
mocked_import_wizard = MagicMock()
importer = OpenLyricsImport(mocked_manager, filenames=[])
importer.import_wizard = mocked_import_wizard
2014-04-16 20:33:06 +00:00
importer.open_lyrics = MagicMock()
importer.open_lyrics.xml_to_song = MagicMock()
2014-04-15 22:23:29 +00:00
# WHEN: Importing each file
importer.import_source = [os.path.join(TEST_PATH, song_file)]
2014-04-16 20:33:06 +00:00
importer.do_import()
2014-04-15 22:23:29 +00:00
2014-04-16 20:33:06 +00:00
# THEN: The xml_to_song() method should have been called
self.assertTrue(importer.open_lyrics.xml_to_song.called)
def process_formatting_tags_test(self):
"""
Test that _process_formatting_tags works
"""
2014-09-25 20:31:31 +00:00
# GIVEN: A OpenLyric XML with formatting tags and a mocked out manager
mocked_manager = MagicMock()
2014-09-25 20:31:31 +00:00
Settings().setValue('formattingTags/html_tags', json.dumps(start_tags))
ol = OpenLyrics(mocked_manager)
parser = etree.XMLParser(remove_blank_text=True)
parsed_file = etree.parse(open(os.path.join(TEST_PATH, 'duchu-tags.xml'), 'rb'), parser)
xml = etree.tostring(parsed_file).decode()
song_xml = objectify.fromstring(xml)
# WHEN: processing the formatting tags
ol._process_formatting_tags(song_xml, False)
# THEN: New tags should have been saved
2014-09-25 20:31:31 +00:00
self.assertListEqual(json.loads(json.dumps(result_tags)),
json.loads(str(Settings().value('formattingTags/html_tags'))),
'The formatting tags should contain both the old and the new')
def process_author_test(self):
"""
Test that _process_authors works
"""
# GIVEN: A OpenLyric XML with authors and a mocked out manager
with patch('openlp.plugins.songs.lib.openlyricsxml.Author') as mocked_author:
mocked_manager = MagicMock()
mocked_manager.get_object_filtered.return_value = None
ol = OpenLyrics(mocked_manager)
properties_xml = objectify.fromstring(author_xml)
mocked_song = MagicMock()
# WHEN: processing the author xml
ol._process_authors(properties_xml, mocked_song)
# THEN: add_author should have been called twice
self.assertEquals(mocked_song.method_calls[0][1][1], 'words+music')
self.assertEquals(mocked_song.method_calls[1][1][1], 'words')
2016-01-30 14:13:35 +00:00
def process_songbooks_test(self):
"""
Test that _process_songbooks works
"""
# GIVEN: A OpenLyric XML with songbooks and a mocked out manager
with patch('openlp.plugins.songs.lib.openlyricsxml.Book'):
mocked_manager = MagicMock()
mocked_manager.get_object_filtered.return_value = None
ol = OpenLyrics(mocked_manager)
properties_xml = objectify.fromstring(songbook_xml)
mocked_song = MagicMock()
# WHEN: processing the songbook xml
ol._process_songbooks(properties_xml, mocked_song)
# THEN: add_songbook_entry should have been called twice
self.assertEquals(mocked_song.method_calls[0][1][1], '48')
self.assertEquals(mocked_song.method_calls[1][1][1], '445 A')