Implementing VERIFY_CVV required action.

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/prefs/pref_service.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/ui/autofill/autofill_dialog_controller_impl.h"
 #include "chrome/browser/ui/autofill/autofill_dialog_view.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/autofill/browser/autofill_common_test.h"
 #include "components/autofill/browser/autofill_metrics.h"
+#include "components/autofill/browser/wallet/full_wallet.h"
 #include "components/autofill/browser/wallet/instrument.h"
 #include "components/autofill/browser/wallet/wallet_address.h"
 #include "components/autofill/browser/wallet/wallet_client.h"
 #include "components/autofill/browser/wallet/wallet_test_util.h"
 #include "components/autofill/common/form_data.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/test/test_browser_thread.h"
 #include "content/public/test/web_contents_tester.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
@@ skipping 61 matching lines @@
   TestWalletClient(net::URLRequestContextGetter* context,
                    wallet::WalletClientDelegate* delegate)
       : wallet::WalletClient(context, delegate) {}
   virtual ~TestWalletClient() {}
 
   MOCK_METHOD3(AcceptLegalDocuments,
       void(const std::vector<wallet::WalletItems::LegalDocument*>& documents,
            const std::string& google_transaction_id,
            const GURL& source_url));
 
+  MOCK_METHOD3(AuthenticateInstrument,
+      void(const std::string& instrument_id,
+           const std::string& card_verification_number,
+           const std::string& obfuscated_gaia_id));
+
   MOCK_METHOD1(GetFullWallet,
       void(const wallet::WalletClient::FullWalletRequest& request));
 
   MOCK_METHOD2(SaveAddress,
       void(const wallet::Address& address, const GURL& source_url));
 
   MOCK_METHOD3(SaveInstrument,
       void(const wallet::Instrument& instrument,
            const std::string& obfuscated_gaia_id,
            const GURL& source_url));
@@ skipping 95 matching lines @@
     field.autocomplete_attribute = "cc-number";
     FormData form_data;
     form_data.fields.push_back(field);
 
     profile()->GetPrefs()->SetBoolean(
         prefs::kAutofillDialogPayWithoutWallet, true);
     profile()->CreateRequestContext();
     test_web_contents_.reset(
         content::WebContentsTester::CreateTestWebContents(profile(), NULL));
 
-    base::Callback<void(const FormStructure*, const std::string&)> callback;
+    base::Callback<void(const FormStructure*, const std::string&)> callback =
+        base::Bind(&AutofillDialogControllerTest::FinishedCallback,
+                   base::Unretained(this));
     controller_ = new TestAutofillDialogController(
         test_web_contents_.get(),
         form_data,
         GURL(),
         metric_logger_,
         DIALOG_TYPE_REQUEST_AUTOCOMPLETE,
         callback);
     controller_->Init(profile());
     controller_->Show();
   }
 
   virtual void TearDown() OVERRIDE {
     controller_->ViewClosed();
   }
 
  protected:
+  size_t CountNotificationsOfType(DialogNotification::Type type) {
+    size_t count = 0;
+    const std::vector<DialogNotification>& notifications =
+        controller()->CurrentNotifications();
+    for (size_t i = 0; i < notifications.size(); ++i) {
+      if (notifications[i].type() == type)
+        ++count;
+    }
+    return count;
+  }
+
   TestAutofillDialogController* controller() { return controller_; }
 
   TestingProfile* profile() { return &profile_; }
 
  private:
+  void FinishedCallback(const FormStructure* form_structure,
+                        const std::string& google_transaction_id) {}
+
   // A bunch of threads are necessary for classes like TestWebContents and
   // URLRequestContextGetter not to fall over.
   MessageLoopForUI loop_;
   content::TestBrowserThread ui_thread_;
   content::TestBrowserThread file_thread_;
   content::TestBrowserThread file_blocking_thread_;
   content::TestBrowserThread io_thread_;
   TestingProfile profile_;
 
   // The controller owns itself.
@@ skipping 74 matching lines @@
   // Respect variants for the email address field only.
   controller()->GetTestingManager()->AddTestingProfile(&full_profile);
   ui::MenuModel* shipping_model =
       controller()->MenuModelForSection(SECTION_SHIPPING);
   EXPECT_EQ(2, shipping_model->GetItemCount());
   ui::MenuModel* email_model =
       controller()->MenuModelForSection(SECTION_EMAIL);
   EXPECT_EQ(3, email_model->GetItemCount());
 
   email_model->ActivatedAt(0);
-  EXPECT_EQ(kEmail1, controller()->SuggestionTextForSection(SECTION_EMAIL));
+  EXPECT_EQ(kEmail1,
+            controller()->SuggestionStateForSection(SECTION_EMAIL).text);
   email_model->ActivatedAt(1);
-  EXPECT_EQ(kEmail2, controller()->SuggestionTextForSection(SECTION_EMAIL));
+  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);
 }
 
 TEST_F(AutofillDialogControllerTest, AcceptLegalDocuments) {
   controller()->set_is_paying_with_wallet(true);
 
   EXPECT_CALL(*controller()->GetView(), ModelChanged()).Times(1);
   EXPECT_CALL(*controller()->GetTestingWalletClient(),
               AcceptLegalDocuments(_, _, _)).Times(1);
   EXPECT_CALL(*controller()->GetTestingWalletClient(),
               GetFullWallet(_)).Times(1);
 
   scoped_ptr<wallet::WalletItems> wallet_items = wallet::GetTestWalletItems();
   wallet_items->AddLegalDocument(wallet::GetTestLegalDocument());
   wallet_items->AddInstrument(wallet::GetTestMaskedInstrument());
   wallet_items->AddAddress(wallet::GetTestShippingAddress());
   controller()->OnDidGetWalletItems(wallet_items.Pass());
-  controller()->OnSubmit();
+  controller()->OnAccept();
 }
 
 TEST_F(AutofillDialogControllerTest, SaveAddress) {
   controller()->set_is_paying_with_wallet(true);
 
   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());
-  controller()->OnSubmit();
+  controller()->OnAccept();
 }
 
 TEST_F(AutofillDialogControllerTest, SaveInstrument) {
   controller()->set_is_paying_with_wallet(true);
 
   EXPECT_CALL(*controller()->GetView(), ModelChanged()).Times(1);
   EXPECT_CALL(*controller()->GetTestingWalletClient(),
               SaveInstrument(_, _, _)).Times(1);
 
   scoped_ptr<wallet::WalletItems> wallet_items = wallet::GetTestWalletItems();
   wallet_items->AddAddress(wallet::GetTestShippingAddress());
   controller()->OnDidGetWalletItems(wallet_items.Pass());
-  controller()->OnSubmit();
+  controller()->OnAccept();
 }
 
 TEST_F(AutofillDialogControllerTest, SaveInstrumentAndAddress) {
   controller()->set_is_paying_with_wallet(true);
 
   EXPECT_CALL(*controller()->GetView(), ModelChanged()).Times(1);
   EXPECT_CALL(*controller()->GetTestingWalletClient(),
               SaveInstrumentAndAddress(_, _, _, _)).Times(1);
 
   controller()->OnDidGetWalletItems(wallet::GetTestWalletItems());
-  controller()->OnSubmit();
+  controller()->OnAccept();
 }
 
-TEST_F(AutofillDialogControllerTest, Cancel) {
+TEST_F(AutofillDialogControllerTest, CancelNoSave) {
   controller()->set_is_paying_with_wallet(true);
+  EXPECT_CALL(*controller()->GetTestingWalletClient(),
+              SaveInstrumentAndAddress(_, _, _, _)).Times(0);
 
   EXPECT_CALL(*controller()->GetView(), ModelChanged()).Times(1);
 
   controller()->OnDidGetWalletItems(wallet::GetTestWalletItems());
   controller()->OnCancel();
 }
 
 TEST_F(AutofillDialogControllerTest, EditClickedCancelled) {
   EXPECT_CALL(*controller()->GetView(), ModelChanged()).Times(1);
 
   AutofillProfile full_profile(autofill_test::GetFullProfile());
   const string16 kEmail = ASCIIToUTF16("first@johndoe.com");
   full_profile.SetRawInfo(EMAIL_ADDRESS, kEmail);
   controller()->GetTestingManager()->AddTestingProfile(&full_profile);
 
   ui::MenuModel* email_model =
       controller()->MenuModelForSection(SECTION_EMAIL);
   EXPECT_EQ(2, email_model->GetItemCount());
 
   // When unedited, the initial_value should be empty.
   email_model->ActivatedAt(0);
   const DetailInputs& inputs0 =
       controller()->RequestedFieldsForSection(SECTION_EMAIL);
   EXPECT_EQ(string16(), inputs0[0].initial_value);
-  EXPECT_EQ(kEmail, controller()->SuggestionTextForSection(SECTION_EMAIL));
+  EXPECT_EQ(kEmail,
+            controller()->SuggestionStateForSection(SECTION_EMAIL).text);
 
   // When edited, the initial_value should contain the value.
   controller()->EditClickedForSection(SECTION_EMAIL);
   const DetailInputs& inputs1 =
       controller()->RequestedFieldsForSection(SECTION_EMAIL);
   EXPECT_EQ(kEmail, inputs1[0].initial_value);
-  EXPECT_EQ(string16(), controller()->SuggestionTextForSection(SECTION_EMAIL));
+  EXPECT_EQ(string16(),
+            controller()->SuggestionStateForSection(SECTION_EMAIL).text);
 
   // When edit is cancelled, the initial_value should be empty.
   controller()->EditCancelledForSection(SECTION_EMAIL);
   const DetailInputs& inputs2 =
       controller()->RequestedFieldsForSection(SECTION_EMAIL);
-  EXPECT_EQ(kEmail, controller()->SuggestionTextForSection(SECTION_EMAIL));
+  EXPECT_EQ(kEmail,
+            controller()->SuggestionStateForSection(SECTION_EMAIL).text);
   EXPECT_EQ(string16(), inputs2[0].initial_value);
 }
 
+TEST_F(AutofillDialogControllerTest, VerifyCvv) {
+  controller()->set_is_paying_with_wallet(true);
+
+  EXPECT_CALL(*controller()->GetView(), ModelChanged()).Times(2);
+  EXPECT_CALL(*controller()->GetTestingWalletClient(),
+              GetFullWallet(_)).Times(1);
+  EXPECT_CALL(*controller()->GetTestingWalletClient(),
+              AuthenticateInstrument(_, _, _)).Times(1);
+
+  scoped_ptr<wallet::WalletItems> wallet_items = wallet::GetTestWalletItems();
+  wallet_items->AddInstrument(wallet::GetTestMaskedInstrument());
+  wallet_items->AddAddress(wallet::GetTestShippingAddress());
+  controller()->OnDidGetWalletItems(wallet_items.Pass());
+  controller()->OnAccept();
+
+  EXPECT_EQ(0U, CountNotificationsOfType(DialogNotification::REQUIRED_ACTION));
+  EXPECT_TRUE(controller()->SectionIsActive(SECTION_EMAIL));
+  EXPECT_TRUE(controller()->SectionIsActive(SECTION_SHIPPING));
+  EXPECT_TRUE(controller()->SectionIsActive(SECTION_CC_BILLING));
+  EXPECT_FALSE(controller()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_OK));
+  EXPECT_TRUE(controller()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_CANCEL));
+
+  SuggestionState suggestion_state =
+      controller()->SuggestionStateForSection(SECTION_CC_BILLING);
+  EXPECT_TRUE(suggestion_state.extra_text.empty());
+
+  base::DictionaryValue dict;
+  scoped_ptr<base::ListValue> list(new base::ListValue());
+  list->AppendString("verify_cvv");
+  dict.Set("required_action", list.release());
+  controller()->OnDidGetFullWallet(wallet::FullWallet::CreateFullWallet(dict));
+
+  EXPECT_EQ(1U, CountNotificationsOfType(DialogNotification::REQUIRED_ACTION));
+  EXPECT_FALSE(controller()->SectionIsActive(SECTION_EMAIL));
+  EXPECT_FALSE(controller()->SectionIsActive(SECTION_SHIPPING));
+  EXPECT_TRUE(controller()->SectionIsActive(SECTION_CC_BILLING));
+
+  suggestion_state =
+      controller()->SuggestionStateForSection(SECTION_CC_BILLING);
+  EXPECT_FALSE(suggestion_state.extra_text.empty());
+  EXPECT_EQ(
+      0, controller()->MenuModelForSection(SECTION_CC_BILLING)->GetItemCount());
+
+  EXPECT_TRUE(controller()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_OK));
+  EXPECT_TRUE(controller()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_CANCEL));
+
+  controller()->OnAccept();
+}
+
 }  // namespace autofill