mirror of
https://github.com/SickGear/SickGear.git
synced 2024-12-11 05:33:37 +00:00
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
|
import unittest
|
||
|
|
import test_lib as test
|
||
|
|
import os.path
|
||
|
|
import datetime
|
||
|
|
from lib.dateutil import tz
|
||
|
|
import sickbeard
|
||
|
|
from sickbeard import network_timezones, helpers
|
||
|
|
import sickbeard.encodingKludge as ek
|
||
|
|
|
||
|
|
|
||
|
|
class NetworkTimezoneTests(test.SickbeardTestDBCase):
|
||
|
|
@classmethod
|
||
|
|
def tearDownClass(cls):
|
||
|
|
super(NetworkTimezoneTests, cls).tearDownClass()
|
||
|
|
cls.remove_zoneinfo()
|
||
|
|
try:
|
||
|
|
os.rmdir(sickbeard.ZONEINFO_DIR)
|
||
|
|
except (StandardError, Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def setUpClass(cls):
|
||
|
|
super(NetworkTimezoneTests, cls).setUpClass()
|
||
|
|
cls.remove_zoneinfo()
|
||
|
|
|
||
|
|
@classmethod
|
||
|
|
def remove_zoneinfo(cls):
|
||
|
|
# delete all existing zoneinfo files
|
||
|
|
for (path, dirs, files) in ek.ek(os.walk, helpers.real_path(sickbeard.ZONEINFO_DIR)):
|
||
|
|
for filename in files:
|
||
|
|
if filename.endswith('.tar.gz'):
|
||
|
|
file_w_path = ek.ek(os.path.join, path, filename)
|
||
|
|
try:
|
||
|
|
ek.ek(os.remove, file_w_path)
|
||
|
|
except (StandardError, Exception):
|
||
|
|
pass
|
||
|
|
|
||
|
|
def test_timezone(self):
|
||
|
|
network_timezones.update_network_dict()
|
||
|
|
network_timezones.sb_timezone = tz.gettz('CET', zoneinfo_priority=True)
|
||
|
|
d = datetime.date(2018, 9, 2).toordinal()
|
||
|
|
t = 'Monday 9:00 PM'
|
||
|
|
network = 'NBC'
|
||
|
|
r = network_timezones.parse_date_time(d, t, network)
|
||
|
|
local_date = datetime.datetime(2018, 9, 3, 3, 0, 0).replace(tzinfo=tz.gettz('CET', zoneinfo_priority=True))
|
||
|
|
self.assertEqual(r, local_date)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
suite = unittest.TestLoader().loadTestsFromTestCase(NetworkTimezoneTests)
|
||
|
|
unittest.TextTestRunner(verbosity=2).run(suite)
|