summaryrefslogtreecommitdiffstats
path: root/chrome/browser/extensions/default_apps_unittest.cc
blob: 4fa9bc2c987e03b541e02e6517a4b0553aceabf8 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
// 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"

// TODO(dpolukhin): On Chrome OS all apps are installed via external extensions,
// and the web store promo is never shown.
#if !defined(OS_CHROMEOS)
TEST(ExtensionDefaultApps, 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());
  EXPECT_EQ(default_app_ids, *default_apps.GetDefaultApps());

  // The promo should not be shown until the default apps have been installed.
  ExtensionIdSet installed_app_ids;
  EXPECT_FALSE(default_apps.CheckShouldShowPromo(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.CheckShouldShowPromo(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.CheckShouldShowPromo(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.CheckShouldShowPromo(default_app_ids));
    default_apps.DidShowPromo();
    EXPECT_EQ(i + 1, default_apps.GetPromoCounter());
  }
  EXPECT_FALSE(default_apps.CheckShouldShowPromo(default_app_ids));
  EXPECT_EQ(DefaultApps::kAppsPromoCounterMax, default_apps.GetPromoCounter());
}

TEST(ExtensionDefaultApps, 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.CheckShouldShowPromo(default_app_ids));
  default_apps.DidShowPromo();
  EXPECT_EQ(1, default_apps.GetPromoCounter());

  default_apps.SetPromoHidden();
  EXPECT_FALSE(default_apps.CheckShouldShowPromo(default_app_ids));
  EXPECT_EQ(DefaultApps::kAppsPromoCounterMax, default_apps.GetPromoCounter());
}

TEST(ExtensionDefaultApps, 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.CheckShouldShowPromo(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.CheckShouldShowPromo(installed_app_ids));
  EXPECT_EQ(DefaultApps::kAppsPromoCounterMax, default_apps.GetPromoCounter());
}

TEST(ExtensionDefaultApps, 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.CheckShouldShowPromo(installed_ids));
  EXPECT_EQ(DefaultApps::kAppsPromoCounterMax, default_apps.GetPromoCounter());
}
#endif  // OS_CHROMEOS