From 344bc9d502469acc5597355d1b832e68b5893e11 Mon Sep 17 00:00:00 2001
From: David Vogt <david.vogt@adfinis-sygroup.ch>
Date: Tue, 15 Oct 2019 14:36:39 +0200
Subject: [PATCH] feat(filters): add test for new order filters

---
 caluma/core/tests/test_ordersets.py | 72 +++++++++++++++++++++++++++++
 1 file changed, 72 insertions(+)
 create mode 100644 caluma/core/tests/test_ordersets.py

diff --git a/caluma/core/tests/test_ordersets.py b/caluma/core/tests/test_ordersets.py
new file mode 100644
index 000000000..406bf11c4
--- /dev/null
+++ b/caluma/core/tests/test_ordersets.py
@@ -0,0 +1,72 @@
+import pytest
+
+from ..relay import extract_global_id
+
+
+@pytest.mark.parametrize("question__type", ["text"])
+def test_inversible_with_vars(db, schema_executor, document_factory, question):
+
+    doc_a, doc_b, doc_c = (
+        document_factory(meta={"foo": "a", "blah": "xxx"}),
+        document_factory(meta={"foo": "b", "blah": "xxx"}),
+        document_factory(meta={"foo": "c", "blah": "blub"}),
+    )
+
+    doc_a.answers.create(question=question, value="44")
+    doc_b.answers.create(question=question, value="33")
+    doc_c.answers.create(question=question, value="22")
+
+    query = """
+        query test_ordering($order: [DocumentOrderSetType]) {
+          allDocuments(order: $order) {
+            edges {
+              node {
+                id
+              }
+            }
+          }
+        }
+    """
+
+    doc_a_id = str(doc_a.id)
+    doc_b_id = str(doc_b.id)
+    doc_c_id = str(doc_c.id)
+
+    def result_ids(result):
+        return [
+            extract_global_id(n["node"]["id"])
+            for n in result.data["allDocuments"]["edges"]
+        ]
+
+    def get_ordered(ordering):
+        result = schema_executor(query, variables={"order": ordering})
+        assert not result.errors
+        return result_ids(result)
+
+    # ordering by answer value, forward.
+    ordered = get_ordered([{"answerValue": question.slug}])
+    assert ordered == [doc_c_id, doc_b_id, doc_a_id]
+
+    # ordering by answer value, backward.
+    ordered = get_ordered([{"answerValue": question.slug, "direction": "DESC"}])
+    assert ordered == [doc_a_id, doc_b_id, doc_c_id]
+
+    # order by meta key "blah" first, which contains the same value twice,
+    # then by the answer value, which orders the ambiguity between
+    # the first two same-value documents
+    ordered = get_ordered(
+        [{"meta": "blah"}, {"answerValue": question.slug, "direction": "DESC"}]
+    )
+    assert ordered == [doc_c_id, doc_a_id, doc_b_id]
+
+    # same, but secondary order goes backwards
+    ordered = get_ordered([{"meta": "blah"}, {"answerValue": question.slug}])
+    assert ordered == [doc_c_id, doc_b_id, doc_a_id]
+
+    # Just for coverage - sort by attribute
+    ordered = get_ordered([{"attribute": "FORM"}])
+
+    assert ordered == [
+        str(doc.id)
+        for doc in sorted([doc_c, doc_b, doc_a], key=lambda doc: doc.form.slug)
+    ]