fix timezone in test_list_tagged

This commit is contained in:
Yuming Zhu 2018-05-23 00:33:16 +00:00 committed by Mike McLean
parent 777fdeba28
commit d70f22ee1b
2 changed files with 17 additions and 1 deletions

View file

@ -8,6 +8,8 @@ try:
import unittest2 as unittest
except ImportError:
import unittest
import time
import koji
import koji.util
from .loadkojid import kojid
@ -55,6 +57,7 @@ class TestBuildNotification(unittest.TestCase):
def setUp(self):
self.original_timezone = os.environ.get('TZ')
os.environ['TZ'] = 'US/Eastern'
time.tzset()
self.tempdir = tempfile.mkdtemp()
self.SMTP = mock.patch('smtplib.SMTP').start()
self.session = mock.MagicMock()
@ -67,6 +70,7 @@ class TestBuildNotification(unittest.TestCase):
del os.environ['TZ']
else:
os.environ['TZ'] = self.original_timezone
time.tzset()
mock.patch.stopall()
def test_build_notification(self):

View file

@ -1,4 +1,6 @@
import sys
import os
import time
import mock
@ -11,6 +13,9 @@ class TestCliListTagged(utils.CliTestCase):
maxDiff = None
def setUp(self):
self.original_timezone = os.environ.get('TZ')
os.environ['TZ'] = 'US/Eastern'
time.tzset()
self.error_format = """Usage: %s list-tagged [options] tag [package]
(Specify the --help global option for a list of other help options)
@ -48,6 +53,13 @@ class TestCliListTagged(utils.CliTestCase):
'tag_name': 'tag',
'owner_name': 'owner'}]
def tearDown(self):
if self.original_timezone is None:
del os.environ['TZ']
else:
os.environ['TZ'] = self.original_timezone
time.tzset()
@mock.patch('koji.util.eventFromOpts', return_value={'id': 1000,
'ts': 1000000.11})
@mock.patch('koji_cli.commands.activate_session')
@ -66,7 +78,7 @@ class TestCliListTagged(utils.CliTestCase):
package='pkg')
self.session.listTaggedRPMS.assert_not_called()
self.assert_console_message(sys.stdout,
'Querying at event 1000 (Mon Jan 12 13:46:40 1970)\n'
'Querying at event 1000 (Mon Jan 12 08:46:40 1970)\n'
'Build Tag Built by\n'
'---------------------------------------- -------------------- ----------------\n'
'n-v-r tag owner\n')