[Payment Request] unit tests for the FullCardRequester

ios/chrome/browser/ui/payments/full_card_requester_unittest.mm

Index: ios/chrome/browser/ui/payments/full_card_requester_unittest.mm
diff --git a/ios/chrome/browser/ui/payments/full_card_requester_unittest.mm b/ios/chrome/browser/ui/payments/full_card_requester_unittest.mm
new file mode 100644
index 0000000000000000000000000000000000000000..13de8778f01255f08584154cc55f877458856aaa
--- /dev/null
+++ b/ios/chrome/browser/ui/payments/full_card_requester_unittest.mm
@@ -0,0 +1,141 @@
+// Copyright 2017 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.
+
+#import "ios/chrome/browser/ui/payments/full_card_requester.h"
+
+#include "base/logging.h"
+#include "base/strings/string16.h"
+#include "base/strings/utf_string_conversions.h"
+#include "base/test/ios/wait_util.h"
+#include "components/autofill/core/browser/autofill_manager.h"
+#include "components/autofill/core/browser/autofill_test_utils.h"
+#include "components/autofill/core/browser/credit_card.h"
+#include "components/autofill/ios/browser/autofill_driver_ios.h"
+#import "ios/chrome/browser/autofill/autofill_agent.h"
+#import "ios/chrome/browser/autofill/autofill_controller.h"
+#include "ios/chrome/browser/browser_state/test_chrome_browser_state.h"
+#include "ios/chrome/browser/infobars/infobar_manager_impl.h"
+#include "ios/chrome/browser/ui/autofill/card_unmask_prompt_view_bridge.h"
+#import "ios/chrome/browser/web/chrome_web_test.h"
+#import "ios/chrome/test/scoped_key_window.h"
+#import "ios/testing/ocmock_complex_type_helper.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "third_party/ocmock/OCMock/OCMock.h"
+#include "third_party/ocmock/gtest_support.h"
+
+#if !defined(__has_feature) || !__has_feature(objc_arc)
+#error "This file requires ARC support."
+#endif
+
+@interface FullCardRequesterConsumerMock
+    : OCMockComplexTypeHelper<FullCardRequesterConsumer>
+@end
+
+@implementation FullCardRequesterConsumerMock
+
+typedef void (^mock_full_card_request_did_succeed)(const autofill::CreditCard&,
+                                                   const base::string16&);
+
+- (void)fullCardRequestDidSucceedWithCard:(const autofill::CreditCard&)card
+                         verificationCode:
+                             (const base::string16&)verificationCode {
+  return static_cast<mock_full_card_request_did_succeed>(
+      [self blockForSelector:_cmd])(card, verificationCode);
+}
+
+@end
+
+class PaymentRequestFullCardRequesterTest : public ChromeWebTest {
+ protected:
+  PaymentRequestFullCardRequesterTest()
+      : credit_card_(autofill::test::GetCreditCard()) {
+    TestChromeBrowserState::Builder test_cbs_builder;
+    chrome_browser_state_ = test_cbs_builder.Build();
+  }
+
+  void SetUp() override {
+    ChromeWebTest::SetUp();
+
+    // Set up what is needed to have an instance of autofill::AutofillManager.
+    AutofillAgent* autofill_agent =
+        [[AutofillAgent alloc] initWithBrowserState:chrome_browser_state_.get()
+                                           webState:web_state()];
+    InfoBarManagerImpl::CreateForWebState(web_state());
+    autofill_controller_ = [[AutofillController alloc]
+             initWithBrowserState:chrome_browser_state_.get()
+                         webState:web_state()
+                    autofillAgent:autofill_agent
+        passwordGenerationManager:nullptr
+                  downloadEnabled:NO];
+  }
+
+  void TearDown() override {
+    [autofill_controller_ detachFromWebState];
+
+    ChromeWebTest::TearDown();
+  }
+
+  autofill::CreditCard credit_card_;
+  std::unique_ptr<TestChromeBrowserState> chrome_browser_state_;
+  // Manages autofill for a single page.
+  AutofillController* autofill_controller_;
+};
+
+// Tests that the FullCardRequester presents and dismisses the card unmask
+// prompt view controller, when the full card is requested and when the user
+// enters the CVC/expiration information respectively.
+TEST_F(PaymentRequestFullCardRequesterTest, PresentAndDismiss) {
+  UIViewController* base_view_controller = [[UIViewController alloc] init];
+  ScopedKeyWindow scoped_key_window_;
+  [scoped_key_window_.Get() setRootViewController:base_view_controller];
+
+  FullCardRequester full_card_requester(nil, base_view_controller,
+                                        chrome_browser_state_.get());
+
+  EXPECT_EQ(nil, base_view_controller.presentedViewController);
+
+  autofill::AutofillManager* autofill_manager =
+      autofill::AutofillDriverIOS::FromWebState(web_state())
+          ->autofill_manager();
+  full_card_requester.GetFullCard(&credit_card_, autofill_manager);
+
+  // Spin the run loop to trigger the animation.
+  base::test::ios::SpinRunLoopWithMaxDelay(base::TimeDelta::FromSecondsD(1.0));
+  EXPECT_TRUE([base_view_controller.presentedViewController
+      isMemberOfClass:[CardUnmaskPromptViewController class]]);
+
+  full_card_requester.OnUnmaskVerificationResult(
+      autofill::AutofillClient::SUCCESS);
+
+  // Wait until the view controller is ordered to be dismissed and the animation
+  // completes.
+  WaitForCondition(^bool() {
+    return !base_view_controller.presentedViewController;
+  });
+  EXPECT_EQ(nil, base_view_controller.presentedViewController);
+}
+
+// Tests that calling the FullCardRequester's delegate method which signals that
+// the full credit card details have been successfully received, causes the
+// FullCardRequester's delegate method to get called.
+TEST_F(PaymentRequestFullCardRequesterTest, FullCardRequestSucceeded) {
+  // Mock the consumer.
+  id consumer =
+      [OCMockObject mockForProtocol:@protocol(FullCardRequesterConsumer)];
+  id consumer_mock([[FullCardRequesterConsumerMock alloc]
+      initWithRepresentedObject:consumer]);
+  SEL selector = @selector(fullCardRequestDidSucceedWithCard:verificationCode:);
+  [consumer_mock onSelector:selector
+       callBlockExpectation:^(const autofill::CreditCard& card,
+                              const base::string16& verificationCode) {
+         EXPECT_EQ(credit_card_, card);
+         EXPECT_EQ(base::ASCIIToUTF16("123"), verificationCode);
+       }];
+
+  FullCardRequester full_card_requester(consumer_mock, nil,
+                                        chrome_browser_state_.get());
+
+  full_card_requester.OnFullCardRequestSucceeded(credit_card_,
+                                                 base::ASCIIToUTF16("123"));
+}