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