def test_selected_files(mock_handle, test_download): """ Test whether the selected files are set correctly """ def mocked_set_file_prios(_): mocked_set_file_prios.called = True mocked_set_file_prios.called = False mocked_file = MockObject() mocked_file.path = 'my/path' mock_torrent_info = MockObject() mock_torrent_info.files = lambda: [mocked_file, mocked_file] test_download.handle.prioritize_files = mocked_set_file_prios test_download.handle.get_torrent_info = lambda: mock_torrent_info test_download.handle.rename_file = lambda *_: None test_download.get_share_mode = lambda: False test_download.tdef.get_infohash = lambda: b'a' * 20 test_download.set_selected_files([0]) assert mocked_set_file_prios.called test_download.get_share_mode = lambda: False mocked_set_file_prios.called = False assert not mocked_set_file_prios.called
def test_selected_files(self): """ Test whether the selected files are set correctly """ def mocked_set_file_prios(_): mocked_set_file_prios.called = True mocked_set_file_prios.called = False mocked_file = MockObject() mocked_file.path = 'my/path' mock_torrent_info = MockObject() mock_torrent_info.files = lambda: [mocked_file, mocked_file] self.download.handle.prioritize_files = mocked_set_file_prios self.download.handle.get_torrent_info = lambda: mock_torrent_info self.download.handle.rename_file = lambda *_: None self.download.get_share_mode = lambda: False self.download.tdef.get_infohash = lambda: b'a' * 20 self.download.set_selected_files([0]) self.assertTrue(mocked_set_file_prios.called) self.download.get_share_mode = lambda: False mocked_set_file_prios.called = False self.assertFalse(mocked_set_file_prios.called)