Multi-account AccountChooser for interactive autocomplete.

chrome/browser/ui/autofill/autofill_dialog_controller_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 "base/guid.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop.h"
 #include "base/prefs/pref_service.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/ui/autofill/autofill_dialog_controller_impl.h"
@@ skipping 124 matching lines @@
   MOCK_METHOD4(SaveInstrumentAndAddress,
       void(const wallet::Instrument& instrument,
            const wallet::Address& address,
            const std::string& obfuscated_gaia_id,
            const GURL& source_url));
 
  private:
   DISALLOW_COPY_AND_ASSIGN(TestWalletClient);
 };
 
+class TestAccountChooserModel : public AccountChooserModel {
+public:
+  TestAccountChooserModel(AccountChooserModelDelegate* delegate,
+                          PrefService* prefs)
+      : AccountChooserModel(delegate, prefs) {
+  }
+
+  using AccountChooserModel::kActiveWalletItemId;
+  using AccountChooserModel::kAutofillItemId;
+};
+
 class TestAutofillDialogController : public AutofillDialogControllerImpl {
  public:
   TestAutofillDialogController(
       content::WebContents* contents,
       const FormData& form_structure,
       const GURL& source_url,
       const AutofillMetrics& metric_logger,
       const DialogType dialog_type,
       const base::Callback<void(const FormStructure*,
                                 const std::string&)>& callback)
@@ skipping 118 matching lines @@
 
   static scoped_ptr<wallet::FullWallet> CreateFullWalletWithVerifyCvv() {
     base::DictionaryValue dict;
     scoped_ptr<base::ListValue> list(new base::ListValue());
     list->AppendString("verify_cvv");
     dict.Set("required_action", list.release());
     return wallet::FullWallet::CreateFullWallet(dict);
   }
 
   void SetUpWallet() {
-    controller()->MenuModelForAccountChooser()->ActivatedAt(
-        AccountChooserModel::kWalletItemId);
     controller()->OnUserNameFetchSuccess("user@example.com");
+    ui::MenuModel* account_model = controller()->MenuModelForAccountChooser();
+    ASSERT_TRUE(account_model);
+    account_model->ActivatedAt(TestAccountChooserModel::kActiveWalletItemId);
   }
 
   TestAutofillDialogController* controller() { return controller_; }
 
   TestingProfile* profile() { return &profile_; }
 
  private:
   void FinishedCallback(const FormStructure* form_structure,
                         const std::string& google_transaction_id) {}
 
@@ skipping 326 matching lines @@
   wallet_items->AddAddress(wallet::GetTestShippingAddress());
   controller()->OnDidGetWalletItems(wallet_items.Pass());
   controller()->OnAccept();
 
   EXPECT_FALSE(controller()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_OK));
   EXPECT_TRUE(controller()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_CANCEL));
 
   ui::MenuModel* account_menu = controller()->MenuModelForAccountChooser();
   ASSERT_TRUE(account_menu);
   ASSERT_GE(2, account_menu->GetItemCount());
-  account_menu->ActivatedAt(AccountChooserModel::kWalletItemId);
-  account_menu->ActivatedAt(AccountChooserModel::kAutofillItemId);
+  account_menu->ActivatedAt(TestAccountChooserModel::kActiveWalletItemId);
+  account_menu->ActivatedAt(TestAccountChooserModel::kAutofillItemId);
 
   EXPECT_TRUE(controller()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_OK));
   EXPECT_TRUE(controller()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_CANCEL));
 }
 
 TEST_F(AutofillDialogControllerTest, ErrorDuringVerifyCvv) {
   SetUpWallet();
 
   EXPECT_CALL(*controller()->GetTestingWalletClient(),
               GetFullWallet(_)).Times(1);
@@ skipping 27 matching lines @@
   controller()->OnDidGetWalletItems(wallet_items.Pass());
   controller()->OnAccept();
   controller()->OnDidGetFullWallet(CreateFullWalletWithVerifyCvv());
 
   ASSERT_TRUE(controller()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_OK));
   ASSERT_TRUE(controller()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_CANCEL));
 
   ui::MenuModel* account_menu = controller()->MenuModelForAccountChooser();
   ASSERT_TRUE(account_menu);
   ASSERT_GE(2, account_menu->GetItemCount());
-  account_menu->ActivatedAt(AccountChooserModel::kWalletItemId);
-  account_menu->ActivatedAt(AccountChooserModel::kAutofillItemId);
+  account_menu->ActivatedAt(TestAccountChooserModel::kActiveWalletItemId);
+  account_menu->ActivatedAt(TestAccountChooserModel::kAutofillItemId);
 
   EXPECT_TRUE(controller()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_OK));
   EXPECT_TRUE(controller()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_CANCEL));
 }
 
 // Test that when a wallet error happens only an error is shown (and no other
 // Wallet-related notifications).
 TEST_F(AutofillDialogControllerTest, WalletErrorNotification) {
   SetUpWallet();
 
@@ skipping 25 matching lines @@
       DialogNotification::EXPLANATORY_MESSAGE).size());
 
   // Wallet notifications are mutually exclusive.
   EXPECT_TRUE(NotificationsOfType(
       DialogNotification::WALLET_USAGE_CONFIRMATION).empty());
   EXPECT_TRUE(NotificationsOfType(
       DialogNotification::WALLET_SIGNIN_PROMO).empty());
 
   // Switch to using Autofill, no explanatory message should show.
   ui::MenuModel* account_menu = controller()->MenuModelForAccountChooser();
-  account_menu->ActivatedAt(AccountChooserModel::kAutofillItemId);
+  ASSERT_TRUE(account_menu);
+  account_menu->ActivatedAt(TestAccountChooserModel::kAutofillItemId);
   EXPECT_TRUE(NotificationsOfType(
       DialogNotification::EXPLANATORY_MESSAGE).empty());
 
   // Switch to Wallet, pretend this isn't first run. No message should show.
-  account_menu->ActivatedAt(AccountChooserModel::kWalletItemId);
+  account_menu->ActivatedAt(TestAccountChooserModel::kActiveWalletItemId);
   controller()->set_is_first_run(false);
   EXPECT_TRUE(NotificationsOfType(
       DialogNotification::EXPLANATORY_MESSAGE).empty());
 }
 
 // Verifies that the "[X] Save details in wallet" notification shows on first
 // run with an incomplete profile, stays showing when switching to Autofill in
 // the account chooser, and continues to show on second+ run when a user's
 // wallet is incomplete. This also tests that submitting disables interactivity.
 TEST_F(AutofillDialogControllerTest, SaveDetailsInWallet) {
@@ skipping 10 matching lines @@
   EXPECT_TRUE(notifications.front().interactive());
 
   // Wallet notifications are mutually exclusive.
   EXPECT_TRUE(NotificationsOfType(
       DialogNotification::WALLET_SIGNIN_PROMO).empty());
   EXPECT_TRUE(NotificationsOfType(
       DialogNotification::EXPLANATORY_MESSAGE).empty());
 
   // Using Autofill on second run, show an interactive, unchecked checkbox.
   ui::MenuModel* account_model = controller()->MenuModelForAccountChooser();
-  account_model->ActivatedAt(AccountChooserModel::kAutofillItemId);
+  ASSERT_TRUE(account_model);
+  account_model->ActivatedAt(TestAccountChooserModel::kAutofillItemId);
   controller()->set_is_first_run(false);
 
   notifications =
       NotificationsOfType(DialogNotification::WALLET_USAGE_CONFIRMATION);
   EXPECT_EQ(1U, notifications.size());
   EXPECT_FALSE(notifications.front().checked());
   EXPECT_TRUE(notifications.front().interactive());
 
   // Notifications shouldn't be interactive while submitting.
-  account_model->ActivatedAt(AccountChooserModel::kWalletItemId);
+  account_model->ActivatedAt(TestAccountChooserModel::kActiveWalletItemId);
   controller()->OnAccept();
   EXPECT_FALSE(NotificationsOfType(
       DialogNotification::WALLET_USAGE_CONFIRMATION).front().interactive());
 }
 
 // Verifies that no Wallet notifications are shown after first run (i.e. no
 // "[X] Save details to wallet" or "These details are from your Wallet") when
 // the user has a complete wallet.
 TEST_F(AutofillDialogControllerTest, NoWalletNotifications) {
   SetUpWallet();
   controller()->set_is_first_run(false);
 
   // Simulate a complete wallet.
   scoped_ptr<wallet::WalletItems> wallet_items = wallet::GetTestWalletItems();
   wallet_items->AddInstrument(wallet::GetTestMaskedInstrument());
   wallet_items->AddAddress(wallet::GetTestShippingAddress());
   controller()->OnDidGetWalletItems(wallet_items.Pass());
 
   EXPECT_TRUE(NotificationsOfType(
       DialogNotification::EXPLANATORY_MESSAGE).empty());
   EXPECT_TRUE(NotificationsOfType(
       DialogNotification::WALLET_USAGE_CONFIRMATION).empty());
 }
 
 }  // namespace autofill