forked from openlp/openlp
461 lines
22 KiB
Python
461 lines
22 KiB
Python
# -*- coding: utf-8 -*-
|
|
# vim: autoindent shiftwidth=4 expandtab textwidth=120 tabstop=4 softtabstop=4
|
|
|
|
###############################################################################
|
|
# OpenLP - Open Source Lyrics Projection #
|
|
# --------------------------------------------------------------------------- #
|
|
# Copyright (c) 2008-2013 Raoul Snyman #
|
|
# Portions copyright (c) 2008-2013 Tim Bentley, Gerald Britton, Jonathan #
|
|
# Corwin, Samuel Findlay, Michael Gorven, Scott Guerrieri, Matthias Hub, #
|
|
# Meinert Jordan, Armin Köhler, Erik Lundin, Edwin Lunando, Brian T. Meyer. #
|
|
# Joshua Miller, Stevan Pettit, Andreas Preikschat, Mattias Põldaru, #
|
|
# Christian Richter, Philip Ridout, Simon Scudder, Jeffrey Smith, #
|
|
# Maikel Stuivenberg, Martin Thompson, Jon Tibble, Dave Warnock, #
|
|
# Frode Woldsund, Martin Zibricky, Patrick Zimmermann #
|
|
# --------------------------------------------------------------------------- #
|
|
# 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 Songs plugin.
|
|
"""
|
|
from unittest import TestCase
|
|
|
|
from openlp.plugins.songs.lib import VerseType, clean_string, clean_title, strip_rtf
|
|
from openlp.plugins.songs.lib.songcompare import songs_probably_equal, _remove_typos, _op_length
|
|
from tests.functional import patch, MagicMock
|
|
|
|
|
|
class TestLib(TestCase):
|
|
"""
|
|
Test the functions in the :mod:`lib` module.
|
|
"""
|
|
def setUp(self):
|
|
"""
|
|
Mock up two songs and provide a set of lyrics for the songs_probably_equal tests.
|
|
"""
|
|
self.full_lyrics ='''amazing grace how sweet the sound that saved a wretch like me i once was lost but now am
|
|
found was blind but now i see twas grace that taught my heart to fear and grace my fears relieved how
|
|
precious did that grace appear the hour i first believed through many dangers toils and snares i have already
|
|
come tis grace that brought me safe thus far and grace will lead me home'''
|
|
self.short_lyrics ='''twas grace that taught my heart to fear and grace my fears relieved how precious did that
|
|
grace appear the hour i first believed'''
|
|
self.error_lyrics ='''amazing how sweet the trumpet that saved a wrench like me i once was losst but now am
|
|
found waf blind but now i see it was grace that taught my heart to fear and grace my fears relieved how
|
|
precious did that grace appppppppear the hour i first believedxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx snares i have
|
|
already come to this grace that brought me safe so far and grace will lead me home'''
|
|
self.different_lyrics='''on a hill far away stood an old rugged cross the emblem of suffering and shame and i love
|
|
that old cross where the dearest and best for a world of lost sinners was slain so ill cherish the old rugged
|
|
cross till my trophies at last i lay down i will cling to the old rugged cross and exchange it some day for a
|
|
crown'''
|
|
self.song1 = MagicMock()
|
|
self.song2 = MagicMock()
|
|
|
|
def clean_string_test(self):
|
|
"""
|
|
Test the clean_string() function
|
|
"""
|
|
# GIVEN: A "dirty" string
|
|
dirty_string = 'Ain\'t gonna find\t you there.'
|
|
|
|
# WHEN: We run the string through the function
|
|
result = clean_string(dirty_string)
|
|
|
|
# THEN: The string should be cleaned up and lower-cased
|
|
self.assertEqual(result, 'aint gonna find you there ', 'The string should be cleaned up properly')
|
|
|
|
def clean_title_test(self):
|
|
"""
|
|
Test the clean_title() function
|
|
"""
|
|
# GIVEN: A "dirty" string
|
|
dirty_string = 'This\u0000 is a\u0014 dirty \u007Fstring\u009F'
|
|
|
|
# WHEN: We run the string through the function
|
|
result = clean_title(dirty_string)
|
|
|
|
# THEN: The string should be cleaned up
|
|
self.assertEqual(result, 'This is a dirty string', 'The title should be cleaned up properly: "%s"' % result)
|
|
|
|
def songs_probably_equal_same_song_test(self):
|
|
"""
|
|
Test the songs_probably_equal function with twice the same song.
|
|
"""
|
|
# GIVEN: Two equal songs.
|
|
self.song1.search_lyrics = self.full_lyrics
|
|
self.song2.search_lyrics = self.full_lyrics
|
|
|
|
# WHEN: We compare those songs for equality.
|
|
result = songs_probably_equal(self.song1, self.song2)
|
|
|
|
# THEN: The result should be True.
|
|
assert result == True, 'The result should be True'
|
|
|
|
def songs_probably_equal_short_song_test(self):
|
|
"""
|
|
Test the songs_probably_equal function with a song and a shorter version of the same song.
|
|
"""
|
|
# GIVEN: A song and a short version of the same song.
|
|
self.song1.search_lyrics = self.full_lyrics
|
|
self.song2.search_lyrics = self.short_lyrics
|
|
|
|
# WHEN: We compare those songs for equality.
|
|
result = songs_probably_equal(self.song1, self.song2)
|
|
|
|
# THEN: The result should be True.
|
|
assert result == True, 'The result should be True'
|
|
|
|
def songs_probably_equal_error_song_test(self):
|
|
"""
|
|
Test the songs_probably_equal function with a song and a very erroneous version of the same song.
|
|
"""
|
|
# GIVEN: A song and the same song with lots of errors.
|
|
self.song1.search_lyrics = self.full_lyrics
|
|
self.song2.search_lyrics = self.error_lyrics
|
|
|
|
# WHEN: We compare those songs for equality.
|
|
result = songs_probably_equal(self.song1, self.song2)
|
|
|
|
# THEN: The result should be True.
|
|
assert result == True, 'The result should be True'
|
|
|
|
def songs_probably_equal_different_song_test(self):
|
|
"""
|
|
Test the songs_probably_equal function with two different songs.
|
|
"""
|
|
# GIVEN: Two different songs.
|
|
self.song1.search_lyrics = self.full_lyrics
|
|
self.song2.search_lyrics = self.different_lyrics
|
|
|
|
# WHEN: We compare those songs for equality.
|
|
result = songs_probably_equal(self.song1, self.song2)
|
|
|
|
# THEN: The result should be False.
|
|
assert result == False, 'The result should be False'
|
|
|
|
def remove_typos_beginning_test(self):
|
|
"""
|
|
Test the _remove_typos function with a typo at the beginning.
|
|
"""
|
|
# GIVEN: A diffset with a difference at the beginning.
|
|
diff = [('replace', 0, 2, 0, 1), ('equal', 2, 11, 1, 10)]
|
|
|
|
# WHEN: We remove the typos in there.
|
|
result = _remove_typos(diff)
|
|
|
|
# THEN: There should be no typos at the beginning anymore.
|
|
assert len(result) == 1, 'The result should contain only one element.'
|
|
assert result[0][0] == 'equal', 'The result should contain an equal element.'
|
|
|
|
def remove_typos_beginning_negated_test(self):
|
|
"""
|
|
Test the _remove_typos function with a large difference at the beginning.
|
|
"""
|
|
# GIVEN: A diffset with a large difference at the beginning.
|
|
diff = [('replace', 0, 20, 0, 1), ('equal', 20, 29, 1, 10)]
|
|
|
|
# WHEN: We remove the typos in there.
|
|
result = _remove_typos(list(diff))
|
|
|
|
# THEN: There diff should not have changed.
|
|
assert result == diff
|
|
|
|
def remove_typos_end_test(self):
|
|
"""
|
|
Test the _remove_typos function with a typo at the end.
|
|
"""
|
|
# GIVEN: A diffset with a difference at the end.
|
|
diff = [('equal', 0, 10, 0, 10), ('replace', 10, 12, 10, 11)]
|
|
|
|
# WHEN: We remove the typos in there.
|
|
result = _remove_typos(diff)
|
|
|
|
# THEN: There should be no typos at the end anymore.
|
|
assert len(result) == 1, 'The result should contain only one element.'
|
|
assert result[0][0] == 'equal', 'The result should contain an equal element.'
|
|
|
|
def remove_typos_end_negated_test(self):
|
|
"""
|
|
Test the _remove_typos function with a large difference at the end.
|
|
"""
|
|
# GIVEN: A diffset with a large difference at the end.
|
|
diff = [('equal', 0, 10, 0, 10), ('replace', 10, 20, 10, 1)]
|
|
|
|
# WHEN: We remove the typos in there.
|
|
result = _remove_typos(list(diff))
|
|
|
|
# THEN: There diff should not have changed.
|
|
assert result == diff
|
|
|
|
def remove_typos_middle_test(self):
|
|
"""
|
|
Test the _remove_typos function with a typo in the middle.
|
|
"""
|
|
# GIVEN: A diffset with a difference in the middle.
|
|
diff = [('equal', 0, 10, 0, 10), ('replace', 10, 12, 10, 11), ('equal', 12, 22, 11, 21)]
|
|
|
|
# WHEN: We remove the typos in there.
|
|
result = _remove_typos(diff)
|
|
|
|
# THEN: There should be no typos in the middle anymore. The remaining equals should have been merged.
|
|
assert len(result) is 1, 'The result should contain only one element.'
|
|
assert result[0][0] == 'equal', 'The result should contain an equal element.'
|
|
assert result[0][1] == 0, 'The start indices should be kept.'
|
|
assert result[0][2] == 22, 'The stop indices should be kept.'
|
|
assert result[0][3] == 0, 'The start indices should be kept.'
|
|
assert result[0][4] == 21, 'The stop indices should be kept.'
|
|
|
|
def remove_typos_beginning_negated_test(self):
|
|
"""
|
|
Test the _remove_typos function with a large difference in the middle.
|
|
"""
|
|
# GIVEN: A diffset with a large difference in the middle.
|
|
diff = [('equal', 0, 10, 0, 10), ('replace', 10, 20, 10, 11), ('equal', 20, 30, 11, 21)]
|
|
|
|
# WHEN: We remove the typos in there.
|
|
result = _remove_typos(list(diff))
|
|
|
|
# THEN: There diff should not have changed.
|
|
assert result == diff
|
|
|
|
def op_length_test(self):
|
|
"""
|
|
Test the _op_length function.
|
|
"""
|
|
# GIVEN: A diff entry.
|
|
diff_entry = ('replace', 0, 2, 4, 14)
|
|
|
|
# WHEN: We calculate the length of that diff.
|
|
result = _op_length(diff_entry)
|
|
|
|
# THEN: The maximum length should be returned.
|
|
assert result == 10, 'The length should be 10.'
|
|
|
|
def strip_rtf_charsets_test(self):
|
|
"""
|
|
Test that the strip_rtf() method properly decodes the supported charsets.
|
|
"""
|
|
test_charset_table = [
|
|
('0', 'weor\\\'F0-myndum \\\'FEah\\par ', 'weorð-myndum þah\n'),
|
|
('128', '\\\'83C\\\'83G\\\'83X\\\'A5\\\'83L\\\'83\\\'8A\\\'83X\\\'83g\\\'A1 '
|
|
'\\\\ \\\'95\\\\ \\\'8E\\} \\\'8E\\{ \\\'A1\\par ', 'イエス･キリスト｡ ¥ 表 枝 施 ｡\n'),
|
|
('129', '\\\'BF\\\'B9\\\'BC\\\'F6 \\\'B1\\\'D7\\\'B8\\\'AE\\\'BD\\\'BA\\\'B5\\\'B5\\par ', '예수 그리스도\n'),
|
|
('134', '\\\'D2\\\'AE\\\'F6\\\'D5\\\'BB\\\'F9\\\'B6\\\'BD\\\'CA\\\'C7\\\'D6\\\'F7\\par ', '耶稣基督是主\n'),
|
|
('161', '\\\'D7\\\'F1\\\'E9\\\'F3\\\'F4\\\'FC\\\'F2\\par ', 'Χριστός\n'),
|
|
('162', 'Hazreti \\\'DDsa\\par ', 'Hazreti İsa\n'),
|
|
('163', 'ph\\\'FD\\\'F5ng\\par ', 'phương\n'),
|
|
('177', '\\\'E1\\\'F8\\\'E0\\\'F9\\\'E9\\\'FA\\par ', 'בראשית\n'),
|
|
('178', '\\\'ED\\\'D3\\\'E6\\\'DA \\\'C7\\\'E1\\\'E3\\\'D3\\\'ED\\\'CD\\par ', 'يسوع المسيح\n'),
|
|
('186', 'J\\\'EBzus Kristus yra Vie\\\'F0pats\\par ', 'Jėzus Kristus yra Viešpats\n'),
|
|
('204', '\\\'D0\\\'EE\\\'F1\\\'F1\\\'E8\\\'FF\\par ', 'Россия\n'),
|
|
('222', '\\\'A4\\\'C3\\\'D4\\\'CA\\\'B5\\\'EC\\par ', 'คริสต์\n'),
|
|
('238', 'Z\\\'E1v\\\'ECre\\\'E8n\\\'E1 zkou\\\'9Aka\\par ', 'Závěrečná zkouška\n')
|
|
]
|
|
|
|
# GIVEN: For each character set and input
|
|
for charset, input, exp_result in test_charset_table:
|
|
|
|
# WHEN: We call strip_rtf on the input RTF
|
|
result, result_enc = strip_rtf(
|
|
'{\\rtf1 \\ansi \\ansicpg1252 {\\fonttbl \\f0 \\fswiss \\fcharset%s Helvetica;}' \
|
|
'{\\colortbl ;\\red0 \\green0 \\blue0 ;}\\pard \\f0 %s}' % (charset, input))
|
|
|
|
# THEN: The stripped text matches thed expected result
|
|
assert result == exp_result, 'The result should be %s' % exp_result
|
|
|
|
|
|
class TestVerseType(TestCase):
|
|
"""
|
|
This is a test case to test various methods in the VerseType enumeration class.
|
|
"""
|
|
|
|
def translated_tag_test(self):
|
|
"""
|
|
Test that the translated_tag() method returns the correct tags
|
|
"""
|
|
# GIVEN: A mocked out translate() function that just returns what it was given
|
|
with patch('openlp.plugins.songs.lib.translate') as mocked_translate:
|
|
mocked_translate.side_effect = lambda x, y: y
|
|
|
|
# WHEN: We run the translated_tag() method with a "verse"
|
|
result = VerseType.translated_tag('v')
|
|
|
|
# THEN: The result should be "V"
|
|
self.assertEqual(result, 'V', 'The result should be "V"')
|
|
|
|
# WHEN: We run the translated_tag() method with a "chorus"
|
|
result = VerseType.translated_tag('c')
|
|
|
|
# THEN: The result should be "C"
|
|
self.assertEqual(result, 'C', 'The result should be "C"')
|
|
|
|
def translated_invalid_tag_test(self):
|
|
"""
|
|
Test that the translated_tag() method returns the default tag when passed an invalid tag
|
|
"""
|
|
# GIVEN: A mocked out translate() function that just returns what it was given
|
|
with patch('openlp.plugins.songs.lib.translate') as mocked_translate:
|
|
mocked_translate.side_effect = lambda x, y: y
|
|
|
|
# WHEN: We run the translated_tag() method with an invalid verse type
|
|
result = VerseType.translated_tag('z')
|
|
|
|
# THEN: The result should be "O"
|
|
self.assertEqual(result, 'O', 'The result should be "O", but was "%s"' % result)
|
|
|
|
def translated_invalid_tag_with_specified_default_test(self):
|
|
"""
|
|
Test that the translated_tag() method returns the specified default tag when passed an invalid tag
|
|
"""
|
|
# GIVEN: A mocked out translate() function that just returns what it was given
|
|
with patch('openlp.plugins.songs.lib.translate') as mocked_translate:
|
|
mocked_translate.side_effect = lambda x, y: y
|
|
|
|
# WHEN: We run the translated_tag() method with an invalid verse type and specify a default
|
|
result = VerseType.translated_tag('q', VerseType.Bridge)
|
|
|
|
# THEN: The result should be "B"
|
|
self.assertEqual(result, 'B', 'The result should be "B", but was "%s"' % result)
|
|
|
|
def translated_invalid_tag_with_invalid_default_test(self):
|
|
"""
|
|
Test that the translated_tag() method returns a sane default tag when passed an invalid default
|
|
"""
|
|
# GIVEN: A mocked out translate() function that just returns what it was given
|
|
with patch('openlp.plugins.songs.lib.translate') as mocked_translate:
|
|
mocked_translate.side_effect = lambda x, y: y
|
|
|
|
# WHEN: We run the translated_tag() method with an invalid verse type and an invalid default
|
|
result = VerseType.translated_tag('q', 29)
|
|
|
|
# THEN: The result should be "O"
|
|
self.assertEqual(result, 'O', 'The result should be "O", but was "%s"' % result)
|
|
|
|
def translated_name_test(self):
|
|
"""
|
|
Test that the translated_name() method returns the correct name
|
|
"""
|
|
# GIVEN: A mocked out translate() function that just returns what it was given
|
|
with patch('openlp.plugins.songs.lib.translate') as mocked_translate:
|
|
mocked_translate.side_effect = lambda x, y: y
|
|
|
|
# WHEN: We run the translated_name() method with a "verse"
|
|
result = VerseType.translated_name('v')
|
|
|
|
# THEN: The result should be "Verse"
|
|
self.assertEqual(result, 'Verse', 'The result should be "Verse"')
|
|
|
|
# WHEN: We run the translated_name() method with a "chorus"
|
|
result = VerseType.translated_name('c')
|
|
|
|
# THEN: The result should be "Chorus"
|
|
self.assertEqual(result, 'Chorus', 'The result should be "Chorus"')
|
|
|
|
def translated_invalid_name_test(self):
|
|
"""
|
|
Test that the translated_name() method returns the default name when passed an invalid tag
|
|
"""
|
|
# GIVEN: A mocked out translate() function that just returns what it was given
|
|
with patch('openlp.plugins.songs.lib.translate') as mocked_translate:
|
|
mocked_translate.side_effect = lambda x, y: y
|
|
|
|
# WHEN: We run the translated_name() method with an invalid verse type
|
|
result = VerseType.translated_name('z')
|
|
|
|
# THEN: The result should be "Other"
|
|
self.assertEqual(result, 'Other', 'The result should be "Other", but was "%s"' % result)
|
|
|
|
def translated_invalid_name_with_specified_default_test(self):
|
|
"""
|
|
Test that the translated_name() method returns the specified default name when passed an invalid tag
|
|
"""
|
|
# GIVEN: A mocked out translate() function that just returns what it was given
|
|
with patch('openlp.plugins.songs.lib.translate') as mocked_translate:
|
|
mocked_translate.side_effect = lambda x, y: y
|
|
|
|
# WHEN: We run the translated_name() method with an invalid verse type and specify a default
|
|
result = VerseType.translated_name('q', VerseType.Bridge)
|
|
|
|
# THEN: The result should be "Bridge"
|
|
self.assertEqual(result, 'Bridge', 'The result should be "Bridge", but was "%s"' % result)
|
|
|
|
def translated_invalid_name_with_invalid_default_test(self):
|
|
"""
|
|
Test that the translated_name() method returns the specified default tag when passed an invalid tag
|
|
"""
|
|
# GIVEN: A mocked out translate() function that just returns what it was given
|
|
with patch('openlp.plugins.songs.lib.translate') as mocked_translate:
|
|
mocked_translate.side_effect = lambda x, y: y
|
|
|
|
# WHEN: We run the translated_name() method with an invalid verse type and specify an invalid default
|
|
result = VerseType.translated_name('q', 29)
|
|
|
|
# THEN: The result should be "Other"
|
|
self.assertEqual(result, 'Other', 'The result should be "Other", but was "%s"' % result)
|
|
|
|
def from_tag_test(self):
|
|
"""
|
|
Test that the from_tag() method returns the correct VerseType.
|
|
"""
|
|
# GIVEN: A mocked out translate() function that just returns what it was given
|
|
with patch('openlp.plugins.songs.lib.translate') as mocked_translate:
|
|
mocked_translate.side_effect = lambda x, y: y
|
|
|
|
# WHEN: We run the from_tag() method with a valid verse type, we get the name back
|
|
result = VerseType.from_tag('v')
|
|
|
|
# THEN: The result should be VerseType.Verse
|
|
self.assertEqual(result, VerseType.Verse, 'The result should be VerseType.Verse, but was "%s"' % result)
|
|
|
|
def from_tag_with_invalid_tag_test(self):
|
|
"""
|
|
Test that the from_tag() method returns the default VerseType when it is passed an invalid tag.
|
|
"""
|
|
# GIVEN: A mocked out translate() function that just returns what it was given
|
|
with patch('openlp.plugins.songs.lib.translate') as mocked_translate:
|
|
mocked_translate.side_effect = lambda x, y: y
|
|
|
|
# WHEN: We run the from_tag() method with a valid verse type, we get the name back
|
|
result = VerseType.from_tag('w')
|
|
|
|
# THEN: The result should be VerseType.Other
|
|
self.assertEqual(result, VerseType.Other, 'The result should be VerseType.Other, but was "%s"' % result)
|
|
|
|
def from_tag_with_specified_default_test(self):
|
|
"""
|
|
Test that the from_tag() method returns the specified default when passed an invalid tag.
|
|
"""
|
|
# GIVEN: A mocked out translate() function that just returns what it was given
|
|
with patch('openlp.plugins.songs.lib.translate') as mocked_translate:
|
|
mocked_translate.side_effect = lambda x, y: y
|
|
|
|
# WHEN: We run the from_tag() method with an invalid verse type, we get the specified default back
|
|
result = VerseType.from_tag('x', VerseType.Chorus)
|
|
|
|
# THEN: The result should be VerseType.Chorus
|
|
self.assertEqual(result, VerseType.Chorus, 'The result should be VerseType.Chorus, but was "%s"' % result)
|
|
|
|
def from_tag_with_invalid_default_test(self):
|
|
"""
|
|
Test that the from_tag() method returns a sane default when passed an invalid tag and an invalid default.
|
|
"""
|
|
# GIVEN: A mocked out translate() function that just returns what it was given
|
|
with patch('openlp.plugins.songs.lib.translate') as mocked_translate:
|
|
mocked_translate.side_effect = lambda x, y: y
|
|
|
|
# WHEN: We run the from_tag() method with an invalid verse type, we get the specified default back
|
|
result = VerseType.from_tag('m', 29)
|
|
|
|
# THEN: The result should be VerseType.Other
|
|
self.assertEqual(result, VerseType.Other, 'The result should be VerseType.Other, but was "%s"' % result)
|