Added tests to handle circular aliases

Materials-Consortia · Mar 25, 2020 · 51cf0cd · 51cf0cd
1 parent d2c8976
commit 51cf0cd
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/tests/filtertransformers/test_mongo.py b/tests/filtertransformers/test_mongo.py
@@ -413,7 +413,10 @@ def test_unaliased_length_operator(self):
     def test_aliased_length_operator(self):
         from optimade.server.mappers import StructureMapper
 
-        transformer = MongoTransformer(mapper=StructureMapper())
+        class MyMapper(StructureMapper):
+            ALIASES = (("elements", "my_elements"), ("nelements", "nelem"))
+
+        transformer = MongoTransformer(mapper=MyMapper())
         parser = LarkParser(version=self.version, variant=self.variant)
 
         self.assertEqual(
@@ -428,6 +431,10 @@ def test_aliased_length_operator(self):
             transformer.transform(parser.parse("cartesian_site_positions LENGTH 3")),
             {"nsites": 3},
         )
+        self.assertEqual(
+            transformer.transform(parser.parse("cartesian_site_positions LENGTH 3")),
+            {"nsites": 3},
+        )
         self.assertEqual(
             transformer.transform(
                 parser.parse("cartesian_site_positions LENGTH >= 10")
@@ -445,6 +452,10 @@ def test_aliased_length_operator(self):
             {"nsites.11": {"$exists": True}},
         )
 
+        self.assertEqual(
+            transformer.transform(parser.parse("elements LENGTH 3")), {"nelem": 3},
+        )
+
     def test_aliases(self):
         """ Test that valid aliases are allowed, but do not effect
         r-values.