@@ -1,11 +1,17 @@ import unittest from django.db import models -from fields import EncryptedCharField + +from fields import EncryptedCharField, PickleField + class EncObject(models.Model): password = EncryptedCharField(max_length=20) +class PickleObject(models.Model): + name = models.CharField(max_length=16) + data = PickleField() + class EncryptTests(unittest.TestCase): def testMaxFieldLength(self): password = 'this is a password!!' @@ -14,3 +20,35 @@ class EncryptTests(unittest.TestCase): obj = EncObject.objects.get(id=obj.id) self.assertEqual(password, obj.password) +class TestPickleField(unittest.TestCase): + def setUp(self): + PickleObject.objects.all().delete() + + def test_not_string_data(self): + items = [ + 'Item 1', 'Item 2', 'Item 3', 'Item 4', 'Item 5' + ] + + obj = PickleObject.objects.create(name='default', data=items) + self.assertEqual(PickleObject.objects.count(), 1) + + self.assertEqual(obj.data, items) + + obj = PickleObject.objects.get(name='default') + self.assertEqual(obj.data, items) + + def test_string_and_unicode_data(self): + DATA = ( + ('string', 'Simple string'), + ('unicode', u'Simple unicode string'), + ) + + for name, data in DATA: + obj = PickleObject.objects.create(name=name, data=data) + self.assertEqual(obj.data, data) + + self.assertEqual(PickleObject.objects.count(), 2) + + for name, data in DATA: + obj = PickleObject.objects.get(name=name) + self.assertEqual(obj.data, data) django_fields/tests.py a1ec5c9d8e12268f2848d37fab585c7c7664b319 ef7b37445c5652ae2bd40e237f31b238250d839d Alexander Artemenko svetlyak.40wt@gmail.com http://github.com/svetlyak40wt/django-fields/commit/b1d1953355db02b033b2b4972f40156e5e62e648 b1d1953355db02b033b2b4972f40156e5e62e648 2009-05-29T05:37:07-07:00 2009-05-29T05:37:07-07:00 Merge branch 'master' of git@github.com:svetlyak40wt/django-fields 95d66cca63d1682d3643bba6bdbd1c51db1178b3 Alexander Artemenko svetlyak.40wt@gmail.com