def test_getters_setters_2(self): """ Testing various getters and setters in DownloadState """ lt_status = MockObject() lt_status.state = 3 lt_status.upload_rate = 123 lt_status.download_rate = 43 lt_status.total_upload = 100 lt_status.total_download = 200 lt_status.all_time_upload = 100 lt_status.all_time_download = 200 lt_status.list_peers = 10 lt_status.list_seeds = 5 lt_status.progress = 0.75 lt_status.error = False lt_status.paused = False lt_status.state = 3 lt_status.num_pieces = 0 lt_status.pieces = [] lt_status.finished_time = 10 download_state = DownloadState(self.mock_download, lt_status, None) self.assertEqual(download_state.get_status(), DLSTATUS_DOWNLOADING) self.assertEqual(download_state.get_current_speed(UPLOAD), 123) self.assertEqual(download_state.get_current_speed(DOWNLOAD), 43) self.assertEqual(download_state.get_total_transferred(UPLOAD), 100) self.assertEqual(download_state.get_total_transferred(DOWNLOAD), 200) self.assertEqual(download_state.get_seeding_ratio(), 0.5) self.assertEqual(download_state.get_eta(), 0.25) self.assertEqual(download_state.get_num_seeds_peers(), (5, 5)) self.assertEqual(download_state.get_pieces_complete(), []) self.assertEqual(download_state.get_pieces_total_complete(), (0, 0)) self.assertEqual(download_state.get_seeding_time(), 10) lt_status.num_pieces = 6 lt_status.pieces = [1, 1, 1, 0, 0, 0] self.assertEqual(download_state.get_pieces_complete(), [1, 1, 1, 0, 0, 0]) self.assertEqual(download_state.get_pieces_total_complete(), (6, 3)) self.mock_download.config = MockObject() self.mock_download.config.get_selected_files = lambda: ['test'] self.assertEqual(download_state.get_selected_files(), ['test']) self.assertEqual(download_state.get_progress(), 0.75)
def create_mock_status(self): status = MockObject() status.state = 3 status.upload_rate = 123 status.download_rate = 43 status.upload_payload_rate = 123 status.download_payload_rate = 43 status.total_upload = 100 status.total_download = 200 status.all_time_upload = 100 status.all_time_download = 200 status.list_peers = 10 status.list_seeds = 5 status.progress = 0.75 status.error = False status.paused = False status.state = 3 status.num_pieces = 0 status.pieces = [] return status
def test_should_upgrade(self): from tribler_core.upgrade import db72_to_pony pony_db = self.session_base_dir / 'pony.db' # Old DB does not exist self.assertFalse(should_upgrade(self.session_base_dir / 'nonexistent.db', None)) # Old DB is not OK db72_to_pony.old_db_version_ok = lambda _: False self.assertFalse(should_upgrade(OLD_DB_SAMPLE, None)) # Pony DB does not exist db72_to_pony.old_db_version_ok = lambda _: True self.assertTrue(should_upgrade(OLD_DB_SAMPLE, pony_db)) mock_logger = MockObject() mock_logger.error = lambda _,a: None # Bad Pony DB with open(pony_db, 'w') as f: f.write("") self.assertFalse(should_upgrade(OLD_DB_SAMPLE, pony_db, logger=mock_logger))