Autofill:requestAutocomplete: Enable prompting for complete address when instrument being used does…

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 183 matching lines @@
   }
 
   TestWalletClient* GetTestingWalletClient() {
     return &test_wallet_client_;
   }
 
   void set_is_first_run(bool is_first_run) { is_first_run_ = is_first_run; }
 
   const GURL& open_tab_url() { return open_tab_url_; }
 
+  bool IsSectionInEditState(DialogSection section) {
+    std::map<DialogSection, bool> state = section_editing_state();
+    return state[section];
+  }
+
  protected:
   virtual PersonalDataManager* GetManager() OVERRIDE {
     return &test_manager_;
   }
 
   virtual wallet::WalletClient* GetWalletClient() OVERRIDE {
     return &test_wallet_client_;
   }
 
   virtual bool IsFirstRun() const OVERRIDE {
@@ skipping 319 matching lines @@
   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));
+  ASSERT_FALSE(controller()->IsSectionInEditState(SECTION_CC_BILLING));
   // "use billing", "add", "manage", and 5 suggestions.
   EXPECT_EQ(8,
       controller()->MenuModelForSection(SECTION_SHIPPING)->GetItemCount());
   EXPECT_TRUE(controller()->MenuModelForSection(SECTION_SHIPPING)->
       IsItemCheckedAt(4));
+  ASSERT_FALSE(controller()->IsSectionInEditState(SECTION_SHIPPING));
 }
 
 TEST_F(AutofillDialogControllerTest, SaveAddress) {
   EXPECT_CALL(*controller()->GetView(), ModelChanged()).Times(1);
   EXPECT_CALL(*controller()->GetTestingWalletClient(),
               SaveAddress(_, _)).Times(1);
 
   scoped_ptr<wallet::WalletItems> wallet_items = wallet::GetTestWalletItems();
   wallet_items->AddInstrument(wallet::GetTestMaskedInstrument());
   controller()->OnDidGetWalletItems(wallet_items.Pass());
@@ skipping 605 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());
 }
 
+// Tests that user is prompted when using instrument with minimal address.
+TEST_F(AutofillDialogControllerTest, UpgradeMinimalAddress) {
+  scoped_ptr<wallet::WalletItems> wallet_items = wallet::GetTestWalletItems();
+  wallet_items->AddInstrument(wallet::GetTestMaskedInstrumentWithIdAndAddress(
+      "id", wallet::GetTestMinimalAddress()));
+  scoped_ptr<wallet::Address> address(wallet::GetTestShippingAddress());
+  address->set_is_complete_address(false);
+  wallet_items->AddAddress(address.Pass());
+  controller()->OnDidGetWalletItems(wallet_items.Pass());
+
+  // Assert that dialog's SECTION_CC_BILLING section is in edit mode.
+  ASSERT_TRUE(controller()->IsSectionInEditState(SECTION_CC_BILLING));
+  // Shipping section should be in edit mode because of
+  // is_minimal_shipping_address.
+  ASSERT_TRUE(controller()->IsSectionInEditState(SECTION_SHIPPING));
+}
+
 }  // namespace autofill