committed by
JackDandy
4 changed files with 59 additions and 1 deletions
@ -0,0 +1,51 @@ |
|||
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) |
Loading…
Reference in new issue