def test_get_index(self):
t = TorrentDef()
t.metainfo_valid = True
t.metainfo = {'info': {'files': [{'path': ['a.txt'], 'length': 123}]}}
self.assertEqual(t.get_index_of_file_in_files('a.txt'), 0)
self.assertRaises(ValueError, t.get_index_of_file_in_files, 'b.txt')
self.assertRaises(ValueError, t.get_index_of_file_in_files, None)
t.metainfo = {'info': {'files': [{'path': ['a.txt'], 'path.utf-8': ['b.txt'], 'length': 123}]}}
self.assertEqual(t.get_index_of_file_in_files('b.txt'), 0)
def test_is_multifile(self):
"""
Test whether a TorrentDef is correctly classified as multifile torrent
"""
tdef = TorrentDef()
self.assertFalse(tdef.is_multifile_torrent())
tdef.metainfo = {}
self.assertFalse(tdef.is_multifile_torrent())
tdef.metainfo = {'info': {'files': ['a']}}
self.assertTrue(tdef.is_multifile_torrent())
def test_get_index(self):
t = TorrentDef()
t.metainfo_valid = True
t.metainfo = {'info': {'files': [{'path': ['a.txt'], 'length': 123}]}}
self.assertEqual(t.get_index_of_file_in_files('a.txt'), 0)
self.assertRaises(ValueError, t.get_index_of_file_in_files, 'b.txt')
self.assertRaises(ValueError, t.get_index_of_file_in_files, None)
t.metainfo = {
'info': {
'files': [{
'path': ['a.txt'],
'path.utf-8': ['b.txt'],
'length': 123
}]
}
}
self.assertEqual(t.get_index_of_file_in_files('b.txt'), 0)
def test_get_nr_pieces(self):
"""
Test getting the number of pieces from a TorrentDef
"""
tdef = TorrentDef()
self.assertEqual(tdef.get_nr_pieces(), 0)
tdef.metainfo = {'info': {'pieces': 'a' * 40}}
self.assertEqual(tdef.get_nr_pieces(), 2)
def test_get_index(self):
"""
Test whether we can successfully get the index of a file in a torrent.
"""
t = TorrentDef()
t.metainfo = {'info': {'files': [{'path': ['a.txt'], 'length': 123}]}}
self.assertEqual(t.get_index_of_file_in_files('a.txt'), 0)
self.assertRaises(ValueError, t.get_index_of_file_in_files, 'b.txt')
self.assertRaises(ValueError, t.get_index_of_file_in_files, None)
t.metainfo = {
'info': {
'files': [{
'path': ['a.txt'],
'path.utf-8': ['b.txt'],
'length': 123
}]
}
}
self.assertEqual(t.get_index_of_file_in_files('b.txt'), 0)
