Move setNonDirectionalSelectionIfNeeded() to SelectionController from FrameSelection

third_party/WebKit/Source/core/editing/FrameSelectionTest.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/FrameSelection.h"
 
 #include "bindings/core/v8/ExceptionStatePlaceholder.h"
 #include "core/dom/Document.h"
 #include "core/dom/Element.h"
 #include "core/dom/Text.h"
 #include "core/editing/EditingTestBase.h"
 #include "core/editing/FrameCaret.h"
+#include "core/editing/SelectionController.h"
 #include "core/frame/FrameView.h"
 #include "core/html/HTMLBodyElement.h"
+#include "core/input/EventHandler.h"
 #include "core/layout/LayoutView.h"
 #include "core/paint/PaintInfo.h"
 #include "core/paint/PaintLayer.h"
 #include "core/testing/DummyPageHolder.h"
 #include "platform/graphics/paint/DrawingRecorder.h"
 #include "platform/graphics/paint/PaintController.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "wtf/PassRefPtr.h"
 #include "wtf/RefPtr.h"
 #include "wtf/StdLibExtras.h"
@@ skipping 241 matching lines @@
                                  createVisiblePosition(Position(text, 2)),
                                  CharacterGranularity);
   EXPECT_EQ_SELECTED_TEXT("o B");
   // "Fo<o B|ar Baz," with the Word granularity.
   selection().moveRangeSelection(createVisiblePosition(Position(text, 5)),
                                  createVisiblePosition(Position(text, 2)),
                                  WordGranularity);
   EXPECT_EQ_SELECTED_TEXT("Foo Bar");
 }
 
-TEST_F(FrameSelectionTest, setNonDirectionalSelectionIfNeeded) {
+// TODO(yosin): We should move |SelectionControllerTest" to
+// "SelectionControllerTest.cpp"
+class SelectionControllerTest : public EditingTestBase {
+ protected:
+  SelectionControllerTest() = default;
+
+  const VisibleSelection& visibleSelectionInDOMTree() const {
+    return selection().selection();
+  }
+
+  const VisibleSelectionInFlatTree& visibleSelectionInFlatTree() const {
+    return selection().selectionInFlatTree();
+  }
+
+  void setNonDirectionalSelectionIfNeeded(const VisibleSelectionInFlatTree&,
+                                          TextGranularity);
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(SelectionControllerTest);
+};
+
+// TODO(yosin): We should move this function to "SelectionControllerTest.cpp"
+void SelectionControllerTest::setNonDirectionalSelectionIfNeeded(
+    const VisibleSelectionInFlatTree& newSelection,
+    TextGranularity granularity) {
+  frame()
+      .eventHandler()
+      .selectionController()
+      .setNonDirectionalSelectionIfNeeded(
+          newSelection, granularity, SelectionController::DoNotAdjustEndpoints);
+}
+
+// TODO(yosin): We should move this test to "SelectionControllerTest.cpp"
+TEST_F(SelectionControllerTest, setNonDirectionalSelectionIfNeeded) {
   const char* bodyContent = "<span id=top>top</span><span id=host></span>";
   const char* shadowContent = "<span id=bottom>bottom</span>";
   setBodyContent(bodyContent);
   ShadowRoot* shadowRoot = setShadowContent(shadowContent, "host");
 
   Node* top = document().getElementById("top")->firstChild();
   Node* bottom = shadowRoot->getElementById("bottom")->firstChild();
   Node* host = document().getElementById("host");
 
   // top to bottom
-  selection().setNonDirectionalSelectionIfNeeded(
+  setNonDirectionalSelectionIfNeeded(
       createVisibleSelection(SelectionInFlatTree::Builder()
                                  .collapse(PositionInFlatTree(top, 1))
                                  .extend(PositionInFlatTree(bottom, 3))
                                  .build()),
       CharacterGranularity);
   EXPECT_EQ(Position(top, 1), visibleSelectionInDOMTree().base());
   EXPECT_EQ(Position::beforeNode(host), visibleSelectionInDOMTree().extent());
   EXPECT_EQ(Position(top, 1), visibleSelectionInDOMTree().start());
   EXPECT_EQ(Position(top, 3), visibleSelectionInDOMTree().end());
 
   EXPECT_EQ(PositionInFlatTree(top, 1), visibleSelectionInFlatTree().base());
   EXPECT_EQ(PositionInFlatTree(bottom, 3),
             visibleSelectionInFlatTree().extent());
   EXPECT_EQ(PositionInFlatTree(top, 1), visibleSelectionInFlatTree().start());
   EXPECT_EQ(PositionInFlatTree(bottom, 3), visibleSelectionInFlatTree().end());
 
   // bottom to top
-  selection().setNonDirectionalSelectionIfNeeded(
+  setNonDirectionalSelectionIfNeeded(
       createVisibleSelection(SelectionInFlatTree::Builder()
                                  .collapse(PositionInFlatTree(bottom, 3))
                                  .extend(PositionInFlatTree(top, 1))
                                  .build()),
       CharacterGranularity);
   EXPECT_EQ(Position(bottom, 3), visibleSelectionInDOMTree().base());
   EXPECT_EQ(Position::beforeNode(bottom->parentNode()),
             visibleSelectionInDOMTree().extent());
   EXPECT_EQ(Position(bottom, 0), visibleSelectionInDOMTree().start());
   EXPECT_EQ(Position(bottom, 3), visibleSelectionInDOMTree().end());
@@ skipping 31 matching lines @@
   selection().updateIfNeeded();
 
   // TODO(yosin): Once lazy canonicalization implemented, selection.start
   // should be Position(HTML, 0).
   EXPECT_EQ(Position(), selection().start())
       << "updateIfNeeded() makes selection to null.";
   EXPECT_EQ(selection().start(), caretPosition().position());
 }
 
 }  // namespace blink