respect default wallet instrument/address

chrome/browser/ui/autofill/autofill_dialog_controller_unittest.cc

Index: chrome/browser/ui/autofill/autofill_dialog_controller_unittest.cc
diff --git a/chrome/browser/ui/autofill/autofill_dialog_controller_unittest.cc b/chrome/browser/ui/autofill/autofill_dialog_controller_unittest.cc
index c1a3c0d735385d9662192de5d4298f3141125855..9209fe85253d7caa575123bfb87d36d77cfb8e37 100644
--- a/chrome/browser/ui/autofill/autofill_dialog_controller_unittest.cc
+++ b/chrome/browser/ui/autofill/autofill_dialog_controller_unittest.cc
@@ -437,6 +437,35 @@ TEST_F(AutofillDialogControllerTest, AcceptLegalDocuments) {
   controller()->OnAccept();
 }
 
+// Makes sure the default object IDs are respected.
+TEST_F(AutofillDialogControllerTest, WalletDefaults) {
+  SetUpWallet();
+
+  scoped_ptr<wallet::WalletItems> wallet_items = wallet::GetTestWalletItems();
+  wallet_items->AddInstrument(wallet::GetTestNonDefaultMaskedInstrument());
+  wallet_items->AddInstrument(wallet::GetTestNonDefaultMaskedInstrument());
+  wallet_items->AddInstrument(wallet::GetTestMaskedInstrument());
+  wallet_items->AddInstrument(wallet::GetTestNonDefaultMaskedInstrument());
+
+  wallet_items->AddAddress(wallet::GetTestNonDefaultShippingAddress());
+  wallet_items->AddAddress(wallet::GetTestNonDefaultShippingAddress());
+  wallet_items->AddAddress(wallet::GetTestNonDefaultShippingAddress());
+  wallet_items->AddAddress(wallet::GetTestShippingAddress());
+  wallet_items->AddAddress(wallet::GetTestNonDefaultShippingAddress());
+
+  controller()->OnDidGetWalletItems(wallet_items.Pass());
+  // "add", "manage", and 4 suggestions.
+  EXPECT_EQ(6,
+      controller()->MenuModelForSection(SECTION_CC_BILLING)->GetItemCount());
+  EXPECT_TRUE(controller()->MenuModelForSection(SECTION_CC_BILLING)->
+      IsItemCheckedAt(2));
+  // "use billing", "add", "manage", and 5 suggestions.
+  EXPECT_EQ(8,
+      controller()->MenuModelForSection(SECTION_SHIPPING)->GetItemCount());
+  EXPECT_TRUE(controller()->MenuModelForSection(SECTION_SHIPPING)->
+      IsItemCheckedAt(4));
+}
+
 TEST_F(AutofillDialogControllerTest, SaveAddress) {
   SetUpWallet();