mirror of
https://github.com/status-im/spiff-arena.git
synced 2025-02-11 01:06:36 +00:00
5f2eab770c
1591ba436 black e0c09896d re-order imports the way backend always used to git-subtree-dir: spiffworkflow-backend git-subtree-split: 1591ba436dab5e1db2b37553de38be12ffdb705f
82 lines
2.8 KiB
Python
82 lines
2.8 KiB
Python
"""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")
|
|
process_model = load_test_spec(
|
|
process_model_id="get_localtime", bpmn_file_name="get_localtime.bpmn"
|
|
)
|
|
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))
|