more query tests

Author: timgraham

.github/workflows/test-python-atlas.yml

@@ -54,8 +54,8 @@ jobs:
         run: bash .github/workflows/start_local_atlas.sh mongodb/mongodb-atlas-local:8.0.15
       - name: Download crypt shared
         run: |
-          wget https://downloads.mongodb.com/linux/mongo_crypt_shared_v1-linux-x86_64-enterprise-ubuntu2404-8.0.15.tgz
-          tar -xvzf mongo_crypt_shared_v1-linux-x86_64-enterprise-ubuntu2404-8.0.15.tgz lib/mongo_crypt_v1.so
+          wget https://downloads.mongodb.com/linux/mongo_crypt_shared_v1-linux-x86_64-enterprise-ubuntu2404-8.2.1.tgz
+          tar -xvzf mongo_crypt_shared_v1-linux-x86_64-enterprise-ubuntu2404-8.2.1.tgz lib/mongo_crypt_v1.so
           ls -d "$PWD"/lib/mongo_crypt_v1.so
       - name: Run tests
         run: python3 django_repo/tests/runtests_.py

tests/encryption_/test_fields.py

@@ -313,27 +313,36 @@ def test_get_or_create(self):
         self.assertEqual(obj1, obj2)
 
     def test_join(self):
-        list(Book.objects.filter(author__name="xxx"))
+        msg = (
+            "Non-empty 'let' field is not allowed in the $lookup aggregation "
+            "stage over an encrypted collection."
+        )
+        with self.assertRaisesMessage(DatabaseError, msg):
+            list(Book.objects.filter(author__name="xxx"))
 
     def test_order_by(self):
         msg = "Cannot add an encrypted field as a prefix of another encrypted field"
         with self.assertRaisesMessage(DatabaseError, msg):
             list(CharModel.objects.order_by("value"))
 
     def test_select_related(self):
-        list(Book.objects.select_related("author"))
+        msg = (
+            "Non-empty 'let' field is not allowed in the $lookup aggregation "
+            "stage over an encrypted collection."
+        )
+        with self.assertRaisesMessage(DatabaseError, msg):
+            list(Book.objects.select_related("author"))
 
     def test_update(self):
         msg = "Multi-document updates are not allowed with Queryable Encryption"
         with self.assertRaisesMessage(DatabaseError, msg):
             self.assertEqual(CharModel.objects.update(value="xyz"), 1)
 
     def test_update_or_create(self):
-        # TODO
-        # msg = "Multi-document updates are not allowed with Queryable Encryption"
-        # with self.assertRaisesMessage(DatabaseError, msg):
-        # obj, created = CharModel.objects.update_or_create(value="xyz"), 1)
-        pass
+        CharModel.objects.create(value="xyz")
+        msg = "Multi-document updates are not allowed with Queryable Encryption"
+        with self.assertRaisesMessage(DatabaseError, msg):
+            CharModel.objects.update_or_create(value="xyz", defaults={"plain": "abc"})
 
     def test_union(self):
         msg = "Aggregation stage $unionWith is not allowed or supported with automatic encryption."