Ensures that the overdrive new titles script does not fail when cover…

…age records are added.

src/palace/manager/core/coverage.py

@@ -383,7 +383,7 @@ def run_once(self, progress, count_as_covered=None):
             transient_failures,
             persistent_failures,
         ), results = self.process_batch_and_handle_results(batch_results)
-
+        self.finalize_batch()
         # Update the running totals so that the service's eventual timestamp
         # will have a useful .achievements.
         progress.successes += successes
@@ -486,9 +486,6 @@ def process_batch_and_handle_results(self, batch):
             num_ignored,
         )
 
-        # Finalize this batch before moving on to the next one.
-        self.finalize_batch()
-
         # For all purposes outside this method, treat an ignored identifier
         # as a transient failure.
         transient_failures += num_ignored
@@ -910,6 +907,7 @@ def run_on_specific_identifiers(self, identifiers):
         while index < len(need_coverage):
             batch = need_coverage[index : index + self.batch_size]
             (s, t, p), r = self.process_batch_and_handle_results(batch)
+            self.finalize_batch()
             successes += s
             transient_failures += t
             persistent_failures += p

tests/manager/core/test_coverage.py

@@ -535,9 +535,6 @@ def test_process_batch_and_handle_results(self, db: DatabaseTransactionFixture):
         class MockProvider1(AlwaysSuccessfulCoverageProvider):
             OPERATION = "i succeed"
 
-            def finalize_batch(self):
-                self.finalized = True
-
         success_provider = MockProvider1(db.session)
 
         batch = [i1, i2]
@@ -546,9 +543,6 @@ def finalize_batch(self):
         # Two successes.
         assert (2, 0, 0) == counts
 
-        # finalize_batch() was called.
-        assert True == success_provider.finalized
-
         # Each represented with a CoverageRecord with status='success'
         assert all(isinstance(x, CoverageRecord) for x in successes)
         assert [CoverageRecord.SUCCESS] * 2 == [x.status for x in successes]