diff options
author | Raphaël Barrois <raphael.barrois@polyconseil.fr> | 2012-05-22 15:59:09 +0200 |
---|---|---|
committer | Raphaël Barrois <raphael.barrois@polyconseil.fr> | 2012-05-22 15:59:09 +0200 |
commit | 966849ba373c14b12effbda624a4982d9813e434 (patch) | |
tree | e380415d3cdfa548ca69808792f8ed62d5017d79 | |
parent | 1a3f2698a0ccde1cc1825186e49563d4dc9dd72c (diff) | |
download | semantic-version-966849ba373c14b12effbda624a4982d9813e434.tar.gz |
Add introspection rules for south
Signed-off-by: Raphaël Barrois <raphael.barrois@polyconseil.fr>
-rw-r--r-- | src/semantic_version/django_fields.py | 21 | ||||
-rw-r--r-- | tests/test_django.py | 51 |
2 files changed, 72 insertions, 0 deletions
diff --git a/src/semantic_version/django_fields.py b/src/semantic_version/django_fields.py index c6ef688..0e1b083 100644 --- a/src/semantic_version/django_fields.py +++ b/src/semantic_version/django_fields.py @@ -59,3 +59,24 @@ class SpecField(BaseSemVerField): if isinstance(value, base.Spec): return value return base.Spec(value) + + +def add_south_rules(): + from south.modelsinspector import add_introspection_rules + + add_introspection_rules([ + ( + (VersionField,), + [], + {'partial': ('partial', {'default': False})}, + ), + ], ["semantic_version\.django_fields"]) + + +try: # pragma: no cover + import south +except ImportError: # pragma: no cover + south = None + +if south: # pragma: no cover + add_south_rules() diff --git a/tests/test_django.py b/tests/test_django.py index b4a3c29..52d419c 100644 --- a/tests/test_django.py +++ b/tests/test_django.py @@ -31,6 +31,13 @@ if django_loaded: # pragma: no cover from django_test_app import models from django.core import serializers +try: # pragma: no cover + import south + import south.creator.freezer + import south.modelsinspector +except ImportError: # pragma: no cover + south = None + @unittest.skipIf(not django_loaded, "Django not installed") class DjangoFieldTestCase(unittest.TestCase): @@ -98,6 +105,50 @@ class DjangoFieldTestCase(unittest.TestCase): self.assertEqual(o2, obj2.object) +@unittest.skipIf(not django_loaded or south is None, "Couldn't import south and django") +class SouthTestCase(unittest.TestCase): + def test_freezing_version_model(self): + frozen = south.modelsinspector.get_model_fields(models.VersionModel) + + self.assertEqual(frozen['version'], + ('semantic_version.django_fields.VersionField', [], {'max_length': '200'})) + + self.assertEqual(frozen['spec'], + ('semantic_version.django_fields.SpecField', [], {'max_length': '200'})) + + def test_freezing_partial_version_model(self): + frozen = south.modelsinspector.get_model_fields(models.PartialVersionModel) + + self.assertEqual(frozen['partial'], + ('semantic_version.django_fields.VersionField', [], {'max_length': '200', 'partial': 'True'})) + + self.assertEqual(frozen['optional'], + ('semantic_version.django_fields.VersionField', [], {'max_length': '200', 'blank': 'True', 'null': 'True'})) + + self.assertEqual(frozen['optional_spec'], + ('semantic_version.django_fields.SpecField', [], {'max_length': '200', 'blank': 'True', 'null': 'True'})) + + def test_freezing_app(self): + frozen = south.creator.freezer.freeze_apps('django_test_app') + + # Test VersionModel + self.assertEqual(frozen['django_test_app.versionmodel']['version'], + ('semantic_version.django_fields.VersionField', [], {'max_length': '200'})) + + self.assertEqual(frozen['django_test_app.versionmodel']['spec'], + ('semantic_version.django_fields.SpecField', [], {'max_length': '200'})) + + # Test PartialVersionModel + self.assertEqual(frozen['django_test_app.partialversionmodel']['partial'], + ('semantic_version.django_fields.VersionField', [], {'max_length': '200', 'partial': 'True'})) + + self.assertEqual(frozen['django_test_app.partialversionmodel']['optional'], + ('semantic_version.django_fields.VersionField', [], {'max_length': '200', 'blank': 'True', 'null': 'True'})) + + self.assertEqual(frozen['django_test_app.partialversionmodel']['optional_spec'], + ('semantic_version.django_fields.SpecField', [], {'max_length': '200', 'blank': 'True', 'null': 'True'})) + + if django_loaded: from django.test import TestCase from django.test.simple import DjangoTestSuiteRunner |