|
3 | 3 |
|
4 | 4 | from django_mongodb_backend.test import MongoTestCaseMixin |
5 | 5 |
|
6 | | -from .models import Author |
| 6 | +from .models import Author, Book |
7 | 7 |
|
8 | 8 |
|
9 | 9 | class SaveDollarPrefixTests(MongoTestCaseMixin, TestCase): |
@@ -45,6 +45,38 @@ def test_update_dollar_prefix_in_value_expression(self): |
45 | 45 | [{"$set": {"name": {"$literal": "$updated"}}}], |
46 | 46 | ) |
47 | 47 |
|
| 48 | + def test_update_dict_value(self): |
| 49 | + """MQL-like dict Value() expressions are correctly handled on update.""" |
| 50 | + obj = Book.objects.create(name="foobar", data={}) |
| 51 | + obj.data = Value({"$concat": ["$name", "-", "$name"]}) |
| 52 | + obj.save() |
| 53 | + obj.refresh_from_db() |
| 54 | + self.assertEqual(obj.data, {"$concat": ["$name", "-", "$name"]}) |
| 55 | + |
| 56 | + def test_update_dict(self): |
| 57 | + """MQL-like dict updates are correctly handled on update.""" |
| 58 | + obj = Book.objects.create(name="foobar") |
| 59 | + obj.data = {"$concat": ["$name", "-", "$name"]} |
| 60 | + obj.save() |
| 61 | + obj.refresh_from_db() |
| 62 | + self.assertEqual(obj.data, {"$concat": ["$name", "-", "$name"]}) |
| 63 | + |
| 64 | + def test_update_tuple(self): |
| 65 | + """MQL-like tuple updates are correctly handled on update.""" |
| 66 | + obj = Book.objects.create(name="foobar") |
| 67 | + obj.data = ("$name", "-", "$name") |
| 68 | + obj.save() |
| 69 | + obj.refresh_from_db() |
| 70 | + self.assertEqual(obj.data, ["$name", "-", "$name"]) |
| 71 | + |
| 72 | + def test_update_tuple_value(self): |
| 73 | + """MQL-like tuple Value() expressions are correctly handled on update.""" |
| 74 | + obj = Book.objects.create(name="foobar") |
| 75 | + obj.data = Value(("$name", "-", "$name")) |
| 76 | + obj.save() |
| 77 | + obj.refresh_from_db() |
| 78 | + self.assertEqual(obj.data, ["$name", "-", "$name"]) |
| 79 | + |
48 | 80 |
|
49 | 81 | class QueryDollarPrefixTests(MongoTestCaseMixin, TestCase): |
50 | 82 | def test_query_injection(self): |
|
0 commit comments