fix inconsistencies in ._changed_fields computation

This commit is contained in:
Bastien Gérard
2020-10-07 00:01:09 +02:00
parent 1f2a5db016
commit aabc18755c
3 changed files with 72 additions and 8 deletions

View File

@@ -537,6 +537,7 @@ class TestDelta(MongoDBTestCase):
{},
)
doc.save()
assert doc._get_changed_fields() == []
doc = doc.reload(10)
assert doc.embedded_field.list_field[0] == "1"
@@ -767,9 +768,7 @@ class TestDelta(MongoDBTestCase):
MyDoc.drop_collection()
mydoc = MyDoc(
name="testcase1", subs={"a": {"b": EmbeddedDoc(name="foo")}}
).save()
MyDoc(name="testcase1", subs={"a": {"b": EmbeddedDoc(name="foo")}}).save()
mydoc = MyDoc.objects.first()
subdoc = mydoc.subs["a"]["b"]
@@ -781,6 +780,35 @@ class TestDelta(MongoDBTestCase):
mydoc._clear_changed_fields()
assert [] == mydoc._get_changed_fields()
def test_nested_nested_fields_db_field_set__gets_mark_as_changed_and_cleaned(self):
class EmbeddedDoc(EmbeddedDocument):
name = StringField(db_field="db_name")
class MyDoc(Document):
embed = EmbeddedDocumentField(EmbeddedDoc, db_field="db_embed")
name = StringField(db_field="db_name")
MyDoc.drop_collection()
MyDoc(name="testcase1", embed=EmbeddedDoc(name="foo")).save()
mydoc = MyDoc.objects.first()
mydoc.embed.name = "foo1"
assert mydoc.embed._get_changed_fields() == ["db_name"]
assert mydoc._get_changed_fields() == ["db_embed.db_name"]
mydoc = MyDoc.objects.first()
embed = EmbeddedDoc(name="foo2")
embed.name = "bar"
mydoc.embed = embed
assert embed._get_changed_fields() == ["db_name"]
assert mydoc._get_changed_fields() == ["db_embed"]
mydoc._clear_changed_fields()
assert mydoc._get_changed_fields() == []
def test_lower_level_mark_as_changed(self):
class EmbeddedDoc(EmbeddedDocument):
name = StringField()

View File

@@ -370,8 +370,7 @@ class FieldTest(unittest.TestCase):
assert Post.objects.all()[0].user_lists == [[u1, u2], [u3]]
def test_circular_reference(self):
"""Ensure you can handle circular references
"""
"""Ensure you can handle circular references"""
class Relation(EmbeddedDocument):
name = StringField()
@@ -426,6 +425,7 @@ class FieldTest(unittest.TestCase):
daughter.relations.append(mother)
daughter.relations.append(daughter)
assert daughter._get_changed_fields() == ["relations"]
daughter.save()
assert "[<Person: Mother>, <Person: Daughter>]" == "%s" % Person.objects()