1 files changed, 118 insertions, 0 deletions
diff --git a/chrome/browser/utility_process_host_unittest.cc b/chrome/browser/utility_process_host_unittest.cc
new file mode 100644
index 0000000..b48d95a
--- /dev/null
+++ b/chrome/browser/utility_process_host_unittest.cc
@@ -0,0 +1,118 @@
+// Copyright (c) 2009 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 "chrome/browser/utility_process_host.h"
+
+#include "base/file_path.h"
+#include "base/file_util.h"
+#include "base/logging.h"
+#include "base/message_loop.h"
+#include "base/path_service.h"
+#include "base/string_util.h"
+#include "build/build_config.h"
+#include "chrome/browser/renderer_host/resource_dispatcher_host.h"
+#include "chrome/common/chrome_constants.h"
+#include "chrome/common/chrome_paths.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace {
+
+class UtilityProcessHostTest : public testing::Test {
+ public:
+  UtilityProcessHostTest() {
+  }
+
+ protected:
+  MessageLoopForIO message_loop_;
+};
+
+class TestUtilityProcessHostClient : public UtilityProcessHost::Client {
+ public:
+  explicit TestUtilityProcessHostClient(MessageLoop* message_loop)
+      : message_loop_(message_loop), success_(false) {
+  }
+
+  // UtilityProcessHost::Client methods.
+  virtual void OnProcessCrashed() {
+    NOTREACHED();
+  }
+
+  virtual void OnUnpackExtensionSucceeded(const DictionaryValue& manifest) {
+    success_ = true;
+    message_loop_->PostTask(FROM_HERE, new MessageLoop::QuitTask);
+  }
+
+  virtual void OnUnpackExtensionFailed(const std::string& error_message) {
+    NOTREACHED();
+  }
+
+  virtual void OnUnpackWebResourceSucceeded(
+      const ListValue& json_data) {
+    NOTREACHED();
+  }
+
+  virtual void OnUnpackWebResourceFailed(
+      const std::string& error_message) {
+    NOTREACHED();
+  }
+
+  bool success() {
+    return success_;
+  }
+
+ private:
+  MessageLoop* message_loop_;
+  bool success_;
+};
+
+class TestUtilityProcessHost : public UtilityProcessHost {
+ public:
+  TestUtilityProcessHost(TestUtilityProcessHostClient* client,
+                         MessageLoop* loop_io)
+      : UtilityProcessHost(new ResourceDispatcherHost(NULL), client, loop_io) {
+  }
+
+ protected:
+  virtual std::wstring GetUtilityProcessCmd() {
+    FilePath exe_path;
+    PathService::Get(base::DIR_EXE, &exe_path);
+    exe_path = exe_path.AppendASCII(WideToASCII(
+        chrome::kBrowserProcessExecutablePath));
+    return exe_path.ToWStringHack();
+  }
+
+  virtual bool UseSandbox() {
+    return false;
+  }
+
+ private:
+};
+
+TEST_F(UtilityProcessHostTest, ExtensionUnpacker) {
+  // Copy the test extension into a temp dir and install from the temp dir.
+  FilePath extension_file;
+  ASSERT_TRUE(PathService::Get(chrome::DIR_TEST_DATA, &extension_file));
+  extension_file = extension_file.AppendASCII("extensions")
+                                 .AppendASCII("theme.crx");
+  FilePath temp_extension_dir;
+  ASSERT_TRUE(PathService::Get(base::DIR_TEMP, &temp_extension_dir));
+  temp_extension_dir = temp_extension_dir.AppendASCII("extension_test");
+  ASSERT_TRUE(file_util::CreateDirectory(temp_extension_dir));
+  ASSERT_TRUE(file_util::CopyFile(extension_file,
+                                  temp_extension_dir.AppendASCII("theme.crx")));
+
+  scoped_refptr<TestUtilityProcessHostClient> client(
+      new TestUtilityProcessHostClient(&message_loop_));
+  TestUtilityProcessHost* process_host = new TestUtilityProcessHost(
+      client.get(), &message_loop_);
+  process_host->StartExtensionUnpacker(
+      temp_extension_dir.AppendASCII("theme.crx"));
+  message_loop_.Run();
+  EXPECT_TRUE(client->success());
+
+  // Clean up the temp dir.
+  file_util::Delete(temp_extension_dir, true);
+}
+
+}  // namespace