def setup_torrents(torrents):
for i, torrent in enumerate(torrents):
mock_status = MockObject()
mock_status.total_upload = i * i
mock_status.active_time = i
mock_handle = MockObject()
mock_handle.status = lambda status=mock_status: status
mock_dl = MockObject()
mock_dl.handle = mock_handle
torrent.download = mock_dl
return torrents
def test_upload_policy(self):
for i, torrent in enumerate(self.torrents):
mock_status = MockObject()
mock_status.total_upload = i * i
mock_status.active_time = i
mock_handle = MockObject()
mock_handle.status = lambda status=mock_status: status
mock_dl = MockObject()
mock_dl.handle = mock_handle
torrent.download = mock_dl
policy = UploadPolicy()
sorted_torrents = policy.sort(self.torrents)
expected_torrents = list(reversed(self.torrents))
self.assertItemsEqual(sorted_torrents, expected_torrents, 'Arrays contains different torrents')
self.assertListEqual(sorted_torrents, expected_torrents, 'Array is not sorted properly')
def test_upload_policy(self):
for i, torrent in enumerate(self.torrents):
mock_status = MockObject()
mock_status.total_upload = i * i
mock_status.active_time = i
mock_handle = MockObject()
mock_handle.status = lambda status=mock_status: status
mock_dl = MockObject()
mock_dl.handle = mock_handle
torrent.download = mock_dl
policy = UploadPolicy()
sorted_torrents = policy.sort(self.torrents)
expected_torrents = list(reversed(self.torrents))
self.assertItemsEqual(sorted_torrents, expected_torrents,
'Arrays contains different torrents')
self.assertListEqual(sorted_torrents, expected_torrents,
'Array is not sorted properly')
