132 lines
4.7 KiB
Python
132 lines
4.7 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
"""Tests for utils module."""
|
|
|
|
import datetime
|
|
import time
|
|
import unittest
|
|
|
|
import mock
|
|
|
|
from satori import utils
|
|
|
|
|
|
class SomeTZ(datetime.tzinfo):
|
|
|
|
"""A random timezone."""
|
|
|
|
def utcoffset(self, dt):
|
|
return datetime.timedelta(minutes=45)
|
|
|
|
def tzname(self, dt):
|
|
return "STZ"
|
|
|
|
def dst(self, dt):
|
|
return datetime.timedelta(0)
|
|
|
|
|
|
class TestTimeUtils(unittest.TestCase):
|
|
|
|
"""Test time formatting functions."""
|
|
|
|
def test_get_formatted_time_string(self):
|
|
some_time = time.gmtime(0)
|
|
with mock.patch.object(utils.time, 'gmtime') as mock_gmt:
|
|
mock_gmt.return_value = some_time
|
|
result = utils.get_time_string()
|
|
self.assertEqual(result, "1970-01-01 00:00:00 +0000")
|
|
|
|
def test_get_formatted_time_string_time_struct(self):
|
|
result = utils.get_time_string(time_obj=time.gmtime(0))
|
|
self.assertEqual(result, "1970-01-01 00:00:00 +0000")
|
|
|
|
def test_get_formatted_time_string_datetime(self):
|
|
result = utils.get_time_string(
|
|
time_obj=datetime.datetime(1970, 2, 1, 1, 2, 3, 0))
|
|
self.assertEqual(result, "1970-02-01 01:02:03 +0000")
|
|
|
|
def test_get_formatted_time_string_datetime_tz(self):
|
|
result = utils.get_time_string(
|
|
time_obj=datetime.datetime(1970, 2, 1, 1, 2, 3, 0, SomeTZ()))
|
|
self.assertEqual(result, "1970-02-01 01:47:03 +0000")
|
|
|
|
def test_parse_time_string(self):
|
|
result = utils.parse_time_string("1970-02-01 01:02:03 +0000")
|
|
self.assertEqual(result, datetime.datetime(1970, 2, 1, 1, 2, 3, 0))
|
|
|
|
def test_parse_time_string_with_tz(self):
|
|
result = utils.parse_time_string("1970-02-01 01:02:03 +1000")
|
|
self.assertEqual(result, datetime.datetime(1970, 2, 1, 11, 2, 3, 0))
|
|
|
|
|
|
class TestGetSource(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
self.function_signature = "def get_my_source_oneline_docstring(self):"
|
|
self.function_oneline_docstring = '"""A beautiful docstring."""'
|
|
self.function_multiline_docstring = ('"""A beautiful docstring.\n\n'
|
|
'Is a terrible thing to '
|
|
'waste.\n"""')
|
|
self.function_body = ['the_problem = "not the problem"',
|
|
'return the_problem']
|
|
|
|
def get_my_source_oneline_docstring(self):
|
|
"""A beautiful docstring."""
|
|
the_problem = "not the problem"
|
|
return the_problem
|
|
|
|
def get_my_source_multiline_docstring(self):
|
|
"""A beautiful docstring.
|
|
|
|
Is a terrible thing to waste.
|
|
"""
|
|
the_problem = "not the problem"
|
|
return the_problem
|
|
|
|
def test_get_source(self):
|
|
nab = utils.get_source_body(self.get_my_source_oneline_docstring)
|
|
self.assertEqual("\n".join(self.function_body), nab)
|
|
|
|
def test_get_source_with_docstring(self):
|
|
nab = utils.get_source_body(self.get_my_source_oneline_docstring,
|
|
with_docstring=True)
|
|
copy = self.function_oneline_docstring + "\n" + "\n".join(
|
|
self.function_body)
|
|
self.assertEqual(copy, nab)
|
|
|
|
def test_get_source_with_multiline_docstring(self):
|
|
nab = utils.get_source_body(self.get_my_source_multiline_docstring,
|
|
with_docstring=True)
|
|
copy = (self.function_multiline_docstring + "\n" + "\n".join(
|
|
self.function_body))
|
|
self.assertEqual(copy, nab)
|
|
|
|
def test_get_definition(self):
|
|
nab = utils.get_source_definition(
|
|
self.get_my_source_oneline_docstring)
|
|
copy = "%s\n \n %s" % (self.function_signature,
|
|
"\n ".join(self.function_body))
|
|
self.assertEqual(copy, nab)
|
|
|
|
def test_get_definition_with_docstring(self):
|
|
nab = utils.get_source_definition(
|
|
self.get_my_source_oneline_docstring, with_docstring=True)
|
|
copy = "%s\n %s\n %s" % (self.function_signature,
|
|
self.function_oneline_docstring,
|
|
"\n ".join(self.function_body))
|
|
self.assertEqual(copy, nab)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|