AU: Update Downloader to support our image formats.

src/platform/update_engine/filesystem_copier_action_unittest.cc

 // Copyright (c) 2010 The Chromium 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 <glib.h>
 #include <set>
 #include <string>
 #include <vector>
 #include <gtest/gtest.h>
 #include "base/string_util.h"
 #include "update_engine/filesystem_copier_action.h"
 #include "update_engine/filesystem_iterator.h"
 #include "update_engine/omaha_hash_calculator.h"
 #include "update_engine/test_utils.h"
 #include "update_engine/utils.h"
 
 using std::set;
 using std::string;
 using std::vector;
 
 namespace chromeos_update_engine {
 
 class FilesystemCopierActionTest : public ::testing::Test {
  protected:
-  void DoTest(bool run_out_of_space, bool terminate_early);
+  void DoTest(bool run_out_of_space,
+              bool terminate_early,
+              bool use_kernel_partition);
   void SetUp() {
   }
   void TearDown() {
   }
 };
 
 class FilesystemCopierActionTestDelegate : public ActionProcessorDelegate {
  public:
   FilesystemCopierActionTestDelegate() : ran_(false), success_(false) {}
   void ExitMainLoop() {
@@ skipping 40 matching lines @@
   processor->StartProcessing();
   if (args->terminate_early) {
     EXPECT_TRUE(args->filesystem_copier_action);
     args->processor->StopProcessing();
   }
   return FALSE;
 }
 
 TEST_F(FilesystemCopierActionTest, RunAsRootSimpleTest) {
   ASSERT_EQ(0, getuid());
-  DoTest(false, false);
+  DoTest(false, false, false);
+
+  DoTest(false, false, true);
 }
 void FilesystemCopierActionTest::DoTest(bool run_out_of_space,
-                                        bool terminate_early) {
+                                        bool terminate_early,
+                                        bool use_kernel_partition) {
   GMainLoop *loop = g_main_loop_new(g_main_context_default(), FALSE);
 
   string a_loop_file;
   string b_loop_file;
   
   EXPECT_TRUE(utils::MakeTempFile("/tmp/a_loop_file.XXXXXX",
                                   &a_loop_file,
                                   NULL));
   ScopedPathUnlinker a_loop_file_unlinker(a_loop_file);
   EXPECT_TRUE(utils::MakeTempFile("/tmp/b_loop_file.XXXXXX",
@@ skipping 25 matching lines @@
   string b_dev = GetUnusedLoopDevice();
   EXPECT_FALSE(b_dev.empty());
   EXPECT_EQ(0, System(StringPrintf("losetup %s %s",
                                    b_dev.c_str(),
                                    b_loop_file.c_str())));
   ScopedLoopbackDeviceReleaser b_dev_releaser(b_dev);
 
   // Set up the action objects
   InstallPlan install_plan;
   install_plan.is_full_update = false;
-  install_plan.install_path = b_dev;
+  if (use_kernel_partition)
+    install_plan.kernel_install_path = b_dev;
+  else
+    install_plan.install_path = b_dev;
 
   ActionProcessor processor;
   FilesystemCopierActionTestDelegate delegate;
   delegate.set_loop(loop);
   processor.set_delegate(&delegate);
 
   ObjectFeederAction<InstallPlan> feeder_action;
-  FilesystemCopierAction copier_action;
+  FilesystemCopierAction copier_action(use_kernel_partition);
   ObjectCollectorAction<InstallPlan> collector_action;
 
   BondActions(&feeder_action, &copier_action);
   BondActions(&copier_action, &collector_action);
 
   processor.EnqueueAction(&feeder_action);
   processor.EnqueueAction(&copier_action);
   processor.EnqueueAction(&collector_action);
 
   copier_action.set_copy_source(a_dev);
@@ skipping 41 matching lines @@
   bool ran_;
   bool success_;
 };
 
 TEST_F(FilesystemCopierActionTest, MissingInputObjectTest) {
   ActionProcessor processor;
   FilesystemCopierActionTest2Delegate delegate;
 
   processor.set_delegate(&delegate);
 
-  FilesystemCopierAction copier_action;
+  FilesystemCopierAction copier_action(false);
   ObjectCollectorAction<InstallPlan> collector_action;
 
   BondActions(&copier_action, &collector_action);
 
   processor.EnqueueAction(&copier_action);
   processor.EnqueueAction(&collector_action);
   processor.StartProcessing();
   EXPECT_FALSE(processor.IsRunning());
   EXPECT_TRUE(delegate.ran_);
   EXPECT_FALSE(delegate.success_);
 }
 
 TEST_F(FilesystemCopierActionTest, FullUpdateTest) {
   ActionProcessor processor;
   FilesystemCopierActionTest2Delegate delegate;
 
   processor.set_delegate(&delegate);
 
   ObjectFeederAction<InstallPlan> feeder_action;
-  InstallPlan install_plan(true, "", "", "");
+  InstallPlan install_plan(true, "", "", "", "");
   feeder_action.set_obj(install_plan);
-  FilesystemCopierAction copier_action;
+  FilesystemCopierAction copier_action(false);
   ObjectCollectorAction<InstallPlan> collector_action;
 
   BondActions(&feeder_action, &copier_action);
   BondActions(&copier_action, &collector_action);
 
   processor.EnqueueAction(&feeder_action);
   processor.EnqueueAction(&copier_action);
   processor.EnqueueAction(&collector_action);
   processor.StartProcessing();
   EXPECT_FALSE(processor.IsRunning());
   EXPECT_TRUE(delegate.ran_);
   EXPECT_TRUE(delegate.success_);
 }
 
 TEST_F(FilesystemCopierActionTest, NonExistentDriveTest) {
   ActionProcessor processor;
   FilesystemCopierActionTest2Delegate delegate;
 
   processor.set_delegate(&delegate);
 
   ObjectFeederAction<InstallPlan> feeder_action;
-  InstallPlan install_plan(false, "", "", "/some/missing/file/path");
+  InstallPlan install_plan(false, "", "", "/no/such/file", "/no/such/file");
   feeder_action.set_obj(install_plan);
-  FilesystemCopierAction copier_action;
+  FilesystemCopierAction copier_action(false);
   ObjectCollectorAction<InstallPlan> collector_action;
 
   BondActions(&copier_action, &collector_action);
 
   processor.EnqueueAction(&feeder_action);
   processor.EnqueueAction(&copier_action);
   processor.EnqueueAction(&collector_action);
   processor.StartProcessing();
   EXPECT_FALSE(processor.IsRunning());
   EXPECT_TRUE(delegate.ran_);
   EXPECT_FALSE(delegate.success_);
 }
 
 TEST_F(FilesystemCopierActionTest, RunAsRootNoSpaceTest) {
   ASSERT_EQ(0, getuid());
-  DoTest(true, false);
+  DoTest(true, false, false);
 }
 
 TEST_F(FilesystemCopierActionTest, RunAsRootTerminateEarlyTest) {
   ASSERT_EQ(0, getuid());
-  DoTest(false, true);
+  DoTest(false, true, false);
 }
 
 }  // namespace chromeos_update_engine