56 lines
2.5 KiB
Python
56 lines
2.5 KiB
Python
from tests.base_test import BaseTest
|
|
from crc.scripts.get_localtime import GetLocaltime
|
|
import dateparser
|
|
import datetime
|
|
|
|
|
|
class TestGetLocaltime(BaseTest):
|
|
|
|
def test_get_localtime(self):
|
|
timestamp = datetime.datetime.utcnow()
|
|
workflow = self.create_workflow('get_localtime')
|
|
|
|
workflow_api = self.get_workflow_api(workflow)
|
|
task = workflow_api.next_task
|
|
|
|
workflow_api = self.complete_form(workflow, task, {'with_timestamp': True,
|
|
'with_timezone': False,
|
|
'timestamp': str(timestamp)})
|
|
task = workflow_api.next_task
|
|
|
|
# The workflow calls get_localtime twice, once with named arguments and once without
|
|
localtime_with = task.data['localtime_with']
|
|
localtime_without = task.data['localtime_without']
|
|
|
|
self.assertEqual(dateparser.parse(localtime_with), GetLocaltime().do_task(None, None, None, timestamp=str(timestamp)))
|
|
self.assertEqual(dateparser.parse(localtime_without), GetLocaltime().do_task(None, None, None, str(timestamp)))
|
|
|
|
def test_get_localtime_with_timezone(self):
|
|
|
|
timestamp = datetime.datetime.utcnow()
|
|
workflow = self.create_workflow('get_localtime')
|
|
|
|
workflow_api = self.get_workflow_api(workflow)
|
|
task = workflow_api.next_task
|
|
|
|
workflow_api = self.complete_form(workflow, task, {'with_timestamp': True,
|
|
'with_timezone': True,
|
|
'timestamp': str(timestamp),
|
|
'timezone': 'US/Eastern'})
|
|
task = workflow_api.next_task
|
|
|
|
# The workflow calls get_localtime twice, once with named arguments and once without
|
|
localtime_with = task.data['localtime_with']
|
|
localtime_without = task.data['localtime_without']
|
|
|
|
self.assertEqual(dateparser.parse(localtime_with), GetLocaltime().do_task(None, None, None, timestamp=str(timestamp), timezone='US/Eastern'))
|
|
self.assertEqual(dateparser.parse(localtime_without), GetLocaltime().do_task(None, None, None, str(timestamp), 'US/Eastern'))
|
|
|
|
def test_get_localtime_no_timestamp(self):
|
|
workflow = self.create_workflow('get_localtime')
|
|
workflow_api = self.get_workflow_api(workflow)
|
|
task = workflow_api.next_task
|
|
|
|
with self.assertRaises(AssertionError):
|
|
self.complete_form(workflow, task, {'with_timestamp': False, 'with_timezone': False})
|