Persist the choice of AutofillDataModel when using the requestAutocomplete dialog.

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 407 matching lines @@
   // A full profile should be picked up.
   AutofillProfile full_profile(test::GetFullProfile());
   full_profile.set_origin(kSettingsOrigin);
   full_profile.SetRawInfo(ADDRESS_HOME_LINE2, string16());
   controller()->GetTestingManager()->AddTestingProfile(&full_profile);
   shipping_model = controller()->MenuModelForSection(SECTION_SHIPPING);
   EXPECT_EQ(4, shipping_model->GetItemCount());
   EXPECT_TRUE(!!controller()->MenuModelForSection(SECTION_EMAIL));
 }
 
+// Makes sure that the choice of which Autofill profile to use for each section
+// is sticky.
+TEST_F(AutofillDialogControllerTest, AutofillProfileDefaults) {
+  AutofillProfile full_profile(test::GetFullProfile());
+  full_profile.set_origin(kSettingsOrigin);
+  controller()->GetTestingManager()->AddTestingProfile(&full_profile);
+  AutofillProfile full_profile2(test::GetFullProfile2());
+  full_profile2.set_origin(kSettingsOrigin);
+  controller()->GetTestingManager()->AddTestingProfile(&full_profile2);
+
+  // Until a selection has been made, the defauilt shipping suggestion is the

[Ilya Sherman, 2013/05/21 03:54:35]

nit: "default"

[Evan Stade, 2013/05/21 17:34:41]

Done.

+  // first one (after "use billing").
+  SuggestionsMenuModel* shipping_model = static_cast<SuggestionsMenuModel*>(
+      controller()->MenuModelForSection(SECTION_SHIPPING));
+  EXPECT_EQ(1, shipping_model->checked_item());
+
+  for (int i = 2; i >= 0; --i) {
+    SuggestionsMenuModel* shipping_model = static_cast<SuggestionsMenuModel*>(
+        controller()->MenuModelForSection(SECTION_SHIPPING));
+    shipping_model->ExecuteCommand(i, 0);
+    controller()->OnAccept();
+
+    TearDown();
+    SetUp();

[Ilya Sherman, 2013/05/21 03:54:35]

Hmm, these methods aren't really meant to be called except by the test harness. 
I'm worried that this will break at some point in the future as the contents of
SetUp() and TearDown() evolve.  Is there perhaps a way to write this test that
doesn't call into these methods?

[Evan Stade, 2013/05/21 17:34:41]

If the contents of SetUp and TearDown evolve, we probably want the changes to
apply here as well. Conveniently, if someone does make a breaking change, we
will notice that this test is failing and be able to address that issue when it
arises (it would probably just consist of splitting out the functionality that's
currently in SetUp/TearDown into other functions that can be called separately).

+    controller()->GetTestingManager()->AddTestingProfile(&full_profile);
+    controller()->GetTestingManager()->AddTestingProfile(&full_profile2);

[Ilya Sherman, 2013/05/21 03:54:35]

Please add a test verifying the behavior when the PersonalDataManager no longer
has the default profile saved.

[Evan Stade, 2013/05/21 17:34:41]

Done. Also improved the behavior in that case (instead of falling back to index
0, fall back to the normal default).

+    shipping_model = static_cast<SuggestionsMenuModel*>(
+        controller()->MenuModelForSection(SECTION_SHIPPING));
+    EXPECT_EQ(i, shipping_model->checked_item());
+  }
+}
+
 TEST_F(AutofillDialogControllerTest, AutofillProfileVariants) {
   EXPECT_CALL(*controller()->GetView(), ModelChanged()).Times(1);
   ui::MenuModel* email_model =
       controller()->MenuModelForSection(SECTION_EMAIL);
   EXPECT_FALSE(email_model);
 
   // Set up some variant data.
   AutofillProfile full_profile(test::GetFullProfile());
   std::vector<string16> names;
   names.push_back(ASCIIToUTF16("John Doe"));
   names.push_back(ASCIIToUTF16("Jane Doe"));
   full_profile.SetRawMultiInfo(EMAIL_ADDRESS, names);
   const string16 kEmail1 = ASCIIToUTF16(kFakeEmail);
   const string16 kEmail2 = ASCIIToUTF16("admin@example.com");
   std::vector<string16> emails;
   emails.push_back(kEmail1);
   emails.push_back(kEmail2);
   full_profile.SetRawMultiInfo(EMAIL_ADDRESS, emails);
 
   // Respect variants for the email address field only.
   controller()->GetTestingManager()->AddTestingProfile(&full_profile);
   ui::MenuModel* shipping_model =
       controller()->MenuModelForSection(SECTION_SHIPPING);
   EXPECT_EQ(4, shipping_model->GetItemCount());
   email_model = controller()->MenuModelForSection(SECTION_EMAIL);
   ASSERT_TRUE(!!email_model);
   EXPECT_EQ(4, email_model->GetItemCount());
 
+  // The first one is the default.
+  SuggestionsMenuModel* email_suggestions = static_cast<SuggestionsMenuModel*>(
+      controller()->MenuModelForSection(SECTION_EMAIL));
+  EXPECT_EQ(0, email_suggestions->checked_item());
+
   email_model->ActivatedAt(0);
   EXPECT_EQ(kEmail1,
             controller()->SuggestionStateForSection(SECTION_EMAIL).text);
   email_model->ActivatedAt(1);
   EXPECT_EQ(kEmail2,
             controller()->SuggestionStateForSection(SECTION_EMAIL).text);
 
   controller()->EditClickedForSection(SECTION_EMAIL);
   const DetailInputs& inputs =
       controller()->RequestedFieldsForSection(SECTION_EMAIL);
   EXPECT_EQ(kEmail2, inputs[0].initial_value);
+
+  // The choice of variant is persisted across runs of the dialog.
+  email_model->ActivatedAt(0);
+  email_model->ActivatedAt(1);
+  controller()->OnAccept();
+
+  TearDown();
+  SetUp();
+  controller()->GetTestingManager()->AddTestingProfile(&full_profile);
+  email_suggestions = static_cast<SuggestionsMenuModel*>(
+      controller()->MenuModelForSection(SECTION_EMAIL));
+  EXPECT_EQ(1, email_suggestions->checked_item());
 }
 
 // Test selecting a shipping address different from billing as address.
 TEST_F(AutofillDialogControllerTest, DontUseBillingAsShipping) {
   AutofillProfile full_profile(test::GetFullProfile());
   AutofillProfile full_profile2(test::GetFullProfile2());
   CreditCard credit_card;
   test::SetCreditCardInfo(&credit_card, "Test User",
                           "4234567890654321", // Visa
                           "11", "2100");
@@ skipping 731 matching lines @@
   EXPECT_FALSE(controller()->ShouldOfferToSaveInChrome());
 
   controller()->MenuModelForSection(SECTION_EMAIL)->ActivatedAt(1);
   EXPECT_TRUE(controller()->ShouldOfferToSaveInChrome());
 
   profile()->set_incognito(true);
   EXPECT_FALSE(controller()->ShouldOfferToSaveInChrome());
 }
 
 }  // namespace autofill