michael@0: import unittest
michael@0: 
michael@0: import sys
michael@0: import os.path
michael@0: import mozunit
michael@0: 
michael@0: from printprereleasesuffix import get_prerelease_suffix
michael@0: 
michael@0: class TestGetPreReleaseSuffix(unittest.TestCase):
michael@0:   """
michael@0:   Unit tests for the get_prerelease_suffix function
michael@0:   """
michael@0: 
michael@0:   def test_alpha_1(self):
michael@0:     """test 1a1 version string"""
michael@0:     self.c = get_prerelease_suffix('1a1')
michael@0:     self.assertEqual(self.c, ' 1 Alpha 1')
michael@0: 
michael@0:   def test_alpha_10(self):
michael@0:     """test 1.2a10 version string"""
michael@0:     self.c = get_prerelease_suffix('1.2a10')
michael@0:     self.assertEqual(self.c, ' 1.2 Alpha 10')
michael@0: 
michael@0:   def test_beta_3(self):
michael@0:     """test 1.2.3b3 version string"""
michael@0:     self.c = get_prerelease_suffix('1.2.3b3')
michael@0:     self.assertEqual(self.c, ' 1.2.3 Beta 3')
michael@0: 
michael@0:   def test_beta_30(self):
michael@0:     """test 1.2.3.4b30 version string"""
michael@0:     self.c = get_prerelease_suffix('1.2.3.4b30')
michael@0:     self.assertEqual(self.c, ' 1.2.3.4 Beta 30')
michael@0: 
michael@0:   def test_release_1(self):
michael@0:     """test 1.2.3.4 version string"""
michael@0:     self.c = get_prerelease_suffix('1.2.3.4')
michael@0:     self.assertEqual(self.c, '')
michael@0: 
michael@0:   def test_alpha_1_pre(self):
michael@0:     """test 1.2a1pre version string"""
michael@0:     self.c = get_prerelease_suffix('1.2a1pre')
michael@0:     self.assertEqual(self.c, '')
michael@0: 
michael@0:   def test_beta_10_pre(self):
michael@0:     """test 3.4b10pre version string"""
michael@0:     self.c = get_prerelease_suffix('3.4b10pre')
michael@0:     self.assertEqual(self.c, '')
michael@0: 
michael@0:   def test_pre_0(self):
michael@0:     """test 1.2pre0 version string"""
michael@0:     self.c = get_prerelease_suffix('1.2pre0')
michael@0:     self.assertEqual(self.c, '')
michael@0: 
michael@0:   def test_pre_1_b(self):
michael@0:     """test 1.2pre1b version string"""
michael@0:     self.c = get_prerelease_suffix('1.2pre1b')
michael@0:     self.assertEqual(self.c, '')
michael@0: 
michael@0:   def test_a_a(self):
michael@0:     """test 1.2aa version string"""
michael@0:     self.c = get_prerelease_suffix('1.2aa')
michael@0:     self.assertEqual(self.c, '')
michael@0: 
michael@0:   def test_b_b(self):
michael@0:     """test 1.2bb version string"""
michael@0:     self.c = get_prerelease_suffix('1.2bb')
michael@0:     self.assertEqual(self.c, '')
michael@0: 
michael@0:   def test_a_b(self):
michael@0:     """test 1.2ab version string"""
michael@0:     self.c = get_prerelease_suffix('1.2ab')
michael@0:     self.assertEqual(self.c, '')
michael@0: 
michael@0:   def test_plus(self):
michael@0:     """test 1.2+ version string """
michael@0:     self.c = get_prerelease_suffix('1.2+')
michael@0:     self.assertEqual(self.c, '')
michael@0: 
michael@0: if __name__ == '__main__':
michael@0:   mozunit.main()