from datetime import UTC, datetime, timedelta
from itertools import product

import pandas as pd
import pytest

# noinspection PyUnresolvedReferences
from generalresearch.incite.mergers.foundations.user_id_product import (
    UserIdProductMergeItem,
)


@pytest.mark.parametrize(
    argnames="offset, duration, start",
    argvalues=list(
        product(
            ["12h", "3D"],
            [timedelta(days=5)],
            [(datetime.now(tz=UTC) - timedelta(days=35)).replace(microsecond=0)],
        )
    ),
)
class TestUserIDProduct:

    @pytest.mark.skip
    def test_base(self, client_no_amm, user_id_product_merge):
        ddf = user_id_product_merge.ddf()
        df = client_no_amm.compute(collections=ddf, sync=True)
        assert isinstance(df, pd.DataFrame)
        assert not df.empty

    @pytest.mark.skip
    def test_base_item(self, client_no_amm, user_id_product_merge, user_collection):
        assert len(user_id_product_merge.items) == 1

        for item in user_id_product_merge.items:
            assert isinstance(item, UserIdProductMergeItem)

            path = item.path

            try:
                modified_time1 = path.stat().st_mtime
            except Exception:
                modified_time1 = 0

            user_id_product_merge.build(client=client_no_amm, user_coll=user_collection)
            modified_time2 = path.stat().st_mtime

            assert modified_time2 > modified_time1

    @pytest.mark.skip
    def test_read(self, client_no_amm, user_id_product_merge):
        users_ddf = user_id_product_merge.ddf()
        df = client_no_amm.compute(collections=users_ddf, sync=True)

        assert isinstance(df, pd.DataFrame)
        assert len(df.columns) == 1
        assert str(df.product_id.dtype) == "category"