Clear the PrintContext in WebLocalFrameImpl::Close().

third_party/WebKit/Source/core/page/PrintContextTest.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/page/PrintContext.h"
 
 #include <memory>
 
 #include "core/dom/Document.h"
 #include "core/frame/FrameView.h"
 #include "core/html/HTMLElement.h"
 #include "core/layout/LayoutTestHelper.h"
 #include "core/layout/LayoutView.h"
 #include "core/paint/PaintLayer.h"
 #include "core/paint/PaintLayerPainter.h"
 #include "core/testing/DummyPageHolder.h"
 #include "platform/graphics/GraphicsContext.h"
 #include "platform/graphics/paint/DrawingRecorder.h"
 #include "platform/graphics/paint/PaintRecordBuilder.h"
 #include "platform/scroll/ScrollbarTheme.h"
 #include "platform/text/TextStream.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "third_party/skia/include/core/SkCanvas.h"
 
 namespace blink {
 
 const int kPageWidth = 800;
 const int kPageHeight = 600;
 
-class MockPrintContext : public PrintContext {
- public:
-  MockPrintContext(LocalFrame* frame) : PrintContext(frame) {}
-
-  void OutputLinkedDestinations(GraphicsContext& context,
-                                const IntRect& page_rect) {
-    PrintContext::OutputLinkedDestinations(context, page_rect);
-  }
-};
-
 class MockCanvas : public SkCanvas {
  public:
   enum OperationType { kDrawRect, kDrawPoint };
 
   struct Operation {
     OperationType type;
     SkRect rect;
   };
 
   MockCanvas() : SkCanvas(kPageWidth, kPageHeight) {}
+  ~MockCanvas() override {}
 
   void onDrawAnnotation(const SkRect& rect,
                         const char key[],
                         SkData* value) override {
     if (rect.width() == 0 && rect.height() == 0) {
       SkPoint point = getTotalMatrix().mapXY(rect.x(), rect.y());
       Operation operation = {kDrawPoint,
                              SkRect::MakeXYWH(point.x(), point.y(), 0, 0)};
       recorded_operations_.push_back(operation);
     } else {
       Operation operation = {kDrawRect, rect};
       getTotalMatrix().mapRect(&operation.rect);
       recorded_operations_.push_back(operation);
     }
   }
 
   const Vector<Operation>& RecordedOperations() const {
     return recorded_operations_;
   }
 
  private:
   Vector<Operation> recorded_operations_;
 };
 
 class PrintContextTest : public RenderingTest {
  protected:
   explicit PrintContextTest(LocalFrameClient* local_frame_client = nullptr)
       : RenderingTest(local_frame_client) {}
+  ~PrintContextTest() override {}
 
   void SetUp() override {
     RenderingTest::SetUp();
-    print_context_ = new MockPrintContext(GetDocument().GetFrame());
+    print_context_ = new PrintContext(GetDocument().GetFrame());
   }
 
-  MockPrintContext& PrintContext() { return *print_context_.Get(); }
+  PrintContext& GetPrintContext() { return *print_context_.Get(); }
 
   void SetBodyInnerHTML(String body_content) {
     GetDocument().body()->setAttribute(HTMLNames::styleAttr, "margin: 0");
     GetDocument().body()->setInnerHTML(body_content);
   }
 
   void PrintSinglePage(MockCanvas& canvas) {
     IntRect page_rect(0, 0, kPageWidth, kPageHeight);
-    PrintContext().BeginPrintMode(page_rect.Width(), page_rect.Height());
+    GetPrintContext().BeginPrintMode(page_rect.Width(), page_rect.Height());
     GetDocument().View()->UpdateAllLifecyclePhases();
     PaintRecordBuilder builder(page_rect);
     GraphicsContext& context = builder.Context();
     context.SetPrinting(true);
     GetDocument().View()->PaintContents(context, kGlobalPaintPrinting,
                                         page_rect);
     {
       DrawingRecorder recorder(context, *GetDocument().GetLayoutView(),
                                DisplayItem::kPrintedContentDestinationLocations,
                                page_rect);
-      PrintContext().OutputLinkedDestinations(context, page_rect);
+      GetPrintContext().OutputLinkedDestinations(context, page_rect);
     }
     builder.EndRecording()->playback(&canvas);
-    PrintContext().EndPrintMode();
+    GetPrintContext().EndPrintMode();
   }
 
   static String AbsoluteBlockHtmlForLink(int x,
                                          int y,
                                          int width,
                                          int height,
                                          const char* url,
                                          const char* children = nullptr) {
     TextStream ts;
     ts << "<a style='position: absolute; left: " << x << "px; top: " << y
@@ skipping 14 matching lines @@
                               const char* name,
                               const char* text_content) {
     TextStream ts;
     ts << "<a name='" << name << "' style='position: absolute; left: " << x
        << "px; top: " << y << "px'>" << text_content << "</a>";
     return ts.Release();
   }
 
  private:
   std::unique_ptr<DummyPageHolder> page_holder_;
-  Persistent<MockPrintContext> print_context_;
+  Persistent<PrintContext> print_context_;
 };
 
 class PrintContextFrameTest : public PrintContextTest {
  public:
   PrintContextFrameTest()
       : PrintContextTest(SingleChildLocalFrameClient::Create()) {}
 };
 
 #define EXPECT_SKRECT_EQ(expectedX, expectedY, expectedWidth, expectedHeight, \
                          actualRect)                                          \
@@ skipping 199 matching lines @@
   EXPECT_EQ(MockCanvas::kDrawRect, operations[0].type);
   EXPECT_SKRECT_EQ(50, 60, 70, 80,
                    operations[0].rect);  // FIXME: the rect should be clipped.
   EXPECT_EQ(MockCanvas::kDrawRect, operations[1].type);
   EXPECT_SKRECT_EQ(150, 160, 170, 180, operations[1].rect);
   EXPECT_EQ(MockCanvas::kDrawRect, operations[2].type);
   EXPECT_SKRECT_EQ(250, 260, 270, 280, operations[2].rect);
 }
 
 }  // namespace blink