Implement newly saved card bubble for realz and update generated card bubble to

chrome/browser/ui/autofill/generated_credit_card_bubble_controller_unittest.cc

+// Copyright 2013 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/basictypes.h"
+#include "base/compiler_specific.h"
+#include "base/logging.h"
+#include "base/memory/scoped_ptr.h"
+#include "base/prefs/pref_service.h"
+#include "base/strings/string16.h"
+#include "base/strings/utf_string_conversions.h"
+#include "chrome/browser/ui/autofill/generated_credit_card_bubble_controller.h"
+#include "chrome/browser/ui/autofill/test_generated_credit_card_bubble_view.h"
+#include "chrome/common/pref_names.h"
+#include "chrome/test/base/testing_profile.h"
+#include "components/autofill/core/browser/autofill_common_test.h"
+#include "content/public/browser/web_contents.h"
+#include "content/public/common/page_transition_types.h"
+#include "content/public/test/test_browser_thread_bundle.h"
+#include "content/public/test/web_contents_tester.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "ui/base/range/range.h"
+
+#if defined(OS_WIN)
+#include "ui/base/win/scoped_ole_initializer.h"
+#endif
+
+namespace autofill {
+
+namespace {
+
+base::string16 BackingCard() {
+  return ASCIIToUTF16("Visa - 1111");
+}
+
+base::string16 FrontingCard() {
+  return ASCIIToUTF16("Mastercard - 4444");
+}
+
+base::string16 RangeOfString(const base::string16& string,
+                             const ui::Range& range) {
+  return string.substr(range.start(), range.end() - range.start());
+}
+
+class TestGeneratedCreditCardBubbleController
+    : public GeneratedCreditCardBubbleController {
+ public:
+  explicit TestGeneratedCreditCardBubbleController(
+      content::WebContents* contents)
+      : GeneratedCreditCardBubbleController(contents) {
+    contents->SetUserData(UserDataKey(), this);
+  }
+
+  virtual ~TestGeneratedCreditCardBubbleController() {}
+
+  bool IsInstalled() const {
+    return web_contents()->GetUserData(UserDataKey()) == this;
+  }
+
+  TestGeneratedCreditCardBubbleView* GetTestingBubble() {
+    return static_cast<TestGeneratedCreditCardBubbleView*>(
+        GeneratedCreditCardBubbleController::bubble().get());
+  }
+
+ protected:
+  virtual base::WeakPtr<GeneratedCreditCardBubbleView> CreateBubble() OVERRIDE {
+    return TestGeneratedCreditCardBubbleView::Create(GetWeakPtr());
+  }
+
+  virtual bool CanShow() const OVERRIDE {
+    return true;
+  }
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(TestGeneratedCreditCardBubbleController);
+};
+
+class GeneratedCreditCardBubbleControllerTest : public testing::Test {
+ public:
+  GeneratedCreditCardBubbleControllerTest()
+      : test_web_contents_(
+            content::WebContentsTester::CreateTestWebContents(
+                &profile_, NULL)) {}
+
+  virtual void SetUp() OVERRIDE {
+    // Attaches immediately to |test_web_contents_| so a test version will exist
+    // before a non-test version can be created.
+    new TestGeneratedCreditCardBubbleController(test_web_contents_.get());
+    ASSERT_TRUE(controller()->IsInstalled());
+  }
+
+ protected:
+  TestGeneratedCreditCardBubbleController* controller() {
+    return static_cast<TestGeneratedCreditCardBubbleController*>(
+        TestGeneratedCreditCardBubbleController::FromWebContents(
+            test_web_contents_.get()));
+  }
+
+  int GeneratedCardBubbleTimesShown() {
+    return profile_.GetPrefs()->GetInteger(
+        ::prefs::kAutofillGeneratedCardBubbleTimesShown);
+  }
+
+  void Show() {
+    ASSERT_TRUE(controller()->IsInstalled());
+    TestGeneratedCreditCardBubbleController::Show(test_web_contents_.get(),
+                                                  BackingCard(),
+                                                  FrontingCard());
+  }
+
+  void Navigate() {
+    NavigateWithTransition(content::PAGE_TRANSITION_LINK);
+  }
+
+  void Redirect() {
+    NavigateWithTransition(content::PAGE_TRANSITION_CLIENT_REDIRECT);
+  }
+
+ private:
+  void NavigateWithTransition(content::PageTransition trans) {
+    content::WebContentsTester::For(test_web_contents_.get())->TestDidNavigate(
+        test_web_contents_->GetRenderViewHost(), 1, GURL("about:blank"), trans);
+  }
+
+  content::TestBrowserThreadBundle thread_bundle_;
+#if defined(OS_WIN)
+  // Without this there will be drag and drop failures. http://crbug.com/227221
+  ui::ScopedOleInitializer ole_initializer_;
+#endif
+  TestingProfile profile_;
+  scoped_ptr<content::WebContents> test_web_contents_;
+};
+
+}  // namespace
+
+TEST_F(GeneratedCreditCardBubbleControllerTest, GeneratedCardBubbleTimesShown) {
+  ASSERT_EQ(0, GeneratedCardBubbleTimesShown());
+
+  // Ensure that showing the generated card UI bumps the persistent count.
+  Show();
+  EXPECT_EQ(1, GeneratedCardBubbleTimesShown());
+  EXPECT_TRUE(controller()->GetTestingBubble()->showing());
+
+  Show();
+  Show();
+  EXPECT_EQ(3, GeneratedCardBubbleTimesShown());
+  EXPECT_TRUE(controller()->GetTestingBubble()->showing());
+}
+
+TEST_F(GeneratedCreditCardBubbleControllerTest, TitleText) {
+  Show();
+  EXPECT_FALSE(controller()->TitleText().empty());
+}
+
+TEST_F(GeneratedCreditCardBubbleControllerTest, ContentsText) {
+  // Ensure that while showing the generated card UI that the bubble's text
+  // contains "Visa - 1111" and "Mastercard - 4444".
+  Show();
+  base::string16 contents_text = controller()->ContentsText();
+  EXPECT_NE(base::string16::npos, contents_text.find(BackingCard()));
+  EXPECT_NE(base::string16::npos, contents_text.find(FrontingCard()));
+
+  // Make sure that |bubble_text_| is regenerated the same way in |Setup()|.
+  Show();
+  EXPECT_EQ(contents_text, controller()->ContentsText());
+}
+
+TEST_F(GeneratedCreditCardBubbleControllerTest, ContentsTextRanges) {
+  // Check that the highlighted ranges in the bubble's text are correct.
+  Show();
+  const base::string16& contents_text = controller()->ContentsText();
+  const std::vector<TextRange>& ranges = controller()->ContentsTextRanges();
+
+  ASSERT_EQ(3U, ranges.size());
+
+  EXPECT_EQ(BackingCard(), RangeOfString(contents_text, ranges[0].range));
+  EXPECT_FALSE(ranges[0].is_link);
+
+  EXPECT_EQ(FrontingCard(), RangeOfString(contents_text, ranges[1].range));
+  EXPECT_FALSE(ranges[1].is_link);
+
+  EXPECT_TRUE(ranges[2].is_link);
+
+  Show();
+  EXPECT_EQ(ranges, controller()->ContentsTextRanges());
+}
+
+TEST_F(GeneratedCreditCardBubbleControllerTest, AnchorIcon) {
+  Show();
+  EXPECT_FALSE(controller()->AnchorIcon().IsEmpty());
+}
+
+TEST_F(GeneratedCreditCardBubbleControllerTest, HideOnNavigate) {
+  // When a user navigates away from a page (or refreshes) normally, the bubble
+  // should be hidden.
+  EXPECT_FALSE(controller()->GetTestingBubble());
+  Show();
+  EXPECT_TRUE(controller()->GetTestingBubble()->showing());
+
+  Navigate();
+  EXPECT_FALSE(controller());
+}
+
+TEST_F(GeneratedCreditCardBubbleControllerTest, StayOnRedirect) {
+  // If a page redirects right after submitting, the bubble should remain.
+  EXPECT_FALSE(controller()->GetTestingBubble());
+  Show();
+  EXPECT_TRUE(controller()->GetTestingBubble()->showing());
+
+  Redirect();
+  EXPECT_TRUE(controller()->GetTestingBubble()->showing());
+}
+
+}  // namespace autofill