AU: Start checkpointing update progress.

omaha_hash_calculator_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 <math.h>
 #include <unistd.h>
 
 #include <vector>
 
 #include <glib.h>
@@ skipping 33 matching lines @@
   OmahaHashCalculator calc;
   calc.Update("h", 1);
   calc.Update("i", 1);
   calc.Finalize();
   EXPECT_EQ(kExpectedHash, calc.hash());
   vector<char> raw_hash(kExpectedRawHash,
                         kExpectedRawHash + arraysize(kExpectedRawHash));
   EXPECT_TRUE(raw_hash == calc.raw_hash());
 }
 
+TEST(OmahaHashCalculatorTest, ContextTest) {
+  OmahaHashCalculator calc;
+  calc.Update("h", 1);
+  OmahaHashCalculator calc_next;
+  calc_next.SetContext(calc.GetContext());
+  calc_next.Update("i", 1);
+  calc_next.Finalize();
+  // Generated by running this on a linux shell:
+  // $ echo -n hi | openssl dgst -sha256 -binary | openssl base64
+  EXPECT_EQ("j0NDRmSPa5bfid2pAcUXaxCm2Dlh3TwayItZstwyeqQ=", calc_next.hash());
+}
+
 TEST(OmahaHashCalculatorTest, BigTest) {
   OmahaHashCalculator calc;
 
   for (int i = 0; i < 1000000; i++) {
     char buf[8];
     ASSERT_EQ(0 == i ? 1 : static_cast<int>(floorf(logf(i) / logf(10))) + 1,
               snprintf(buf, sizeof(buf), "%d", i)) << " i = " << i;
     calc.Update(buf, strlen(buf));
   }
   calc.Finalize();
@@ skipping 12 matching lines @@
   {
     OmahaHashCalculator calc;
   }
   {
     OmahaHashCalculator calc;
     calc.Update("h", 1);
   }
 }
 
 }  // namespace chromeos_update_engine