aboutsummaryrefslogtreecommitdiff
path: root/tests/grliq/managers/test_forensic_results.py
blob: 68db7328a138fc3b7cb13569ebcbd004cd6e1ded (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
from typing import TYPE_CHECKING

if TYPE_CHECKING:
    from generalresearch.grliq.managers.forensic_data import GrlIqDataManager
    from generalresearch.grliq.managers.forensic_results import (
        GrlIqCategoryResultsReader,
    )


class TestGrlIqCategoryResultsReader:

    def test_filter_category_results(
        self, grliq_dm: "GrlIqDataManager", grliq_crr: "GrlIqCategoryResultsReader"
    ):
        from generalresearch.grliq.models.forensic_result import (
            GrlIqForensicCategoryResult,
            Phase,
        )

        # this is just testing that it doesn't fail
        grliq_dm.create_dummy(is_attempt_allowed=True)
        grliq_dm.create_dummy(is_attempt_allowed=True)

        res = grliq_crr.filter_category_results(limit=2, phase=Phase.OFFERWALL_ENTER)[0]
        assert res.get("category_result")
        assert isinstance(res["category_result"], GrlIqForensicCategoryResult)
        assert res["user_agent"].os.family