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