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 106 matching lines @@
         DIALOG_TYPE_REQUEST_AUTOCOMPLETE,
         callback);
     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:
   // 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_;
@@ skipping 103 matching lines @@
   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, 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_FALSE(controller()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_OK));
+  EXPECT_TRUE(controller()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_CANCEL));
+
+  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_TRUE(controller()->IsSubmitPausedOn(wallet::VERIFY_CVV));
+  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));
+  EXPECT_TRUE(
+      !controller()->SuggestionTextForSection(SECTION_CC_BILLING).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