Add a regression test for the CardUnmaskPrompt.

chrome/browser/ui/autofill/card_unmask_prompt_view_browsertest.cc

 // Copyright (c) 2015 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/message_loop/message_loop.h"
+#include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/ui/autofill/card_unmask_prompt_controller_impl.h"
+#include "chrome/browser/ui/autofill/card_unmask_prompt_view_tester.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "components/autofill/core/browser/autofill_test_utils.h"
 #include "components/autofill/core/browser/card_unmask_delegate.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/test/test_utils.h"
 
 namespace autofill {
 
 namespace {
 
 class TestCardUnmaskDelegate : public CardUnmaskDelegate {
  public:
   TestCardUnmaskDelegate() : weak_factory_(this) {}
 
   virtual ~TestCardUnmaskDelegate() {}
 
   // CardUnmaskDelegate implementation.
-  void OnUnmaskResponse(const UnmaskResponse& response) override {}
+  void OnUnmaskResponse(const UnmaskResponse& response) override {
+    response_ = response;
+  }
   void OnUnmaskPromptClosed() override {}
 
   base::WeakPtr<TestCardUnmaskDelegate> GetWeakPtr() {
     return weak_factory_.GetWeakPtr();
   }
 
+  UnmaskResponse response() { return response_; }
+
  private:
+  UnmaskResponse response_;
+
   base::WeakPtrFactory<TestCardUnmaskDelegate> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(TestCardUnmaskDelegate);
 };
 
 class TestCardUnmaskPromptController : public CardUnmaskPromptControllerImpl {
  public:
   TestCardUnmaskPromptController(
       content::WebContents* contents,
       scoped_refptr<content::MessageLoopRunner> runner)
       : CardUnmaskPromptControllerImpl(contents),
         runner_(runner),
         weak_factory_(this) {}
 
   // CardUnmaskPromptControllerImpl implementation.
-  void OnDidLoadRiskFingerprint(const std::string& risk_data) override {
-    CardUnmaskPromptControllerImpl::OnDidLoadRiskFingerprint(risk_data);
-
-    // Call Quit() from here rather than from OnUnmaskDialogClosed().
-    // FingerprintDataLoader starts several async tasks that take a while to
-    // complete. If Quit() is called before FingerprintDataLoader is all done
-    // then LeakTracker will detect that some resources have not been freed
-    // and cause the browser test to fail. This is not a real leak though -
-    // normally the async tasks would complete and encounter weak callbacks.
-    runner_->Quit();
+  base::TimeDelta GetSuccessMessageDuration() const override {
+    return base::TimeDelta::FromMilliseconds(10);
   }
 
-  void RunMessageLoop() { runner_->Run(); }
+  void LoadRiskFingerprint() override {
+    OnDidLoadRiskFingerprint(std::string());
+  }
 
   base::WeakPtr<TestCardUnmaskPromptController> GetWeakPtr() {
     return weak_factory_.GetWeakPtr();
   }
 
+  using CardUnmaskPromptControllerImpl::view;
+
  private:
   scoped_refptr<content::MessageLoopRunner> runner_;
   base::WeakPtrFactory<TestCardUnmaskPromptController> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(TestCardUnmaskPromptController);
 };
 
 class CardUnmaskPromptViewBrowserTest : public InProcessBrowserTest {
  public:
   CardUnmaskPromptViewBrowserTest() : InProcessBrowserTest() {}
 
   ~CardUnmaskPromptViewBrowserTest() override {}
 
   void SetUpOnMainThread() override {
     runner_ = new content::MessageLoopRunner;
     controller_.reset(new TestCardUnmaskPromptController(
         browser()->tab_strip_model()->GetActiveWebContents(), runner_));
     delegate_.reset(new TestCardUnmaskDelegate());
   }
 
   TestCardUnmaskPromptController* controller() { return controller_.get(); }
   TestCardUnmaskDelegate* delegate() { return delegate_.get(); }
 
- private:
+ protected:
   // This member must outlive the controller.
   scoped_refptr<content::MessageLoopRunner> runner_;
 
+ private:
   scoped_ptr<TestCardUnmaskPromptController> controller_;
   scoped_ptr<TestCardUnmaskDelegate> delegate_;
 
   DISALLOW_COPY_AND_ASSIGN(CardUnmaskPromptViewBrowserTest);
 };
 
 IN_PROC_BROWSER_TEST_F(CardUnmaskPromptViewBrowserTest, DisplayUI) {
-  CreditCard credit_card(CreditCard::MASKED_SERVER_CARD, "a123");
-  test::SetCreditCardInfo(&credit_card, "Bonnie Parker",
-                          "2109" /* Mastercard */, "12", "2012");
-  credit_card.SetTypeForMaskedCard(kMasterCard);
+  controller()->ShowPrompt(test::GetMaskedServerCard(),
+                           delegate()->GetWeakPtr());
+}
 
-  controller()->ShowPrompt(credit_card, delegate()->GetWeakPtr());
-  controller()->RunMessageLoop();
+// TODO(bondd): bring up on Mac.
+#if !defined(OS_MACOSX)
+// Makes sure the user can close the dialog while the verification success
+// message is showing.
+IN_PROC_BROWSER_TEST_F(CardUnmaskPromptViewBrowserTest,
+                       EarlyCloseAfterSuccess) {
+  controller()->ShowPrompt(test::GetMaskedServerCard(),
+                           delegate()->GetWeakPtr());
+  controller()->OnUnmaskResponse(base::ASCIIToUTF16("123"),
+                                 base::ASCIIToUTF16("10"),
+                                 base::ASCIIToUTF16("19"), false);
+  DCHECK_EQ(delegate()->response().cvc, base::ASCIIToUTF16("123"));
+  controller()->OnVerificationResult(AutofillClient::SUCCESS);
+
+  // Simulate the user clicking [x] before the "Success!" message disappears.
+  CardUnmaskPromptViewTester::For(controller()->view())->Close();
+  // Wait a little while; there should be no crash.

[groby-ooo-7-16, 2015/03/19 23:47:56]

I suppose there's no way we can do without a fuzzy guess here? (I.e. there's no
hook for "done" where we can post the Quit from)

[Evan Stade, 2015/03/20 00:22:42]

The nature of the crash was that a stale closure was being run some amount of
time after the dialog was closed. The solution is to cancel that closure. So no,
there's no function to hook into.

+  base::MessageLoop::current()->PostDelayedTask(
+      FROM_HERE, base::Bind(&content::MessageLoopRunner::Quit,
+                            base::Unretained(runner_.get())),
+      2 * controller()->GetSuccessMessageDuration());
+  runner_->Run();
 }
+#endif
 
 }  // namespace
 
 }  // namespace autofill