Fixes pickling issue with choice fields

Removes the dynamic __get_field_display partials before pickling
This commit is contained in:
Ross Lawley
2011-06-16 12:50:45 +01:00
parent 22a7ee5885
commit cae3f3eeff
2 changed files with 51 additions and 27 deletions

View File

@@ -15,7 +15,7 @@ class PickleEmbedded(EmbeddedDocument):
class PickleTest(Document):
number = IntField()
string = StringField()
string = StringField(choices=(('One', '1'), ('Two', '2')))
embedded = EmbeddedDocumentField(PickleEmbedded)
lists = ListField(StringField())
@@ -1516,7 +1516,7 @@ class DocumentTest(unittest.TestCase):
def test_picklable(self):
pickle_doc = PickleTest(number=1, string="OH HAI", lists=['1', '2'])
pickle_doc = PickleTest(number=1, string="One", lists=['1', '2'])
pickle_doc.embedded = PickleEmbedded()
pickle_doc.save()
@@ -1525,7 +1525,7 @@ class DocumentTest(unittest.TestCase):
self.assertEquals(resurrected, pickle_doc)
resurrected.string = "Working"
resurrected.string = "Two"
resurrected.save()
pickle_doc.reload()