AU: Delta Performer: properly detect idempotent operations

delta_performer_unittest.cc

 // Copyright (c) 2009 The Chromium OS Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include <sys/mount.h>
 #include <inttypes.h>
 
 #include <algorithm>
 #include <string>
 #include <vector>
 
 #include <base/scoped_ptr.h>
 #include <base/string_util.h>
 #include <google/protobuf/repeated_field.h>
 #include <gtest/gtest.h>
 
 #include "update_engine/delta_diff_generator.h"
 #include "update_engine/delta_performer.h"
+#include "update_engine/extent_ranges.h"
 #include "update_engine/graph_types.h"
 #include "update_engine/payload_signer.h"
 #include "update_engine/prefs_mock.h"
 #include "update_engine/test_utils.h"
 #include "update_engine/update_metadata.pb.h"
 #include "update_engine/utils.h"
 
 namespace chromeos_update_engine {
 
 using std::min;
@@ skipping 311 matching lines @@
     EXPECT_EQ(kChunkSize / kBlockSize,
               graph[i].op.dst_extents(0).num_blocks());
     if (graph[i].op.type() !=
         DeltaArchiveManifest_InstallOperation_Type_REPLACE) {
       EXPECT_EQ(DeltaArchiveManifest_InstallOperation_Type_REPLACE_BZ,
                 graph[i].op.type());
     }
   }
 }
 
+TEST(DeltaPerformerTest, IsIdempotentOperationTest) {
+  DeltaArchiveManifest_InstallOperation op;
+  EXPECT_TRUE(DeltaPerformer::IsIdempotentOperation(op));
+  *(op.add_dst_extents()) = ExtentForRange(0, 5);
+  EXPECT_TRUE(DeltaPerformer::IsIdempotentOperation(op));
+  *(op.add_src_extents()) = ExtentForRange(4, 1);
+  EXPECT_FALSE(DeltaPerformer::IsIdempotentOperation(op));
+  op.clear_src_extents();
+  *(op.add_src_extents()) = ExtentForRange(5, 3);
+  EXPECT_TRUE(DeltaPerformer::IsIdempotentOperation(op));
+  *(op.add_dst_extents()) = ExtentForRange(20, 6);
+  EXPECT_TRUE(DeltaPerformer::IsIdempotentOperation(op));
+  *(op.add_src_extents()) = ExtentForRange(19, 2);
+  EXPECT_FALSE(DeltaPerformer::IsIdempotentOperation(op));
+}
+
 }  // namespace chromeos_update_engine