third_party/WebKit/Source/platform/graphics/paint/DrawingDisplayItemTest.cpp

// Copyright 2017 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 "platform/graphics/paint/DrawingDisplayItem.h"

#include "SkTypes.h"
#include "platform/graphics/paint/PaintRecorder.h"
#include "platform/graphics/skia/SkiaUtils.h"
#include "platform/testing/FakeDisplayItemClient.h"
#include "public/platform/WebDisplayItemList.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace blink {
namespace {

using ::testing::_;

class DrawingDisplayItemTest : public ::testing::Test {
 protected:
  FakeDisplayItemClient client_;
};

class MockWebDisplayItemList : public WebDisplayItemList {
 public:
  MOCK_METHOD2(AppendDrawingItem,
               void(const WebRect& visual_rect, sk_sp<const cc::PaintRecord>));
};

static sk_sp<PaintRecord> CreateRectRecord(const FloatRect& record_bounds) {
  PaintRecorder recorder;
  PaintCanvas* canvas =
      recorder.beginRecording(record_bounds.Width(), record_bounds.Height());
  canvas->drawRect(record_bounds, PaintFlags());
  return recorder.finishRecordingAsPicture();
}

static sk_sp<PaintRecord> CreateRectRecordWithTranslate(
    const FloatRect& record_bounds,
    float dx,
    float dy) {
  PaintRecorder recorder;
  PaintCanvas* canvas =
      recorder.beginRecording(record_bounds.Width(), record_bounds.Height());
  canvas->save();
  canvas->translate(dx, dy);
  canvas->drawRect(record_bounds, PaintFlags());
  canvas->restore();
  return recorder.finishRecordingAsPicture();
}

TEST_F(DrawingDisplayItemTest, VisualRectAndDrawingBounds) {
  FloatRect record_bounds(5.5, 6.6, 7.7, 8.8);
  LayoutRect drawing_bounds(record_bounds);
  client_.SetVisualRect(drawing_bounds);

  DrawingDisplayItem item(client_, DisplayItem::Type::kDocumentBackground,
                          CreateRectRecord(record_bounds));
  EXPECT_EQ(drawing_bounds, item.VisualRect());

  MockWebDisplayItemList list1;
  WebRect expected_rect = EnclosingIntRect(drawing_bounds);
  EXPECT_CALL(list1, AppendDrawingItem(expected_rect, _)).Times(1);
  item.AppendToWebDisplayItemList(LayoutSize(), &list1);

  LayoutSize offset(LayoutUnit(2.1), LayoutUnit(3.6));
  LayoutRect visual_rect_with_offset = drawing_bounds;
  visual_rect_with_offset.Move(-offset);
  WebRect expected_visual_rect = EnclosingIntRect(visual_rect_with_offset);
  MockWebDisplayItemList list2;
  EXPECT_CALL(list2, AppendDrawingItem(expected_visual_rect, _)).Times(1);
  item.AppendToWebDisplayItemList(offset, &list2);
}

TEST_F(DrawingDisplayItemTest, Equals) {
  FloatRect bounds1(100.1, 100.2, 100.3, 100.4);
  client_.SetVisualRect(LayoutRect(bounds1));
  DrawingDisplayItem item1(client_, DisplayItem::kDocumentBackground,
                           CreateRectRecord(bounds1));
  DrawingDisplayItem translated(client_, DisplayItem::kDocumentBackground,
                                CreateRectRecordWithTranslate(bounds1, 10, 20));
  // This item contains a DrawingRecord that is different from but visually
  // equivalent to item1's.
  DrawingDisplayItem zero_translated(
      client_, DisplayItem::kDocumentBackground,
      CreateRectRecordWithTranslate(bounds1, 0, 0));

  FloatRect bounds2(100.5, 100.6, 100.7, 100.8);
  client_.SetVisualRect(LayoutRect(bounds2));
  DrawingDisplayItem item2(client_, DisplayItem::kDocumentBackground,
                           CreateRectRecord(bounds2));

  DrawingDisplayItem empty_item(client_, DisplayItem::kDocumentBackground,
                                nullptr);

  EXPECT_TRUE(item1.Equals(item1));
  EXPECT_FALSE(item1.Equals(item2));
  EXPECT_FALSE(item1.Equals(translated));
  EXPECT_TRUE(item1.Equals(zero_translated));
  EXPECT_FALSE(item1.Equals(empty_item));

  EXPECT_FALSE(item2.Equals(item1));
  EXPECT_TRUE(item2.Equals(item2));
  EXPECT_FALSE(item2.Equals(translated));
  EXPECT_FALSE(item2.Equals(zero_translated));
  EXPECT_FALSE(item2.Equals(empty_item));

  EXPECT_FALSE(translated.Equals(item1));
  EXPECT_FALSE(translated.Equals(item2));
  EXPECT_TRUE(translated.Equals(translated));
  EXPECT_FALSE(translated.Equals(zero_translated));
  EXPECT_FALSE(translated.Equals(empty_item));

  EXPECT_TRUE(zero_translated.Equals(item1));
  EXPECT_FALSE(zero_translated.Equals(item2));
  EXPECT_FALSE(zero_translated.Equals(translated));
  EXPECT_TRUE(zero_translated.Equals(zero_translated));
  EXPECT_FALSE(zero_translated.Equals(empty_item));

  EXPECT_FALSE(empty_item.Equals(item1));
  EXPECT_FALSE(empty_item.Equals(item2));
  EXPECT_FALSE(empty_item.Equals(translated));
  EXPECT_FALSE(empty_item.Equals(zero_translated));
  EXPECT_TRUE(empty_item.Equals(empty_item));
}

}  // namespace
}  // namespace blink