app_list: Drive app integration.

chrome/browser/apps/drive/drive_app_provider_browsertest.cc

Index: chrome/browser/apps/drive/drive_app_provider_browsertest.cc
diff --git a/chrome/browser/apps/drive/drive_app_provider_browsertest.cc b/chrome/browser/apps/drive/drive_app_provider_browsertest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..075714587245550ba821c1f99720930c7202d5ce
--- /dev/null
+++ b/chrome/browser/apps/drive/drive_app_provider_browsertest.cc
@@ -0,0 +1,405 @@
+// Copyright 2014 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/apps/drive/drive_app_provider.h"
+
+#include "base/logging.h"
+#include "base/macros.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/path_service.h"
+#include "base/timer/timer.h"
+#include "chrome/browser/apps/drive/drive_app_mapping.h"
+#include "chrome/browser/apps/drive/drive_service_bridge.h"
+#include "chrome/browser/drive/drive_app_registry.h"
+#include "chrome/browser/drive/fake_drive_service.h"
+#include "chrome/browser/extensions/extension_browsertest.h"
+#include "chrome/browser/extensions/install_tracker.h"
+#include "chrome/browser/extensions/install_tracker_factory.h"
+#include "chrome/common/chrome_paths.h"
+#include "chrome/common/extensions/manifest_handlers/app_launch_info.h"
+#include "content/public/test/test_utils.h"
+#include "extensions/browser/extension_registry.h"
+#include "sync/api/string_ordinal.h"
+#include "ui/app_list/app_list_folder_item.h"
+#include "ui/app_list/app_list_item.h"
+#include "ui/app_list/app_list_item_list.h"
+#include "ui/app_list/test/app_list_test_model.h"
+
+using extensions::AppLaunchInfo;
+using extensions::Extension;
+using extensions::ExtensionRegistry;
+
+namespace {
+
+const char kDriveAppId[] = "drive_app_id";
+const char kDriveAppName[] = "Fake Drive App";
+const char kLaunchUrl[] = "http://example.com/drive";
+
+// App id of hosted_app.crx.
+const char kChromeAppId[] = "kbmnembihfiondgfjekmnmcbddelicoi";
+
+// Stub drive service bridge.
+class TestDriveServiceBridge : public DriveServiceBridge {
+ public:
+  explicit TestDriveServiceBridge(drive::DriveAppRegistry* registry)
+      : registry_(registry) {}
+  virtual ~TestDriveServiceBridge() {}
+
+  virtual drive::DriveAppRegistry* GetAppRegistry() OVERRIDE {
+    return registry_;
+  }
+
+ private:
+  drive::DriveAppRegistry* registry_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestDriveServiceBridge);
+};
+
+// A model builder that creates and deletes items of a given model when app
+// is installed and uninstalled in the given profile.
+class TestModelBuilder : public extensions::InstallObserver {
+ public:
+  TestModelBuilder(Profile* profile, app_list::test::AppListTestModel* model)
+      : profile_(profile), model_(model) {
+    extensions::InstallTracker::Get(profile_)->AddObserver(this);
+  }
+  virtual ~TestModelBuilder() {
+    extensions::InstallTracker::Get(profile_)->RemoveObserver(this);
+  }
+
+  // extensions::InstallObserver
+  virtual void OnExtensionLoaded(const Extension* extension) OVERRIDE {
+    model_->CreateAndAddItem(extension->id());
+  }
+  virtual void OnExtensionUninstalled(const Extension* extension) OVERRIDE {
+    model_->DeleteItem(extension->id());
+  }
+
+ private:
+  Profile* profile_;
+  app_list::test::AppListTestModel* model_;
+
+  DISALLOW_COPY_AND_ASSIGN(TestModelBuilder);
+};
+
+}  // namespace
+
+class DriveAppProviderTest : public ExtensionBrowserTest {
+ public:
+  DriveAppProviderTest() {}
+  virtual ~DriveAppProviderTest() {}
+
+  // ExtensionBrowserTest:
+  virtual void SetUpOnMainThread() OVERRIDE {
+    ExtensionBrowserTest::SetUpOnMainThread();
+
+    fake_drive_service_.reset(new drive::FakeDriveService);
+    fake_drive_service_->LoadAppListForDriveApi("drive/applist_empty.json");
+    apps_registry_.reset(
+        new drive::DriveAppRegistry(fake_drive_service_.get()));
+
+    model_.reset(new app_list::test::AppListTestModel);
+    model_builder_.reset(new TestModelBuilder(profile(), model_.get()));
+
+    provider_.reset(new DriveAppProvider(profile(), model_.get()));
+    provider_->SetDriveServiceBridgeForTest(
+        make_scoped_ptr(new TestDriveServiceBridge(apps_registry_.get()))
+            .PassAs<DriveServiceBridge>());
+  }
+
+  virtual void CleanUpOnMainThread() OVERRIDE {
+    provider_.reset();
+    apps_registry_.reset();
+    fake_drive_service_.reset();
+    model_builder_.reset();
+
+    ExtensionBrowserTest::CleanUpOnMainThread();
+  }
+
+  const Extension* InstallChromeApp(int expected_change) {
+    base::FilePath test_data_path;
+    if (!PathService::Get(chrome::DIR_TEST_DATA, &test_data_path)) {
+      ADD_FAILURE();
+      return NULL;
+    }
+    test_data_path =
+        test_data_path.AppendASCII("extensions").AppendASCII("hosted_app.crx");
+    const Extension* extension =
+        InstallExtension(test_data_path, expected_change);
+    return extension;
+  }
+
+  void RefreshDriveAppRegistry() {
+    apps_registry_->Update();
+    content::RunAllPendingInMessageLoop();
+  }
+
+  void WaitForPendingDriveAppConverters() {
+    DCHECK(!runner_);
+
+    if (provider_->pending_converters_.empty())
+      return;
+
+    runner_ = new content::MessageLoopRunner;
+
+    pending_drive_app_converter_check_timer_.Start(
+        FROM_HERE,
+        base::TimeDelta::FromMilliseconds(50),
+        base::Bind(&DriveAppProviderTest::OnPendingDriveAppConverterCheckTimer,
+                   base::Unretained(this)));
+
+    runner_->Run();
+
+    pending_drive_app_converter_check_timer_.Stop();
+    runner_ = NULL;
+  }
+
+  drive::FakeDriveService* fake_drive_service() {
+    return fake_drive_service_.get();
+  }
+  app_list::test::AppListTestModel* model() { return model_.get(); }
+  DriveAppProvider* provider() { return provider_.get(); }
+  DriveAppMapping* mapping() { return provider_->mapping_.get(); }
+
+ private:
+  void OnPendingDriveAppConverterCheckTimer() {
+    if (provider_->pending_converters_.empty())
+      runner_->Quit();
+  }
+
+  scoped_ptr<drive::FakeDriveService> fake_drive_service_;
+  scoped_ptr<drive::DriveAppRegistry> apps_registry_;
+  scoped_ptr<TestModelBuilder> model_builder_;
+  scoped_ptr<app_list::test::AppListTestModel> model_;
+  scoped_ptr<DriveAppProvider> provider_;
+
+  base::RepeatingTimer<DriveAppProviderTest>
+      pending_drive_app_converter_check_timer_;
+  scoped_refptr<content::MessageLoopRunner> runner_;
+
+  DISALLOW_COPY_AND_ASSIGN(DriveAppProviderTest);
+};
+
+// A Drive app maps to an existing Chrome app that has a matching id.
+// Uninstalling the chrome app would also disconnect the drive app.
+IN_PROC_BROWSER_TEST_F(DriveAppProviderTest, ExistingChromeApp) {
+  // Prepare an existing chrome app.
+  const Extension* chrome_app = InstallChromeApp(1);
+  ASSERT_TRUE(chrome_app);
+
+  // Prepare a Drive app that matches the chrome app id.
+  fake_drive_service()->AddApp(
+      kDriveAppId, kDriveAppName, chrome_app->id(), kLaunchUrl);
+  RefreshDriveAppRegistry();
+  WaitForPendingDriveAppConverters();
+
+  // The Drive app should use the matching chrome app.
+  EXPECT_EQ(chrome_app->id(), mapping()->GetChromeApp(kDriveAppId));
+
+  // Unintalling chrome app should disconnect the Drive app on server.
+  EXPECT_TRUE(fake_drive_service()->HasApp(kDriveAppId));
+  UninstallExtension(chrome_app->id());
+  EXPECT_FALSE(fake_drive_service()->HasApp(kDriveAppId));
+}
+
+// A Drive app creates an URL app when no matching Chrome app presents.
+IN_PROC_BROWSER_TEST_F(DriveAppProviderTest, CreateUrlApp) {
+  // Prepare a Drive app with no underlying chrome app.
+  fake_drive_service()->AddApp(kDriveAppId, kDriveAppName, "", kLaunchUrl);
+  RefreshDriveAppRegistry();
+  WaitForPendingDriveAppConverters();
+
+  // An Url app should be created.
+  const Extension* chrome_app =
+      ExtensionRegistry::Get(profile())->GetExtensionById(
+          mapping()->GetChromeApp(kDriveAppId), ExtensionRegistry::EVERYTHING);
+  ASSERT_TRUE(chrome_app);
+  EXPECT_EQ(kDriveAppName, chrome_app->name());
+  EXPECT_TRUE(chrome_app->is_hosted_app());
+  EXPECT_TRUE(chrome_app->from_bookmark());
+  EXPECT_EQ(GURL(kLaunchUrl), AppLaunchInfo::GetLaunchWebURL(chrome_app));
+
+  EXPECT_EQ(chrome_app->id(), mapping()->GetChromeApp(kDriveAppId));
+
+  // Unintalling the chrome app should disconnect the Drive app on server.
+  EXPECT_TRUE(fake_drive_service()->HasApp(kDriveAppId));
+  UninstallExtension(chrome_app->id());
+  EXPECT_FALSE(fake_drive_service()->HasApp(kDriveAppId));
+}
+
+// A matching Chrome app replaces the created URL app.
+IN_PROC_BROWSER_TEST_F(DriveAppProviderTest, MatchingChromeAppInstalled) {
+  // Prepare a Drive app that matches the not-yet-installed kChromeAppId.
+  fake_drive_service()->AddApp(
+      kDriveAppId, kDriveAppName, kChromeAppId, kLaunchUrl);
+  RefreshDriveAppRegistry();
+  WaitForPendingDriveAppConverters();
+
+  // An Url app should be created.
+  const Extension* url_app =
+      ExtensionRegistry::Get(profile())->GetExtensionById(
+          mapping()->GetChromeApp(kDriveAppId), ExtensionRegistry::EVERYTHING);
+  EXPECT_TRUE(url_app->is_hosted_app());
+  EXPECT_TRUE(url_app->from_bookmark());
+
+  const std::string url_app_id = url_app->id();
+  EXPECT_NE(kChromeAppId, url_app_id);
+  EXPECT_EQ(url_app_id, mapping()->GetChromeApp(kDriveAppId));
+
+  // Installs a chrome app with matching id.
+  InstallChromeApp(0);
+
+  // The Drive app should be mapped to chrome app.
+  EXPECT_EQ(kChromeAppId, mapping()->GetChromeApp(kDriveAppId));
+
+  // Url app should be auto uninstalled.
+  EXPECT_FALSE(ExtensionRegistry::Get(profile())->GetExtensionById(
+      url_app_id, ExtensionRegistry::EVERYTHING));
+}
+
+// Tests that the corresponding URL app is uninstalled when a Drive app is
+// disconnected.
+IN_PROC_BROWSER_TEST_F(DriveAppProviderTest,
+                       DisconnectDriveAppUninstallUrlApp) {
+  // Prepare a Drive app that matches the not-yet-installed kChromeAppId.
+  fake_drive_service()->AddApp(
+      kDriveAppId, kDriveAppName, kChromeAppId, kLaunchUrl);
+  RefreshDriveAppRegistry();
+  WaitForPendingDriveAppConverters();
+
+  // Url app is created.
+  const std::string url_app_id = mapping()->GetChromeApp(kDriveAppId);
+  EXPECT_TRUE(ExtensionRegistry::Get(profile())->GetExtensionById(
+      url_app_id, ExtensionRegistry::EVERYTHING));
+
+  fake_drive_service()->RemoveAppByProductId(kChromeAppId);
+  RefreshDriveAppRegistry();
+
+  // Url app is auto uninstalled.
+  EXPECT_FALSE(ExtensionRegistry::Get(profile())->GetExtensionById(
+      url_app_id, ExtensionRegistry::EVERYTHING));
+}
+
+// Tests that the matching Chrome app is preserved when a Drive app is
+// disconnected.
+IN_PROC_BROWSER_TEST_F(DriveAppProviderTest,
+                       DisconnectDriveAppPreserveChromeApp) {
+  // Prepare an existing chrome app.
+  const Extension* chrome_app = InstallChromeApp(1);
+  ASSERT_TRUE(chrome_app);
+
+  // Prepare a Drive app that matches the chrome app id.
+  fake_drive_service()->AddApp(
+      kDriveAppId, kDriveAppName, kChromeAppId, kLaunchUrl);
+  RefreshDriveAppRegistry();
+  WaitForPendingDriveAppConverters();
+
+  fake_drive_service()->RemoveAppByProductId(kChromeAppId);
+  RefreshDriveAppRegistry();
+
+  // Chrome app is still present after the Drive app is disconnected.
+  EXPECT_TRUE(ExtensionRegistry::Get(profile())->GetExtensionById(
+      kChromeAppId, ExtensionRegistry::EVERYTHING));
+}
+
+// A new URL app replaces the existing one and keeps existing// position when a
+// Drive app changes its name or URL.
+IN_PROC_BROWSER_TEST_F(DriveAppProviderTest, DriveAppChanged) {
+  // Prepare a Drive app with no underlying chrome app.
+  fake_drive_service()->AddApp(
+      kDriveAppId, kDriveAppName, kChromeAppId, kLaunchUrl);
+  RefreshDriveAppRegistry();
+  WaitForPendingDriveAppConverters();
+
+  // An Url app should be created.
+  const std::string url_app_id = mapping()->GetChromeApp(kDriveAppId);
+  const Extension* url_app =
+      ExtensionRegistry::Get(profile())
+          ->GetExtensionById(url_app_id, ExtensionRegistry::EVERYTHING);
+  ASSERT_TRUE(url_app);
+  EXPECT_EQ(kDriveAppName, url_app->name());
+  EXPECT_TRUE(url_app->is_hosted_app());
+  EXPECT_TRUE(url_app->from_bookmark());
+  EXPECT_EQ(GURL(kLaunchUrl), AppLaunchInfo::GetLaunchWebURL(url_app));
+
+  // Put url app into a test folder position.
+  const char kTestFolder[] = "test-folder";
+  const size_t kExpectedIndex = 3;
+  for (size_t i = 0; i < kExpectedIndex; ++i) {
+    model()->AddItemToFolder(
+        model()->CreateItem(model()->GetItemName(static_cast<int>(i))),
+        kTestFolder);
+  }
+  app_list::AppListFolderItem* folder = model()->FindFolderItem(kTestFolder);
+  ASSERT_TRUE(folder);
+  model()->MoveItemToFolderAt(model()->FindItem(url_app->id()),
+                              kTestFolder,
+                              folder->item_list()
+                                  ->item_at(kExpectedIndex - 1)
+                                  ->position()
+                                  .CreateAfter());
+
+  // Register the Drive app with a different name and URL.
+  const char kAnotherName[] = "Another drive app name";
+  const char kAnotherLaunchUrl[] = "http://example.com/another_end_point";
+  fake_drive_service()->RemoveAppByProductId(kChromeAppId);
+  fake_drive_service()->AddApp(
+      kDriveAppId, kAnotherName, kChromeAppId, kAnotherLaunchUrl);
+  RefreshDriveAppRegistry();
+  WaitForPendingDriveAppConverters();
+
+  // Old URL app should be auto uninstalled.
+  url_app = ExtensionRegistry::Get(profile())
+                ->GetExtensionById(url_app_id, ExtensionRegistry::EVERYTHING);
+  EXPECT_FALSE(url_app);
+
+  // New URL app should be used.
+  const std::string new_url_app_id = mapping()->GetChromeApp(kDriveAppId);
+  EXPECT_NE(new_url_app_id, url_app_id);
+
+  const Extension* new_url_app =
+      ExtensionRegistry::Get(profile())
+          ->GetExtensionById(new_url_app_id, ExtensionRegistry::EVERYTHING);
+  ASSERT_TRUE(new_url_app);
+  EXPECT_EQ(kAnotherName, new_url_app->name());
+  EXPECT_TRUE(new_url_app->is_hosted_app());
+  EXPECT_TRUE(new_url_app->from_bookmark());
+  EXPECT_EQ(GURL(kAnotherLaunchUrl),
+            AppLaunchInfo::GetLaunchWebURL(new_url_app));
+
+  app_list::AppListItem* item = model()->FindItem(new_url_app_id);
+  ASSERT_TRUE(item);
+  EXPECT_EQ(kTestFolder, item->folder_id());
+  size_t item_index;
+  ASSERT_TRUE(folder->item_list()->FindItemIndex(new_url_app_id, &item_index));
+  EXPECT_EQ(kExpectedIndex, item_index);
+}
+
+// An existing URL app is not changed when underlying drive app data (name and
+// URL) is not changed.
+IN_PROC_BROWSER_TEST_F(DriveAppProviderTest, NoChange) {
+  // Prepare one Drive app.
+  fake_drive_service()->AddApp(
+      kDriveAppId, kDriveAppName, kChromeAppId, kLaunchUrl);
+  RefreshDriveAppRegistry();
+  WaitForPendingDriveAppConverters();
+
+  const std::string url_app_id = mapping()->GetChromeApp(kDriveAppId);
+  const Extension* url_app =
+      ExtensionRegistry::Get(profile())
+          ->GetExtensionById(url_app_id, ExtensionRegistry::EVERYTHING);
+
+  // Refresh with no actual change.
+  RefreshDriveAppRegistry();
+  WaitForPendingDriveAppConverters();
+
+  // Url app should remain unchanged.
+  const std::string new_url_app_id = mapping()->GetChromeApp(kDriveAppId);
+  EXPECT_EQ(new_url_app_id, url_app_id);
+
+  const Extension* new_url_app =
+      ExtensionRegistry::Get(profile())
+          ->GetExtensionById(new_url_app_id, ExtensionRegistry::EVERYTHING);
+  EXPECT_EQ(url_app, new_url_app);
+}