1 files changed, 79 insertions, 0 deletions
diff --git a/config/tests/unit-printprereleasesuffix.py b/config/tests/unit-printprereleasesuffix.py
new file mode 100644
index 0000000000..cebf48883c
--- /dev/null
+++ b/config/tests/unit-printprereleasesuffix.py
@@ -0,0 +1,79 @@
+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()