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

chrome/browser/ui/autofill/autofill_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/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/autofill_credit_card_bubble_controller.h"
-#include "chrome/browser/ui/autofill/test_autofill_credit_card_bubble.h"
-#include "chrome/common/pref_names.h"
-#include "chrome/test/base/testing_profile.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 - 5888");
-}
-base::string16 NewCard() {
-  return ASCIIToUTF16("Discover - 7582");
-}
-
-base::string16 RangeOfString(const base::string16& string,
-                             const ui::Range& range) {
-  return string.substr(range.start(), range.end() - range.start());
-}
-
-class TestAutofillCreditCardBubbleController
-    : public AutofillCreditCardBubbleController {
- public:
-  explicit TestAutofillCreditCardBubbleController(
-      content::WebContents* contents)
-      : AutofillCreditCardBubbleController(contents) {
-    contents->SetUserData(UserDataKey(), this);
-    EXPECT_TRUE(IsInstalled());
-  }
-  virtual ~TestAutofillCreditCardBubbleController() {}
-
-  bool IsInstalled() const {
-    return web_contents()->GetUserData(UserDataKey()) == this;
-  }
-
-  TestAutofillCreditCardBubble* GetTestingBubble() {
-    return static_cast<TestAutofillCreditCardBubble*>(
-        AutofillCreditCardBubbleController::bubble().get());
-  }
-
- protected:
-  virtual base::WeakPtr<AutofillCreditCardBubble> CreateBubble() OVERRIDE {
-    return TestAutofillCreditCardBubble::Create(GetWeakPtr());
-  }
-
-  virtual bool CanShow() const OVERRIDE {
-    return true;
-  }
-
- private:
-  DISALLOW_COPY_AND_ASSIGN(TestAutofillCreditCardBubbleController);
-};
-
-class AutofillCreditCardBubbleControllerTest : public testing::Test {
- public:
-  AutofillCreditCardBubbleControllerTest()
-      : 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 TestAutofillCreditCardBubbleController(test_web_contents_.get());
-  }
-
- protected:
-  TestAutofillCreditCardBubbleController* controller() {
-    return static_cast<TestAutofillCreditCardBubbleController*>(
-        TestAutofillCreditCardBubbleController::FromWebContents(
-            test_web_contents_.get()));
-  }
-
-  int GeneratedCardBubbleTimesShown() {
-    return profile()->GetPrefs()->GetInteger(
-        ::prefs::kAutofillGeneratedCardBubbleTimesShown);
-  }
-
-  Profile* profile() { return &profile_; }
-
-  content::WebContentsTester* test_web_contents() {
-    return content::WebContentsTester::For(test_web_contents_.get());
-  }
-
-  void ShowGeneratedCardUI() {
-    ASSERT_TRUE(controller()->IsInstalled());
-    TestAutofillCreditCardBubbleController::ShowGeneratedCardUI(
-        test_web_contents_.get(), BackingCard(), FrontingCard());
-  }
-
-  void ShowNewCardSavedBubble() {
-    EXPECT_TRUE(controller()->IsInstalled());
-    TestAutofillCreditCardBubbleController::ShowNewCardSavedBubble(
-        test_web_contents_.get(), NewCard());
-  }
-
-  void Navigate() {
-    NavigateWithTransition(content::PAGE_TRANSITION_LINK);
-  }
-
-  void Redirect() {
-    NavigateWithTransition(content::PAGE_TRANSITION_CLIENT_REDIRECT);
-  }
-
- private:
-  void NavigateWithTransition(content::PageTransition trans) {
-    test_web_contents()->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(AutofillCreditCardBubbleControllerTest, GeneratedCardBubbleTimesShown) {
-  ASSERT_EQ(0, GeneratedCardBubbleTimesShown());
-
-  ShowGeneratedCardUI();
-  EXPECT_EQ(1, GeneratedCardBubbleTimesShown());
-  EXPECT_TRUE(controller()->GetTestingBubble()->showing());
-
-  ShowGeneratedCardUI();
-  ShowGeneratedCardUI();
-  EXPECT_EQ(3, GeneratedCardBubbleTimesShown());
-  EXPECT_TRUE(controller()->GetTestingBubble()->showing());
-}
-
-TEST_F(AutofillCreditCardBubbleControllerTest, BubbleText) {
-  ShowGeneratedCardUI();
-  base::string16 generated_text = controller()->BubbleText();
-  EXPECT_NE(generated_text.find(BackingCard()), base::string16::npos);
-  EXPECT_NE(generated_text.find(FrontingCard()), base::string16::npos);
-  EXPECT_EQ(generated_text.find(NewCard()), base::string16::npos);
-
-  ShowNewCardSavedBubble();
-  base::string16 new_text = controller()->BubbleText();
-  EXPECT_NE(new_text, generated_text);
-  EXPECT_EQ(new_text.find(BackingCard()), base::string16::npos);
-  EXPECT_EQ(new_text.find(FrontingCard()), base::string16::npos);
-  EXPECT_NE(new_text.find(NewCard()), base::string16::npos);
-
-  ShowGeneratedCardUI();
-  EXPECT_EQ(generated_text, controller()->BubbleText());
-
-  ShowNewCardSavedBubble();
-  EXPECT_EQ(new_text, controller()->BubbleText());
-}
-
-TEST_F(AutofillCreditCardBubbleControllerTest, BubbleTextRanges) {
-  ShowGeneratedCardUI();
-  base::string16 text = controller()->BubbleText();
-  std::vector<ui::Range> ranges = controller()->BubbleTextRanges();
-
-  ASSERT_EQ(ranges.size(), 2U);
-  EXPECT_EQ(BackingCard(), RangeOfString(text, ranges[0]));
-  EXPECT_EQ(FrontingCard(), RangeOfString(text, ranges[1]));
-
-  ShowNewCardSavedBubble();
-  text = controller()->BubbleText();
-  ranges = controller()->BubbleTextRanges();
-
-  ASSERT_EQ(ranges.size(), 1U);
-  EXPECT_EQ(NewCard(), RangeOfString(text, ranges[0]));
-}
-
-TEST_F(AutofillCreditCardBubbleControllerTest, HideOnNavigate) {
-  EXPECT_FALSE(controller()->GetTestingBubble());
-  ShowGeneratedCardUI();
-  EXPECT_TRUE(controller()->GetTestingBubble()->showing());
-
-  Navigate();
-  EXPECT_FALSE(controller());
-
-  SetUp();
-
-  EXPECT_FALSE(controller()->GetTestingBubble());
-  ShowNewCardSavedBubble();
-  EXPECT_TRUE(controller()->GetTestingBubble()->showing());
-
-  Navigate();
-  EXPECT_FALSE(controller());
-}
-
-TEST_F(AutofillCreditCardBubbleControllerTest, StayOnRedirect) {
-  EXPECT_FALSE(controller()->GetTestingBubble());
-  ShowGeneratedCardUI();
-  EXPECT_TRUE(controller()->GetTestingBubble()->showing());
-
-  Redirect();
-  EXPECT_TRUE(controller()->GetTestingBubble()->showing());
-
-  SetUp();
-
-  EXPECT_FALSE(controller()->GetTestingBubble());
-  ShowNewCardSavedBubble();
-  EXPECT_TRUE(controller()->GetTestingBubble()->showing());
-
-  Redirect();
-  EXPECT_TRUE(controller()->GetTestingBubble()->showing());
-}
-
-}  // namespace autofill