chrome/browser/ui/autofill/account_chooser_model_unittest.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/prefs/pref_service.h"
   6 #include "base/strings/utf_string_conversions.h"
   7 #include "chrome/browser/ui/autofill/account_chooser_model.h"
   8 #include "chrome/common/pref_names.h"
   9 #include "chrome/test/base/testing_profile.h"
  10 #include "components/autofill/core/browser/autofill_metrics.h"
  11 #include "testing/gmock/include/gmock/gmock.h"
  12 #include "testing/gtest/include/gtest/gtest.h"
  13
  14 using base::ASCIIToUTF16;
  15
  16 namespace autofill {
  17
  18 namespace {
  19
  20 class TestAccountChooserModel : public AccountChooserModel {
  21  public:
  22   TestAccountChooserModel(AccountChooserModelDelegate* delegate,
  23                           Profile* profile,
  24                           bool disable_wallet,
  25                           const AutofillMetrics& metric_logger)
  26       : AccountChooserModel(delegate, profile, disable_wallet, metric_logger) {}
  27   virtual ~TestAccountChooserModel() {}
  28
  29   using AccountChooserModel::kWalletAccountsStartId;
  30   using AccountChooserModel::kWalletAddAccountId;
  31   using AccountChooserModel::kAutofillItemId;
  32
  33  private:
  34   DISALLOW_COPY_AND_ASSIGN(TestAccountChooserModel);
  35 };
  36
  37 class MockAccountChooserModelDelegate : public AccountChooserModelDelegate {
  38  public:
  39   MockAccountChooserModelDelegate() {}
  40   virtual ~MockAccountChooserModelDelegate() {}
  41
  42   MOCK_METHOD0(AccountChooserWillShow, void());
  43   MOCK_METHOD0(AccountChoiceChanged, void());
  44   MOCK_METHOD0(AddAccount, void());
  45   MOCK_METHOD0(UpdateAccountChooserView, void());
  46 };
  47
  48 class AccountChooserModelTest : public testing::Test {
  49  public:
  50   AccountChooserModelTest()
  51       : model_(&delegate_, &profile_, false, metric_logger_) {}
  52   virtual ~AccountChooserModelTest() {}
  53
  54   TestingProfile* profile() { return &profile_; }
  55   MockAccountChooserModelDelegate* delegate() { return &delegate_; }
  56   TestAccountChooserModel* model() { return &model_; }
  57   const AutofillMetrics& metric_logger() { return metric_logger_; }
  58
  59  private:
  60   TestingProfile profile_;
  61   testing::NiceMock<MockAccountChooserModelDelegate> delegate_;
  62   TestAccountChooserModel model_;
  63   AutofillMetrics metric_logger_;
  64 };
  65
  66 }  // namespace
  67
  68 TEST_F(AccountChooserModelTest, ObeysPref) {
  69   // When "Pay without wallet" is false, use Wallet by default.
  70   {
  71     profile()->GetPrefs()->SetBoolean(
  72         ::prefs::kAutofillDialogPayWithoutWallet, false);
  73     TestAccountChooserModel model(delegate(),
  74                                   profile(),
  75                                   false,
  76                                   metric_logger());
  77     EXPECT_TRUE(model.WalletIsSelected());
  78   }
  79   // When the user chose to "Pay without wallet", use Autofill.
  80   {
  81     profile()->GetPrefs()->SetBoolean(
  82         ::prefs::kAutofillDialogPayWithoutWallet, true);
  83     TestAccountChooserModel model(delegate(),
  84                                   profile(),
  85                                   false,
  86                                   metric_logger());
  87     EXPECT_FALSE(model.WalletIsSelected());
  88   }
  89   // When the |disable_wallet| argument is true, use Autofill regardless
  90   // of the pref.
  91   {
  92     profile()->GetPrefs()->SetBoolean(
  93         ::prefs::kAutofillDialogPayWithoutWallet, false);
  94     TestAccountChooserModel model(delegate(), profile(), true, metric_logger());
  95     EXPECT_FALSE(model.WalletIsSelected());
  96   }
  97   // In incognito, use local data regardless of the pref.
  98   {
  99     TestingProfile::Builder builder;
 100     builder.SetIncognito();
 101     scoped_ptr<TestingProfile> incognito = builder.Build();
 102     incognito->SetOriginalProfile(profile());
 103     profile()->GetPrefs()->SetBoolean(
 104         ::prefs::kAutofillDialogPayWithoutWallet, false);
 105     incognito->GetPrefs()->SetBoolean(
 106         ::prefs::kAutofillDialogPayWithoutWallet, false);
 107
 108     TestAccountChooserModel model(delegate(),
 109                                   incognito.get(),
 110                                   false,
 111                                   metric_logger());
 112     EXPECT_FALSE(model.WalletIsSelected());
 113   }
 114 }
 115
 116 TEST_F(AccountChooserModelTest, IgnoresPrefChanges) {
 117   ASSERT_FALSE(profile()->GetPrefs()->GetBoolean(
 118       ::prefs::kAutofillDialogPayWithoutWallet));
 119   EXPECT_TRUE(model()->WalletIsSelected());
 120
 121   // Check that nothing changes while this dialog is running if a pref changes
 122   // (this could cause subtle bugs or annoyances if a user closes another
 123   // running dialog).
 124   profile()->GetPrefs()->SetBoolean(
 125       ::prefs::kAutofillDialogPayWithoutWallet, true);
 126   EXPECT_TRUE(model()->WalletIsSelected());
 127 }
 128
 129 TEST_F(AccountChooserModelTest, HandlesError) {
 130   EXPECT_CALL(*delegate(), AccountChoiceChanged()).Times(1);
 131   EXPECT_CALL(*delegate(), UpdateAccountChooserView()).Times(1);
 132
 133   ASSERT_TRUE(model()->WalletIsSelected());
 134   ASSERT_TRUE(model()->IsCommandIdEnabled(
 135       TestAccountChooserModel::kWalletAccountsStartId));
 136
 137   model()->SetHadWalletError();
 138   EXPECT_FALSE(model()->WalletIsSelected());
 139   EXPECT_FALSE(model()->IsCommandIdEnabled(
 140       TestAccountChooserModel::kWalletAccountsStartId));
 141 }
 142
 143 TEST_F(AccountChooserModelTest, HandlesSigninError) {
 144   EXPECT_CALL(*delegate(), AccountChoiceChanged()).Times(1);
 145   EXPECT_CALL(*delegate(), UpdateAccountChooserView()).Times(2);
 146
 147   // 0. "Unknown" wallet account, we don't know if the user is signed-in yet.
 148   ASSERT_TRUE(model()->WalletIsSelected());
 149   ASSERT_TRUE(model()->IsCommandIdEnabled(
 150       TestAccountChooserModel::kWalletAccountsStartId));
 151   ASSERT_TRUE(model()->WalletIsSelected());
 152   ASSERT_FALSE(model()->HasAccountsToChoose());
 153   ASSERT_EQ(3, model()->GetItemCount());
 154   EXPECT_EQ(base::string16(), model()->GetActiveWalletAccountName());
 155
 156   // 1. "Known" wallet account (e.g. after active/passive/automatic sign-in).
 157   // Calls UpdateAccountChooserView.
 158   std::vector<std::string> accounts;
 159   accounts.push_back("john.doe@gmail.com");
 160   model()->SetWalletAccounts(accounts, 0U);
 161   ASSERT_TRUE(model()->WalletIsSelected());
 162   ASSERT_TRUE(model()->IsCommandIdEnabled(
 163       TestAccountChooserModel::kWalletAccountsStartId));
 164   ASSERT_TRUE(model()->WalletIsSelected());
 165   ASSERT_TRUE(model()->HasAccountsToChoose());
 166   EXPECT_EQ(3, model()->GetItemCount());
 167   EXPECT_EQ(ASCIIToUTF16(accounts[0]), model()->GetActiveWalletAccountName());
 168
 169   // 2. Sign-in failure.
 170   // Autofill data should be selected and be the only valid choice.
 171   // Calls UpdateAccountChooserView.
 172   // Calls AccountChoiceChanged.
 173   model()->SetHadWalletSigninError();
 174   EXPECT_FALSE(model()->WalletIsSelected());
 175   EXPECT_TRUE(model()->IsCommandIdEnabled(
 176       TestAccountChooserModel::kWalletAccountsStartId));
 177   EXPECT_FALSE(model()->WalletIsSelected());
 178   EXPECT_FALSE(model()->HasAccountsToChoose());
 179   EXPECT_EQ(2, model()->GetItemCount());
 180   EXPECT_EQ(base::string16(), model()->GetActiveWalletAccountName());
 181 }
 182
 183 TEST_F(AccountChooserModelTest, RespectsUserChoice) {
 184   EXPECT_CALL(*delegate(), AccountChoiceChanged()).Times(2);
 185
 186   model()->ExecuteCommand(TestAccountChooserModel::kAutofillItemId, 0);
 187   EXPECT_FALSE(model()->WalletIsSelected());
 188
 189   EXPECT_CALL(*delegate(), AddAccount());
 190   model()->ExecuteCommand(TestAccountChooserModel::kWalletAddAccountId, 0);
 191   EXPECT_FALSE(model()->WalletIsSelected());
 192
 193   model()->ExecuteCommand(TestAccountChooserModel::kWalletAccountsStartId, 0);
 194   EXPECT_TRUE(model()->WalletIsSelected());
 195 }
 196
 197 TEST_F(AccountChooserModelTest, HandlesMultipleAccounts) {
 198   EXPECT_FALSE(model()->HasAccountsToChoose());
 199
 200   std::vector<std::string> accounts;
 201   accounts.push_back("john.doe@gmail.com");
 202   accounts.push_back("jane.smith@gmail.com");
 203   model()->SetWalletAccounts(accounts, 0U);
 204   EXPECT_TRUE(model()->HasAccountsToChoose());
 205   EXPECT_TRUE(model()->WalletIsSelected());
 206   ASSERT_EQ(4, model()->GetItemCount());
 207   EXPECT_TRUE(model()->IsCommandIdEnabled(
 208       TestAccountChooserModel::kWalletAccountsStartId));
 209   EXPECT_TRUE(model()->IsCommandIdEnabled(
 210       TestAccountChooserModel::kWalletAccountsStartId + 1));
 211   EXPECT_EQ(ASCIIToUTF16(accounts[0]), model()->GetActiveWalletAccountName());
 212   model()->SetWalletAccounts(accounts, 1U);
 213   EXPECT_EQ(ASCIIToUTF16(accounts[1]), model()->GetActiveWalletAccountName());
 214
 215   model()->ExecuteCommand(TestAccountChooserModel::kWalletAccountsStartId, 0);
 216   EXPECT_EQ(ASCIIToUTF16(accounts[0]), model()->GetActiveWalletAccountName());
 217   model()->ExecuteCommand(TestAccountChooserModel::kWalletAccountsStartId + 1,
 218                           0);
 219   EXPECT_EQ(ASCIIToUTF16(accounts[1]), model()->GetActiveWalletAccountName());
 220
 221   // Setting the wallet accounts forces the switch to wallet.
 222   model()->ExecuteCommand(TestAccountChooserModel::kAutofillItemId, 0);
 223   EXPECT_FALSE(model()->WalletIsSelected());
 224   model()->SetWalletAccounts(accounts, 1U);
 225   EXPECT_TRUE(model()->WalletIsSelected());
 226 }
 227
 228 }  // namespace autofill