This commit is contained in:
Samuel Mehrbrodt 2014-12-24 14:21:17 +01:00
parent f13a9d00b5
commit c5d716f565
1 changed files with 16 additions and 2 deletions

View File

@ -158,9 +158,9 @@ class TestDB(TestCase):
# THEN: It should return only the name
self.assertEqual("John Doe", display_name)
def test_author_get_display_name_with_type(self):
def test_author_get_display_name_with_type_words(self):
"""
Test that the display name of an author with a type is correct
Test that the display name of an author with a type is correct (Words)
"""
# GIVEN: An author
author = Author()
@ -172,6 +172,20 @@ class TestDB(TestCase):
# THEN: It should return the name with the type in brackets
self.assertEqual("John Doe (Words)", display_name)
def test_author_get_display_name_with_type_translation(self):
"""
Test that the display name of an author with a type is correct (Translation)
"""
# GIVEN: An author
author = Author()
author.display_name = "John Doe"
# WHEN: We call the get_display_name() function
display_name = author.get_display_name(AuthorType.Translation)
# THEN: It should return the name with the type in brackets
self.assertEqual("John Doe (Translation)", display_name)
def test_upgrade_old_song_db(self):
"""
Test that we can upgrade an old song db to the current schema