PNaCl on-demand installs: Make a separate async IPC to check if PNaCl is installed

chrome/browser/nacl_host/nacl_file_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 "chrome/browser/nacl_host/nacl_file_host.h"
 
 #include "base/basictypes.h"
 #include "base/files/file_path.h"
 #include "base/files/scoped_temp_dir.h"
-#include "base/path_service.h"
+#include "base/run_loop.h"
 #include "base/test/scoped_path_override.h"
+#include "base/threading/sequenced_worker_pool.h"
 #include "chrome/browser/nacl_host/nacl_browser.h"
 #include "components/nacl/common/nacl_browser_delegate.h"
+#include "content/public/browser/browser_thread.h"
+#include "content/public/test/test_browser_thread_bundle.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using nacl_file_host::PnaclCanOpenFile;
+using nacl_file_host::EnsurePnaclInstalled;
 
-class DummyNaClBrowserDelegate : public NaClBrowserDelegate {
+class TestNaClBrowserDelegate : public NaClBrowserDelegate {
  public:
+
+  TestNaClBrowserDelegate() : should_pnacl_install_succeed_(false) { }
+
   virtual void ShowNaClInfobar(int render_process_id,
                                int render_view_id,
                                int error_id) OVERRIDE {
   }
 
   virtual bool DialogsAreSuppressed() OVERRIDE {
     return false;
   }
 
   virtual bool GetCacheDirectory(base::FilePath* cache_dir) OVERRIDE {
@@ skipping 17 matching lines @@
     return std::string();
   }
 
   virtual ppapi::host::HostFactory* CreatePpapiHostFactory(
       content::BrowserPpapiHost* ppapi_host) OVERRIDE {
     return NULL;
   }
 
   virtual void TryInstallPnacl(
       const base::Callback<void(bool)>& installed) OVERRIDE {
-    NOTREACHED();
+    installed.Run(should_pnacl_install_succeed_);
   }
 
   void SetPnaclDirectory(const base::FilePath& pnacl_dir) {
     pnacl_path_ = pnacl_dir;
   }
 
+  // Indicate if we should mock the PNaCl install as succeeding
+  // or failing for the next test.
+  void SetShouldPnaclInstallSucceed(bool succeed) {
+    should_pnacl_install_succeed_ = succeed;
+  }
+
  private:
   base::FilePath pnacl_path_;
+  bool should_pnacl_install_succeed_;
 };
 
-// Try to pass a few funny filenames with a dummy pnacl directory set.
-TEST(PnaclFileHostTest, TestFilenamesWithPnaclPath) {
+class NaClFileHostTest : public testing::Test {
+ protected:
+  NaClFileHostTest();
+  virtual ~NaClFileHostTest();
+
+  virtual void SetUp() OVERRIDE {
+    nacl_browser_delegate_ = new TestNaClBrowserDelegate;
+    NaClBrowser::SetDelegate(nacl_browser_delegate_);
+  }
+
+  virtual void TearDown() OVERRIDE {
+    NaClBrowser::SetDelegate(NULL);  // This deletes nacl_browser_delegate_
+  }
+
+  TestNaClBrowserDelegate* nacl_browser_delegate() {
+    return nacl_browser_delegate_;
+  }
+
+  bool install_success() { return install_success_; }
+  size_t install_call_count() {
+    return std::count(events_.begin(), events_.end(), INSTALL_DONE);
+  }
+  size_t progress_call_count() {
+    return std::count(events_.begin(), events_.end(), INSTALL_PROGRESS);
+  }
+  bool events_in_correct_order() {
+    // INSTALL_DONE should be the last thing.
+    // The rest should be progress events.
+    size_t size = events_.size();
+    return size > 0 && events_[size - 1] == INSTALL_DONE
+        && progress_call_count() == (size - 1);
+  }
+
+ public:  // Allow classes to bind these callback methods.
+  void CallbackInstall(bool success) {
+    install_success_ = success;
+    events_.push_back(INSTALL_DONE);
+  }
+
+  void CallbackProgress(int64 current_progress,
+                        int64 total_progress) {
+    events_.push_back(INSTALL_PROGRESS);
+    // TODO(jvoung): be able to check that current_progress
+    // goes up monotonically and hits total_progress at the end,
+    // when we actually get real progress events.
+  }
+
+ private:
+  enum EventType {
+    INSTALL_DONE,
+    INSTALL_PROGRESS
+  };
+  TestNaClBrowserDelegate* nacl_browser_delegate_;
+  bool install_success_;
+  std::vector<EventType> events_;
+  content::TestBrowserThreadBundle thread_bundle_;
+  DISALLOW_COPY_AND_ASSIGN(NaClFileHostTest);
+};
+
+NaClFileHostTest::NaClFileHostTest()
+    : nacl_browser_delegate_(NULL),
+      install_success_(false),
+      thread_bundle_(content::TestBrowserThreadBundle::IO_MAINLOOP) {
+}
+
+NaClFileHostTest::~NaClFileHostTest() {
+}
+
+// Try to pass a few funny filenames with a dummy PNaCl directory set.
+TEST_F(NaClFileHostTest, TestFilenamesWithPnaclPath) {
   base::ScopedTempDir scoped_tmp_dir;
   ASSERT_TRUE(scoped_tmp_dir.CreateUniqueTempDir());
 
-  base::FilePath kDummyPnaclPath = scoped_tmp_dir.path();
+  base::FilePath kTestPnaclPath = scoped_tmp_dir.path();
 
-  DummyNaClBrowserDelegate* nacl_browser_delegate =
-      new DummyNaClBrowserDelegate;
-  nacl_browser_delegate->SetPnaclDirectory(kDummyPnaclPath);
-  NaClBrowser::SetDelegate(nacl_browser_delegate);
-  ASSERT_TRUE(NaClBrowser::GetDelegate()->GetPnaclDirectory(&kDummyPnaclPath));
+  nacl_browser_delegate()->SetPnaclDirectory(kTestPnaclPath);
+  ASSERT_TRUE(NaClBrowser::GetDelegate()->GetPnaclDirectory(&kTestPnaclPath));
 
   // Check allowed strings, and check that the expected prefix is added.
   base::FilePath out_path;
   EXPECT_TRUE(PnaclCanOpenFile("pnacl_json", &out_path));
-  base::FilePath expected_path = kDummyPnaclPath.Append(
+  base::FilePath expected_path = kTestPnaclPath.Append(
       FILE_PATH_LITERAL("pnacl_public_pnacl_json"));
-  EXPECT_EQ(out_path, expected_path);
+  EXPECT_EQ(expected_path, out_path);
 
   EXPECT_TRUE(PnaclCanOpenFile("x86_32_llc", &out_path));
-  expected_path = kDummyPnaclPath.Append(
+  expected_path = kTestPnaclPath.Append(
       FILE_PATH_LITERAL("pnacl_public_x86_32_llc"));
-  EXPECT_EQ(out_path, expected_path);
+  EXPECT_EQ(expected_path, out_path);
 
   // Check character ranges.
   EXPECT_FALSE(PnaclCanOpenFile(".xchars", &out_path));
   EXPECT_FALSE(PnaclCanOpenFile("/xchars", &out_path));
   EXPECT_FALSE(PnaclCanOpenFile("x/chars", &out_path));
   EXPECT_FALSE(PnaclCanOpenFile("\\xchars", &out_path));
   EXPECT_FALSE(PnaclCanOpenFile("x\\chars", &out_path));
   EXPECT_FALSE(PnaclCanOpenFile("$xchars", &out_path));
   EXPECT_FALSE(PnaclCanOpenFile("%xchars", &out_path));
   EXPECT_FALSE(PnaclCanOpenFile("CAPS", &out_path));
@@ skipping 13 matching lines @@
 #if defined(OS_WIN)
   EXPECT_FALSE(PnaclCanOpenFile("..\\llc", &out_path));
   EXPECT_FALSE(PnaclCanOpenFile("%SystemRoot%", &out_path));
   EXPECT_FALSE(PnaclCanOpenFile("%SystemRoot%\\explorer.exe", &out_path));
 #else
   EXPECT_FALSE(PnaclCanOpenFile("../llc", &out_path));
   EXPECT_FALSE(PnaclCanOpenFile("/bin/sh", &out_path));
   EXPECT_FALSE(PnaclCanOpenFile("$HOME", &out_path));
   EXPECT_FALSE(PnaclCanOpenFile("$HOME/.bashrc", &out_path));
 #endif
-  NaClBrowser::SetDelegate(NULL);
 }
+
+// Test that callbacks return success when PNaCl looks like it is
+// already installed. No intermediate progress events are expected.
+TEST_F(NaClFileHostTest, TestEnsureInstalledAlreadyInstalled) {
+  base::ScopedTempDir scoped_tmp_dir;
+  ASSERT_TRUE(scoped_tmp_dir.CreateUniqueTempDir());
+
+  base::FilePath kTestPnaclPath = scoped_tmp_dir.path();
+  nacl_browser_delegate()->SetPnaclDirectory(kTestPnaclPath);
+  ASSERT_TRUE(NaClBrowser::GetDelegate()->GetPnaclDirectory(&kTestPnaclPath));
+
+  EnsurePnaclInstalled(
+      base::Bind(&NaClFileHostTest::CallbackInstall, base::Unretained(this)),
+      base::Bind(&NaClFileHostTest::CallbackProgress, base::Unretained(this)));
+  content::BrowserThread::GetBlockingPool()->FlushForTesting();
+  base::RunLoop().RunUntilIdle();
+
+  EXPECT_TRUE(install_success());
+  EXPECT_EQ(1u, install_call_count());
+  EXPECT_EQ(0u, progress_call_count());
+  EXPECT_TRUE(events_in_correct_order());
+}
+
+// Test that final callback is called with success, when PNaCl does not
+// look like it's already installed, but mock installer says success.
+// Also check that intermediate progress events are called.
+TEST_F(NaClFileHostTest, TestEnsureInstalledNotInstalledSuccess) {
+  base::FilePath kTestPnaclPath;
+  nacl_browser_delegate()->SetPnaclDirectory(kTestPnaclPath);
+  nacl_browser_delegate()->SetShouldPnaclInstallSucceed(true);
+
+  EnsurePnaclInstalled(
+      base::Bind(&NaClFileHostTest::CallbackInstall, base::Unretained(this)),
+      base::Bind(&NaClFileHostTest::CallbackProgress, base::Unretained(this)));
+  content::BrowserThread::GetBlockingPool()->FlushForTesting();
+  base::RunLoop().RunUntilIdle();
+
+  EXPECT_TRUE(install_success());
+  EXPECT_EQ(1u, install_call_count());
+  EXPECT_GE(progress_call_count(), 1u);
+  EXPECT_TRUE(events_in_correct_order());
+}
+
+// Test that final callback is called with error, when PNaCl does not look
+// like it's already installed, but mock installer says error.
+// Also check that intermediate progress events are called.
+TEST_F(NaClFileHostTest, TestEnsureInstalledNotInstalledError) {
+  base::FilePath kTestPnaclPath;
+  nacl_browser_delegate()->SetPnaclDirectory(kTestPnaclPath);
+  nacl_browser_delegate()->SetShouldPnaclInstallSucceed(false);
+
+  EnsurePnaclInstalled(
+      base::Bind(&NaClFileHostTest::CallbackInstall, base::Unretained(this)),
+      base::Bind(&NaClFileHostTest::CallbackProgress, base::Unretained(this)));
+  content::BrowserThread::GetBlockingPool()->FlushForTesting();
+  base::RunLoop().RunUntilIdle();
+
+  EXPECT_FALSE(install_success());
+  EXPECT_EQ(1u, install_call_count());
+  EXPECT_GE(progress_call_count(), 1u);
+  EXPECT_TRUE(events_in_correct_order());
+}