Initial commitv3.3.4

1 files changed, 247 insertions, 0 deletions

diff --git a/test_utils/incite/mergers/conftest.py b/test_utils/incite/mergers/conftest.py
new file mode 100644
index 0000000..e4e3bdd
--- /dev/null
+++ b/test_utils/incite/mergers/conftest.py
@@ -0,0 +1,247 @@
+from datetime import timedelta, datetime
+from typing import TYPE_CHECKING, Optional, Callable
+
+import pytest
+
+from test_utils.conftest import clear_directory
+from test_utils.incite.conftest import mnt_filepath
+
+if TYPE_CHECKING:
+    from generalresearch.incite.mergers import MergeType
+    from generalresearch.incite.mergers.ym_wall_summary import (
+        YMWallSummaryMerge,
+        YMWallSummaryMergeItem,
+    )
+    from generalresearch.incite.mergers.pop_ledger import PopLedgerMerge
+    from generalresearch.incite.mergers.ym_survey_wall import YMSurveyWallMerge
+    from generalresearch.incite.base import GRLDatasets
+    from generalresearch.incite.mergers.foundations.enriched_session import (
+        EnrichedSessionMerge,
+    )
+    from generalresearch.incite.mergers.foundations.enriched_task_adjust import (
+        EnrichedTaskAdjustMerge,
+    )
+    from generalresearch.incite.mergers.foundations.enriched_wall import (
+        EnrichedWallMerge,
+    )
+    from generalresearch.incite.mergers.foundations.user_id_product import (
+        UserIdProductMerge,
+    )
+    from generalresearch.incite.mergers.ym_survey_wall import (
+        YMSurveyWallMergeCollectionItem,
+    )
+
+
+# --------------------------
+#      Merges
+# --------------------------
+
+
+@pytest.fixture(scope="function")
+def rm_pop_ledger_merge(pop_ledger_merge) -> Callable:
+    def _rm_pop_ledger_merge():
+        clear_directory(pop_ledger_merge.archive_path)
+
+    return _rm_pop_ledger_merge
+
+
+@pytest.fixture(scope="function")
+def pop_ledger_merge(
+    mnt_filepath: "GRLDatasets",
+    offset: str,
+    start: datetime,
+    duration: timedelta,
+) -> "PopLedgerMerge":
+    from generalresearch.incite.mergers.pop_ledger import PopLedgerMerge
+    from generalresearch.incite.mergers import MergeType
+
+    return PopLedgerMerge(
+        start=start,
+        finished=start + duration if duration else None,
+        offset=offset,
+        archive_path=mnt_filepath.archive_path(enum_type=MergeType.POP_LEDGER),
+    )
+
+
+@pytest.fixture(scope="function")
+def pop_ledger_merge_item(
+    start,
+    pop_ledger_merge,
+) -> "PopLedgerMergeItem":
+    from generalresearch.incite.mergers.pop_ledger import PopLedgerMergeItem
+
+    return PopLedgerMergeItem(
+        start=start,
+        _collection=pop_ledger_merge,
+    )
+
+
+@pytest.fixture(scope="function")
+def ym_survey_wall_merge(
+    mnt_filepath: "GRLDatasets",
+    start: datetime,
+) -> "YMSurveyWallMerge":
+    from generalresearch.incite.mergers.ym_survey_wall import YMSurveyWallMerge
+    from generalresearch.incite.mergers import MergeType
+
+    return YMSurveyWallMerge(
+        start=None,
+        offset="10D",
+        archive_path=mnt_filepath.archive_path(enum_type=MergeType.YM_SURVEY_WALL),
+    )
+
+
+@pytest.fixture(scope="function")
+def ym_survey_wall_merge_item(
+    start, ym_survey_wall_merge
+) -> "YMSurveyWallMergeCollectionItem":
+    from generalresearch.incite.mergers.ym_survey_wall import (
+        YMSurveyWallMergeCollectionItem,
+    )
+
+    return YMSurveyWallMergeCollectionItem(
+        start=start,
+        _collection=pop_ledger_merge,
+    )
+
+
+@pytest.fixture(scope="function")
+def ym_wall_summary_merge(
+    mnt_filepath: "GRLDatasets",
+    offset: str,
+    duration: timedelta,
+    start: datetime,
+) -> "YMWallSummaryMerge":
+    from generalresearch.incite.mergers.ym_wall_summary import YMWallSummaryMerge
+    from generalresearch.incite.mergers import MergeType
+
+    return YMWallSummaryMerge(
+        start=start,
+        finished=start + duration,
+        offset=offset,
+        archive_path=mnt_filepath.archive_path(enum_type=MergeType.POP_LEDGER),
+    )
+
+
+def ym_wall_summary_merge_item(
+    start, ym_wall_summary_merge
+) -> "YMWallSummaryMergeItem":
+    from generalresearch.incite.mergers.ym_wall_summary import (
+        YMWallSummaryMergeItem,
+    )
+
+    return YMWallSummaryMergeItem(
+        start=start,
+        _collection=pop_ledger_merge,
+    )
+
+
+# --------------------------
+#      Merges: Foundations
+# --------------------------
+
+
+@pytest.fixture(scope="function")
+def enriched_session_merge(
+    mnt_filepath: "GRLDatasets",
+    offset: str,
+    duration: timedelta,
+    start: datetime,
+) -> "EnrichedSessionMerge":
+    from generalresearch.incite.mergers.foundations.enriched_session import (
+        EnrichedSessionMerge,
+    )
+    from generalresearch.incite.mergers import MergeType
+
+    return EnrichedSessionMerge(
+        start=start,
+        finished=start + duration if duration else None,
+        offset=offset,
+        archive_path=mnt_filepath.archive_path(enum_type=MergeType.ENRICHED_SESSION),
+    )
+
+
+@pytest.fixture(scope="function")
+def enriched_task_adjust_merge(
+    mnt_filepath: "GRLDatasets",
+    offset: str,
+    duration: timedelta,
+    start: datetime,
+) -> "EnrichedTaskAdjustMerge":
+    from generalresearch.incite.mergers.foundations.enriched_task_adjust import (
+        EnrichedTaskAdjustMerge,
+    )
+    from generalresearch.incite.mergers import MergeType
+
+    return EnrichedTaskAdjustMerge(
+        start=start,
+        finished=start + duration,
+        offset=offset,
+        archive_path=mnt_filepath.archive_path(
+            enum_type=MergeType.ENRICHED_TASK_ADJUST
+        ),
+    )
+
+
+@pytest.fixture(scope="function")
+def enriched_wall_merge(
+    mnt_filepath: "GRLDatasets",
+    offset: str,
+    duration: timedelta,
+    start: datetime,
+) -> "EnrichedWallMerge":
+    from generalresearch.incite.mergers import MergeType
+    from generalresearch.incite.mergers.foundations.enriched_wall import (
+        EnrichedWallMerge,
+    )
+
+    return EnrichedWallMerge(
+        start=start,
+        finished=start + duration if duration else None,
+        offset=offset,
+        archive_path=mnt_filepath.archive_path(enum_type=MergeType.ENRICHED_WALL),
+    )
+
+
+@pytest.fixture(scope="function")
+def user_id_product_merge(
+    mnt_filepath: "GRLDatasets",
+    duration: timedelta,
+    offset,
+    start: datetime,
+) -> "UserIdProductMerge":
+    from generalresearch.incite.mergers.foundations.user_id_product import (
+        UserIdProductMerge,
+    )
+    from generalresearch.incite.mergers import MergeType
+
+    return UserIdProductMerge(
+        start=start,
+        finished=start + duration,
+        offset=None,
+        archive_path=mnt_filepath.archive_path(enum_type=MergeType.USER_ID_PRODUCT),
+    )
+
+
+# --------------------------
+#      Generic / Base
+# --------------------------
+
+
+@pytest.fixture(scope="function")
+def merge_collection(
+    mnt_filepath,
+    merge_type: "MergeType",
+    offset,
+    duration,
+    start,
+):
+    from generalresearch.incite.mergers import MergeCollection
+
+    return MergeCollection(
+        merge_type=merge_type,
+        start=start,
+        finished=start + duration,
+        offset=offset,
+        archive_path=mnt_filepath.archive_path(enum_type=merge_type),
+    )