Refactor DocumentMarkerController

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

 // Copyright 2014 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/InputMethodController.h"
 
 #include <memory>
 #include "core/dom/Document.h"
 #include "core/dom/Element.h"
 #include "core/dom/Range.h"
@@ skipping 1504 matching lines @@
 }
 
 TEST_F(InputMethodControllerTest,
        Marker_WhitespaceFixupAroundMarkerNotContainingSpace) {
   Element* div = insertHTMLElement(
       "<div id='sample' contenteditable>Initial text blah</div>", "sample");
 
   // Add marker under "text" (use TextMatch since Composition markers don't
   // persist across editing operations)
   EphemeralRange markerRange = PlainTextRange(8, 12).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
+
   // Delete "Initial"
   Vector<CompositionUnderline> emptyUnderlines;
   controller().setCompositionFromExistingText(emptyUnderlines, 0, 7);
   controller().commitText(String(""), emptyUnderlines, 0);
 
   // Delete "blah"
   controller().setCompositionFromExistingText(emptyUnderlines, 6, 10);
   controller().commitText(String(""), emptyUnderlines, 0);
 
   // Check that the marker is still attached to "text" and doesn't include
   // either space around it
   EXPECT_EQ(1u, document().markers().markersFor(div->firstChild()).size());
   EXPECT_STREQ(
       "text",
       getMarkedText(document().markers(), div->firstChild(), 0).utf8().data());
 }
 
 TEST_F(InputMethodControllerTest,
        Marker_WhitespaceFixupAroundMarkerBeginningWithSpace) {
   Element* div = insertHTMLElement(
       "<div id='sample' contenteditable>Initial text blah</div>", "sample");
 
   // Add marker under " text" (use TextMatch since Composition markers don't
   // persist across editing operations)
   EphemeralRange markerRange = PlainTextRange(7, 12).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
+
   // Delete "Initial"
   Vector<CompositionUnderline> emptyUnderlines;
   controller().setCompositionFromExistingText(emptyUnderlines, 0, 7);
   controller().commitText(String(""), emptyUnderlines, 0);
 
   // Delete "blah"
   controller().setCompositionFromExistingText(emptyUnderlines, 6, 10);
   controller().commitText(String(""), emptyUnderlines, 0);
 
   // Check that the marker is still attached to " text" and includes the space
   // before "text" but not the space after
   EXPECT_EQ(1u, document().markers().markers().size());
   ASSERT_STREQ(
       "\xC2\xA0text",
       getMarkedText(document().markers(), div->firstChild(), 0).utf8().data());
 }
 
 TEST_F(InputMethodControllerTest,
        Marker_WhitespaceFixupAroundMarkerEndingWithSpace) {
   Element* div = insertHTMLElement(
       "<div id='sample' contenteditable>Initial text blah</div>", "sample");
 
   // Add marker under "text " (use TextMatch since Composition markers don't
   // persist across editing operations)
   EphemeralRange markerRange = PlainTextRange(8, 13).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
+
   // Delete "Initial"
   Vector<CompositionUnderline> emptyUnderlines;
   controller().setCompositionFromExistingText(emptyUnderlines, 0, 7);
   controller().commitText(String(""), emptyUnderlines, 0);
 
   // Delete "blah"
   controller().setCompositionFromExistingText(emptyUnderlines, 6, 10);
   controller().commitText(String(""), emptyUnderlines, 0);
 
   // Check that the marker is still attached to "text " and includes the space
   // after "text" but not the space before
   EXPECT_EQ(1u, document().markers().markers().size());
   ASSERT_STREQ(
       "text\xC2\xA0",
       getMarkedText(document().markers(), div->firstChild(), 0).utf8().data());
 }
 
 TEST_F(InputMethodControllerTest,
        Marker_WhitespaceFixupAroundMarkerBeginningAndEndingWithSpaces) {
   Element* div = insertHTMLElement(
       "<div id='sample' contenteditable>Initial text blah</div>", "sample");
 
   // Add marker under " text " (use TextMatch since Composition markers don't
   // persist across editing operations)
   EphemeralRange markerRange = PlainTextRange(7, 13).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   // Delete "Initial"
   Vector<CompositionUnderline> emptyUnderlines;
   controller().setCompositionFromExistingText(emptyUnderlines, 0, 7);
   controller().commitText(String(""), emptyUnderlines, 0);
 
   // Delete "blah"
   controller().setCompositionFromExistingText(emptyUnderlines, 6, 10);
   controller().commitText(String(""), emptyUnderlines, 0);
 
   // Check that the marker is still attached to " text " and includes both the
   // space before "text" and the space after
   EXPECT_EQ(1u, document().markers().markers().size());
   ASSERT_STREQ(
       "\xC2\xA0text\xC2\xA0",
       getMarkedText(document().markers(), div->firstChild(), 0).utf8().data());
 }
 
 TEST_F(InputMethodControllerTest, Marker_ReplaceStartOfMarker) {
   Element* div = insertHTMLElement(
       "<div id='sample' contenteditable>Initial text</div>", "sample");
 
   // Add marker under "Initial text"
   EphemeralRange markerRange = PlainTextRange(0, 12).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   // Replace "Initial" with "Original"
   Vector<CompositionUnderline> emptyUnderlines;
   controller().setCompositionFromExistingText(emptyUnderlines, 0, 7);
   controller().commitText(String("Original"), emptyUnderlines, 0);
 
   // Verify marker is under "Original text"
   EXPECT_EQ(1u, document().markers().markers().size());
   ASSERT_STREQ(
       "Original text",
       getMarkedText(document().markers(), div->firstChild(), 0).utf8().data());
 }
 
 TEST_F(InputMethodControllerTest, Marker_ReplaceTextContainsStartOfMarker) {
   Element* div = insertHTMLElement(
       "<div id='sample' contenteditable>This is some initial text</div>",
       "sample");
 
   // Add marker under "initial text"
   EphemeralRange markerRange = PlainTextRange(13, 25).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   // Replace "some initial" with "boring"
   Vector<CompositionUnderline> emptyUnderlines;
   controller().setCompositionFromExistingText(emptyUnderlines, 8, 20);
   controller().commitText(String("boring"), emptyUnderlines, 0);
 
   // Verify marker is under " text"
   EXPECT_EQ(1u, document().markers().markers().size());
   EXPECT_STREQ(
       " text",
       getMarkedText(document().markers(), div->firstChild(), 0).utf8().data());
 }
 
 TEST_F(InputMethodControllerTest, Marker_ReplaceEndOfMarker) {
   Element* div = insertHTMLElement(
       "<div id='sample' contenteditable>Initial text</div>", "sample");
 
   // Add marker under "Initial text"
   EphemeralRange markerRange = PlainTextRange(0, 12).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   // Replace "text" with "string"
   Vector<CompositionUnderline> emptyUnderlines;
   controller().setCompositionFromExistingText(emptyUnderlines, 8, 12);
   controller().commitText(String("string"), emptyUnderlines, 0);
 
   // Verify marker is under "Initial string"
   EXPECT_EQ(1u, document().markers().markers().size());
   ASSERT_STREQ(
       "Initial string",
       getMarkedText(document().markers(), div->firstChild(), 0).utf8().data());
 }
 
 TEST_F(InputMethodControllerTest, Marker_ReplaceTextContainsEndOfMarker) {
   Element* div = insertHTMLElement(
       "<div id='sample' contenteditable>This is some initial text</div>",
       "sample");
 
   // Add marker under "some initial"
   EphemeralRange markerRange = PlainTextRange(8, 20).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   // Replace "initial text" with "content"
   Vector<CompositionUnderline> emptyUnderlines;
   controller().setCompositionFromExistingText(emptyUnderlines, 13, 25);
   controller().commitText(String("content"), emptyUnderlines, 0);
 
   EXPECT_STREQ("This is some content", div->innerHTML().utf8().data());
 
   // Verify marker is under "some "
   EXPECT_EQ(1u, document().markers().markers().size());
   EXPECT_STREQ(
       "some ",
       getMarkedText(document().markers(), div->firstChild(), 0).utf8().data());
 }
 
 TEST_F(InputMethodControllerTest, Marker_ReplaceEntireMarker) {
   Element* div = insertHTMLElement(
       "<div id='sample' contenteditable>Initial text</div>", "sample");
 
   // Add marker under "text"
   EphemeralRange markerRange = PlainTextRange(8, 12).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   // Replace "text" with "string"
   Vector<CompositionUnderline> emptyUnderlines;
   controller().setCompositionFromExistingText(emptyUnderlines, 8, 12);
   controller().commitText(String("string"), emptyUnderlines, 0);
 
   // Verify marker is under "string"
   EXPECT_EQ(1u, document().markers().markers().size());
   ASSERT_STREQ(
       "string",
       getMarkedText(document().markers(), div->firstChild(), 0).utf8().data());
 }
 
 TEST_F(InputMethodControllerTest, Marker_ReplaceTextWithMarkerAtBeginning) {
   Element* div = insertHTMLElement(
       "<div id='sample' contenteditable>Initial text</div>", "sample");
 
   // Add marker under "Initial"
   EphemeralRange markerRange = PlainTextRange(0, 7).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   EXPECT_EQ(1u, document().markers().markers().size());
 
   // Replace "Initial text" with "New string"
   Vector<CompositionUnderline> emptyUnderlines;
   controller().setCompositionFromExistingText(emptyUnderlines, 0, 12);
   controller().commitText(String("New string"), emptyUnderlines, 0);
 
   // Verify marker was removed
   EXPECT_EQ(0u, document().markers().markers().size());
 }
 
 TEST_F(InputMethodControllerTest, Marker_ReplaceTextWithMarkerAtEnd) {
   Element* div = insertHTMLElement(
       "<div id='sample' contenteditable>Initial text</div>", "sample");
 
   // Add marker under "text"
   EphemeralRange markerRange = PlainTextRange(8, 12).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   EXPECT_EQ(1u, document().markers().markers().size());
 
   // Replace "Initial text" with "New string"
   Vector<CompositionUnderline> emptyUnderlines;
   controller().setCompositionFromExistingText(emptyUnderlines, 0, 12);
   controller().commitText(String("New string"), emptyUnderlines, 0);
 
   // Verify marker was removed
   EXPECT_EQ(0u, document().markers().markers().size());
 }
 
 TEST_F(InputMethodControllerTest, Marker_Deletions) {
   Element* div = insertHTMLElement(
       "<div id='sample' contenteditable>1111122222333334444455555</div>",
       "sample");
 
   EphemeralRange markerRange = PlainTextRange(0, 5).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   markerRange = PlainTextRange(5, 10).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   markerRange = PlainTextRange(10, 15).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   markerRange = PlainTextRange(15, 20).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   markerRange = PlainTextRange(20, 25).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   EXPECT_EQ(5u, document().markers().markers().size());
 
   // Delete third marker and portions of second and fourth
   Vector<CompositionUnderline> emptyUnderlines;
   controller().setCompositionFromExistingText(emptyUnderlines, 8, 17);
   controller().commitText(String(""), emptyUnderlines, 0);
 
   // Verify markers were updated correctly
   EXPECT_EQ(4u, document().markers().markers().size());
 
   EXPECT_EQ(0u, document().markers().markers()[0]->startOffset());
   EXPECT_EQ(5u, document().markers().markers()[0]->endOffset());
 
   EXPECT_EQ(5u, document().markers().markers()[1]->startOffset());
   EXPECT_EQ(8u, document().markers().markers()[1]->endOffset());
 
   EXPECT_EQ(8u, document().markers().markers()[2]->startOffset());
   EXPECT_EQ(11u, document().markers().markers()[2]->endOffset());
 
   EXPECT_EQ(11u, document().markers().markers()[3]->startOffset());
   EXPECT_EQ(16u, document().markers().markers()[3]->endOffset());
 
   document().markers().clear();
 
   markerRange = PlainTextRange(5, 10).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   // Delete exactly on a marker
   controller().setCompositionFromExistingText(emptyUnderlines, 5, 10);
   controller().commitText(String(""), emptyUnderlines, 0);
   EXPECT_EQ(0u, document().markers().markers().size());
 
   markerRange = PlainTextRange(5, 10).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   // Delete middle of marker
   controller().setCompositionFromExistingText(emptyUnderlines, 6, 9);
   controller().commitText(String(""), emptyUnderlines, 0);
 
-  EXPECT_EQ(2u, document().markers().markers().size());
+  EXPECT_EQ(1u, document().markers().markers().size());
 
   EXPECT_EQ(5u, document().markers().markers()[0]->startOffset());
-  EXPECT_EQ(6u, document().markers().markers()[0]->endOffset());
-
-  EXPECT_EQ(6u, document().markers().markers()[1]->startOffset());
-  EXPECT_EQ(7u, document().markers().markers()[1]->endOffset());
+  EXPECT_EQ(7u, document().markers().markers()[0]->endOffset());
 }
 
 TEST_F(InputMethodControllerTest, Marker_Insertions) {
   Element* div = insertHTMLElement(
       "<div id='sample' contenteditable>1111122222333334444455555</div>",
       "sample");
 
   EphemeralRange markerRange = PlainTextRange(0, 5).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   markerRange = PlainTextRange(5, 10).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   markerRange = PlainTextRange(10, 15).createRange(*div);
-  document().markers().addMarker(markerRange.startPosition(),
-                                 markerRange.endPosition(),
-                                 DocumentMarker::TextMatch);
+  document().markers().addTextMatchMarker(markerRange, false);
 
   // insert in middle of second marker
   Vector<CompositionUnderline> emptyUnderlines;
   controller().setComposition("", emptyUnderlines, 7, 7);
   controller().commitText(String("66666"), emptyUnderlines, -7);
 
   EXPECT_EQ(3u, document().markers().markers().size());
 
   EXPECT_EQ(0u, document().markers().markers()[0]->startOffset());
   EXPECT_EQ(5u, document().markers().markers()[0]->endOffset());
@@ skipping 13 matching lines @@
   EXPECT_EQ(5u, document().markers().markers()[0]->endOffset());
 
   EXPECT_EQ(10u, document().markers().markers()[1]->startOffset());
   EXPECT_EQ(20u, document().markers().markers()[1]->endOffset());
 
   EXPECT_EQ(20u, document().markers().markers()[2]->startOffset());
   EXPECT_EQ(25u, document().markers().markers()[2]->endOffset());
 }
 
 }  // namespace blink