ParseDateStringActionTestCase

Complexity

Total Complexity

3

Size/Duplication

Total Lines	31
Duplicated Lines	0 %

3 Methods

Rating	Name	Duplication	Size	Complexity
A	test_run_success()	0	19	1
A	test_run_invalid_date_string()	0	3	1
A	assertTimestampMatches()	0	6	1

import datetime

import arrow

from st2tests.base import BaseActionTestCase

from parse_date_string import ParseDateStringAction


class ParseDateStringActionTestCase(BaseActionTestCase):
    def test_run_success(self):
        action = ParseDateStringAction()

        result = action.run(date_string='now')
        expected = arrow.utcnow().timestamp
        self.assertTimestampMatches(result, expected)

        result = action.run(date_string='1 hour ago')
        expected = arrow.utcnow().replace(hours=-1).timestamp
        self.assertTimestampMatches(result, expected)

        result = action.run(date_string='3 days ago')
        expected = arrow.utcnow().replace(days=-3).timestamp
        self.assertTimestampMatches(result, expected)

        result = action.run(date_string='2013-05-12')
        expected = datetime.datetime(2013, 5, 12)
        expected = arrow.get(expected).timestamp
        self.assertEqual(result, expected)

    def test_run_invalid_date_string(self):
        action = ParseDateStringAction()
        self.assertRaises(ValueError, action.run, date_string='some invalid string')

    def assertTimestampMatches(self, actual_ts, expected_ts):
        """
        Custom assert function which allows actual result to drift from the expected one
        for +/- 1 second (this is to account for the time between the function runs).
        """
        self.assertTrue(actual_ts in [expected_ts, (expected_ts + 1), (expected_ts - 1)])