def test_update_progress(self):
tracker = ProgressTracker()
progress = tracker.get_progress()
tracker.add_item("http://test1", 100, "rpm")
tracker.update_progress_download("http://test1", 100, 5)
progress = tracker.get_progress()
self.assertEquals(progress["total_size_bytes"], 100)
self.assertEquals(progress["remaining_bytes"], 95)
self.assertEquals(progress["total_num_items"], 1)
self.assertEquals(progress["remaining_num_items"], 1)
self.assertTrue(progress["type_info"].has_key("rpm"))
self.assertEquals(progress["type_info"]["rpm"]["total_size_bytes"], 100)
self.assertEquals(progress["type_info"]["rpm"]["size_left"], 95)
self.assertEquals(progress["type_info"]["rpm"]["total_count"], 1)
self.assertEquals(progress["type_info"]["rpm"]["items_left"], 1)
self.assertEquals(progress["type_info"]["rpm"]["num_success"], 0)
self.assertEquals(progress["type_info"]["rpm"]["num_error"], 0)
tracker.update_progress_download("http://test1", 100, 90)
progress = tracker.get_progress()
self.assertEquals(progress["total_size_bytes"], 100)
self.assertEquals(progress["remaining_bytes"], 10)
self.assertEquals(progress["total_num_items"], 1)
self.assertEquals(progress["remaining_num_items"], 1)
self.assertTrue(progress["type_info"].has_key("rpm"))
self.assertEquals(progress["type_info"]["rpm"]["total_size_bytes"], 100)
self.assertEquals(progress["type_info"]["rpm"]["size_left"], 10)
self.assertEquals(progress["type_info"]["rpm"]["total_count"], 1)
self.assertEquals(progress["type_info"]["rpm"]["items_left"], 1)
self.assertEquals(progress["type_info"]["rpm"]["num_success"], 0)
self.assertEquals(progress["type_info"]["rpm"]["num_error"], 0)
tracker.update_progress_download("http://test1", 100, 100)
progress = tracker.get_progress()
self.assertEquals(progress["total_size_bytes"], 100)
self.assertEquals(progress["remaining_bytes"], 0)
self.assertEquals(progress["total_num_items"], 1)
self.assertEquals(progress["remaining_num_items"], 1)
self.assertTrue(progress["type_info"].has_key("rpm"))
self.assertEquals(progress["type_info"]["rpm"]["total_size_bytes"], 100)
self.assertEquals(progress["type_info"]["rpm"]["size_left"], 0)
self.assertEquals(progress["type_info"]["rpm"]["total_count"], 1)
self.assertEquals(progress["type_info"]["rpm"]["items_left"], 1)
self.assertEquals(progress["type_info"]["rpm"]["num_success"], 0)
self.assertEquals(progress["type_info"]["rpm"]["num_error"], 0)
tracker.item_complete("http://test1", True)
progress = tracker.get_progress()
self.assertEquals(progress["total_size_bytes"], 100)
self.assertEquals(progress["remaining_bytes"], 0)
self.assertEquals(progress["total_num_items"], 1)
self.assertEquals(progress["remaining_num_items"], 0)
self.assertTrue(progress["type_info"].has_key("rpm"))
self.assertEquals(progress["type_info"]["rpm"]["total_size_bytes"], 100)
self.assertEquals(progress["type_info"]["rpm"]["size_left"], 0)
self.assertEquals(progress["type_info"]["rpm"]["total_count"], 1)
self.assertEquals(progress["type_info"]["rpm"]["items_left"], 0)
self.assertEquals(progress["type_info"]["rpm"]["num_success"], 1)
self.assertEquals(progress["type_info"]["rpm"]["num_error"], 0)
def test_progress_after_refetch_of_item(self):
tracker = ProgressTracker()
progress = tracker.get_progress()
fetchURL = "http://test1"
tracker.add_item(fetchURL, 100, "rpm")
tracker.update_progress_download(fetchURL, 80, 5)
tracker.reset_progress(fetchURL)
progress = tracker.get_progress()
self.assertEquals(progress["total_size_bytes"], 80)
self.assertEquals(progress["remaining_bytes"], 80)
self.assertEquals(progress["total_num_items"], 1)
self.assertEquals(progress["remaining_num_items"], 1)
self.assertTrue(progress["type_info"].has_key("rpm"))
self.assertEquals(progress["type_info"]["rpm"]["total_size_bytes"], 80)
self.assertEquals(progress["type_info"]["rpm"]["size_left"], 80)
self.assertEquals(progress["type_info"]["rpm"]["total_count"], 1)
self.assertEquals(progress["type_info"]["rpm"]["items_left"], 1)
self.assertEquals(progress["type_info"]["rpm"]["num_success"], 0)
self.assertEquals(progress["type_info"]["rpm"]["num_error"], 0)
tracker.update_progress_download(fetchURL, 80, 5)
progress = tracker.get_progress()
self.assertEquals(progress["total_size_bytes"], 80)
self.assertEquals(progress["remaining_bytes"], 75)
tracker.item_complete("http://test1", True)
progress = tracker.get_progress()
self.assertEquals(progress["total_size_bytes"], 80)
self.assertEquals(progress["remaining_bytes"], 0)
self.assertEquals(progress["total_num_items"], 1)
self.assertEquals(progress["remaining_num_items"], 0)
self.assertTrue(progress["type_info"].has_key("rpm"))
self.assertEquals(progress["type_info"]["rpm"]["total_size_bytes"], 80)
self.assertEquals(progress["type_info"]["rpm"]["size_left"], 0)
self.assertEquals(progress["type_info"]["rpm"]["total_count"], 1)
self.assertEquals(progress["type_info"]["rpm"]["items_left"], 0)
self.assertEquals(progress["type_info"]["rpm"]["num_success"], 1)
self.assertEquals(progress["type_info"]["rpm"]["num_error"], 0)
