AU: Start checkpointing update progress.

download_action_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 <string>
 #include <vector>
+
 #include <glib.h>
 #include <gmock/gmock.h>
 #include <gtest/gtest.h>
+
 #include "update_engine/action_pipe.h"
 #include "update_engine/download_action.h"
 #include "update_engine/mock_http_fetcher.h"
 #include "update_engine/omaha_hash_calculator.h"
+#include "update_engine/prefs_mock.h"
 #include "update_engine/test_utils.h"
 #include "update_engine/utils.h"
 
 namespace chromeos_update_engine {
 
 using std::string;
 using std::vector;
 using testing::_;
 using testing::AtLeast;
 using testing::InSequence;
@@ skipping 75 matching lines @@
       OmahaHashCalculator::OmahaHashOfData(data);
   uint64_t size = data.size() + (size_test ? 1 : 0);
   InstallPlan install_plan(true,
                            "",
                            size,
                            hash,
                            output_temp_file.GetPath(),
                            "");
   ObjectFeederAction<InstallPlan> feeder_action;
   feeder_action.set_obj(install_plan);
-  DownloadAction download_action(new MockHttpFetcher(&data[0],
-                                                     data.size()));
+  PrefsMock prefs;
+  DownloadAction download_action(&prefs, new MockHttpFetcher(&data[0],
+                                                             data.size()));
   download_action.SetTestFileWriter(&writer);
   BondActions(&feeder_action, &download_action);
   DownloadActionDelegateMock download_delegate;
   if (use_download_delegate) {
     InSequence s;
     download_action.set_delegate(&download_delegate);
     EXPECT_CALL(download_delegate, SetDownloadStatus(true)).Times(1);
     EXPECT_CALL(download_delegate, BytesReceived(_, _)).Times(AtLeast(1));
     EXPECT_CALL(download_delegate, SetDownloadStatus(false)).Times(1);
   }
@@ skipping 97 matching lines @@
   memset(&data[0], 0, data.size());
 
   ScopedTempFile temp_file;
   {
     DirectFileWriter writer;
 
     // takes ownership of passed in HttpFetcher
     ObjectFeederAction<InstallPlan> feeder_action;
     InstallPlan install_plan(true, "", 0, "", temp_file.GetPath(), "");
     feeder_action.set_obj(install_plan);
-    DownloadAction download_action(new MockHttpFetcher(&data[0], data.size()));
+    PrefsMock prefs;
+    DownloadAction download_action(&prefs,
+                                   new MockHttpFetcher(&data[0], data.size()));
     download_action.SetTestFileWriter(&writer);
     DownloadActionDelegateMock download_delegate;
     if (use_download_delegate) {
       InSequence s;
       download_action.set_delegate(&download_delegate);
       EXPECT_CALL(download_delegate, SetDownloadStatus(true)).Times(1);
       EXPECT_CALL(download_delegate, SetDownloadStatus(false)).Times(1);
     }
     TerminateEarlyTestProcessorDelegate delegate;
     delegate.loop_ = loop;
@@ skipping 81 matching lines @@
 
   // takes ownership of passed in HttpFetcher
   InstallPlan install_plan(true,
                            "",
                            1,
                            OmahaHashCalculator::OmahaHashOfString("x"),
                            "/dev/null",
                            "/dev/null");
   ObjectFeederAction<InstallPlan> feeder_action;
   feeder_action.set_obj(install_plan);
-  DownloadAction download_action(new MockHttpFetcher("x", 1));
+  PrefsMock prefs;
+  DownloadAction download_action(&prefs, new MockHttpFetcher("x", 1));
   download_action.SetTestFileWriter(&writer);
 
   DownloadActionTestAction test_action;
   test_action.expected_input_object_ = install_plan;
   BondActions(&feeder_action, &download_action);
   BondActions(&download_action, &test_action);
 
   ActionProcessor processor;
   PassObjectOutTestProcessorDelegate delegate;
   delegate.loop_ = loop;
@@ skipping 12 matching lines @@
 TEST(DownloadActionTest, BadOutFileTest) {
   GMainLoop *loop = g_main_loop_new(g_main_context_default(), FALSE);
 
   const string path("/fake/path/that/cant/be/created/because/of/missing/dirs");
   DirectFileWriter writer;
 
   // takes ownership of passed in HttpFetcher
   InstallPlan install_plan(true, "", 0, "", path, "");
   ObjectFeederAction<InstallPlan> feeder_action;
   feeder_action.set_obj(install_plan);
-  DownloadAction download_action(new MockHttpFetcher("x", 1));
+  PrefsMock prefs;
+  DownloadAction download_action(&prefs, new MockHttpFetcher("x", 1));
   download_action.SetTestFileWriter(&writer);
 
   BondActions(&feeder_action, &download_action);
 
   ActionProcessor processor;
   processor.EnqueueAction(&feeder_action);
   processor.EnqueueAction(&download_action);
   processor.StartProcessing();
   ASSERT_FALSE(processor.IsRunning());
 
   g_main_loop_unref(loop);
 }
 
 }  // namespace chromeos_update_engine