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])
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)