Remote Assistance on Chrome OS Part III - NativeMessageHost

chrome/browser/extensions/api/messaging/native_message_process_host_unittest.cc

 // Copyright (c) 2012 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 "base/bind.h"
 #include "base/files/file.h"
 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/files/scoped_file.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/json/json_reader.h"
 #include "base/memory/scoped_ptr.h"
-#include "base/memory/weak_ptr.h"
 #include "base/message_loop/message_loop.h"
 #include "base/rand_util.h"
 #include "base/run_loop.h"
 #include "base/strings/stringprintf.h"
 #include "base/test/test_timeouts.h"
 #include "base/threading/platform_thread.h"
 #include "base/threading/sequenced_worker_pool.h"
 #include "base/time/time.h"
 #include "chrome/browser/extensions/api/messaging/native_message_process_host.h"
 #include "chrome/browser/extensions/api/messaging/native_messaging_test_util.h"
@@ skipping 62 matching lines @@
                  base::kNullProcessHandle,
                  read_file_.Pass(), write_file_.Pass());
   }
 
  private:
   mutable base::File read_file_;
   mutable base::File write_file_;
 };
 
 class NativeMessagingTest : public ::testing::Test,
-                            public NativeMessageProcessHost::Client,
+                            public NativeMessageHost::Client,
                             public base::SupportsWeakPtr<NativeMessagingTest> {
  protected:
   NativeMessagingTest()
       : current_channel_(chrome::VersionInfo::CHANNEL_DEV),
         thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP),
         channel_closed_(false) {}
 
   virtual void SetUp() override {
     ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
   }
 
   virtual void TearDown() override {
-    if (native_message_process_host_.get()) {
-      BrowserThread::DeleteSoon(BrowserThread::IO, FROM_HERE,
-                                native_message_process_host_.release());
+    if (native_message_host_.get()) {
+      BrowserThread::DeleteSoon(
+          BrowserThread::IO, FROM_HERE, native_message_host_.release());
     }
     base::RunLoop().RunUntilIdle();
   }
 
-  virtual void PostMessageFromNativeProcess(
-      int port_id,
-      const std::string& message) override  {
+  virtual void PostMessageFromNativeHost(const std::string& message) override {
     last_message_ = message;
 
     // Parse the message.
     base::Value* parsed = base::JSONReader::Read(message);
     base::DictionaryValue* dict_value;
     if (parsed && parsed->GetAsDictionary(&dict_value)) {
       last_message_parsed_.reset(dict_value);
     } else {
       LOG(ERROR) << "Failed to parse " << message;
       last_message_parsed_.reset();
       delete parsed;
     }
 
     if (run_loop_)
       run_loop_->Quit();
   }
 
-  virtual void CloseChannel(int port_id,
-                            const std::string& error_message) override {
+  virtual void CloseChannel(const std::string& error_message) override {
     channel_closed_ = true;
     if (run_loop_)
       run_loop_->Quit();
   }
 
  protected:
   std::string FormatMessage(const std::string& message) {
     uint32_t length = message.length();
     return std::string(reinterpret_cast<char*>(&length), 4).append(message);
   }
 
   base::FilePath CreateTempFileWithMessage(const std::string& message) {
     base::FilePath filename;
     if (!base::CreateTemporaryFileInDir(temp_dir_.path(), &filename))
       return base::FilePath();
 
     std::string message_with_header = FormatMessage(message);
     int bytes_written = base::WriteFile(
         filename, message_with_header.data(), message_with_header.size());
     if (bytes_written < 0 ||
         (message_with_header.size() != static_cast<size_t>(bytes_written))) {
       return base::FilePath();
     }
     return filename;
   }
 
   base::ScopedTempDir temp_dir_;
   // Force the channel to be dev.
   ScopedCurrentChannel current_channel_;
-  scoped_ptr<NativeMessageProcessHost> native_message_process_host_;
+  scoped_ptr<NativeMessageHost> native_message_host_;
   scoped_ptr<base::RunLoop> run_loop_;
   content::TestBrowserThreadBundle thread_bundle_;
   std::string last_message_;
   scoped_ptr<base::DictionaryValue> last_message_parsed_;
   bool channel_closed_;
 };
 
 // Read a single message from a local file.
 TEST_F(NativeMessagingTest, SingleSendMessageRead) {
   base::FilePath temp_output_file = temp_dir_.path().AppendASCII("output");
   base::FilePath temp_input_file = CreateTempFileWithMessage(kTestMessage);
   ASSERT_FALSE(temp_input_file.empty());
 
   scoped_ptr<NativeProcessLauncher> launcher =
       FakeLauncher::Create(temp_input_file, temp_output_file).Pass();
-  native_message_process_host_ = NativeMessageProcessHost::CreateWithLauncher(
-      AsWeakPtr(), ScopedTestNativeMessagingHost::kExtensionId, "empty_app.py",
-      0, launcher.Pass());
-  ASSERT_TRUE(native_message_process_host_.get());
+  native_message_host_ = NativeMessageProcessHost::CreateWithLauncher(
+      ScopedTestNativeMessagingHost::kExtensionId,
+      "empty_app.py",
+      launcher.Pass());
+  native_message_host_->Start(this);
+  ASSERT_TRUE(native_message_host_.get());
   run_loop_.reset(new base::RunLoop());
   run_loop_->RunUntilIdle();
 
   if (last_message_.empty()) {
     run_loop_.reset(new base::RunLoop());
+    scoped_ptr<NativeMessageProcessHost> native_message_process_host_(
+        static_cast<NativeMessageProcessHost*>(native_message_host_.release()));
     native_message_process_host_->ReadNowForTesting();
     run_loop_->Run();
   }
   EXPECT_EQ(kTestMessage, last_message_);
 }
 
 // Tests sending a single message. The message should get written to
 // |temp_file| and should match the contents of single_message_request.msg.
 TEST_F(NativeMessagingTest, SingleSendMessageWrite) {
   base::FilePath temp_output_file = temp_dir_.path().AppendASCII("output");
@@ skipping 17 matching lines @@
 #else  // defined(OS_WIN)
   base::PlatformFile pipe_handles[2];
   ASSERT_EQ(0, pipe(pipe_handles));
   read_file = base::File(pipe_handles[0]);
   base::File write_file(pipe_handles[1]);
 #endif  // !defined(OS_WIN)
 
   scoped_ptr<NativeProcessLauncher> launcher =
       FakeLauncher::CreateWithPipeInput(read_file.Pass(),
                                         temp_output_file).Pass();
-  native_message_process_host_ = NativeMessageProcessHost::CreateWithLauncher(
-      AsWeakPtr(), ScopedTestNativeMessagingHost::kExtensionId, "empty_app.py",
-      0, launcher.Pass());
-  ASSERT_TRUE(native_message_process_host_.get());
+  native_message_host_ = NativeMessageProcessHost::CreateWithLauncher(
+      ScopedTestNativeMessagingHost::kExtensionId,
+      "empty_app.py",
+      launcher.Pass());
+  native_message_host_->Start(this);
+  ASSERT_TRUE(native_message_host_.get());
   base::RunLoop().RunUntilIdle();
 
-  native_message_process_host_->Send(kTestMessage);
+  native_message_host_->OnMessage(kTestMessage);
   base::RunLoop().RunUntilIdle();
 
   std::string output;
   base::TimeTicks start_time = base::TimeTicks::Now();
   while (base::TimeTicks::Now() - start_time < TestTimeouts::action_timeout()) {
     ASSERT_TRUE(base::ReadFileToString(temp_output_file, &output));
     if (!output.empty())
       break;
     base::PlatformThread::YieldCurrentThread();
   }
 
   EXPECT_EQ(FormatMessage(kTestMessage), output);
 }
 
 // Test send message with a real client. The client just echo's back the text
 // it received.
 TEST_F(NativeMessagingTest, EchoConnect) {
   ScopedTestNativeMessagingHost test_host;
   ASSERT_NO_FATAL_FAILURE(test_host.RegisterTestHost(false));
+  std::string error_message;
+  native_message_host_ = NativeMessageProcessHost::Create(
+      NULL,
+      ScopedTestNativeMessagingHost::kExtensionId,
+      ScopedTestNativeMessagingHost::kHostName,
+      false,
+      &error_message);
+  native_message_host_->Start(this);
+  ASSERT_TRUE(native_message_host_.get());
 
-  native_message_process_host_ = NativeMessageProcessHost::Create(
-      NULL, AsWeakPtr(), ScopedTestNativeMessagingHost::kExtensionId,
-      ScopedTestNativeMessagingHost::kHostName, 0, false);
-  ASSERT_TRUE(native_message_process_host_.get());
-
-  native_message_process_host_->Send("{\"text\": \"Hello.\"}");
+  native_message_host_->OnMessage("{\"text\": \"Hello.\"}");
   run_loop_.reset(new base::RunLoop());
   run_loop_->Run();
   ASSERT_FALSE(last_message_.empty());
   ASSERT_TRUE(last_message_parsed_);
 
   std::string expected_url = std::string("chrome-extension://") +
       ScopedTestNativeMessagingHost::kExtensionId + "/";
   int id;
   EXPECT_TRUE(last_message_parsed_->GetInteger("id", &id));
   EXPECT_EQ(1, id);
   std::string text;
   EXPECT_TRUE(last_message_parsed_->GetString("echo.text", &text));
   EXPECT_EQ("Hello.", text);
   std::string url;
   EXPECT_TRUE(last_message_parsed_->GetString("caller_url", &url));
   EXPECT_EQ(expected_url, url);
 
-  native_message_process_host_->Send("{\"foo\": \"bar\"}");
+  native_message_host_->OnMessage("{\"foo\": \"bar\"}");
   run_loop_.reset(new base::RunLoop());
   run_loop_->Run();
   EXPECT_TRUE(last_message_parsed_->GetInteger("id", &id));
   EXPECT_EQ(2, id);
   EXPECT_TRUE(last_message_parsed_->GetString("echo.foo", &text));
   EXPECT_EQ("bar", text);
   EXPECT_TRUE(last_message_parsed_->GetString("caller_url", &url));
   EXPECT_EQ(expected_url, url);
 }
 
 TEST_F(NativeMessagingTest, UserLevel) {
   ScopedTestNativeMessagingHost test_host;
   ASSERT_NO_FATAL_FAILURE(test_host.RegisterTestHost(true));
 
-  native_message_process_host_ = NativeMessageProcessHost::Create(
-      NULL, AsWeakPtr(), ScopedTestNativeMessagingHost::kExtensionId,
-      ScopedTestNativeMessagingHost::kHostName, 0, true);
-  ASSERT_TRUE(native_message_process_host_.get());
+  std::string error_message;
+  native_message_host_ = NativeMessageProcessHost::Create(
+      NULL,
+      ScopedTestNativeMessagingHost::kExtensionId,
+      ScopedTestNativeMessagingHost::kHostName,
+      true,
+      &error_message);
+  native_message_host_->Start(this);
+  ASSERT_TRUE(native_message_host_.get());
 
-  native_message_process_host_->Send("{\"text\": \"Hello.\"}");
+  native_message_host_->OnMessage("{\"text\": \"Hello.\"}");
   run_loop_.reset(new base::RunLoop());
   run_loop_->Run();
   ASSERT_FALSE(last_message_.empty());
   ASSERT_TRUE(last_message_parsed_);
 }
 
 TEST_F(NativeMessagingTest, DisallowUserLevel) {
   ScopedTestNativeMessagingHost test_host;
   ASSERT_NO_FATAL_FAILURE(test_host.RegisterTestHost(true));
 
-  native_message_process_host_ = NativeMessageProcessHost::Create(
-      NULL, AsWeakPtr(), ScopedTestNativeMessagingHost::kExtensionId,
-      ScopedTestNativeMessagingHost::kHostName, 0, false);
-  ASSERT_TRUE(native_message_process_host_.get());
+  std::string error_message;
+  native_message_host_ = NativeMessageProcessHost::Create(
+      NULL,
+      ScopedTestNativeMessagingHost::kExtensionId,
+      ScopedTestNativeMessagingHost::kHostName,
+      false,
+      &error_message);
+  native_message_host_->Start(this);
+  ASSERT_TRUE(native_message_host_.get());
   run_loop_.reset(new base::RunLoop());
   run_loop_->Run();
 
   // The host should fail to start.
   ASSERT_TRUE(channel_closed_);
 }
 
 }  // namespace extensions