From b0d047ded1a9e80ca66a44c11b178cf20cd99974 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Frederik=20=E2=80=9CFreso=E2=80=9D=20S=2E=20Olesen?= Date: Sat, 29 Apr 2017 15:10:08 +0200 Subject: [PATCH] Add unittest for new Program.addDisambiguation() method --- whipper/test/test_common_program.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/whipper/test/test_common_program.py b/whipper/test/test_common_program.py index d31bd67..9c33c86 100644 --- a/whipper/test/test_common_program.py +++ b/whipper/test/test_common_program.py @@ -179,3 +179,25 @@ class PathTestCase(unittest.TestCase): for template, expected_path in templates.iteritems(): path = prog.getPath(u'/tmp', template, 'mbdiscid', 0, disambiguate=True) self.assertEquals(path, u'/tmp/' + expected_path) + + def testAddDisambiguationUnitTest(self): + """Unit test for Program.addDisambiguation().""" + prog = program.Program(config.Config()) + md = mbngs.DiscMetadata() + + # No relevant disambiguation metadata + self.assertEquals( + prog.addDisambiguation(u'Test', md), + u'Test') + + # Only barcode available + md.barcode = '033651008927' + self.assertEquals( + prog.addDisambiguation(u'Test', md), + u'Test (033651008927)') + + # Both catalog number and barcode available + md.catalogNumber = 'RHR CD 89' + self.assertEquals( + prog.addDisambiguation(u'Test', md), + u'Test (RHR CD 89)')