aboutsummaryrefslogtreecommitdiff
path: root/test_utils/grliq/conftest.py
blob: edd777ebd5989c384f1f9747e6c8c2d114ee2048 (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
28
from datetime import datetime, timedelta, timezone
from typing import TYPE_CHECKING, Optional
from uuid import uuid4

import pytest

if TYPE_CHECKING:
    from generalresearch.config import GRLBaseSettings
    from generalresearch.grliq.models.forensic_data import GrlIqData


@pytest.fixture(scope="function")
def mnt_grliq_archive_dir(settings: "GRLBaseSettings") -> Optional[str]:
    return settings.mnt_grliq_archive_dir


@pytest.fixture(scope="function")
def grliq_data() -> "GrlIqData":
    from generalresearch.grliq.managers import DUMMY_GRLIQ_DATA
    from generalresearch.grliq.models.forensic_data import GrlIqData

    g: GrlIqData = DUMMY_GRLIQ_DATA[1]["data"]

    g.id = None
    g.uuid = uuid4().hex
    g.created_at = datetime.now(tz=timezone.utc)
    g.timestamp = g.created_at - timedelta(seconds=10)
    return g