diff options
Diffstat (limited to 'tests/grliq/managers/test_forensic_data.py')
| -rw-r--r-- | tests/grliq/managers/test_forensic_data.py | 212 |
1 files changed, 212 insertions, 0 deletions
diff --git a/tests/grliq/managers/test_forensic_data.py b/tests/grliq/managers/test_forensic_data.py new file mode 100644 index 0000000..ed4da80 --- /dev/null +++ b/tests/grliq/managers/test_forensic_data.py @@ -0,0 +1,212 @@ +from datetime import timedelta +from uuid import uuid4 + +import pytest + +from generalresearch.grliq.models.events import TimingData, MouseEvent +from generalresearch.grliq.models.forensic_data import GrlIqData +from generalresearch.grliq.models.forensic_result import ( + GrlIqCheckerResults, + GrlIqForensicCategoryResult, +) + +try: + from psycopg.errors import UniqueViolation +except ImportError: + pass + + +class TestGrlIqDataManager: + + def test_create_dummy(self, grliq_dm): + from generalresearch.grliq.managers.forensic_data import GrlIqDataManager + from generalresearch.grliq.models.forensic_data import GrlIqData + + grliq_dm: GrlIqDataManager + gd1: GrlIqData = grliq_dm.create_dummy(is_attempt_allowed=True) + + assert isinstance(gd1, GrlIqData) + assert isinstance(gd1.results, GrlIqCheckerResults) + assert isinstance(gd1.category_result, GrlIqForensicCategoryResult) + + def test_create(self, grliq_data, grliq_dm): + grliq_dm.create(grliq_data) + assert grliq_data.id is not None + + with pytest.raises(UniqueViolation): + grliq_dm.create(grliq_data) + + @pytest.mark.skip(reason="todo") + def test_set_result(self): + pass + + @pytest.mark.skip(reason="todo") + def test_update_fingerprint(self): + pass + + @pytest.mark.skip(reason="todo") + def test_update_fingerprint(self): + pass + + @pytest.mark.skip(reason="todo") + def test_update_data(self): + pass + + def test_get_id(self, grliq_data, grliq_dm): + grliq_dm.create(grliq_data) + + res = grliq_dm.get_data(forensic_id=grliq_data.id) + assert res == grliq_data + + def test_get_uuid(self, grliq_data, grliq_dm): + grliq_dm.create(grliq_data) + + res = grliq_dm.get_data(forensic_uuid=grliq_data.uuid) + assert res == grliq_data + + @pytest.mark.skip(reason="todo") + def test_filter_timing_data(self): + pass + + @pytest.mark.skip(reason="todo") + def test_get_unique_user_count_by_fingerprint(self): + pass + + def test_filter_data(self, grliq_data, grliq_dm): + grliq_dm.create(grliq_data) + res = grliq_dm.filter_data(uuids=[grliq_data.uuid])[0] + assert res == grliq_data + now = res.created_at + res = grliq_dm.filter_data( + created_after=now, created_before=now + timedelta(minutes=1) + ) + assert len(res) == 1 + res = grliq_dm.filter_data( + created_after=now + timedelta(seconds=1), + created_before=now + timedelta(minutes=1), + ) + assert len(res) == 0 + + @pytest.mark.skip(reason="todo") + def test_filter_results(self): + pass + + @pytest.mark.skip(reason="todo") + def test_filter_category_results(self): + pass + + @pytest.mark.skip(reason="todo") + def test_make_filter_str(self): + pass + + def test_filter_count(self, grliq_dm, product): + res = grliq_dm.filter_count(product_id=product.uuid) + + assert isinstance(res, int) + + @pytest.mark.skip(reason="todo") + def test_filter(self): + pass + + @pytest.mark.skip(reason="todo") + def test_temporary_add_missing_fields(self): + pass + + +class TestForensicDataGetAndFilter: + + def test_events(self, grliq_dm, grliq_em): + """If load_events=True, the events and mouse_events attributes should + be an array no matter what. An empty array means that the events were + loaded, but there were no events available. + + If loaded_eventsFalse, the events and mouse_events attributes should + be None + """ + # Load Events == False + forensic_uuid = uuid4().hex + grliq_dm.create_dummy(is_attempt_allowed=True, uuid=forensic_uuid) + + instance = grliq_dm.filter_data(uuids=[forensic_uuid])[0] + assert isinstance(instance, GrlIqData) + + assert instance.events is None + assert instance.mouse_events is None + + # Load Events == True + instance = grliq_dm.get_data(forensic_uuid=forensic_uuid, load_events=True) + assert isinstance(instance, GrlIqData) + # This one doesn't have any events though + assert len(instance.events) == 0 + assert len(instance.mouse_events) == 0 + + def test_timing(self, grliq_dm, grliq_em): + forensic_uuid = uuid4().hex + grliq_dm.create_dummy(is_attempt_allowed=True, uuid=forensic_uuid) + + instance = grliq_dm.filter_data(uuids=[forensic_uuid])[0] + + grliq_em.update_or_create_timing( + session_uuid=instance.mid, + timing_data=TimingData( + client_rtts=[100, 200, 150], server_rtts=[150, 120, 120] + ), + ) + instance = grliq_dm.get_data(forensic_uuid=forensic_uuid, load_events=True) + assert isinstance(instance, GrlIqData) + assert isinstance(instance.events, list) + assert isinstance(instance.mouse_events, list) + assert isinstance(instance.timing_data, TimingData) + + def test_events_events(self, grliq_dm, grliq_em): + forensic_uuid = uuid4().hex + grliq_dm.create_dummy(is_attempt_allowed=True, uuid=forensic_uuid) + + instance = grliq_dm.filter_data(uuids=[forensic_uuid])[0] + + grliq_em.update_or_create_events( + session_uuid=instance.mid, + events=[{"a": "b"}], + mouse_events=[], + event_start=instance.created_at, + event_end=instance.created_at + timedelta(minutes=1), + ) + instance = grliq_dm.get_data(forensic_uuid=forensic_uuid, load_events=True) + assert isinstance(instance, GrlIqData) + assert isinstance(instance.events, list) + assert isinstance(instance.mouse_events, list) + assert instance.timing_data is None + assert instance.events == [{"a": "b"}] + assert len(instance.mouse_events) == 0 + assert len(instance.pointer_move_events) == 0 + assert len(instance.keyboard_events) == 0 + + def test_events_click(self, grliq_dm, grliq_em): + forensic_uuid = uuid4().hex + grliq_dm.create_dummy(is_attempt_allowed=True, uuid=forensic_uuid) + instance = grliq_dm.get_data(forensic_uuid=forensic_uuid, load_events=True) + + click_event = { + "type": "click", + "pageX": 0, + "pageY": 0, + "timeStamp": 123, + "pointerType": "mouse", + } + me = MouseEvent.from_dict(click_event) + grliq_em.update_or_create_events( + session_uuid=instance.mid, + events=[click_event], + mouse_events=[], + event_start=instance.created_at, + event_end=instance.created_at + timedelta(minutes=1), + ) + instance = grliq_dm.get_data(forensic_uuid=forensic_uuid, load_events=True) + assert isinstance(instance, GrlIqData) + assert isinstance(instance.events, list) + assert isinstance(instance.mouse_events, list) + assert instance.timing_data is None + assert instance.events == [click_event] + assert instance.mouse_events == [me] + assert len(instance.pointer_move_events) == 0 + assert len(instance.keyboard_events) == 0 |