def test_torrents_missing_episode_or_tvdb_imdb_id(self):
torrents = [{}]
result = ti.deduplicate_torrents(torrents)
self.assertEqual(result, [])
torrents[0]['episode_info'] = {}
result = ti.deduplicate_torrents(torrents)
self.assertEqual(result, [])
torrents[0]['episode_info'] = {'tvdb': '12345'}
self.assertEqual(result, [])
def test_torrents_are_not_episodes(self):
torrents = [{
'title': 'FooBar',
'episode_info': {
'tvdb': '12345',
'imdb': 'tt12345'
}
}]
result = ti.deduplicate_torrents(torrents)
self.assertEqual(result, [])
def test_torrents_are_episodes(self):
torrents1 = [{
'title': 'FooBar',
'episode_info': {
'tvdb': '12345',
'imdb': 'tt12345',
'seasonnum': '01',
'epnum': '01'
}
}]
result = ti.deduplicate_torrents(torrents1)
self.assertEqual(result, torrents1)
torrents2 = [{
'title': 'Foo.s01e01.mp4',
'episode_info': {
'tvdb': '12345',
'imdb': 'tt12345'
}
}]
result = ti.deduplicate_torrents(torrents2)
self.assertEqual(result[0]['episode_info']['seasonnum'], '01')
self.assertEqual(result[0]['episode_info']['epnum'], '01')
def test_differentiating_sd_and_hd_torrents(self):
torrents = [
{
'title': 'Foo.s01e01.mp4',
'episode_info': {
'tvdb': '12345',
'imdb': 'tt12345'
},
'seeders': 10
},
{
'title': 'Foo.s01e01.720p.mkv',
'episode_info': {
'tvdb': '12345',
'imdb': 'tt12345'
},
'seeders': 20
},
]
result = ti.deduplicate_torrents(torrents)
self.assertEqual(len(result), 2)
def test_deduplication_based_on_repack(self):
torrents = [
{
'title': 'Foo.s01e01.mp4',
'episode_info': {
'tvdb': '12345',
'imdb': 'tt12345'
},
'seeders': 20
},
{
'title': 'Foo.s01e01.REPACK.mp4',
'episode_info': {
'tvdb': '12345',
'imdb': 'tt12345'
},
'seeders': 10
},
]
result = ti.deduplicate_torrents(torrents)
self.assertEqual(len(result), 1)
self.assertEqual(result[0]['seeders'], 10)