update tests for new error message after $facet removal

Author: timgraham

tests/encryption_/test_fields.py

@@ -4,6 +4,7 @@
 from operator import attrgetter
 
 from bson import ObjectId
+from django.conf import settings
 from django.db import DatabaseError
 from django.db.models import Avg, F, Q
 
@@ -245,9 +246,11 @@ def test_time(self):
 class QueryTests(EncryptionTestCase):
     def test_aggregate(self):
         msg = (
-            "Aggregation stage $internalFacetTeeConsumer is not allowed or "
-            "supported with automatic encryption."
+            'csfle "analyze_query" failed: Missing encryption schema for '
+            f"namespace: {settings.DATABASES['encrypted']['NAME']}.$cmd.aggregate"
         )
+        IntegerModel.objects.create(value=1)
+        IntegerModel.objects.create(value=3)
         with self.assertRaisesMessage(DatabaseError, msg):
             list(IntegerModel.objects.aggregate(Avg("value")))
 
@@ -291,12 +294,14 @@ def test_contains(self):
         self.assertIs(CharModel.objects.contains(obj), True)
 
     def test_count(self):
+        CharModel.objects.create(value="a")
+        CharModel.objects.create(value="b")
         msg = (
-            "Aggregation stage $internalFacetTeeConsumer is not allowed or "
-            "supported with automatic encryption."
+            'csfle "analyze_query" failed: Missing encryption schema for '
+            f"namespace: {settings.DATABASES['encrypted']['NAME']}.$cmd.aggregate"
         )
         with self.assertRaisesMessage(DatabaseError, msg):
-            list(CharModel.objects.count())
+            self.assertEqual(CharModel.objects.count(), 2)
 
     def test_dates(self):
         msg = (