Move search test to atlas_search_ test app

Author: WaVEV

tests/atlas_search_/__init__.py

@@ -0,0 +1,25 @@
+from django.db import models
+
+from django_mongodb_backend.fields import (
+    ArrayField,
+    EmbeddedModelField,
+)
+from django_mongodb_backend.models import EmbeddedModel
+
+
+class Writer(EmbeddedModel):
+    name = models.CharField(max_length=10)
+
+
+class Location(EmbeddedModel):
+    type = models.CharField(default="Point", max_length=50)
+    coordinates = ArrayField(models.FloatField(), max_size=2)
+
+
+class Article(models.Model):
+    headline = models.CharField(max_length=100)
+    number = models.IntegerField()
+    body = models.TextField()
+    location = EmbeddedModelField(Location, null=True)
+    plot_embedding = ArrayField(models.FloatField(), size=3, null=True)
+    writer = EmbeddedModelField(Writer, null=True)

tests/atlas_search_/models.py

@@ -70,7 +70,7 @@ def wrapper(self, fetch, *args, **kwargs):
 
 @skipUnlessDBFeature("supports_atlas_search")
 class SearchUtilsMixin(TransactionTestCase):
-    available_apps = ["queries_"]
+    available_apps = None
 
     assertCountEqual = _delayed_assertion(timeout=2)(TransactionTestCase.assertCountEqual)
     assertListEqual = _delayed_assertion(timeout=2)(TransactionTestCase.assertListEqual)
@@ -93,7 +93,6 @@ def drop_index():
         cls.addClassCleanup(drop_index)
 
 
-@skipUnlessDBFeature("supports_atlas_search")
 class SearchEqualsTests(SearchUtilsMixin):
     @classmethod
     def setUpClass(cls):
@@ -155,7 +154,6 @@ def test_function_score(self):
         self.assertAlmostEqual(scored.score, 1.0, places=2)
 
 
-@skipUnlessDBFeature("supports_atlas_search")
 class SearchAutocompleteTests(SearchUtilsMixin):
     @classmethod
     def setUpClass(cls):
@@ -235,7 +233,6 @@ def test_constant_score(self):
         self.assertAlmostEqual(scored.score, 10.0, places=2)
 
 
-@skipUnlessDBFeature("supports_atlas_search")
 class SearchExistsTests(SearchUtilsMixin):
     @classmethod
     def setUpClass(cls):
@@ -261,7 +258,6 @@ def test_constant_score(self):
         self.assertAlmostEqual(scored.score, 10.0, places=2)
 
 
-@skipUnlessDBFeature("supports_atlas_search")
 class SearchInTests(SearchUtilsMixin):
     @classmethod
     def setUpClass(cls):
@@ -290,7 +286,6 @@ def test_constant_score(self):
         self.assertAlmostEqual(scored.score, 10.0, places=2)
 
 
-@skipUnlessDBFeature("supports_atlas_search")
 class SearchPhraseTests(SearchUtilsMixin):
     @classmethod
     def setUpClass(cls):
@@ -321,7 +316,6 @@ def test_constant_score(self):
         self.assertAlmostEqual(scored.score, 10.0, places=2)
 
 
-@skipUnlessDBFeature("supports_atlas_search")
 class SearchRangeTests(SearchUtilsMixin):
     @classmethod
     def setUpClass(cls):
@@ -350,7 +344,6 @@ def test_constant_score(self):
         self.assertAlmostEqual(scored.score, 10.0, places=2)
 
 
-@skipUnlessDBFeature("supports_atlas_search")
 class SearchRegexTests(SearchUtilsMixin):
     @classmethod
     def setUpClass(cls):
@@ -388,7 +381,6 @@ def test_constant_score(self):
         self.assertAlmostEqual(scored.score, 10.0, places=2)
 
 
-@skipUnlessDBFeature("supports_atlas_search")
 class SearchTextTests(SearchUtilsMixin):
     @classmethod
     def setUpClass(cls):
@@ -437,7 +429,6 @@ def test_constant_score(self):
         self.assertAlmostEqual(scored.score, 10.0, places=2)
 
 
-@skipUnlessDBFeature("supports_atlas_search")
 class SearchWildcardTests(SearchUtilsMixin):
     @classmethod
     def setUpClass(cls):
@@ -471,7 +462,6 @@ def test_constant_score(self):
         self.assertAlmostEqual(scored.score, 10.0, places=2)
 
 
-@skipUnlessDBFeature("supports_atlas_search")
 class SearchGeoShapeTests(SearchUtilsMixin):
     @classmethod
     def setUpClass(cls):
@@ -524,7 +514,6 @@ def test_constant_score(self):
         self.assertAlmostEqual(scored.score, 10.0, places=2)
 
 
-@skipUnlessDBFeature("supports_atlas_search")
 class SearchGeoWithinTests(SearchUtilsMixin):
     @classmethod
     def setUpClass(cls):
@@ -579,7 +568,6 @@ def test_constant_score(self):
         self.assertAlmostEqual(scored.score, 10.0, places=2)
 
 
-@skipUnlessDBFeature("supports_atlas_search")
 @unittest.expectedFailure
 class SearchMoreLikeThisTests(SearchUtilsMixin):
     @classmethod
@@ -620,7 +608,6 @@ def test_search_more_like_this(self):
         self.assertQuerySetEqual(qs.all, [self.article1, self.article2], lambda a: a.headline)
 
 
-@skipUnlessDBFeature("supports_atlas_search")
 class CompoundSearchTests(SearchUtilsMixin):
     @classmethod
     def setUpClass(cls):
@@ -784,7 +771,6 @@ def test_search_and_filter(self):
         self.assertCountEqual(qs.all, [self.icy_moons])
 
 
-@skipUnlessDBFeature("supports_atlas_search")
 class SearchVectorTests(SearchUtilsMixin):
     @classmethod
     def setUpClass(cls):

tests/queries_/test_search.py renamed to tests/atlas_search_/test_search.py

@@ -1,12 +1,9 @@
 from django.db import models
 
 from django_mongodb_backend.fields import (
-    ArrayField,
-    EmbeddedModelField,
     ObjectIdAutoField,
     ObjectIdField,
 )
-from django_mongodb_backend.models import EmbeddedModel
 
 
 class Author(models.Model):
@@ -59,21 +56,3 @@ class Meta:
 
     def __str__(self):
         return str(self.pk)
-
-
-class Writer(EmbeddedModel):
-    name = models.CharField(max_length=10)
-
-
-class Location(EmbeddedModel):
-    type = models.CharField(default="Point", max_length=50)
-    coordinates = ArrayField(models.FloatField(), max_size=2)
-
-
-class Article(models.Model):
-    headline = models.CharField(max_length=100)
-    number = models.IntegerField()
-    body = models.TextField()
-    location = EmbeddedModelField(Location, null=True)
-    plot_embedding = ArrayField(models.FloatField(), size=3, null=True)
-    writer = EmbeddedModelField(Writer, null=True)