commit e4235b0ae49ec069c056653dd3764793f2f8a82f
author Kelvin Zhang <zhangkelvin@google.com> Mon Nov 23 13:57:51 2020 -0500
committer Kelvin Zhang <zhangkelvin@google.com> Tue Dec 01 22:13:05 2020 +0000
tree ff0aa9a3dafcf0fbea80edd38dcf6cbaa8ca3ab4
parent 4ce01107651dd3418279fb6aae2193d298cfc96a

Add unittest for resume update

Test: treehugger
Bug: 173392916
Change-Id: I4b320685af93b20fc1f4abcdd6a416f1c97731ab

payload_consumer/delta_performer.cc

diff --git a/payload_consumer/delta_performer.cc b/payload_consumer/delta_performer.cc
index c3a321e..be39542 100644
--- a/payload_consumer/delta_performer.cc
+++ b/payload_consumer/delta_performer.cc
@@ -1443,7 +1443,14 @@
       TEST_AND_RETURN_FALSE(
           prefs_->SetInt64(kPrefsUpdateStateNextDataLength, 0));
     }
-    partition_writer_->CheckpointUpdateProgress(GetPartitionOperationNum());
+    if (partition_writer_) {
+      partition_writer_->CheckpointUpdateProgress(GetPartitionOperationNum());
+    } else {
+      CHECK_EQ(next_operation_num_, num_total_operations_)
+          << "Partition writer is null, we are expected to finish all "
+             "operations: "
+          << next_operation_num_ << "/" << num_total_operations_;
+    }
   }
   TEST_AND_RETURN_FALSE(
       prefs_->SetInt64(kPrefsUpdateStateNextOperation, next_operation_num_));