1 files changed, 132 insertions, 0 deletions
diff --git a/chrome/browser/extensions/default_apps_unittest.cc b/chrome/browser/extensions/default_apps_unittest.cc
new file mode 100644
index 0000000..9e7b0d5
--- /dev/null
+++ b/chrome/browser/extensions/default_apps_unittest.cc
@@ -0,0 +1,132 @@
+// Copyright (c) 2010 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/extensions/default_apps.h"
+#include "chrome/common/extensions/extension.h"
+#include "chrome/test/testing_pref_service.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+TEST(DefaultApps, Basics) {
+  TestingPrefService pref_service;
+  DefaultApps::RegisterUserPrefs(&pref_service);
+  DefaultApps default_apps(&pref_service);
+
+  ExtensionIdSet default_app_ids = *default_apps.GetAppsToInstall();
+  ASSERT_GT(default_app_ids.size(), 0u);
+  EXPECT_FALSE(default_apps.GetDefaultAppsInstalled());
+  EXPECT_EQ(0, default_apps.GetPromoCounter());
+
+  // The promo should not be shown until the default apps have been installed.
+  ExtensionIdSet installed_app_ids;
+  EXPECT_FALSE(default_apps.ShouldShowPromo(installed_app_ids));
+
+  // Simulate installing the apps one by one and notifying default_apps after
+  // each intallation. Nothing should change until we have installed all the
+  // default apps.
+  ExtensionIdSet extension_id_sets[] = {
+    default_app_ids,
+    default_app_ids,
+    default_app_ids
+  };
+  extension_id_sets[0].clear();
+  extension_id_sets[1].erase(extension_id_sets[1].begin());
+  extension_id_sets[2].erase(extension_id_sets[2].begin(),
+                             ++extension_id_sets[2].begin());
+  for (size_t i = 0; i < arraysize(extension_id_sets); ++i) {
+    default_apps.DidInstallApp(extension_id_sets[i]);
+    EXPECT_TRUE(default_app_ids == *default_apps.GetAppsToInstall());
+    EXPECT_FALSE(default_apps.GetDefaultAppsInstalled());
+    EXPECT_FALSE(default_apps.ShouldShowPromo(extension_id_sets[i]));
+  }
+
+  // Simulate all the default apps being installed. Now we should stop getting
+  // default apps to install.
+  default_apps.DidInstallApp(default_app_ids);
+  EXPECT_EQ(NULL, default_apps.GetAppsToInstall());
+  EXPECT_TRUE(default_apps.GetDefaultAppsInstalled());
+
+  // And the promo should become available.
+  EXPECT_TRUE(default_apps.ShouldShowPromo(default_app_ids));
+
+  // The promo should be available up to the max allowed times, then stop.
+  for (int i = 0; i < DefaultApps::kAppsPromoCounterMax; ++i) {
+    EXPECT_TRUE(default_apps.ShouldShowPromo(default_app_ids));
+    default_apps.DidShowPromo();
+    EXPECT_EQ(i + 1, default_apps.GetPromoCounter());
+  }
+  EXPECT_FALSE(default_apps.ShouldShowPromo(default_app_ids));
+  EXPECT_EQ(DefaultApps::kAppsPromoCounterMax, default_apps.GetPromoCounter());
+}
+
+TEST(DefaultApps, HidePromo) {
+  TestingPrefService pref_service;
+  DefaultApps::RegisterUserPrefs(&pref_service);
+  DefaultApps default_apps(&pref_service);
+
+  ExtensionIdSet default_app_ids = *default_apps.GetAppsToInstall();
+  default_apps.DidInstallApp(default_app_ids);
+
+  EXPECT_TRUE(default_apps.ShouldShowPromo(default_app_ids));
+  default_apps.DidShowPromo();
+  EXPECT_EQ(1, default_apps.GetPromoCounter());
+
+  default_apps.SetPromoHidden();
+  EXPECT_FALSE(default_apps.ShouldShowPromo(default_app_ids));
+  EXPECT_EQ(DefaultApps::kAppsPromoCounterMax, default_apps.GetPromoCounter());
+}
+
+TEST(DefaultApps, InstallingAnAppHidesPromo) {
+  TestingPrefService pref_service;
+  DefaultApps::RegisterUserPrefs(&pref_service);
+  DefaultApps default_apps(&pref_service);
+
+  ExtensionIdSet default_app_ids = *default_apps.GetAppsToInstall();
+  ExtensionIdSet installed_app_ids = default_app_ids;
+  default_apps.DidInstallApp(installed_app_ids);
+
+  EXPECT_TRUE(default_apps.ShouldShowPromo(installed_app_ids));
+  default_apps.DidShowPromo();
+  EXPECT_EQ(1, default_apps.GetPromoCounter());
+
+  // Now simulate a new extension being installed. This should cause the promo
+  // to be hidden.
+  installed_app_ids.insert("foo");
+  EXPECT_FALSE(default_apps.ShouldShowPromo(installed_app_ids));
+  EXPECT_EQ(DefaultApps::kAppsPromoCounterMax, default_apps.GetPromoCounter());
+}
+
+TEST(DefaultApps, ManualAppInstalledWhileInstallingDefaultApps) {
+  // It is possible to have apps manually installed while the default apps are
+  // being installed. The network or server might be down, causing the default
+  // app installation to fail. The updater might take awhile to get around to
+  // updating, giving the user a chance to manually intall.
+  //
+  // In these cases, we should keep trying to install default apps until we have
+  // them all, and then stop, even if at that point, we have more apps than just
+  // the default ones.
+  TestingPrefService pref_service;
+  DefaultApps::RegisterUserPrefs(&pref_service);
+  DefaultApps default_apps(&pref_service);
+
+  // Simulate an app getting installed before the complete set of default apps.
+  // This shouldn't affect us installing default apps. We should keep trying.
+  ExtensionIdSet installed_ids;
+  installed_ids.insert("foo");
+  default_apps.DidInstallApp(installed_ids);
+  EXPECT_FALSE(default_apps.GetDefaultAppsInstalled());
+  EXPECT_TRUE(default_apps.GetAppsToInstall());
+
+  // Now add all the default apps in addition to the extra app. We should stop
+  // trying to install default apps.
+  installed_ids = *default_apps.GetAppsToInstall();
+  installed_ids.insert("foo");
+  default_apps.DidInstallApp(installed_ids);
+  EXPECT_TRUE(default_apps.GetDefaultAppsInstalled());
+  EXPECT_FALSE(default_apps.GetAppsToInstall());
+
+  // The promo shouldn't turn on though, because it would look weird with the
+  // user's extra, manually installed extensions.
+  EXPECT_FALSE(default_apps.ShouldShowPromo(installed_ids));
+  EXPECT_EQ(DefaultApps::kAppsPromoCounterMax, default_apps.GetPromoCounter());
+}