Omit "Manage your X..." menu model items when user is new.

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 1386 matching lines @@
   controller()->EditCancelledForSection(SECTION_EMAIL);
   EXPECT_FALSE(controller()->ShouldOfferToSaveInChrome());
 
   controller()->MenuModelForSection(SECTION_EMAIL)->ActivatedAt(1);
   EXPECT_TRUE(controller()->ShouldOfferToSaveInChrome());
 
   profile()->set_incognito(true);
   EXPECT_FALSE(controller()->ShouldOfferToSaveInChrome());
 }
 
+TEST_F(AutofillDialogControllerTest, NoManageMenuItemForNewWalletUsers) {
+  // Make sure the menu model item is created for a returning Wallet user.
+  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(controller()->MenuModelForSection(SECTION_CC_BILLING));
+  // "Same as billing", "123 address", "Add address...", and "Manage addresses".
+  EXPECT_EQ(
+      4, controller()->MenuModelForSection(SECTION_SHIPPING)->GetItemCount());
+
+  // Make sure the menu model item is not created for new Wallet users.
+  base::DictionaryValue dict;
+  scoped_ptr<base::ListValue> required_actions(new base::ListValue);
+  required_actions->AppendString("setup_wallet");
+  dict.Set("required_action", required_actions.release());
+  controller()->OnDidGetWalletItems(
+      wallet::WalletItems::CreateWalletItems(dict).Pass());
+
+  EXPECT_FALSE(controller()->MenuModelForSection(SECTION_CC_BILLING));
+  // "Same as billing" and "Add address...".
+  EXPECT_EQ(
+      2, controller()->MenuModelForSection(SECTION_SHIPPING)->GetItemCount());
+}
+
 }  // namespace autofill