def test___init__(self):
"""
Test __init__
"""
self.assertEqual(
TorrentProvider('Test Provider').provider_type,
GenericProvider.TORRENT)
def test__get_storage_dir(self):
"""
Test _get_storage_dir
"""
test_cases = [
None, 123, 12.3, '',
os.path.join('some', 'path', 'to', 'folder')
]
for torrent_dir in test_cases:
sickbeard.TORRENT_DIR = torrent_dir
self.assertEqual(
TorrentProvider('Test Provider')._get_storage_dir(),
torrent_dir)
def test_is_active(self):
"""
Test is_active
"""
test_cases = {
(False, False): False,
(False, None): False,
(False, True): False,
(None, False): False,
(None, None): False,
(None, True): False,
(True, False): False,
(True, None): False,
(True, True): True,
}
for ((use_torrents, enabled), result) in test_cases.iteritems():
sickbeard.USE_TORRENTS = use_torrents
provider = TorrentProvider('Test Provider')
provider.enabled = enabled
self.assertEqual(provider.is_active(), result)
def test__get_size(self):
"""
Test _get_size
"""
items_list = [
None, {}, {
'size': None
}, {
'size': ''
}, {
'size': '0'
}, {
'size': '123'
}, {
'size': '12.3'
}, {
'size': '-123'
}, {
'size': '-12.3'
}, {
'size': '1100000'
}, {
'size': 0
}, {
'size': 123
}, {
'size': 12.3
}, {
'size': -123
}, {
'size': -12.3
}, {
'size': 1100000
}, [], [None], [1100000], [None, None, None], [None, None, ''],
[None, None, '0'], [None, None, '123'], [None, None, '12.3'],
[None, None, '-123'],
[None, None, '-12.3'], [None, None, '1100000'], [None, None, 0],
[None, None, 123], [None, None, 12.3], [None, None, -123],
[None, None, -12.3], [None, None, 1100000], (), (None, None, None),
(None, None, ''), (None, None, '0'), (None, None, '123'),
(None, None, '12.3'), (None, None, '-123'), (None, None, '-12.3'),
(None, None, '1100000'), '', '0', '123', '12.3', '-123', '-12.3',
'1100000', 0, 123, 12.3, -123, -12.3, 1100000
]
results_list = [
-1, -1, -1, -1, 0, 123, -1, -123, -1, 1100000, -1, -1, -1, -1, -1,
1100000, -1, -1, -1, -1, -1, 0, 123, -1, -123, -1, 1100000, -1, -1,
-1, -1, -1, 1100000, -1, -1, -1, 0, 123, -1, -123, -1, 1100000, -1,
-1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1, -1
]
unicode_items_list = [{
u'size': None
}, {
u'size': u''
}, {
u'size': u'0'
}, {
u'size': u'123'
}, {
u'size': u'12.3'
}, {
u'size': u'-123'
}, {
u'size': u'-12.3'
}, {
u'size': u'1100000'
}, {
u'size': 0
}, {
u'size': 123
}, {
u'size': 12.3
}, {
u'size': -123
}, {
u'size': -12.3
}, {
u'size': 1100000
}, [None, None, u''], [None, None, u'0'], [None, None, u'123'],
[None, None, u'12.3'], [None, None, u'-123'],
[None, None, u'-12.3'], [None, None, u'1100000'],
(None, None, u''), (None, None, u'0'),
(None, None, u'123'), (None, None, u'12.3'),
(None, None, u'-123'), (None, None, u'-12.3'),
(None, None, u'1100000'), u'', u'0', u'123',
u'12.3', u'-123', u'-12.3', u'1100000']
unicode_results_list = [
-1, -1, 0, 123, -1, -123, -1, 1100000, -1, -1, -1, -1, -1, 1100000,
-1, 0, 123, -1, -123, -1, 1100000, -1, 0, 123, -1, -123, -1,
1100000, -1, -1, -1, -1, -1, -1, -1
]
self.assertEqual(
len(items_list), len(results_list),
'Number of parameters (%d) and results (%d) does not match' %
(len(items_list), len(results_list)))
self.assertEqual(
len(unicode_items_list), len(unicode_results_list),
'Number of parameters (%d) and results (%d) does not match' %
(len(unicode_items_list), len(unicode_results_list)))
for (index, item) in enumerate(items_list):
self.assertEqual(
TorrentProvider('Test Provider')._get_size(item),
results_list[index])
for (index, item) in enumerate(unicode_items_list):
self.assertEqual(
TorrentProvider('Test Provider')._get_size(item),
unicode_results_list[index])
