#21: Test FilterFactory

ral-facilities · Sep 27, 2019 · edad675 · edad675
1 parent 1fdf883
commit edad675
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/test/test_database_helpers.py b/test/test_database_helpers.py
@@ -0,0 +1,30 @@
+from unittest import TestCase
+
+from common.database_helpers import QueryFilterFactory, OrderFilter, LimitFilter, SkipFilter, WhereFilter, \
+    IncludeFilter, DistinctFieldFilter
+
+
+class TestQueryFilterFactory(TestCase):
+    def test_order_filter(self):
+        self.assertIs(OrderFilter, type(QueryFilterFactory.get_query_filter({"order": "ID DESC"})))
+
+    def test_limit_filter(self):
+        self.assertIs(LimitFilter, type(QueryFilterFactory.get_query_filter({"limit": 10})))
+
+    def test_skip_filter(self):
+        self.assertIs(SkipFilter, type(QueryFilterFactory.get_query_filter({"skip": 10})))
+
+    def test_where_filter(self):
+        self.assertIs(WhereFilter, type(QueryFilterFactory.get_query_filter({"where": {"ID": {"eq": "1"}}})))
+        self.assertIs(WhereFilter, type(QueryFilterFactory.get_query_filter({"where": {"ID": {"lte": "1"}}})))
+        self.assertIs(WhereFilter, type(QueryFilterFactory.get_query_filter({"where": {"ID": {"gte": "1"}}})))
+        self.assertIs(WhereFilter, type(QueryFilterFactory.get_query_filter({"where": {"ID": {"like": "3"}}})))
+
+    def test_include_filter(self):
+        self.assertIs(IncludeFilter, type(QueryFilterFactory.get_query_filter({"include": "DATAFILE"})))
+        self.assertIs(IncludeFilter, type(QueryFilterFactory.get_query_filter({"include": ["TEST"]})))
+        self.assertIs(IncludeFilter,
+                      type(QueryFilterFactory.get_query_filter({"include": {"Test": ["TEST1", "Test2"]}})))
+
+    def test_distinct_filter(self):
+        self.assertIs(DistinctFieldFilter, type(QueryFilterFactory.get_query_filter({"distinct": "TEST"})))