Commits

James Bennett  committed e80c4ec

Unit tests for version-retrieval, plus a bugfix they uncovered.

  • Participants
  • Parent commits 04bc7d4

Comments (0)

Files changed (2)

File registration/__init__.py

         version = '%s pre-alpha' % version
     else:
         if VERSION[3] != 'final':
-            version = '%s %s %s' % (version, VERSION[3], VERSION[4])
+            version = "%s %s" % (version, VERSION[3])
+            if VERSION[4] != 0:
+                version = '%s %s' % (version, VERSION[4])
     return version

File registration/tests/__init__.py

+from django.test import TestCase
+
+import registration
+
 from registration.tests.backends import *
 from registration.tests.forms import *
 from registration.tests.models import *
 from registration.tests.views import *
+
+
+class RegistrationInfrastructureTests(TestCase):
+    """
+    Test django-registration's internal infrastructure, including
+    version information and backend retrieval.
+    
+    """
+    def setUp(self):
+        self.version = registration.VERSION
+
+    def tearDown(self):
+        registration.VERSION = self.version
+    
+    def test_get_version(self):
+        """
+        Test the version-info reporting.
+        
+        """
+        versions = [
+            {'version': (1, 0, 0, 'alpha', 0),
+             'expected': "1.0 pre-alpha"},
+            {'version': (1, 0, 1, 'alpha', 1),
+             'expected': "1.0.1 alpha 1"},
+            {'version': (1, 1, 0, 'beta', 2),
+             'expected': "1.1 beta 2"},
+            {'version': (1, 2, 1, 'rc', 3),
+             'expected': "1.2.1 rc 3"},
+            {'version': (1, 3, 0, 'final', 0),
+             'expected': "1.3"},
+            {'version': (1, 4, 1, 'beta', 0),
+             'expected': "1.4.1 beta"},
+            ]
+        
+        for version_dict in versions:
+            registration.VERSION = version_dict['version']
+            self.assertEqual(registration.get_version(), version_dict['expected'])