"""Test_get_localtime."""
import datetime

import pytz
from flask.app import Flask
from flask.testing import FlaskClient
from tests.spiffworkflow_backend.helpers.base_test import BaseTest
from tests.spiffworkflow_backend.helpers.test_data import load_test_spec

from spiffworkflow_backend.models.script_attributes_context import (
    ScriptAttributesContext,
)
from spiffworkflow_backend.scripts.get_localtime import GetLocaltime
from spiffworkflow_backend.services.process_instance_processor import (
    ProcessInstanceProcessor,
)
from spiffworkflow_backend.services.process_instance_service import (
    ProcessInstanceService,
)


class TestGetLocaltime(BaseTest):
    """TestProcessAPi."""

    def test_get_localtime_script_directly(self) -> None:
        """Test_get_localtime_script_directly."""
        current_time = datetime.datetime.now()
        timezone = "US/Pacific"
        process_model_identifier = "test_process_model"
        process_instance_id = 1
        script_attributes_context = ScriptAttributesContext(
            task=None,
            environment_identifier="testing",
            process_instance_id=process_instance_id,
            process_model_identifier=process_model_identifier,
        )
        result = GetLocaltime().run(
            script_attributes_context,
            datetime=current_time,
            timezone=timezone,
        )
        assert result == current_time.astimezone(pytz.timezone(timezone))

    def test_get_localtime_script_through_bpmn(
        self,
        app: Flask,
        client: FlaskClient,
        with_db_and_bpmn_file_cleanup: None,
    ) -> None:
        """Test_process_instance_run."""
        initiator_user = self.find_or_create_user("initiator_user")
        self.add_permissions_to_user(
            initiator_user,
            target_uri="/v1.0/process-groups",
            permission_names=["read", "create"],
        )
        self.create_process_group(
            client=client, user=initiator_user, process_group_id="test_group"
        )
        process_model = load_test_spec(
            process_model_id="test_group/get_localtime",
            bpmn_file_name="get_localtime.bpmn",
            process_model_source_directory="get_localtime",
        )
        process_instance = self.create_process_instance_from_process_model(
            process_model=process_model, user=initiator_user
        )
        processor = ProcessInstanceProcessor(process_instance)

        processor.do_engine_steps(save=True)
        active_task = process_instance.active_tasks[0]
        spiff_task = processor.__class__.get_task_by_bpmn_identifier(
            active_task.task_name, processor.bpmn_process_instance
        )

        ProcessInstanceService.complete_form_task(
            processor, spiff_task, {"timezone": "US/Pacific"}, initiator_user
        )

        active_task = process_instance.active_tasks[0]
        spiff_task = processor.__class__.get_task_by_bpmn_identifier(
            active_task.task_name, processor.bpmn_process_instance
        )

        assert spiff_task
        data = spiff_task.data
        some_time = data["some_time"]
        localtime = data["localtime"]
        timezone = data["timezone"]

        assert localtime == some_time.astimezone(pytz.timezone(timezone))