Make associatedLayoutObjectOf() to handle ::first-letter pseudo-element correctly

third_party/WebKit/Source/core/editing/VisibleUnitsTest.cpp

 // Copyright 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 "core/editing/VisibleUnits.h"
 
 #include "core/dom/Text.h"
 #include "core/editing/EditingTestBase.h"
 #include "core/editing/VisiblePosition.h"
 #include "core/html/HTMLTextFormControlElement.h"
+#include "core/layout/LayoutTextFragment.h"
 #include "core/layout/line/InlineBox.h"
 #include <ostream> // NOLINT
 
 namespace blink {
 
 namespace {
 
 PositionWithAffinity positionWithAffinityInDOMTree(Node& anchor, int offset, TextAffinity affinity = TextAffinity::Downstream)
 {
     return PositionWithAffinity(canonicalPositionOf(Position(&anchor, offset)), affinity);
@@ skipping 38 matching lines @@
     RefPtrWillBeRawPtr<Element> body = document().body();
     RefPtrWillBeRawPtr<Element> one = body->querySelector("#one", ASSERT_NO_EXCEPTION);
 
     IntRect boundsInDOMTree = absoluteCaretBoundsOf(createVisiblePosition(Position(one.get(), 0)));
     IntRect boundsInComposedTree = absoluteCaretBoundsOf(createVisiblePosition(PositionInComposedTree(one.get(), 0)));
 
     EXPECT_FALSE(boundsInDOMTree.isEmpty());
     EXPECT_EQ(boundsInDOMTree, boundsInComposedTree);
 }
 
+TEST_F(VisibleUnitsTest, associatedLayoutObjectOfFirstLetterPunctuations)
+{
+    const char* bodyContent = "<style>p:first-letter {color:red;}</style><p id=sample>(a)bc</p>";
+    setBodyContent(bodyContent);
+    updateLayoutAndStyleForPainting();
+
+    Node* sample = document().getElementById("sample");
+    Node* text = sample->firstChild();
+
+    LayoutTextFragment* layoutObject0 = toLayoutTextFragment(associatedLayoutObjectOf(*text, 0));
+    EXPECT_FALSE(layoutObject0->isRemainingTextLayoutObject());
+
+    LayoutTextFragment* layoutObject1 = toLayoutTextFragment(associatedLayoutObjectOf(*text, 1));
+    EXPECT_EQ(layoutObject0, layoutObject1) << "A character 'a' should be part of first letter.";
+
+    LayoutTextFragment* layoutObject2 = toLayoutTextFragment(associatedLayoutObjectOf(*text, 2));
+    EXPECT_EQ(layoutObject0, layoutObject2) << "close parenthesis should be part of first letter.";
+
+    LayoutTextFragment* layoutObject3 = toLayoutTextFragment(associatedLayoutObjectOf(*text, 3));
+    EXPECT_TRUE(layoutObject3->isRemainingTextLayoutObject());
+}
+
+TEST_F(VisibleUnitsTest, associatedLayoutObjectOfFirstLetterSplit)
+{
+    const char* bodyContent = "<style>p:first-letter {color:red;}</style><p id=sample>abc</p>";
+    setBodyContent(bodyContent);
+    updateLayoutAndStyleForPainting();
+
+    Node* sample = document().getElementById("sample");
+    Node* firstLetter = sample->firstChild();
+    // Split "abc" into "a" "bc"
+    toText(firstLetter)->splitText(1, ASSERT_NO_EXCEPTION);
+    updateLayoutAndStyleForPainting();
+
+    LayoutTextFragment* layoutObject0 = toLayoutTextFragment(associatedLayoutObjectOf(*firstLetter, 0));
+    EXPECT_FALSE(layoutObject0->isRemainingTextLayoutObject());
+
+    LayoutTextFragment* layoutObject1 = toLayoutTextFragment(associatedLayoutObjectOf(*firstLetter, 1));
+    EXPECT_EQ(layoutObject0, layoutObject1);
+}
+
+TEST_F(VisibleUnitsTest, associatedLayoutObjectOfFirstLetterWithTrailingWhitespace)
+{
+    const char* bodyContent = "<style>div:first-letter {color:red;}</style><div id=sample>a\n <div></div></div>";
+    setBodyContent(bodyContent);
+    updateLayoutAndStyleForPainting();
+
+    Node* sample = document().getElementById("sample");
+    Node* text = sample->firstChild();
+
+    LayoutTextFragment* layoutObject0 = toLayoutTextFragment(associatedLayoutObjectOf(*text, 0));
+    EXPECT_FALSE(layoutObject0->isRemainingTextLayoutObject());
+
+    LayoutTextFragment* layoutObject1 = toLayoutTextFragment(associatedLayoutObjectOf(*text, 1));
+    EXPECT_TRUE(layoutObject1->isRemainingTextLayoutObject());
+
+    LayoutTextFragment* layoutObject2 = toLayoutTextFragment(associatedLayoutObjectOf(*text, 2));
+    EXPECT_EQ(layoutObject1, layoutObject2);
+}
+
 TEST_F(VisibleUnitsTest, caretMinOffset)
 {
     const char* bodyContent = "<p id=one>one</p>";
     setBodyContent(bodyContent);
 
     RefPtrWillBeRawPtr<Element> one = document().getElementById("one");
 
     EXPECT_EQ(0, caretMinOffset(one->firstChild()));
 }
 
@@ skipping 717 matching lines @@
     EXPECT_EQ(Position(sample, 3), mostBackwardCaretPosition(Position(sample, 3)));
     EXPECT_EQ(Position(sample, 4), mostBackwardCaretPosition(Position(sample, 4)));
     EXPECT_EQ(Position(sample, 5), mostBackwardCaretPosition(Position(sample, 5)));
     EXPECT_EQ(Position(sample, 6), mostBackwardCaretPosition(Position(sample, 6)));
     EXPECT_EQ(Position(sample, 6), mostBackwardCaretPosition(Position(sample, 7)));
     EXPECT_EQ(Position(sample, 6), mostBackwardCaretPosition(Position::lastPositionInNode(sample->parentNode())));
     EXPECT_EQ(Position(sample, 6), mostBackwardCaretPosition(Position::afterNode(sample->parentNode())));
     EXPECT_EQ(Position::lastPositionInNode(document().body()), mostBackwardCaretPosition(Position::lastPositionInNode(document().body())));
 }
 
+TEST_F(VisibleUnitsTest, mostBackwardCaretPositionFirstLetterSplit)
+{
+    const char* bodyContent = "<style>p:first-letter {color:red;}</style><p id=sample>abc</p>";
+    setBodyContent(bodyContent);
+    updateLayoutAndStyleForPainting();
+
+    Node* sample = document().getElementById("sample");
+    Node* firstLetter = sample->firstChild();
+    // Split "abc" into "a" "bc"
+    RefPtrWillBeRawPtr<Text> remaining = toText(firstLetter)->splitText(1, ASSERT_NO_EXCEPTION);
+    updateLayoutAndStyleForPainting();
+
+    EXPECT_EQ(Position(sample, 0), mostBackwardCaretPosition(Position(firstLetter, 0)));
+    EXPECT_EQ(Position(firstLetter, 1), mostBackwardCaretPosition(Position(firstLetter, 1)));
+    EXPECT_EQ(Position(firstLetter, 1), mostBackwardCaretPosition(Position(remaining, 0)));
+    EXPECT_EQ(Position(remaining, 1), mostBackwardCaretPosition(Position(remaining, 1)));
+    EXPECT_EQ(Position(remaining, 2), mostBackwardCaretPosition(Position(remaining, 2)));
+    EXPECT_EQ(Position(remaining, 2), mostBackwardCaretPosition(Position::lastPositionInNode(sample)));
+    EXPECT_EQ(Position(remaining, 2), mostBackwardCaretPosition(Position::afterNode(sample)));
+}
+
 TEST_F(VisibleUnitsTest, mostForwardCaretPositionAfterAnchor)
 {
     const char* bodyContent = "<p id='host'><b id='one'>1</b></p>";
     const char* shadowContent = "<b id='two'>22</b><content select=#one></content><b id='three'>333</b>";
     setBodyContent(bodyContent);
     RefPtrWillBeRawPtr<ShadowRoot> shadowRoot = setShadowContent(shadowContent, "host");
     updateLayoutAndStyleForPainting();
 
     RefPtrWillBeRawPtr<Element> host = document().getElementById("host");
     RefPtrWillBeRawPtr<Element> one = document().getElementById("one");
@@ skipping 327 matching lines @@
     EXPECT_EQ(PositionInComposedTree(space, 1), startOfWord(createVisiblePositionInComposedTree(*three, 1)).deepEquivalent());
 
     EXPECT_EQ(Position(four, 0), startOfWord(createVisiblePositionInDOMTree(*four, 1)).deepEquivalent());
     EXPECT_EQ(PositionInComposedTree(four, 0), startOfWord(createVisiblePositionInComposedTree(*four, 1)).deepEquivalent());
 
     EXPECT_EQ(Position(space, 1), startOfWord(createVisiblePositionInDOMTree(*five, 1)).deepEquivalent());
     EXPECT_EQ(PositionInComposedTree(space, 1), startOfWord(createVisiblePositionInComposedTree(*five, 1)).deepEquivalent());
 }
 
 } // namespace blink