#223: Add test cases for the bug fixed in previous commit

ral-facilities · May 17, 2021 · f16cb15 · f16cb15
1 parent 9021fc8
commit f16cb15
Showing 1 changed file with 29 additions and 2 deletions.
diff --git a/test/db/endpoints/test_get_with_filters.py b/test/db/endpoints/test_get_with_filters.py
@@ -54,6 +54,19 @@ def test_valid_get_with_filters_multiple_distinct(
                 [{"title": "Title for DataGateway API Testing (DB) 0"}],
                 id="Single unrelated distinct field",
             ),
+            pytest.param(
+                '"investigationInstruments.createTime"',
+                [
+                    {
+                        "investigationInstruments": {
+                            "createTime": DateHandler.datetime_object_to_str(
+                                Constants.TEST_MOD_CREATE_DATETIME,
+                            ),
+                        },
+                    },
+                ],
+                id="Single related distinct field",
+            ),
             pytest.param(
                 '["createTime", "investigationInstruments.createTime"]',
                 [
@@ -68,7 +81,21 @@ def test_valid_get_with_filters_multiple_distinct(
                         },
                     },
                 ],
-                id="List containing related distinct field",
+                id="Single related distinct field with unrelated field",
+            ),
+            pytest.param(
+                '["investigationInstruments.createTime", "facility.id"]',
+                [
+                    {
+                        "facility": {"id": 1},
+                        "investigationInstruments": {
+                            "createTime": DateHandler.datetime_object_to_str(
+                                Constants.TEST_MOD_CREATE_DATETIME,
+                            ),
+                        },
+                    },
+                ],
+                id="Multiple related distinct fields",
             ),
             pytest.param(
                 '["createTime", "investigationInstruments.createTime", "facility.id"]',
@@ -85,7 +112,7 @@ def test_valid_get_with_filters_multiple_distinct(
                         },
                     },
                 ],
-                id="Multiple related distinct fields",
+                id="Multiple related distinct fields with unrelated field",
             ),
         ],
     )