from unittest.mock import patch

from django.test.utils import override_settings

import pytest

from baserow.core.action.registries import ActionType
from baserow.core.action.signals import ActionCommandType
from baserow.core.posthog import capture_event, capture_event_action_done


class TestActionType(ActionType):
    type = "test"
    analytics_params = ["must_be_kept"]

    class Params:
        must_be_kept: str
        must_be_removed: str

    def do(cls, *args, **kwargs):
        pass

    def scope(cls, *args, **kwargs):
        pass


@pytest.mark.django_db
@override_settings(POSTHOG_ENABLED=False)
@patch("baserow.core.posthog.posthog")
def test_not_capture_event_if_not_enabled(mock_posthog, data_fixture):
    user = data_fixture.create_user()
    capture_event(user, "test", {})
    mock_posthog.capture.assert_not_called()


@pytest.mark.django_db
@override_settings(POSTHOG_ENABLED=True)
@patch("baserow.core.posthog.posthog")
def test_capture_event_if_enabled(mock_posthog, data_fixture):
    user = data_fixture.create_user()
    workspace = data_fixture.create_workspace()
    capture_event(user, "test", {}, session="session", workspace=workspace)
    mock_posthog.capture.assert_called_once_with(
        distinct_id=user.id,
        event="test",
        properties={
            "user_email": user.email,
            "user_session": "session",
            "workspace_id": workspace.id,
        },
    )


@pytest.mark.django_db
@patch("baserow.core.posthog.capture_event")
def test_capture_event_action_done(mock_capture_event, data_fixture):
    user = data_fixture.create_user()

    test_action = TestActionType()
    params = {
        "must_be_kept": "yes",
        "must_be_removed": "yes",
    }
    capture_event_action_done(
        sender=None,
        user=user,
        action_type=test_action,
        action_params=params,
        action_timestamp=None,
        action_command_type=ActionCommandType.DO,
        workspace=None,
        session="session",
    )
    mock_capture_event.assert_called_once_with(
        user,
        "test",
        {"must_be_kept": "yes"},
        workspace=None,
        session="session",
    )