Exemplo n.º 1
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)