WIP

Author: NoahStapp

tests/performance/perftest/models.py

@@ -301,7 +301,7 @@ class LargeFlatModel(models.Model):
 
 
 class StringEmbeddedModel(EmbeddedModel):
-    unique_id = ObjectIdField()
+    unique_field = models.CharField(max_length=100)
     field1 = models.CharField(max_length=100)
     field2 = models.CharField(max_length=100)
     field3 = models.CharField(max_length=100)
@@ -320,7 +320,6 @@ class StringEmbeddedModel(EmbeddedModel):
 
 
 class IntegerEmbeddedModel(EmbeddedModel):
-    unique_id = ObjectIdField()
     field1 = models.IntegerField()
     field2 = models.IntegerField()
     field3 = models.IntegerField()

tests/performance/perftest/settings.py

@@ -68,7 +68,7 @@
 # }
 
 settings = django_mongodb_backend.parse_uri(
-    f"mongodb+srv://{USER}:{PASSWORD}@intpython736.oztdp.mongodb-dev.net/benchmarking?retryWrites=true&w=majority&appName=INTPYTHON736"
+    f"mongodb://localhost:27017/benchmarking",
 )
 DATABASES = {
     "default": settings,

tests/performance/perftest/tests.py

@@ -308,16 +308,15 @@ def create_model(self):
                     array_models = []
                     for item in v:
                         embedded_str_model = StringEmbeddedModel(**item)
-                        embedded_str_model.unique_id = ObjectId()
+                        embedded_str_model.unique_field = str(ObjectId())
                         array_models.append(embedded_str_model)
                     setattr(model, k, array_models)
                 elif "str" in k:
                     embedded_str_model = StringEmbeddedModel(**v)
-                    embedded_str_model.unique_id = ObjectId()
+                    embedded_str_model.unique_field = str(ObjectId())
                     setattr(model, k, embedded_str_model)
                 else:
                     embedded_int_model = IntegerEmbeddedModel(**v)
-                    embedded_int_model.unique_id = ObjectId()
                     setattr(model, k, embedded_int_model)
             model.save()
 
@@ -353,12 +352,12 @@ def setUp(self):
         super().setUp()
         self.create_model()
         self.ids = [
-            model.embedded_str_doc_1.unique_id for model in list(LargeNestedModel.objects.all())
+            model.embedded_str_doc_1.unique_field for model in list(LargeNestedModel.objects.all())
         ]
 
     def do_task(self):
         for _id in self.ids:
-            list(LargeNestedModel.objects.filter(embedded_str_doc_1__unique_id=_id))
+            list(LargeNestedModel.objects.filter(embedded_str_doc_1__id=_id))
 
     def tearDown(self):
         super().tearDown()
@@ -370,13 +369,13 @@ def setUp(self):
         super().setUp()
         self.create_model()
         self.ids = [
-            model.embedded_str_doc_array[0].unique_id
+            model.embedded_str_doc_array[0].id
             for model in list(LargeNestedModel.objects.all())
         ]
 
     def do_task(self):
         for _id in self.ids:
-            list(LargeNestedModel.objects.filter(embedded_str_doc_array__unique_id__in=[_id]))
+            list(LargeNestedModel.objects.filter(embedded_str_doc_array__id__in=[_id]))
 
     def tearDown(self):
         super().tearDown()
@@ -429,16 +428,10 @@ def tearDown(self):
 class TestLargeFlatDocFilterPkByIn(LargeFlatDocTest, unittest.TestCase):
     def setUp(self):
         super().setUp()
-        # LargeFlatModel.objects.all().delete()
-        # self.ids = []
-        # models = []
-        # # self.fields = [str(ObjectId()) for _ in range(NUM_DOCS // 10)]
-        # for doc in self.documents:
-        #     models.append(LargeFlatModel(**doc))
-        # # for model in models:
-        # #     model.field1 = self.fields[random.randint(0, len(self.fields) - 1)]
-        #
-        # LargeFlatModel.objects.bulk_create(models)
+        models = []
+        for doc in self.documents:
+            models.append(LargeFlatModel(**doc))
+        LargeFlatModel.objects.bulk_create(models)
         self.id = LargeFlatModel.objects.first().id
         print(LargeFlatModel.objects.filter(id__in=[self.id]).explain())
 
@@ -447,4 +440,4 @@ def do_task(self):
 
     def tearDown(self):
         super().tearDown()
-        # LargeFlatModel.objects.all().delete()
+        LargeFlatModel.objects.all().delete()