Use cached composition bounds for firstRectForCharacterRange

content/browser/renderer_host/render_widget_host_view_mac_unittest.mm

 // Copyright (c) 2012 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 "content/browser/renderer_host/render_widget_host_view_mac.h"
 
 #include "base/mac/scoped_nsautorelease_pool.h"
 #include "content/browser/browser_thread_impl.h"
 #include "content/browser/renderer_host/test_render_view_host.h"
 #include "content/common/gpu/gpu_messages.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/base/test/cocoa_test_event_utils.h"
 #import "ui/base/test/ui_cocoa_test_helper.h"
 #include "webkit/plugins/npapi/webplugin.h"
 
 using content::BrowserThread;
 using content::BrowserThreadImpl;
 using content::RenderViewHostImplTestHarness;
 
+namespace {
+
+// Generates the |length| of composition rectangle vector and save them to
+// |output|. It starts from |origin| and each rectangle contains |unit_size|.
+void GenerateCompositionRectArray(const gfx::Point& origin,
+                                  const gfx::Size& unit_size,
+                                  size_t length,
+                                  const std::vector<size_t>& break_points,
+                                  std::vector<gfx::Rect>* output) {
+  DCHECK(output);
+  output->clear();
+
+  std::queue<int> break_point_queue;
+  for (size_t i = 0; i < break_points.size(); ++i)
+    break_point_queue.push(break_points[i]);
+  break_point_queue.push(length);
+  size_t next_break_point = break_point_queue.front();
+  break_point_queue.pop();
+
+  gfx::Rect current_rect(origin, unit_size);
+  for (size_t i = 0; i < length; ++i) {
+    if (i == next_break_point) {
+      current_rect.set_x(origin.x());
+      current_rect.set_y(current_rect.y() + current_rect.height());
+      next_break_point = break_point_queue.front();
+      break_point_queue.pop();
+    }
+    output->push_back(current_rect);
+    current_rect.set_x(current_rect.right());
+  }
+}
+
+gfx::Rect GetExpectedRect(const gfx::Point& origin,
+                          const gfx::Size& size,
+                          const ui::Range& range,
+                          int line_no) {
+  return gfx::Rect(
+      origin.x() + range.start() * size.width(),
+      origin.y() + line_no * size.height(),
+      range.length() * size.width(),
+      size.height());
+}
+
+}  // namespace
+
 class RenderWidgetHostViewMacTest : public RenderViewHostImplTestHarness {
  public:
   RenderWidgetHostViewMacTest() : old_rwhv_(NULL), rwhv_mac_(NULL) {}
 
   virtual void SetUp() {
     RenderViewHostImplTestHarness::SetUp();
 
     // TestRenderViewHost's destruction assumes that its view is a
     // TestRenderWidgetHostView, so store its view and reset it back to the
     // stored view in |TearDown()|.
@@ skipping 169 matching lines @@
   EXPECT_EQ(rwhv_cocoa_.get(), [window firstResponder]);
 
   // Clean up.
   rwhv_mac_->DestroyFakePluginWindowHandle(accelerated_handle);
 }
 
 TEST_F(RenderWidgetHostViewMacTest, Fullscreen) {
   rwhv_mac_->InitAsFullscreen(NULL);
   EXPECT_TRUE(rwhv_mac_->pepper_fullscreen_window());
 }
+
+TEST_F(RenderWidgetHostViewMacTest, UpdateCompositionNormalCase) {
+  const gfx::Point kOrigin(10, 11);
+  const gfx::Size kBoundsUnit(10, 20);
+  rwhv_mac_->SelectionBoundsChanged(kCaretRect, kCaretRect);
+
+  // If there are no update from renderer, always returned caret position.
+  NSRect rect;
+  EXPECT_FALSE(rwhv_mac_->GetCachedFirstRectForCharacterRange(
+      ui::Range(0, 0).ToNSRange(),
+      &rect));
+  EXPECT_FALSE(rwhv_mac_->GetCachedFirstRectForCharacterRange(
+      ui::Range(0, 1).ToNSRange(),
+      &rect));
+  EXPECT_FALSE(rwhv_mac_->GetCachedFirstRectForCharacterRange(
+      ui::Range(1, 0).ToNSRange(),
+      &rect));
+  EXPECT_FALSE(rwhv_mac_->GetCachedFirstRectForCharacterRange(
+      ui::Range(1, 1).ToNSRange(),
+      &rect));
+  EXPECT_FALSE(rwhv_mac_->GetCachedFirstRectForCharacterRange(
+      ui::Range(1, 2).ToNSRange(),
+      &rect));
+
+  const int kCompositionLength = 10;
+  std::vector<gfx::Rect> composition_bounds;
+  const int kCompositionStart = 3;
+  const ui::Range kCompositionRange(kCompositionStart,
+                                    kCompositionStart + kCompositionLength);
+  GenerateCompositionRectArray(kOrigin,
+                               kBoundsUnit,
+                               kCompositionLength,
+                               std::vector<size_t>(),
+                               &composition_bounds);
+  rwhv_mac_->ImeCompositionRangeChanged(kCompositionRange, composition_bounds);
+
+  // Out of range requests will return caret position.
+  EXPECT_FALSE(rwhv_mac_->GetCachedFirstRectForCharacterRange(
+      ui::Range(0, 0).ToNSRange(),
+      &rect));
+  EXPECT_FALSE(rwhv_mac_->GetCachedFirstRectForCharacterRange(
+      ui::Range(1, 1).ToNSRange(),
+      &rect));
+  EXPECT_FALSE(rwhv_mac_->GetCachedFirstRectForCharacterRange(
+      ui::Range(1, 2).ToNSRange(),
+      &rect));
+  EXPECT_FALSE(rwhv_mac_->GetCachedFirstRectForCharacterRange(
+      ui::Range(2, 2).ToNSRange(),
+      &rect));
+  EXPECT_FALSE(rwhv_mac_->GetCachedFirstRectForCharacterRange(
+      ui::Range(13, 14).ToNSRange(),
+      &rect));
+  EXPECT_FALSE(rwhv_mac_->GetCachedFirstRectForCharacterRange(
+      ui::Range(14, 15).ToNSRange(),
+      &rect));
+
+  for (int i = 0; i < kCompositionLength; ++i) {
+    for (int j = 0; j < kCompositionLength - i; ++j) {
+      const ui::Range range(i, i + j);
+      const gfx::Rect expected_rect = GetExpectedRect(kOrigin,
+                                                      kBoundsUnit,
+                                                      range,
+                                                      0);
+      const NSRange request_range = ui::Range(
+          kCompositionStart + range.start(),
+          kCompositionStart + range.end()).ToNSRange();
+      EXPECT_TRUE(rwhv_mac_->GetCachedFirstRectForCharacterRange(request_range,
+                                                                 &rect));
+      EXPECT_EQ(expected_rect, gfx::Rect(NSRectToCGRect(rect)));
+    }
+  }
+}
+
+TEST_F(RenderWidgetHostViewMacTest, UpdateCompositionMultilineCase) {
+  const gfx::Point kOrigin(10, 11);
+  const gfx::Size kBoundsUnit(10, 20);
+  NSRect rect;
+
+  const int kCompositionLength = 30;
+  std::vector<gfx::Rect> composition_bounds;
+  const ui::Range kCompositionRange(0, kCompositionLength);
+  // Set breaking point at 10 and 20.
+  std::vector<size_t> break_points;
+  break_points.push_back(10);
+  break_points.push_back(20);
+  GenerateCompositionRectArray(kOrigin,
+                               kBoundsUnit,
+                               kCompositionLength,
+                               break_points,
+                               &composition_bounds);
+  rwhv_mac_->ImeCompositionRangeChanged(kCompositionRange, composition_bounds);
+
+  // Range doesn't contain line breaking point.
+  ui::Range range;
+  range = ui::Range(5, 8);
+  EXPECT_TRUE(rwhv_mac_->GetCachedFirstRectForCharacterRange(range.ToNSRange(),
+                                                             &rect));
+  EXPECT_EQ(
+      GetExpectedRect(kOrigin, kBoundsUnit, range, 0),
+      gfx::Rect(NSRectToCGRect(rect)));
+  range = ui::Range(15, 18);
+  EXPECT_TRUE(rwhv_mac_->GetCachedFirstRectForCharacterRange(range.ToNSRange(),
+                                                             &rect));
+  EXPECT_EQ(
+      GetExpectedRect(kOrigin, kBoundsUnit, ui::Range(5, 8), 1),
+      gfx::Rect(NSRectToCGRect(rect)));
+  range = ui::Range(25, 28);
+  EXPECT_TRUE(rwhv_mac_->GetCachedFirstRectForCharacterRange(range.ToNSRange(),
+                                                             &rect));
+  EXPECT_EQ(
+      GetExpectedRect(kOrigin, kBoundsUnit, ui::Range(5, 8), 2),
+      gfx::Rect(NSRectToCGRect(rect)));
+
+  // Range contains line breaking point.
+  range = ui::Range(8, 12);
+  EXPECT_TRUE(rwhv_mac_->GetCachedFirstRectForCharacterRange(range.ToNSRange(),
+                                                             &rect));
+  EXPECT_EQ(
+      GetExpectedRect(kOrigin, kBoundsUnit, ui::Range(8, 10), 0),
+      gfx::Rect(NSRectToCGRect(rect)));
+  range = ui::Range(18, 22);
+  EXPECT_TRUE(rwhv_mac_->GetCachedFirstRectForCharacterRange(range.ToNSRange(),
+                                                             &rect));
+  EXPECT_EQ(
+      GetExpectedRect(kOrigin, kBoundsUnit, ui::Range(8, 10), 1),
+      gfx::Rect(NSRectToCGRect(rect)));
+
+  // Start point is line breaking point.
+  range = ui::Range(10, 12);
+  EXPECT_TRUE(rwhv_mac_->GetCachedFirstRectForCharacterRange(range.ToNSRange(),
+                                                             &rect));
+  EXPECT_EQ(
+      GetExpectedRect(kOrigin, kBoundsUnit, ui::Range(0, 2), 1),
+      gfx::Rect(NSRectToCGRect(rect)));
+  range = ui::Range(20, 22);
+  EXPECT_TRUE(rwhv_mac_->GetCachedFirstRectForCharacterRange(range.ToNSRange(),
+                                                             &rect));
+  EXPECT_EQ(
+      GetExpectedRect(kOrigin, kBoundsUnit, ui::Range(0, 2), 2),
+      gfx::Rect(NSRectToCGRect(rect)));
+
+  // End point is line breaking point.
+  range = ui::Range(5, 10);
+  EXPECT_TRUE(rwhv_mac_->GetCachedFirstRectForCharacterRange(range.ToNSRange(),
+                                                             &rect));
+  EXPECT_EQ(
+      GetExpectedRect(kOrigin, kBoundsUnit, ui::Range(5, 10), 0),
+      gfx::Rect(NSRectToCGRect(rect)));
+  range = ui::Range(15, 20);
+  EXPECT_TRUE(rwhv_mac_->GetCachedFirstRectForCharacterRange(range.ToNSRange(),
+                                                             &rect));
+  EXPECT_EQ(
+      GetExpectedRect(kOrigin, kBoundsUnit, ui::Range(5, 10), 1),
+      gfx::Rect(NSRectToCGRect(rect)));
+
+  // Start and end point are same line breaking point.
+  range = ui::Range(10, 10);
+  EXPECT_TRUE(rwhv_mac_->GetCachedFirstRectForCharacterRange(range.ToNSRange(),
+                                                             &rect));
+  EXPECT_EQ(
+      GetExpectedRect(kOrigin, kBoundsUnit, ui::Range(0, 0), 1),
+      gfx::Rect(NSRectToCGRect(rect)));
+  range = ui::Range(20, 20);
+  EXPECT_TRUE(rwhv_mac_->GetCachedFirstRectForCharacterRange(range.ToNSRange(),
+                                                             &rect));
+  EXPECT_EQ(
+      GetExpectedRect(kOrigin, kBoundsUnit, ui::Range(0, 0), 2),
+      gfx::Rect(NSRectToCGRect(rect)));
+
+  // Start and end point are different line breaking point.
+  range = ui::Range(10, 20);
+  EXPECT_TRUE(rwhv_mac_->GetCachedFirstRectForCharacterRange(range.ToNSRange(),
+                                                             &rect));
+  EXPECT_EQ(
+      GetExpectedRect(kOrigin, kBoundsUnit, ui::Range(0, 10), 1),
+      gfx::Rect(NSRectToCGRect(rect)));
+}