1 files changed, 382 insertions, 0 deletions
diff --git a/tests/models/thl/test_marketplace_condition.py b/tests/models/thl/test_marketplace_condition.py
new file mode 100644
index 0000000..217616d
--- /dev/null
+++ b/tests/models/thl/test_marketplace_condition.py
@@ -0,0 +1,382 @@
+import pytest
+from pydantic import ValidationError
+
+
+class TestMarketplaceCondition:
+
+    def test_list_or(self):
+        from generalresearch.models import LogicalOperator
+        from generalresearch.models.thl.survey.condition import (
+            MarketplaceCondition,
+            ConditionValueType,
+        )
+
+        user_qas = {"q1": {"a2"}}
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=False,
+            value_type=ConditionValueType.LIST,
+            values=["a1", "a2", "a3"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=False,
+            value_type=ConditionValueType.LIST,
+            values=["a2"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=False,
+            value_type=ConditionValueType.LIST,
+            values=["a1", "a3"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert not c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q2",
+            negate=False,
+            value_type=ConditionValueType.LIST,
+            values=["a1", "a2", "a3"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert c.evaluate_criterion(user_qas) is None
+
+    def test_list_or_negate(self):
+        from generalresearch.models import LogicalOperator
+        from generalresearch.models.thl.survey.condition import (
+            MarketplaceCondition,
+            ConditionValueType,
+        )
+
+        user_qas = {"q1": {"a2"}}
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=True,
+            value_type=ConditionValueType.LIST,
+            values=["a1", "a2", "a3"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert not c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=True,
+            value_type=ConditionValueType.LIST,
+            values=["a2"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert not c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=True,
+            value_type=ConditionValueType.LIST,
+            values=["a1", "a3"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q2",
+            negate=True,
+            value_type=ConditionValueType.LIST,
+            values=["a1", "a2", "a3"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert c.evaluate_criterion(user_qas) is None
+
+    def test_list_and(self):
+        from generalresearch.models import LogicalOperator
+        from generalresearch.models.thl.survey.condition import (
+            MarketplaceCondition,
+            ConditionValueType,
+        )
+
+        user_qas = {"q1": {"a1", "a2"}}
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=False,
+            value_type=ConditionValueType.LIST,
+            values=["a1", "a2", "a3"],
+            logical_operator=LogicalOperator.AND,
+        )
+        assert not c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=False,
+            value_type=ConditionValueType.LIST,
+            values=["a2"],
+            logical_operator=LogicalOperator.AND,
+        )
+        assert c.evaluate_criterion(user_qas)
+        user_qas = {"q1": {"a1"}}
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=False,
+            value_type=ConditionValueType.LIST,
+            values=["a1", "a2"],
+            logical_operator=LogicalOperator.AND,
+        )
+        assert not c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=False,
+            value_type=ConditionValueType.LIST,
+            values=["a1", "a3"],
+            logical_operator=LogicalOperator.AND,
+        )
+        assert not c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q2",
+            negate=False,
+            value_type=ConditionValueType.LIST,
+            values=["a1", "a2", "a3"],
+            logical_operator=LogicalOperator.AND,
+        )
+        assert c.evaluate_criterion(user_qas) is None
+
+    def test_list_and_negate(self):
+        from generalresearch.models import LogicalOperator
+        from generalresearch.models.thl.survey.condition import (
+            MarketplaceCondition,
+            ConditionValueType,
+        )
+
+        user_qas = {"q1": {"a1", "a2"}}
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=True,
+            value_type=ConditionValueType.LIST,
+            values=["a1", "a2", "a3"],
+            logical_operator=LogicalOperator.AND,
+        )
+        assert c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=True,
+            value_type=ConditionValueType.LIST,
+            values=["a2"],
+            logical_operator=LogicalOperator.AND,
+        )
+        assert not c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=True,
+            value_type=ConditionValueType.LIST,
+            values=["a1", "a3"],
+            logical_operator=LogicalOperator.AND,
+        )
+        assert c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q2",
+            negate=True,
+            value_type=ConditionValueType.LIST,
+            values=["a1", "a2", "a3"],
+            logical_operator=LogicalOperator.AND,
+        )
+        assert c.evaluate_criterion(user_qas) is None
+
+    def test_ranges(self):
+        from generalresearch.models import LogicalOperator
+        from generalresearch.models.thl.survey.condition import (
+            MarketplaceCondition,
+            ConditionValueType,
+        )
+
+        user_qas = {"q1": {"2", "50"}}
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=False,
+            value_type=ConditionValueType.RANGE,
+            values=["1-4", "10-20"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=False,
+            value_type=ConditionValueType.RANGE,
+            values=["1-4"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=False,
+            value_type=ConditionValueType.RANGE,
+            values=["10-20"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert not c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q2",
+            negate=False,
+            value_type=ConditionValueType.RANGE,
+            values=["1-4", "10-20"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert c.evaluate_criterion(user_qas) is None
+        # --- negate
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=True,
+            value_type=ConditionValueType.RANGE,
+            values=["1-4", "10-20"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert not c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=True,
+            value_type=ConditionValueType.RANGE,
+            values=["10-20"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert c.evaluate_criterion(user_qas)
+        # --- AND
+        with pytest.raises(expected_exception=ValidationError):
+            c = MarketplaceCondition(
+                question_id="q1",
+                negate=False,
+                value_type=ConditionValueType.RANGE,
+                values=["1-4", "10-20"],
+                logical_operator=LogicalOperator.AND,
+            )
+
+    def test_ranges_to_list(self):
+        from generalresearch.models import LogicalOperator
+        from generalresearch.models.thl.survey.condition import (
+            MarketplaceCondition,
+            ConditionValueType,
+        )
+
+        user_qas = {"q1": {"2", "50"}}
+        MarketplaceCondition._CONVERT_LIST_TO_RANGE = ["q1"]
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=False,
+            value_type=ConditionValueType.RANGE,
+            values=["1-4", "10-12", "3-5"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert c.evaluate_criterion(user_qas)
+        assert ConditionValueType.LIST == c.value_type
+        assert ["1", "10", "11", "12", "2", "3", "4", "5"] == c.values
+
+    def test_ranges_infinity(self):
+        from generalresearch.models import LogicalOperator
+        from generalresearch.models.thl.survey.condition import (
+            MarketplaceCondition,
+            ConditionValueType,
+        )
+
+        user_qas = {"q1": {"2", "50"}}
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=False,
+            value_type=ConditionValueType.RANGE,
+            values=["1-4", "10-inf"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert c.evaluate_criterion(user_qas)
+        user_qas = {"q1": {"5", "50"}}
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=False,
+            value_type=ConditionValueType.RANGE,
+            values=["1-4", "60-inf"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert not c.evaluate_criterion(user_qas)
+
+        # need to test negative infinity!
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=False,
+            value_type=ConditionValueType.RANGE,
+            values=["inf-40"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert c.evaluate_criterion({"q1": {"5", "50"}})
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=False,
+            value_type=ConditionValueType.RANGE,
+            values=["inf-40"],
+            logical_operator=LogicalOperator.OR,
+        )
+        assert not c.evaluate_criterion({"q1": {"50"}})
+
+    def test_answered(self):
+        from generalresearch.models.thl.survey.condition import (
+            MarketplaceCondition,
+            ConditionValueType,
+        )
+
+        user_qas = {"q1": {"a2"}}
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=False,
+            value_type=ConditionValueType.ANSWERED,
+            values=[],
+        )
+        assert c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q2",
+            negate=False,
+            value_type=ConditionValueType.ANSWERED,
+            values=[],
+        )
+        assert not c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q1",
+            negate=True,
+            value_type=ConditionValueType.ANSWERED,
+            values=[],
+        )
+        assert not c.evaluate_criterion(user_qas)
+        c = MarketplaceCondition(
+            question_id="q2",
+            negate=True,
+            value_type=ConditionValueType.ANSWERED,
+            values=[],
+        )
+        assert c.evaluate_criterion(user_qas)
+
+    def test_invite(self):
+        from generalresearch.models.thl.survey.condition import (
+            MarketplaceCondition,
+            ConditionValueType,
+        )
+
+        user_groups = {"g1", "g2", "g3"}
+        c = MarketplaceCondition(
+            question_id=None,
+            negate=False,
+            value_type=ConditionValueType.RECONTACT,
+            values=["g1", "g4"],
+        )
+        assert c.evaluate_criterion(user_qas=dict(), user_groups=user_groups)
+        c = MarketplaceCondition(
+            question_id=None,
+            negate=False,
+            value_type=ConditionValueType.RECONTACT,
+            values=["g4"],
+        )
+        assert not c.evaluate_criterion(user_qas=dict(), user_groups=user_groups)
+
+        c = MarketplaceCondition(
+            question_id=None,
+            negate=True,
+            value_type=ConditionValueType.RECONTACT,
+            values=["g1", "g4"],
+        )
+        assert not c.evaluate_criterion(user_qas=dict(), user_groups=user_groups)
+        c = MarketplaceCondition(
+            question_id=None,
+            negate=True,
+            value_type=ConditionValueType.RECONTACT,
+            values=["g4"],
+        )
+        assert c.evaluate_criterion(user_qas=dict(), user_groups=user_groups)