# Copyright 2021-2024 Avaiga Private Limited
#
# Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with
# the License. You may obtain a copy of the License at
#
#        http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
# an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the
# specific language governing permissions and limitations under the License.

import pytest

from taipy.core.exceptions.exceptions import InvalidEventOperation
from taipy.core.notification._topic import _Topic
from taipy.core.notification.event import EventEntityType, EventOperation


def test_general_topic_creation():
    topic_1 = _Topic(None, None, None, None)
    assert topic_1.entity_type is None
    assert topic_1.entity_id is None
    assert topic_1.operation is None
    assert topic_1.attribute_name is None

    topic_2 = _Topic(EventEntityType.SCENARIO, "scenario_id")
    assert topic_2.entity_type == EventEntityType.SCENARIO
    assert topic_2.entity_id == "scenario_id"
    assert topic_2.operation is None
    assert topic_2.attribute_name is None

    topic_3 = _Topic(None, None, EventOperation.CREATION)
    assert topic_3.entity_type is None
    assert topic_3.entity_id is None
    assert topic_3.operation == EventOperation.CREATION
    assert topic_3.attribute_name is None

    topic_4 = _Topic(None, None, EventOperation.UPDATE, "properties")
    assert topic_4.entity_type is None
    assert topic_4.entity_id is None
    assert topic_4.operation == EventOperation.UPDATE
    assert topic_4.attribute_name == "properties"

    topic_5 = _Topic(entity_type=EventEntityType.JOB, operation=EventOperation.DELETION)
    assert topic_5.entity_type == EventEntityType.JOB
    assert topic_5.entity_id is None
    assert topic_5.operation == EventOperation.DELETION
    assert topic_5.attribute_name is None

    topic_6 = _Topic(entity_type=EventEntityType.SEQUENCE)
    assert topic_6.entity_type == EventEntityType.SEQUENCE
    assert topic_6.entity_id is None
    assert topic_6.operation is None
    assert topic_6.attribute_name is None


def test_topic_creation_cycle():
    topic_1 = _Topic(EventEntityType.CYCLE, "cycle_id", EventOperation.CREATION)
    assert topic_1.entity_type == EventEntityType.CYCLE
    assert topic_1.entity_id == "cycle_id"
    assert topic_1.operation == EventOperation.CREATION
    assert topic_1.attribute_name is None

    topic_2 = _Topic(EventEntityType.CYCLE, "cycle_id", EventOperation.UPDATE, "frequency")
    assert topic_2.entity_type == EventEntityType.CYCLE
    assert topic_2.entity_id == "cycle_id"
    assert topic_2.operation == EventOperation.UPDATE
    assert topic_2.attribute_name == "frequency"

    topic_3 = _Topic(EventEntityType.CYCLE, "cycle_id", EventOperation.DELETION)
    assert topic_3.entity_type == EventEntityType.CYCLE
    assert topic_3.entity_id == "cycle_id"
    assert topic_3.operation == EventOperation.DELETION
    assert topic_3.attribute_name is None

    topic_4 = _Topic(EventEntityType.CYCLE, "cycle_id", EventOperation.CREATION)
    assert topic_4.entity_type == EventEntityType.CYCLE
    assert topic_4.entity_id == "cycle_id"
    assert topic_4.operation == EventOperation.CREATION
    assert topic_4.attribute_name is None

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.CYCLE, "cycle_id", EventOperation.CREATION, "frequency")

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.CYCLE, "cycle_id", EventOperation.DELETION, "frequency")

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.CYCLE, "cycle_id", attribute_name="frequency")

    with pytest.raises(InvalidEventOperation):
        _ = _Topic(EventEntityType.CYCLE, "cycle_id", EventOperation.SUBMISSION)

    with pytest.raises(InvalidEventOperation):
        _ = _Topic(EventEntityType.CYCLE, "cycle_id", EventOperation.SUBMISSION, "frequency")


def test_topic_creation_scenario():
    topic_1 = _Topic(EventEntityType.SCENARIO, "scenario_id", EventOperation.CREATION)
    assert topic_1.entity_type == EventEntityType.SCENARIO
    assert topic_1.entity_id == "scenario_id"
    assert topic_1.operation == EventOperation.CREATION
    assert topic_1.attribute_name is None

    topic_2 = _Topic(EventEntityType.SCENARIO, "scenario_id", EventOperation.UPDATE, "is_primary")
    assert topic_2.entity_type == EventEntityType.SCENARIO
    assert topic_2.entity_id == "scenario_id"
    assert topic_2.operation == EventOperation.UPDATE
    assert topic_2.attribute_name == "is_primary"

    topic_3 = _Topic(EventEntityType.SCENARIO, "scenario_id", EventOperation.DELETION)
    assert topic_3.entity_type == EventEntityType.SCENARIO
    assert topic_3.entity_id == "scenario_id"
    assert topic_3.operation == EventOperation.DELETION
    assert topic_3.attribute_name is None

    topic_4 = _Topic(EventEntityType.SCENARIO, "scenario_id", EventOperation.SUBMISSION)
    assert topic_4.entity_type == EventEntityType.SCENARIO
    assert topic_4.entity_id == "scenario_id"
    assert topic_4.operation == EventOperation.SUBMISSION
    assert topic_4.attribute_name is None

    topic_5 = _Topic(EventEntityType.SCENARIO, "scenario_id", EventOperation.UPDATE, "properties")
    assert topic_5.entity_type == EventEntityType.SCENARIO
    assert topic_5.entity_id == "scenario_id"
    assert topic_5.operation == EventOperation.UPDATE
    assert topic_5.attribute_name == "properties"

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.SCENARIO, "scenario_id", EventOperation.CREATION, "is_primary")

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.SCENARIO, "scenario_id", EventOperation.DELETION, "is_primary")

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.SCENARIO, "scenario_id", EventOperation.SUBMISSION, "is_primary")

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.SCENARIO, "scenario_id", attribute_name="is_primary")


def test_topic_creation_sequence():
    topic_1 = _Topic(EventEntityType.SEQUENCE, "sequence_id", EventOperation.CREATION)
    assert topic_1.entity_type == EventEntityType.SEQUENCE
    assert topic_1.entity_id == "sequence_id"
    assert topic_1.operation == EventOperation.CREATION
    assert topic_1.attribute_name is None

    topic_2 = _Topic(EventEntityType.SEQUENCE, "sequence_id", EventOperation.UPDATE, "subscribers")
    assert topic_2.entity_type == EventEntityType.SEQUENCE
    assert topic_2.entity_id == "sequence_id"
    assert topic_2.operation == EventOperation.UPDATE
    assert topic_2.attribute_name == "subscribers"

    topic_3 = _Topic(EventEntityType.SEQUENCE, "sequence_id", EventOperation.DELETION)
    assert topic_3.entity_type == EventEntityType.SEQUENCE
    assert topic_3.entity_id == "sequence_id"
    assert topic_3.operation == EventOperation.DELETION
    assert topic_3.attribute_name is None

    topic_4 = _Topic(EventEntityType.SEQUENCE, "sequence_id", EventOperation.SUBMISSION)
    assert topic_4.entity_type == EventEntityType.SEQUENCE
    assert topic_4.entity_id == "sequence_id"
    assert topic_4.operation == EventOperation.SUBMISSION
    assert topic_4.attribute_name is None

    topic_5 = _Topic(EventEntityType.SEQUENCE, "sequence_id", EventOperation.DELETION)
    assert topic_5.entity_type == EventEntityType.SEQUENCE
    assert topic_5.entity_id == "sequence_id"
    assert topic_5.operation == EventOperation.DELETION
    assert topic_5.attribute_name is None

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.SEQUENCE, "sequence_id", EventOperation.CREATION, "subscribers")

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.SEQUENCE, "sequence_id", EventOperation.DELETION, "subscribers")

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.SEQUENCE, "sequence_id", EventOperation.SUBMISSION, "subscribers")

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.SEQUENCE, "sequence_id", attribute_name="subscribers")


def test_topic_creation_task():
    topic_1 = _Topic(EventEntityType.TASK, "task_id", EventOperation.CREATION)
    assert topic_1.entity_type == EventEntityType.TASK
    assert topic_1.entity_id == "task_id"
    assert topic_1.operation == EventOperation.CREATION
    assert topic_1.attribute_name is None

    topic_2 = _Topic(EventEntityType.TASK, "task_id", EventOperation.UPDATE, "function")
    assert topic_2.entity_type == EventEntityType.TASK
    assert topic_2.entity_id == "task_id"
    assert topic_2.operation == EventOperation.UPDATE
    assert topic_2.attribute_name == "function"

    topic_3 = _Topic(EventEntityType.TASK, "task_id", EventOperation.DELETION)
    assert topic_3.entity_type == EventEntityType.TASK
    assert topic_3.entity_id == "task_id"
    assert topic_3.operation == EventOperation.DELETION
    assert topic_3.attribute_name is None

    topic_4 = _Topic(EventEntityType.TASK, "task_id", EventOperation.SUBMISSION)
    assert topic_4.entity_type == EventEntityType.TASK
    assert topic_4.entity_id == "task_id"
    assert topic_4.operation == EventOperation.SUBMISSION
    assert topic_4.attribute_name is None

    topic_5 = _Topic(EventEntityType.TASK, "task_id", EventOperation.SUBMISSION)
    assert topic_5.entity_type == EventEntityType.TASK
    assert topic_5.entity_id == "task_id"
    assert topic_5.operation == EventOperation.SUBMISSION
    assert topic_5.attribute_name is None

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.TASK, "task_id", EventOperation.CREATION, "function")

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.TASK, "task_id", EventOperation.DELETION, "function")

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.TASK, "task_id", EventOperation.SUBMISSION, "function")

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.TASK, "task_id", attribute_name="function")


def test_topic_creation_datanode():
    topic_1 = _Topic(EventEntityType.DATA_NODE, "dn_id", EventOperation.CREATION)
    assert topic_1.entity_type == EventEntityType.DATA_NODE
    assert topic_1.entity_id == "dn_id"
    assert topic_1.operation == EventOperation.CREATION
    assert topic_1.attribute_name is None

    topic_2 = _Topic(EventEntityType.DATA_NODE, "dn_id", EventOperation.UPDATE, "properties")
    assert topic_2.entity_type == EventEntityType.DATA_NODE
    assert topic_2.entity_id == "dn_id"
    assert topic_2.operation == EventOperation.UPDATE
    assert topic_2.attribute_name == "properties"

    topic_3 = _Topic(EventEntityType.DATA_NODE, "dn_id", EventOperation.DELETION)
    assert topic_3.entity_type == EventEntityType.DATA_NODE
    assert topic_3.entity_id == "dn_id"
    assert topic_3.operation == EventOperation.DELETION
    assert topic_3.attribute_name is None

    topic_4 = _Topic(None, "dn_id", EventOperation.UPDATE, "scope")
    assert topic_4.entity_type is None
    assert topic_4.entity_id == "dn_id"
    assert topic_4.operation == EventOperation.UPDATE
    assert topic_4.attribute_name == "scope"

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.DATA_NODE, "dn_id", EventOperation.CREATION, "properties")

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.DATA_NODE, "dn_id", EventOperation.DELETION, "properties")

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.DATA_NODE, "dn_id", attribute_name="properties")

    with pytest.raises(InvalidEventOperation):
        _ = _Topic(EventEntityType.DATA_NODE, "dn_id", EventOperation.SUBMISSION)

    # with pytest.raises(InvalidEventOperation):
    #     _ = Topic(EventEntityType.DATA_NODE, "dn_id", EventOperation.SUBMISSION, "properties")


def test_topic_creation_job():
    topic_1 = _Topic(EventEntityType.JOB, "job_id", EventOperation.CREATION)
    assert topic_1.entity_type == EventEntityType.JOB
    assert topic_1.entity_id == "job_id"
    assert topic_1.operation == EventOperation.CREATION
    assert topic_1.attribute_name is None

    topic_2 = _Topic(EventEntityType.JOB, "job_id", EventOperation.UPDATE, "force")
    assert topic_2.entity_type == EventEntityType.JOB
    assert topic_2.entity_id == "job_id"
    assert topic_2.operation == EventOperation.UPDATE
    assert topic_2.attribute_name == "force"

    topic_3 = _Topic(EventEntityType.JOB, "job_id", EventOperation.DELETION)
    assert topic_3.entity_type == EventEntityType.JOB
    assert topic_3.entity_id == "job_id"
    assert topic_3.operation == EventOperation.DELETION
    assert topic_3.attribute_name is None

    topic_4 = _Topic(EventEntityType.JOB, "job_id", EventOperation.CREATION)
    assert topic_4.entity_type == EventEntityType.JOB
    assert topic_4.entity_id == "job_id"
    assert topic_4.operation == EventOperation.CREATION
    assert topic_4.attribute_name is None

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.JOB, "job_id", EventOperation.CREATION, "force")

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.JOB, "job_id", EventOperation.DELETION, "force")

    # with pytest.raises(InvalidEventAttributeName):
    #     _ = Topic(EventEntityType.JOB, "job_id", attribute_name="force")

    with pytest.raises(InvalidEventOperation):
        _ = _Topic(EventEntityType.JOB, "job_id", EventOperation.SUBMISSION)

    # with pytest.raises(InvalidEventOperation):
    #     _ = Topic(EventEntityType.JOB, "job_id", EventOperation.SUBMISSION, "force")


def test_topic_equal():
    assert _Topic() == _Topic()
    assert _Topic(EventEntityType.SCENARIO) == _Topic(EventEntityType.SCENARIO)
    assert _Topic(entity_id="sequence_id") == _Topic(entity_id="sequence_id")
    assert _Topic(operation=EventOperation.SUBMISSION) == _Topic(operation=EventOperation.SUBMISSION)
    assert _Topic(EventEntityType.JOB, "JOB_id", EventOperation.UPDATE, "status") == _Topic(
        EventEntityType.JOB, "JOB_id", EventOperation.UPDATE, "status"
    )


def test_print():
    topic = _Topic(EventEntityType.TASK, "task_id", EventOperation.UPDATE, "foo")
    assert "TASK" in str(topic)
    assert "task_id" in str(topic)
    assert "UPDATE" in str(topic)
    assert "foo" in str(topic)

    topic_2 = _Topic(EventEntityType.SCENARIO, None, EventOperation.CREATION, None)
    assert "SCENARIO" in str(topic_2)
    assert "CREATION" in str(topic_2)