filesystem_copier_action_unittest.cc - chromiumos/platform/update_engine - Git at Google

 // Copyright (c) 2010 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 <fcntl.h>

 #include <set>
 #include <string>
 #include <vector>

 #include <base/eintr_wrapper.h>
 #include <base/string_util.h>
 #include <glib.h>
 #include <gtest/gtest.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:
   // |verify_hash|: 0 - no hash verification, 1 -- successful hash verification,
   // 2 -- hash verification failure.
   void DoTest(bool run_out_of_space,
               bool terminate_early,
               bool use_kernel_partition,
               int verify_hash);
   void SetUp() {
   }
   void TearDown() {
   }
 };

 class FilesystemCopierActionTestDelegate : public ActionProcessorDelegate {
  public:
   FilesystemCopierActionTestDelegate() : ran_(false), code_(kActionCodeError) {}
   void ExitMainLoop() {
     while (g_main_context_pending(NULL)) {
       g_main_context_iteration(NULL, false);
     }
     g_main_loop_quit(loop_);
   }
   void ProcessingDone(const ActionProcessor* processor, ActionExitCode code) {
     ExitMainLoop();
   }
   void ProcessingStopped(const ActionProcessor* processor) {
     ExitMainLoop();
   }
   void ActionCompleted(ActionProcessor* processor,
                        AbstractAction* action,
                        ActionExitCode code) {
     if (action->Type() == FilesystemCopierAction::StaticType()) {
       ran_ = true;
       code_ = code;
     }
   }
   void set_loop(GMainLoop* loop) {
     loop_ = loop;
   }
   bool ran() { return ran_; }
   ActionExitCode code() { return code_; }
  private:
   GMainLoop* loop_;
   bool ran_;
   ActionExitCode code_;
 };

 struct StartProcessorCallbackArgs {
   ActionProcessor* processor;
   FilesystemCopierAction* filesystem_copier_action;
   bool terminate_early;
 };

 gboolean StartProcessorInRunLoop(gpointer data) {
   StartProcessorCallbackArgs* args =
       reinterpret_cast<StartProcessorCallbackArgs*>(data);
   ActionProcessor* processor = args->processor;
   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, false, 0);
   DoTest(false, false, true, 0);
 }

 void FilesystemCopierActionTest::DoTest(bool run_out_of_space,
                                         bool terminate_early,
                                         bool use_kernel_partition,
                                         int verify_hash) {
   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",
                                   &b_loop_file,
                                   NULL));
   ScopedPathUnlinker b_loop_file_unlinker(b_loop_file);

   // Make random data for a, zero filled data for b.
   const size_t kLoopFileSize = 10 * 1024 * 1024 + 512;
   vector<char> a_loop_data(kLoopFileSize);
   FillWithData(&a_loop_data);
   vector<char> b_loop_data(run_out_of_space ?
                            (kLoopFileSize - 1) :
                            kLoopFileSize,
                            '\0');  // Fill with 0s

   // Write data to disk
   EXPECT_TRUE(WriteFileVector(a_loop_file, a_loop_data));
   EXPECT_TRUE(WriteFileVector(b_loop_file, b_loop_data));

   // Make loop devices for the files
   string a_dev = GetUnusedLoopDevice();
   EXPECT_FALSE(a_dev.empty());
   EXPECT_EQ(0, System(StringPrintf("losetup %s %s",
                                    a_dev.c_str(),
                                    a_loop_file.c_str())));
   ScopedLoopbackDeviceReleaser a_dev_releaser(a_dev);

   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;
   if (verify_hash) {
     if (use_kernel_partition) {
       install_plan.kernel_install_path = a_dev;
       install_plan.kernel_size =
           kLoopFileSize - ((verify_hash == 2) ? 1 : 0);
       EXPECT_TRUE(OmahaHashCalculator::RawHashOfData(
           a_loop_data,
           &install_plan.kernel_hash));
     } else {
       install_plan.install_path = a_dev;
       install_plan.rootfs_size =
           kLoopFileSize - ((verify_hash == 2) ? 1 : 0);
       EXPECT_TRUE(OmahaHashCalculator::RawHashOfData(
           a_loop_data,
           &install_plan.rootfs_hash));
     }
   } else {
     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(use_kernel_partition,
                                        verify_hash != 0);
   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);

   if (!verify_hash) {
     copier_action.set_copy_source(a_dev);
   }
   feeder_action.set_obj(install_plan);

   StartProcessorCallbackArgs start_callback_args;
   start_callback_args.processor = &processor;
   start_callback_args.filesystem_copier_action = &copier_action;
   start_callback_args.terminate_early = terminate_early;

   g_timeout_add(0, &StartProcessorInRunLoop, &start_callback_args);
   g_main_loop_run(loop);
   g_main_loop_unref(loop);

   if (!terminate_early)
     EXPECT_TRUE(delegate.ran());
   if (run_out_of_space || terminate_early) {
     EXPECT_EQ(kActionCodeError, delegate.code());
     return;
   }
   if (verify_hash == 2) {
     EXPECT_EQ(use_kernel_partition ?
               kActionCodeNewKernelVerificationError :
               kActionCodeNewRootfsVerificationError,
               delegate.code());
     return;
   }
   EXPECT_EQ(kActionCodeSuccess, delegate.code());

   // Make sure everything in the out_image is there
   vector<char> a_out;
   EXPECT_TRUE(utils::ReadFile(a_dev, &a_out));
   if (!verify_hash) {
     vector<char> b_out;
     EXPECT_TRUE(utils::ReadFile(b_dev, &b_out));
     EXPECT_TRUE(ExpectVectorsEq(a_out, b_out));
   }
   EXPECT_TRUE(ExpectVectorsEq(a_loop_data, a_out));

   EXPECT_TRUE(collector_action.object() == install_plan);
 }

 class FilesystemCopierActionTest2Delegate : public ActionProcessorDelegate {
  public:
   void ActionCompleted(ActionProcessor* processor,
                        AbstractAction* action,
                        ActionExitCode code) {
     if (action->Type() == FilesystemCopierAction::StaticType()) {
       ran_ = true;
       code_ = code;
     }
   }
   GMainLoop *loop_;
   bool ran_;
   ActionExitCode code_;
 };

 TEST_F(FilesystemCopierActionTest, MissingInputObjectTest) {
   ActionProcessor processor;
   FilesystemCopierActionTest2Delegate delegate;

   processor.set_delegate(&delegate);

   FilesystemCopierAction copier_action(false, 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_EQ(kActionCodeError, delegate.code_);
 }

 TEST_F(FilesystemCopierActionTest, FullUpdateTest) {
   ActionProcessor processor;
   FilesystemCopierActionTest2Delegate delegate;

   processor.set_delegate(&delegate);

   ObjectFeederAction<InstallPlan> feeder_action;
   const char* kUrl = "http://some/url";
   InstallPlan install_plan(true, false, kUrl, 0, "", "", "");
   feeder_action.set_obj(install_plan);
   FilesystemCopierAction copier_action(false, 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_EQ(kActionCodeSuccess, delegate.code_);
   EXPECT_EQ(kUrl, collector_action.object().download_url);
 }

 TEST_F(FilesystemCopierActionTest, ResumeTest) {
   ActionProcessor processor;
   FilesystemCopierActionTest2Delegate delegate;

   processor.set_delegate(&delegate);

   ObjectFeederAction<InstallPlan> feeder_action;
   const char* kUrl = "http://some/url";
   InstallPlan install_plan(false, true, kUrl, 0, "", "", "");
   feeder_action.set_obj(install_plan);
   FilesystemCopierAction copier_action(false, 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_EQ(kActionCodeSuccess, delegate.code_);
   EXPECT_EQ(kUrl, collector_action.object().download_url);
 }

 TEST_F(FilesystemCopierActionTest, NonExistentDriveTest) {
   ActionProcessor processor;
   FilesystemCopierActionTest2Delegate delegate;

   processor.set_delegate(&delegate);

   ObjectFeederAction<InstallPlan> feeder_action;
   InstallPlan install_plan(false,
                            false,
                            "",
                            0,
                            "",
                            "/no/such/file",
                            "/no/such/file");
   feeder_action.set_obj(install_plan);
   FilesystemCopierAction copier_action(false, 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_EQ(kActionCodeError, delegate.code_);
 }

 TEST_F(FilesystemCopierActionTest, RunAsRootVerifyHashTest) {
   ASSERT_EQ(0, getuid());
   DoTest(false, false, false, 1);
   DoTest(false, false, true, 1);
 }

 TEST_F(FilesystemCopierActionTest, RunAsRootVerifyHashFailTest) {
   ASSERT_EQ(0, getuid());
   DoTest(false, false, false, 2);
   DoTest(false, false, true, 2);
 }

 TEST_F(FilesystemCopierActionTest, RunAsRootNoSpaceTest) {
   ASSERT_EQ(0, getuid());
   DoTest(true, false, false, 0);
 }

 TEST_F(FilesystemCopierActionTest, RunAsRootTerminateEarlyTest) {
   ASSERT_EQ(0, getuid());
   DoTest(false, true, false, 0);
 }

 TEST_F(FilesystemCopierActionTest, RunAsRootDetermineFilesystemSizeTest) {
   string img;
   EXPECT_TRUE(utils::MakeTempFile("/tmp/img.XXXXXX", &img, NULL));
   ScopedPathUnlinker img_unlinker(img);
   CreateExtImageAtPath(img, NULL);
   // Extend the "partition" holding the file system from 10MiB to 20MiB.
   EXPECT_EQ(0, System(StringPrintf(
       "dd if=/dev/zero of=%s seek=20971519 bs=1 count=1",
       img.c_str())));
   EXPECT_EQ(20 * 1024 * 1024, utils::FileSize(img));

   for (int i = 0; i < 2; ++i) {
     bool is_kernel = i == 1;
     FilesystemCopierAction action(is_kernel, false);
     EXPECT_EQ(kint64max, action.filesystem_size_);
     {
       int fd = HANDLE_EINTR(open(img.c_str(), O_RDONLY));
       EXPECT_TRUE(fd > 0);
       ScopedFdCloser fd_closer(&fd);
       action.DetermineFilesystemSize(fd);
     }
     EXPECT_EQ(is_kernel ? kint64max : 10 * 1024 * 1024,
               action.filesystem_size_);
   }
 }


 }  // namespace chromeos_update_engine
	// Copyright (c) 2010 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 <fcntl.h>

	#include <set>
	#include <string>
	#include <vector>

	#include <base/eintr_wrapper.h>
	#include <base/string_util.h>
	#include <glib.h>
	#include <gtest/gtest.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:
	// \|verify_hash\|: 0 - no hash verification, 1 -- successful hash verification,
	// 2 -- hash verification failure.
	void DoTest(bool run_out_of_space,
	bool terminate_early,
	bool use_kernel_partition,
	int verify_hash);
	void SetUp() {
	}
	void TearDown() {
	}
	};

	class FilesystemCopierActionTestDelegate : public ActionProcessorDelegate {
	public:
	FilesystemCopierActionTestDelegate() : ran_(false), code_(kActionCodeError) {}
	void ExitMainLoop() {
	while (g_main_context_pending(NULL)) {
	g_main_context_iteration(NULL, false);
	}
	g_main_loop_quit(loop_);
	}
	void ProcessingDone(const ActionProcessor* processor, ActionExitCode code) {
	ExitMainLoop();
	}
	void ProcessingStopped(const ActionProcessor* processor) {
	ExitMainLoop();
	}
	void ActionCompleted(ActionProcessor* processor,
	AbstractAction* action,
	ActionExitCode code) {
	if (action->Type() == FilesystemCopierAction::StaticType()) {
	ran_ = true;
	code_ = code;
	}
	}
	void set_loop(GMainLoop* loop) {
	loop_ = loop;
	}
	bool ran() { return ran_; }
	ActionExitCode code() { return code_; }
	private:
	GMainLoop* loop_;
	bool ran_;
	ActionExitCode code_;
	};

	struct StartProcessorCallbackArgs {
	ActionProcessor* processor;
	FilesystemCopierAction* filesystem_copier_action;
	bool terminate_early;
	};

	gboolean StartProcessorInRunLoop(gpointer data) {
	StartProcessorCallbackArgs* args =
	reinterpret_cast<StartProcessorCallbackArgs*>(data);
	ActionProcessor* processor = args->processor;
	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, false, 0);
	DoTest(false, false, true, 0);
	}

	void FilesystemCopierActionTest::DoTest(bool run_out_of_space,
	bool terminate_early,
	bool use_kernel_partition,
	int verify_hash) {
	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",
	&b_loop_file,
	NULL));
	ScopedPathUnlinker b_loop_file_unlinker(b_loop_file);

	// Make random data for a, zero filled data for b.
	const size_t kLoopFileSize = 10 * 1024 * 1024 + 512;
	vector<char> a_loop_data(kLoopFileSize);
	FillWithData(&a_loop_data);
	vector<char> b_loop_data(run_out_of_space ?
	(kLoopFileSize - 1) :
	kLoopFileSize,
	'\0'); // Fill with 0s

	// Write data to disk
	EXPECT_TRUE(WriteFileVector(a_loop_file, a_loop_data));
	EXPECT_TRUE(WriteFileVector(b_loop_file, b_loop_data));

	// Make loop devices for the files
	string a_dev = GetUnusedLoopDevice();
	EXPECT_FALSE(a_dev.empty());
	EXPECT_EQ(0, System(StringPrintf("losetup %s %s",
	a_dev.c_str(),
	a_loop_file.c_str())));
	ScopedLoopbackDeviceReleaser a_dev_releaser(a_dev);

	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;
	if (verify_hash) {
	if (use_kernel_partition) {
	install_plan.kernel_install_path = a_dev;
	install_plan.kernel_size =
	kLoopFileSize - ((verify_hash == 2) ? 1 : 0);
	EXPECT_TRUE(OmahaHashCalculator::RawHashOfData(
	a_loop_data,
	&install_plan.kernel_hash));
	} else {
	install_plan.install_path = a_dev;
	install_plan.rootfs_size =
	kLoopFileSize - ((verify_hash == 2) ? 1 : 0);
	EXPECT_TRUE(OmahaHashCalculator::RawHashOfData(
	a_loop_data,
	&install_plan.rootfs_hash));
	}
	} else {
	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(use_kernel_partition,
	verify_hash != 0);
	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);

	if (!verify_hash) {
	copier_action.set_copy_source(a_dev);
	}
	feeder_action.set_obj(install_plan);

	StartProcessorCallbackArgs start_callback_args;
	start_callback_args.processor = &processor;
	start_callback_args.filesystem_copier_action = &copier_action;
	start_callback_args.terminate_early = terminate_early;

	g_timeout_add(0, &StartProcessorInRunLoop, &start_callback_args);
	g_main_loop_run(loop);
	g_main_loop_unref(loop);

	if (!terminate_early)
	EXPECT_TRUE(delegate.ran());
	if (run_out_of_space \|\| terminate_early) {
	EXPECT_EQ(kActionCodeError, delegate.code());
	return;
	}
	if (verify_hash == 2) {
	EXPECT_EQ(use_kernel_partition ?
	kActionCodeNewKernelVerificationError :
	kActionCodeNewRootfsVerificationError,
	delegate.code());
	return;
	}
	EXPECT_EQ(kActionCodeSuccess, delegate.code());

	// Make sure everything in the out_image is there
	vector<char> a_out;
	EXPECT_TRUE(utils::ReadFile(a_dev, &a_out));
	if (!verify_hash) {
	vector<char> b_out;
	EXPECT_TRUE(utils::ReadFile(b_dev, &b_out));
	EXPECT_TRUE(ExpectVectorsEq(a_out, b_out));
	}
	EXPECT_TRUE(ExpectVectorsEq(a_loop_data, a_out));

	EXPECT_TRUE(collector_action.object() == install_plan);
	}

	class FilesystemCopierActionTest2Delegate : public ActionProcessorDelegate {
	public:
	void ActionCompleted(ActionProcessor* processor,
	AbstractAction* action,
	ActionExitCode code) {
	if (action->Type() == FilesystemCopierAction::StaticType()) {
	ran_ = true;
	code_ = code;
	}
	}
	GMainLoop *loop_;
	bool ran_;
	ActionExitCode code_;
	};

	TEST_F(FilesystemCopierActionTest, MissingInputObjectTest) {
	ActionProcessor processor;
	FilesystemCopierActionTest2Delegate delegate;

	processor.set_delegate(&delegate);

	FilesystemCopierAction copier_action(false, 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_EQ(kActionCodeError, delegate.code_);
	}

	TEST_F(FilesystemCopierActionTest, FullUpdateTest) {
	ActionProcessor processor;
	FilesystemCopierActionTest2Delegate delegate;

	processor.set_delegate(&delegate);

	ObjectFeederAction<InstallPlan> feeder_action;
	const char* kUrl = "http://some/url";
	InstallPlan install_plan(true, false, kUrl, 0, "", "", "");
	feeder_action.set_obj(install_plan);
	FilesystemCopierAction copier_action(false, 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_EQ(kActionCodeSuccess, delegate.code_);
	EXPECT_EQ(kUrl, collector_action.object().download_url);
	}

	TEST_F(FilesystemCopierActionTest, ResumeTest) {
	ActionProcessor processor;
	FilesystemCopierActionTest2Delegate delegate;

	processor.set_delegate(&delegate);

	ObjectFeederAction<InstallPlan> feeder_action;
	const char* kUrl = "http://some/url";
	InstallPlan install_plan(false, true, kUrl, 0, "", "", "");
	feeder_action.set_obj(install_plan);
	FilesystemCopierAction copier_action(false, 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_EQ(kActionCodeSuccess, delegate.code_);
	EXPECT_EQ(kUrl, collector_action.object().download_url);
	}

	TEST_F(FilesystemCopierActionTest, NonExistentDriveTest) {
	ActionProcessor processor;
	FilesystemCopierActionTest2Delegate delegate;

	processor.set_delegate(&delegate);

	ObjectFeederAction<InstallPlan> feeder_action;
	InstallPlan install_plan(false,
	false,
	"",
	0,
	"",
	"/no/such/file",
	"/no/such/file");
	feeder_action.set_obj(install_plan);
	FilesystemCopierAction copier_action(false, 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_EQ(kActionCodeError, delegate.code_);
	}

	TEST_F(FilesystemCopierActionTest, RunAsRootVerifyHashTest) {
	ASSERT_EQ(0, getuid());
	DoTest(false, false, false, 1);
	DoTest(false, false, true, 1);
	}

	TEST_F(FilesystemCopierActionTest, RunAsRootVerifyHashFailTest) {
	ASSERT_EQ(0, getuid());
	DoTest(false, false, false, 2);
	DoTest(false, false, true, 2);
	}

	TEST_F(FilesystemCopierActionTest, RunAsRootNoSpaceTest) {
	ASSERT_EQ(0, getuid());
	DoTest(true, false, false, 0);
	}

	TEST_F(FilesystemCopierActionTest, RunAsRootTerminateEarlyTest) {
	ASSERT_EQ(0, getuid());
	DoTest(false, true, false, 0);
	}

	TEST_F(FilesystemCopierActionTest, RunAsRootDetermineFilesystemSizeTest) {
	string img;
	EXPECT_TRUE(utils::MakeTempFile("/tmp/img.XXXXXX", &img, NULL));
	ScopedPathUnlinker img_unlinker(img);
	CreateExtImageAtPath(img, NULL);
	// Extend the "partition" holding the file system from 10MiB to 20MiB.
	EXPECT_EQ(0, System(StringPrintf(
	"dd if=/dev/zero of=%s seek=20971519 bs=1 count=1",
	img.c_str())));
	EXPECT_EQ(20 * 1024 * 1024, utils::FileSize(img));

	for (int i = 0; i < 2; ++i) {
	bool is_kernel = i == 1;
	FilesystemCopierAction action(is_kernel, false);
	EXPECT_EQ(kint64max, action.filesystem_size_);
	{
	int fd = HANDLE_EINTR(open(img.c_str(), O_RDONLY));
	EXPECT_TRUE(fd > 0);
	ScopedFdCloser fd_closer(&fd);
	action.DetermineFilesystemSize(fd);
	}
	EXPECT_EQ(is_kernel ? kint64max : 10 * 1024 * 1024,
	action.filesystem_size_);
	}
	}


	} // namespace chromeos_update_engine