# -*- coding: utf-8 -*-

import sys
import os
import unittest

from SpiffWorkflow.bpmn.PythonScriptEngine import PythonScriptEngine
from SpiffWorkflow.bpmn.workflow import BpmnWorkflow

sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..', '..', '..'))
from tests.SpiffWorkflow.bpmn.BpmnWorkflowTestCase import BpmnWorkflowTestCase

__author__ = 'danfunk'


class PythonScriptEngineTest(BpmnWorkflowTestCase):

    def setUp(self):
        self.expressionEngine = PythonScriptEngine()

        # All this, just so we have a task object, not using anything in the Script.
        spec, subprocesses = self.load_workflow_spec('ScriptTest.bpmn', 'ScriptTest')
        workflow = BpmnWorkflow(spec, subprocesses)
        workflow.do_engine_steps()
        self.task = workflow.last_task

    def testDateTimeExpressions(self):
        """Basically, assure that we can use datime, dateutils, and pytz"""
        script = """
# Create Current Date as UTC
now_utc = datetime.datetime.now(datetime.timezone.utc)
# Create Current Date at EST
now_est = now_utc.astimezone(pytz.timezone('US/Eastern'))

# Format a date from a date String in UTC
datestr = "2021-09-23 16:11:00 -0000"  # 12 pm EST,  4pm UTC
dt = dateparser.parse(datestr)
localtime = dt.astimezone(pytz.timezone('US/Eastern'))
localtime_str = localtime.strftime("%Y-%m-%d %H:%M:%S")
        """
        self.expressionEngine.execute(self.task, script)
        self.assertEqual(self.task.data['now_utc'].utcoffset().days, 0)
        self.assertEqual(self.task.data['now_est'].tzinfo.zone, "US/Eastern")
        self.assertEqual(self.task.data['localtime_str'], "2021-09-23 12:11:00")
        self.assertTrue(True)

    def testFunctionsAndGlobalsAreRemoved(self):
        self.assertIn('testvar', self.task.data)
        self.assertIn('testvar2', self.task.data)
        self.assertIn('sample', self.task.data)
        self.assertNotIn('my_function', self.task.data)
        self.assertNotIn('datetime', self.task.data)

def suite():
    return unittest.TestLoader().loadTestsFromTestCase(PythonScriptEngineTest)

if __name__ == '__main__':
    unittest.TextTestRunner(verbosity=2).run(suite())